Java DelegationKey.readFields方法代码示例

import org.apache.hadoop.security.token.delegation.DelegationKey; //导入方法依赖的package包/类
/**
 * Private helper method to load delegation keys from fsimage.
 * @throws IOException on error
 */
private synchronized void loadAllKeys(DataInput in) throws IOException {
  StartupProgress prog = NameNode.getStartupProgress();
  Step step = new Step(StepType.DELEGATION_KEYS);
  prog.beginStep(Phase.LOADING_FSIMAGE, step);
  int numberOfKeys = in.readInt();
  prog.setTotal(Phase.LOADING_FSIMAGE, step, numberOfKeys);
  Counter counter = prog.getCounter(Phase.LOADING_FSIMAGE, step);
  for (int i = 0; i < numberOfKeys; i++) {
    DelegationKey value = new DelegationKey();
    value.readFields(in);
    addKey(value);
    counter.increment();
  }
  prog.endStep(Phase.LOADING_FSIMAGE, step);
}
 

import org.apache.hadoop.security.token.delegation.DelegationKey; //导入方法依赖的package包/类
private DelegationKey loadDelegationKey(byte[] data) throws IOException {
  DelegationKey key = new DelegationKey();
  DataInputStream in = new DataInputStream(new ByteArrayInputStream(data));
  try {
    key.readFields(in);
  } finally {
    IOUtils.cleanup(LOG, in);
  }
  return key;
}
 

import org.apache.hadoop.security.token.delegation.DelegationKey; //导入方法依赖的package包/类
private void loadRMDelegationKeyState(RMState rmState) throws Exception {
  List<String> childNodes =
      getChildrenWithRetries(dtMasterKeysRootPath, false);
  for (String childNodeName : childNodes) {
    String childNodePath = getNodePath(dtMasterKeysRootPath, childNodeName);
    byte[] childData = getDataWithRetries(childNodePath, false);

    if (childData == null) {
      LOG.warn("Content of " + childNodePath + " is broken.");
      continue;
    }

    ByteArrayInputStream is = new ByteArrayInputStream(childData);
    DataInputStream fsIn = new DataInputStream(is);

    try {
      if (childNodeName.startsWith(DELEGATION_KEY_PREFIX)) {
        DelegationKey key = new DelegationKey();
        key.readFields(fsIn);
        rmState.rmSecretManagerState.masterKeyState.add(key);
        if (LOG.isDebugEnabled()) {
          LOG.debug("Loaded delegation key: keyId=" + key.getKeyId()
              + ", expirationDate=" + key.getExpiryDate());
        }
      }
    } finally {
      is.close();
    }
  }
}
 

import org.apache.hadoop.security.token.delegation.DelegationKey; //导入方法依赖的package包/类
private static void loadTokenMasterKeyData(TimelineServiceState state,
    byte[] keyData)
    throws IOException {
  DelegationKey key = new DelegationKey();
  DataInputStream in =
      new DataInputStream(new ByteArrayInputStream(keyData));
  try {
    key.readFields(in);
  } finally {
    IOUtils.cleanup(LOG, in);
  }
  state.tokenMasterKeyState.add(key);
}
 

import org.apache.hadoop.security.token.delegation.DelegationKey; //导入方法依赖的package包/类
private void loadTokenMasterKey(HistoryServerState state, byte[] data)
    throws IOException {
  DelegationKey key = new DelegationKey();
  DataInputStream in =
      new DataInputStream(new ByteArrayInputStream(data));
  try {
    key.readFields(in);
  } finally {
    IOUtils.cleanup(LOG, in);
  }
  state.tokenMasterKeyState.add(key);
}
 

import org.apache.hadoop.security.token.delegation.DelegationKey; //导入方法依赖的package包/类
private void loadTokenMasterKey(HistoryServerState state, Path keyFile,
    long numKeyFileBytes) throws IOException {
  DelegationKey key = new DelegationKey();
  byte[] keyData = readFile(keyFile, numKeyFileBytes);
  DataInputStream in =
      new DataInputStream(new ByteArrayInputStream(keyData));
  try {
    key.readFields(in);
  } finally {
    IOUtils.cleanup(LOG, in);
  }
  state.tokenMasterKeyState.add(key);
}
 

import org.apache.hadoop.security.token.delegation.DelegationKey; //导入方法依赖的package包/类
private void loadRMDelegationKeyState(RMState rmState) throws Exception {
  List<String> childNodes =
      getChildren(dtMasterKeysRootPath);
  for (String childNodeName : childNodes) {
    String childNodePath = getNodePath(dtMasterKeysRootPath, childNodeName);
    byte[] childData = getData(childNodePath);

    if (childData == null) {
      LOG.warn("Content of " + childNodePath + " is broken.");
      continue;
    }

    ByteArrayInputStream is = new ByteArrayInputStream(childData);
    DataInputStream fsIn = new DataInputStream(is);

    try {
      if (childNodeName.startsWith(DELEGATION_KEY_PREFIX)) {
        DelegationKey key = new DelegationKey();
        key.readFields(fsIn);
        rmState.rmSecretManagerState.masterKeyState.add(key);
        if (LOG.isDebugEnabled()) {
          LOG.debug("Loaded delegation key: keyId=" + key.getKeyId()
              + ", expirationDate=" + key.getExpiryDate());
        }
      }
    } finally {
      is.close();
    }
  }
}
 

import org.apache.hadoop.security.token.delegation.DelegationKey; //导入方法依赖的package包/类
/**
 * Process the Delegation Token related section in fsimage.
 * 
 * @param in DataInputStream to process
 * @param v Visitor to walk over records
 */
private void processDelegationTokens(DataInputStream in, ImageVisitor v)
    throws IOException {
  v.visit(ImageElement.CURRENT_DELEGATION_KEY_ID, in.readInt());
  int numDKeys = in.readInt();
  v.visitEnclosingElement(ImageElement.DELEGATION_KEYS,
      ImageElement.NUM_DELEGATION_KEYS, numDKeys);
  for(int i =0; i < numDKeys; i++) {
    DelegationKey key = new DelegationKey();
    key.readFields(in);
    v.visit(ImageElement.DELEGATION_KEY, key.toString());
  }
  v.leaveEnclosingElement();
  v.visit(ImageElement.DELEGATION_TOKEN_SEQUENCE_NUMBER, in.readInt());
  int numDTokens = in.readInt();
  v.visitEnclosingElement(ImageElement.DELEGATION_TOKENS,
      ImageElement.NUM_DELEGATION_TOKENS, numDTokens);
  for(int i=0; i<numDTokens; i++){
    DelegationTokenIdentifier id = new  DelegationTokenIdentifier();
    id.readFields(in);
    long expiryTime = in.readLong();
    v.visitEnclosingElement(ImageElement.DELEGATION_TOKEN_IDENTIFIER);
    v.visit(ImageElement.DELEGATION_TOKEN_IDENTIFIER_KIND,
        id.getKind().toString());
    v.visit(ImageElement.DELEGATION_TOKEN_IDENTIFIER_SEQNO,
        id.getSequenceNumber());
    v.visit(ImageElement.DELEGATION_TOKEN_IDENTIFIER_OWNER,
        id.getOwner().toString());
    v.visit(ImageElement.DELEGATION_TOKEN_IDENTIFIER_RENEWER,
        id.getRenewer().toString());
    v.visit(ImageElement.DELEGATION_TOKEN_IDENTIFIER_REALUSER,
        id.getRealUser().toString());
    v.visit(ImageElement.DELEGATION_TOKEN_IDENTIFIER_ISSUE_DATE,
        id.getIssueDate());
    v.visit(ImageElement.DELEGATION_TOKEN_IDENTIFIER_MAX_DATE,
        id.getMaxDate());
    v.visit(ImageElement.DELEGATION_TOKEN_IDENTIFIER_EXPIRY_TIME,
        expiryTime);
    v.visit(ImageElement.DELEGATION_TOKEN_IDENTIFIER_MASTER_KEY_ID,
        id.getMasterKeyId());
    v.leaveEnclosingElement(); // DELEGATION_TOKEN_IDENTIFIER
  }
  v.leaveEnclosingElement(); // DELEGATION_TOKENS
}