org.apache.hadoop.security.proto.SecurityProtos.CancelDelegationTokenRequestProto Java Examples

/**
* Test CancelDelegationTokenRequestPBImpl.
* Test a transformation to prototype and back
*/
@Test
public void testCancelDelegationTokenRequestPBImpl() {

  Token token = getDelegationToken();

  CancelDelegationTokenRequestPBImpl original = new CancelDelegationTokenRequestPBImpl();
  original.setDelegationToken(token);
  CancelDelegationTokenRequestProto protoType = original.getProto();

  CancelDelegationTokenRequestPBImpl copy = new CancelDelegationTokenRequestPBImpl(protoType);
  assertNotNull(copy.getDelegationToken());
  //compare source and converted
  assertEquals(token, copy.getDelegationToken());

}
 

/**
* Test CancelDelegationTokenRequestPBImpl.
* Test a transformation to prototype and back
*/
@Test
public void testCancelDelegationTokenRequestPBImpl() {

  Token token = getDelegationToken();

  CancelDelegationTokenRequestPBImpl original = new CancelDelegationTokenRequestPBImpl();
  original.setDelegationToken(token);
  CancelDelegationTokenRequestProto protoType = original.getProto();

  CancelDelegationTokenRequestPBImpl copy = new CancelDelegationTokenRequestPBImpl(protoType);
  assertNotNull(copy.getDelegationToken());
  //compare source and converted
  assertEquals(token, copy.getDelegationToken());

}
 

/**
 * Cancel an existing delegation token.
 *
 * @param token delegation token
 */
@Override
public void cancelDelegationToken(Token<OzoneTokenIdentifier> token)
    throws OMException {
  CancelDelegationTokenRequestProto req = CancelDelegationTokenRequestProto
      .newBuilder()
      .setToken(OMPBHelper.convertToTokenProto(token))
      .build();

  OMRequest omRequest = createOMRequest(Type.CancelDelegationToken)
      .setCancelDelegationTokenRequest(req)
      .build();

  final CancelDelegationTokenResponseProto resp;
  try {
    handleError(submitRequest(omRequest));
  } catch (IOException e) {
    if(e instanceof OMException) {
      throw (OMException)e;
    }
    throw new OMException("Cancel delegation token failed.", e,
        TOKEN_ERROR_OTHER);
  }
}
 

@Override
public void cancelDelegationToken(Token<DelegationTokenIdentifier> token)
    throws IOException {
  CancelDelegationTokenRequestProto req = CancelDelegationTokenRequestProto
      .newBuilder()
      .setToken(PBHelper.convert(token))
      .build();
  try {
    rpcProxy.cancelDelegationToken(null, req);
  } catch (ServiceException e) {
    throw ProtobufHelper.getRemoteException(e);
  }
}
 

@Override
public void cancelDelegationToken(Token<DelegationTokenIdentifier> token)
    throws IOException {
  CancelDelegationTokenRequestProto req = CancelDelegationTokenRequestProto
      .newBuilder()
      .setToken(PBHelper.convert(token))
      .build();
  try {
    rpcProxy.cancelDelegationToken(null, req);
  } catch (ServiceException e) {
    throw ProtobufHelper.getRemoteException(e);
  }
}
 

@Override
public CancelDelegationTokenResponseProto cancelDelegationToken(
    RpcController controller, CancelDelegationTokenRequestProto req)
    throws ServiceException {
  try {
    server.cancelDelegationToken(PBHelper.convertDelegationToken(req
        .getToken()));
    return VOID_CANCELDELEGATIONTOKEN_RESPONSE;
  } catch (IOException e) {
    throw new ServiceException(e);
  }
}
 

@Override
public CancelDelegationTokenResponse cancelDelegationToken(
    CancelDelegationTokenRequest request) throws IOException {
  CancelDelegationTokenRequestProto requestProto =
      ((CancelDelegationTokenRequestPBImpl) request).getProto();
  try {
    return new CancelDelegationTokenResponsePBImpl(
        proxy.cancelDelegationToken(null, requestProto));

  } catch (ServiceException e) {
    throw unwrapAndThrowException(e);
  }
}
 

@Override
public CancelDelegationTokenResponseProto cancelDelegationToken(
    RpcController controller, CancelDelegationTokenRequestProto proto)
    throws ServiceException {
  CancelDelegationTokenRequestPBImpl request =
      new CancelDelegationTokenRequestPBImpl(proto);
    try {
      CancelDelegationTokenResponse response = real.cancelDelegationToken(request);
      return ((CancelDelegationTokenResponsePBImpl)response).getProto();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
}
 

@Override
public CancelDelegationTokenRequestProto getProto() {
  mergeLocalToProto();
  proto = viaProto ? proto : builder.build();
  viaProto = true;
  return proto;
}
 

@Override
public CancelDelegationTokenResponse cancelDelegationToken(
    CancelDelegationTokenRequest request) throws YarnException,
    IOException {
  CancelDelegationTokenRequestProto requestProto =
      ((CancelDelegationTokenRequestPBImpl) request).getProto();
  try {
    return new CancelDelegationTokenResponsePBImpl(
        proxy.cancelDelegationToken(null, requestProto));

  } catch (ServiceException e) {
    RPCUtil.unwrapAndThrowException(e);
    return null;
  }
}
 

@Override
public CancelDelegationTokenResponse cancelDelegationToken(
    CancelDelegationTokenRequest request) throws YarnException, IOException {
  CancelDelegationTokenRequestProto requestProto =
      ((CancelDelegationTokenRequestPBImpl) request).getProto();
  try {
    return new CancelDelegationTokenResponsePBImpl(
      proxy.cancelDelegationToken(null, requestProto));

  } catch (ServiceException e) {
    RPCUtil.unwrapAndThrowException(e);
    return null;
  }
}
 

@Override
public CancelDelegationTokenResponseProto cancelDelegationToken(
    RpcController controller, CancelDelegationTokenRequestProto req)
    throws ServiceException {
  try {
    server.cancelDelegationToken(PBHelper.convertDelegationToken(req
        .getToken()));
    return VOID_CANCELDELEGATIONTOKEN_RESPONSE;
  } catch (IOException e) {
    throw new ServiceException(e);
  }
}
 

@Override
public CancelDelegationTokenResponse cancelDelegationToken(
    CancelDelegationTokenRequest request) throws IOException {
  CancelDelegationTokenRequestProto requestProto =
      ((CancelDelegationTokenRequestPBImpl) request).getProto();
  try {
    return new CancelDelegationTokenResponsePBImpl(
        proxy.cancelDelegationToken(null, requestProto));

  } catch (ServiceException e) {
    throw unwrapAndThrowException(e);
  }
}
 

@Override
public CancelDelegationTokenResponse cancelDelegationToken(
    CancelDelegationTokenRequest request) throws YarnException,
    IOException {
  CancelDelegationTokenRequestProto requestProto =
      ((CancelDelegationTokenRequestPBImpl) request).getProto();
  try {
    return new CancelDelegationTokenResponsePBImpl(
        proxy.cancelDelegationToken(null, requestProto));

  } catch (ServiceException e) {
    RPCUtil.unwrapAndThrowException(e);
    return null;
  }
}
 

public Token<OzoneTokenIdentifier> getToken() {
  CancelDelegationTokenRequestProto cancelDelegationTokenRequest =
      getOmRequest().getCancelDelegationTokenRequest();

  return OMPBHelper.convertToDelegationToken(
      cancelDelegationTokenRequest.getToken());
}
 

@Override
public CancelDelegationTokenResponse cancelDelegationToken(
    CancelDelegationTokenRequest request) throws YarnException, IOException {
  CancelDelegationTokenRequestProto requestProto =
      ((CancelDelegationTokenRequestPBImpl) request).getProto();
  try {
    return new CancelDelegationTokenResponsePBImpl(
      proxy.cancelDelegationToken(null, requestProto));

  } catch (ServiceException e) {
    RPCUtil.unwrapAndThrowException(e);
    return null;
  }
}
 

@Override
public CancelDelegationTokenResponseProto cancelDelegationToken(
    RpcController controller, CancelDelegationTokenRequestProto proto)
    throws ServiceException {
  CancelDelegationTokenRequestPBImpl request =
      new CancelDelegationTokenRequestPBImpl(proto);
    try {
      CancelDelegationTokenResponse response = real.cancelDelegationToken(request);
      return ((CancelDelegationTokenResponsePBImpl)response).getProto();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
}
 

@Override
public CancelDelegationTokenRequestProto getProto() {
  mergeLocalToProto();
  proto = viaProto ? proto : builder.build();
  viaProto = true;
  return proto;
}
 

public CancelDelegationTokenRequestPBImpl() {
  builder = CancelDelegationTokenRequestProto.newBuilder();
}
 

public CancelDelegationTokenRequestPBImpl(
    CancelDelegationTokenRequestProto proto) {
  this.proto = proto;
  viaProto = true;
}
 

public CancelDelegationTokenRequestProto getProto() {
  mergeLocalToProto();
  proto = viaProto ? proto : builder.build();
  viaProto = true;
  return proto;
}
 

private void maybeInitBuilder() {
  if (viaProto || builder == null) {
    builder = CancelDelegationTokenRequestProto.newBuilder(proto);
  }
  viaProto = false;
}
 

public CancelDelegationTokenRequestPBImpl() {
  this.builder = CancelDelegationTokenRequestProto.newBuilder();
}
 

public CancelDelegationTokenRequestPBImpl (
    CancelDelegationTokenRequestProto proto) {
  this.proto = proto;
  this.viaProto = true;
}
 

public CancelDelegationTokenRequestPBImpl() {
  this.builder = CancelDelegationTokenRequestProto.newBuilder();
}
 

public CancelDelegationTokenRequestPBImpl (
    CancelDelegationTokenRequestProto proto) {
  this.proto = proto;
  this.viaProto = true;
}
 

private void maybeInitBuilder() {
  if (viaProto || builder == null) {
    builder = CancelDelegationTokenRequestProto.newBuilder(proto);
  }
  viaProto = false;
}
 

private void maybeInitBuilder() {
  if (viaProto || builder == null) {
    builder = CancelDelegationTokenRequestProto.newBuilder(proto);
  }
  viaProto = false;
}
 

public CancelDelegationTokenRequestProto getProto() {
  mergeLocalToProto();
  proto = viaProto ? proto : builder.build();
  viaProto = true;
  return proto;
}
 

public CancelDelegationTokenRequestPBImpl(
    CancelDelegationTokenRequestProto proto) {
  this.proto = proto;
  viaProto = true;
}