private void initApplicationACLs() { if (this.applicationACLS != null) { return; } RegisterApplicationMasterResponseProtoOrBuilder p = viaProto ? proto : builder; List<ApplicationACLMapProto> list = p.getApplicationACLsList(); this.applicationACLS = new HashMap<ApplicationAccessType, String>(list .size()); for (ApplicationACLMapProto aclProto : list) { this.applicationACLS.put(ProtoUtils.convertFromProtoFormat(aclProto .getAccessType()), aclProto.getAcl()); } }
private void addApplicationACLs() { maybeInitBuilder(); builder.clearApplicationACLs(); if (applicationACLS == null) { return; } Iterable<? extends ApplicationACLMapProto> values = new Iterable<ApplicationACLMapProto>() { @Override public Iterator<ApplicationACLMapProto> iterator() { return new Iterator<ApplicationACLMapProto>() { Iterator<ApplicationAccessType> aclsIterator = applicationACLS .keySet().iterator(); @Override public boolean hasNext() { return aclsIterator.hasNext(); } @Override public ApplicationACLMapProto next() { ApplicationAccessType key = aclsIterator.next(); return ApplicationACLMapProto.newBuilder().setAcl( applicationACLS.get(key)).setAccessType( ProtoUtils.convertToProtoFormat(key)).build(); } @Override public void remove() { throw new UnsupportedOperationException(); } }; } }; this.builder.addAllApplicationACLs(values); }
private void mergeLocalToBuilder() { if (this.states != null) { maybeInitBuilder(); builder.clearNodeStates(); Iterable<NodeStateProto> iterable = new Iterable<NodeStateProto>() { @Override public Iterator<NodeStateProto> iterator() { return new Iterator<NodeStateProto>() { Iterator<NodeState> iter = states.iterator(); @Override public boolean hasNext() { return iter.hasNext(); } @Override public NodeStateProto next() { return ProtoUtils.convertToProtoFormat(iter.next()); } @Override public void remove() { throw new UnsupportedOperationException(); } }; } }; builder.addAllNodeStates(iterable); } }
private void initNodeStates() { if (this.states != null) { return; } GetClusterNodesRequestProtoOrBuilder p = viaProto ? proto : builder; List<NodeStateProto> list = p.getNodeStatesList(); this.states = EnumSet.noneOf(NodeState.class); for (NodeStateProto c : list) { this.states.add(ProtoUtils.convertFromProtoFormat(c)); } }
private void initApplicationStates() { if (this.applicationStates != null) { return; } GetApplicationsRequestProtoOrBuilder p = viaProto ? proto : builder; List<YarnApplicationStateProto> appStatesList = p.getApplicationStatesList(); this.applicationStates = EnumSet.noneOf(YarnApplicationState.class); for (YarnApplicationStateProto c : appStatesList) { this.applicationStates.add(ProtoUtils.convertFromProtoFormat(c)); } }
private void initScope() { if (this.scope != null) { return; } GetApplicationsRequestProtoOrBuilder p = viaProto ? proto : builder; this.scope = ProtoUtils.convertFromProtoFormat(p.getScope()); }
@Override public synchronized AMCommand getAMCommand() { AllocateResponseProtoOrBuilder p = viaProto ? proto : builder; if (!p.hasAMCommand()) { return null; } return ProtoUtils.convertFromProtoFormat(p.getAMCommand()); }
@Override public synchronized void setAMCommand(AMCommand command) { maybeInitBuilder(); if (command == null) { builder.clearAMCommand(); return; } builder.setAMCommand(ProtoUtils.convertToProtoFormat(command)); }
private void addSystemCredentialsToProto() { maybeInitBuilder(); builder.clearSystemCredentialsForApps(); for (Map.Entry<ApplicationId, ByteBuffer> entry : systemCredentials.entrySet()) { builder.addSystemCredentialsForApps(SystemCredentialsForAppsProto.newBuilder() .setAppId(convertToProtoFormat(entry.getKey())) .setCredentialsForApp(ProtoUtils.convertToProtoFormat( entry.getValue().duplicate()))); } }
private void initSystemCredentials() { NodeHeartbeatResponseProtoOrBuilder p = viaProto ? proto : builder; List<SystemCredentialsForAppsProto> list = p.getSystemCredentialsForAppsList(); this.systemCredentials = new HashMap<ApplicationId, ByteBuffer> (); for (SystemCredentialsForAppsProto c : list) { ApplicationId appId = convertFromProtoFormat(c.getAppId()); ByteBuffer byteBuffer = ProtoUtils.convertFromProtoFormat(c.getCredentialsForApp()); this.systemCredentials.put(appId, byteBuffer); } }
private ContainerManagerApplicationProto buildAppProto(ApplicationId appId, String user, Credentials credentials, Map<ApplicationAccessType, String> appAcls, LogAggregationContext logAggregationContext) { ContainerManagerApplicationProto.Builder builder = ContainerManagerApplicationProto.newBuilder(); builder.setId(((ApplicationIdPBImpl) appId).getProto()); builder.setUser(user); if (logAggregationContext != null) { builder.setLogAggregationContext(( (LogAggregationContextPBImpl)logAggregationContext).getProto()); } builder.clearCredentials(); if (credentials != null) { DataOutputBuffer dob = new DataOutputBuffer(); try { credentials.writeTokenStorageToStream(dob); builder.setCredentials(ByteString.copyFrom(dob.getData())); } catch (IOException e) { // should not occur LOG.error("Cannot serialize credentials", e); } } builder.clearAcls(); if (appAcls != null) { for (Map.Entry<ApplicationAccessType, String> acl : appAcls.entrySet()) { ApplicationACLMapProto p = ApplicationACLMapProto.newBuilder() .setAccessType(ProtoUtils.convertToProtoFormat(acl.getKey())) .setAcl(acl.getValue()) .build(); builder.addAcls(p); } } return builder.build(); }
private void mergeLocalToBuilder() { if (this.attemptId != null) { builder.setAttemptId(((ApplicationAttemptIdPBImpl)attemptId).getProto()); } if(this.masterContainer != null) { builder.setMasterContainer(((ContainerPBImpl)masterContainer).getProto()); } if(this.appAttemptTokens != null) { builder.setAppAttemptTokens(ProtoUtils.convertToProtoFormat( this.appAttemptTokens)); } }
@Override public Credentials getAppAttemptTokens() { ApplicationAttemptStateDataProtoOrBuilder p = viaProto ? proto : builder; if(appAttemptTokens != null) { return convertCredentialsFromByteBuffer(appAttemptTokens); } if(!p.hasAppAttemptTokens()) { return null; } this.appAttemptTokens = ProtoUtils.convertFromProtoFormat( p.getAppAttemptTokens()); return convertCredentialsFromByteBuffer(appAttemptTokens); }