Java MasterKey类代码示例

import org.apache.hadoop.yarn.server.api.records.MasterKey; //导入依赖的package包/类
@Override
public RegisterNodeManagerResponse registerNodeManager(
    RegisterNodeManagerRequest request) throws YarnException,
    IOException {
  RegisterNodeManagerResponse response = recordFactory
      .newRecordInstance(RegisterNodeManagerResponse.class);
  MasterKey masterKey = new MasterKeyPBImpl();
  masterKey.setKeyId(123);
  masterKey.setBytes(ByteBuffer.wrap(new byte[] { new Integer(123)
    .byteValue() }));
  response.setContainerTokenMasterKey(masterKey);
  response.setNMTokenMasterKey(masterKey);
  return response;
}
 

import org.apache.hadoop.yarn.server.api.records.MasterKey; //导入依赖的package包/类
@Override
public boolean equals(Object obj) {
  if (this == obj)
    return true;
  if (!(obj instanceof MasterKey)) {
    return false;
  }
  MasterKey other = (MasterKey) obj;
  if (this.getKeyId() != other.getKeyId()) {
    return false;
  }
  if (!this.getBytes().equals(other.getBytes())) {
    return false;
  }
  return true;
}
 

import org.apache.hadoop.yarn.server.api.records.MasterKey; //导入依赖的package包/类
public static NodeHeartbeatResponse newNodeHeartbeatResponse(int responseId,
    NodeAction action, List<ContainerId> containersToCleanUp,
    List<ApplicationId> applicationsToCleanUp,
    MasterKey containerTokenMasterKey, MasterKey nmTokenMasterKey,
    long nextHeartbeatInterval) {
  NodeHeartbeatResponse response = recordFactory
      .newRecordInstance(NodeHeartbeatResponse.class);
  response.setResponseId(responseId);
  response.setNodeAction(action);
  response.setContainerTokenMasterKey(containerTokenMasterKey);
  response.setNMTokenMasterKey(nmTokenMasterKey);
  response.setNextHeartBeatInterval(nextHeartbeatInterval);
  if(containersToCleanUp != null) {
    response.addAllContainersToCleanup(containersToCleanUp);
  }
  if(applicationsToCleanUp != null) {
    response.addAllApplicationsToCleanup(applicationsToCleanUp);
  }
  return response;
}
 

import org.apache.hadoop.yarn.server.api.records.MasterKey; //导入依赖的package包/类
private void populateKeys(NodeHeartbeatRequest request,
    NodeHeartbeatResponse nodeHeartBeatResponse) {

  // Check if node's masterKey needs to be updated and if the currentKey has
  // roller over, send it across

  // ContainerTokenMasterKey

  MasterKey nextMasterKeyForNode =
      this.containerTokenSecretManager.getNextKey();
  if (nextMasterKeyForNode != null
      && (request.getLastKnownContainerTokenMasterKey().getKeyId()
          != nextMasterKeyForNode.getKeyId())) {
    nodeHeartBeatResponse.setContainerTokenMasterKey(nextMasterKeyForNode);
  }

  // NMTokenMasterKey

  nextMasterKeyForNode = this.nmTokenSecretManager.getNextKey();
  if (nextMasterKeyForNode != null
      && (request.getLastKnownNMTokenMasterKey().getKeyId() 
          != nextMasterKeyForNode.getKeyId())) {
    nodeHeartBeatResponse.setNMTokenMasterKey(nextMasterKeyForNode);
  }
}
 

import org.apache.hadoop.yarn.server.api.records.MasterKey; //导入依赖的package包/类
public void recover(RMState state) {
  if (state.getAMRMTokenSecretManagerState() != null) {
    // recover the current master key
    MasterKey currentKey =
        state.getAMRMTokenSecretManagerState().getCurrentMasterKey();
    this.currentMasterKey =
        new MasterKeyData(currentKey, createSecretKey(currentKey.getBytes()
          .array()));

    // recover the next master key if not null
    MasterKey nextKey =
        state.getAMRMTokenSecretManagerState().getNextMasterKey();
    if (nextKey != null) {
      this.nextMasterKey =
          new MasterKeyData(nextKey, createSecretKey(nextKey.getBytes()
            .array()));
      this.timer.schedule(new NextKeyActivator(), this.activationDelay);
    }
  }
}
 

import org.apache.hadoop.yarn.server.api.records.MasterKey; //导入依赖的package包/类
public static NodeHeartbeatResponse newNodeHeartbeatResponse(int responseId,
    NodeAction action, List<ContainerId> containersToCleanUp,
    List<ApplicationId> applicationsToCleanUp,List<NodeContainerUpdate> listNodeContainerUpdate,
    MasterKey containerTokenMasterKey, MasterKey nmTokenMasterKey,
    long nextHeartbeatInterval) {
  NodeHeartbeatResponse response = recordFactory
      .newRecordInstance(NodeHeartbeatResponse.class);
  response.setResponseId(responseId);
  response.setNodeAction(action);
  response.setContainerTokenMasterKey(containerTokenMasterKey);
  response.setNMTokenMasterKey(nmTokenMasterKey);
  response.setNextHeartBeatInterval(nextHeartbeatInterval);
  
  if(containersToCleanUp != null) {
    response.addAllContainersToCleanup(containersToCleanUp);
  }
  if(applicationsToCleanUp != null) {
    response.addAllApplicationsToCleanup(applicationsToCleanUp);
  }
  if(listNodeContainerUpdate != null){
    response.addNodeContainersToUpdate(listNodeContainerUpdate);	
  }
  return response;
}
 

import org.apache.hadoop.yarn.server.api.records.MasterKey; //导入依赖的package包/类
public MasterKeyData(int serialNo, SecretKey secretKey) {
  this.masterKeyRecord = Records.newRecord(MasterKey.class);
  this.masterKeyRecord.setKeyId(serialNo);
  this.generatedSecretKey = secretKey;
  this.masterKeyRecord.setBytes(ByteBuffer.wrap(generatedSecretKey
    .getEncoded()));
}
 

import org.apache.hadoop.yarn.server.api.records.MasterKey; //导入依赖的package包/类
@Private
public MasterKey getCurrentKey() {
  this.readLock.lock();
  try {
    return this.currentMasterKey.getMasterKey();
  } finally {
    this.readLock.unlock();
  }
}
 

import org.apache.hadoop.yarn.server.api.records.MasterKey; //导入依赖的package包/类
@Private
public MasterKey getCurrentKey() {
  this.readLock.lock();
  try {
  return this.currentMasterKey.getMasterKey();
  } finally {
    this.readLock.unlock();
  }
}
 

import org.apache.hadoop.yarn.server.api.records.MasterKey; //导入依赖的package包/类
public static NodeHeartbeatRequest newInstance(NodeStatus nodeStatus,
    MasterKey lastKnownContainerTokenMasterKey,
    MasterKey lastKnownNMTokenMasterKey) {
  NodeHeartbeatRequest nodeHeartbeatRequest =
      Records.newRecord(NodeHeartbeatRequest.class);
  nodeHeartbeatRequest.setNodeStatus(nodeStatus);
  nodeHeartbeatRequest
      .setLastKnownContainerTokenMasterKey(lastKnownContainerTokenMasterKey);
  nodeHeartbeatRequest
      .setLastKnownNMTokenMasterKey(lastKnownNMTokenMasterKey);
  return nodeHeartbeatRequest;
}
 

import org.apache.hadoop.yarn.server.api.records.MasterKey; //导入依赖的package包/类
@Override
public MasterKey getLastKnownContainerTokenMasterKey() {
  NodeHeartbeatRequestProtoOrBuilder p = viaProto ? proto : builder;
  if (this.lastKnownContainerTokenMasterKey != null) {
    return this.lastKnownContainerTokenMasterKey;
  }
  if (!p.hasLastKnownContainerTokenMasterKey()) {
    return null;
  }
  this.lastKnownContainerTokenMasterKey =
      convertFromProtoFormat(p.getLastKnownContainerTokenMasterKey());
  return this.lastKnownContainerTokenMasterKey;
}
 

import org.apache.hadoop.yarn.server.api.records.MasterKey; //导入依赖的package包/类
@Override
public void setLastKnownContainerTokenMasterKey(MasterKey masterKey) {
  maybeInitBuilder();
  if (masterKey == null) 
    builder.clearLastKnownContainerTokenMasterKey();
  this.lastKnownContainerTokenMasterKey = masterKey;
}
 

import org.apache.hadoop.yarn.server.api.records.MasterKey; //导入依赖的package包/类
@Override
public MasterKey getLastKnownNMTokenMasterKey() {
  NodeHeartbeatRequestProtoOrBuilder p = viaProto ? proto : builder;
  if (this.lastKnownNMTokenMasterKey != null) {
    return this.lastKnownNMTokenMasterKey;
  }
  if (!p.hasLastKnownNmTokenMasterKey()) {
    return null;
  }
  this.lastKnownNMTokenMasterKey =
      convertFromProtoFormat(p.getLastKnownNmTokenMasterKey());
  return this.lastKnownNMTokenMasterKey;
}
 

import org.apache.hadoop.yarn.server.api.records.MasterKey; //导入依赖的package包/类
@Override
public void setLastKnownNMTokenMasterKey(MasterKey masterKey) {
  maybeInitBuilder();
  if (masterKey == null) 
    builder.clearLastKnownNmTokenMasterKey();
  this.lastKnownNMTokenMasterKey = masterKey;
}
 

import org.apache.hadoop.yarn.server.api.records.MasterKey; //导入依赖的package包/类
@Override
public MasterKey getContainerTokenMasterKey() {
  RegisterNodeManagerResponseProtoOrBuilder p = viaProto ? proto : builder;
  if (this.containerTokenMasterKey != null) {
    return this.containerTokenMasterKey;
  }
  if (!p.hasContainerTokenMasterKey()) {
    return null;
  }
  this.containerTokenMasterKey =
      convertFromProtoFormat(p.getContainerTokenMasterKey());
  return this.containerTokenMasterKey;
}