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

private void setupRequest() {
  GetDelegationTokenRequestProto getDelegationTokenRequestProto =
      GetDelegationTokenRequestProto.newBuilder()
      .setRenewer(identifier.getRenewer().toString())
      .build();

  originalRequest = OMRequest.newBuilder()
      .setClientId(UUID.randomUUID().toString())
      .setCmdType(Type.GetDelegationToken)
      .setGetDelegationTokenRequest(getDelegationTokenRequestProto)
      .build();

  omGetDelegationTokenRequest =
      new OMGetDelegationTokenRequest(originalRequest);

  modifiedRequest = null;
}
 

@Before
public void setupGetDelegationToken() {
  Text tester = new Text("tester");
  identifier = new OzoneTokenIdentifier(tester, tester, tester);
  identifier.setOmCertSerialId("certID");

  GetDelegationTokenRequestProto getDelegationTokenRequestProto =
      GetDelegationTokenRequestProto.newBuilder()
      .setRenewer(identifier.getRenewer().toString())
      .build();

  OMRequest omRequest = OMRequest.newBuilder()
      .setClientId(UUID.randomUUID().toString())
      .setCmdType(Type.GetDelegationToken)
      .setGetDelegationTokenRequest(getDelegationTokenRequestProto)
      .build();

  updateGetDelegationTokenRequest =
      new OMGetDelegationTokenRequest(omRequest)
          .getOmRequest()
          .getUpdateGetDelegationTokenRequest();
}
 

@Override
public GetDelegationTokenResponseProto getDelegationToken(
    RpcController controller, GetDelegationTokenRequestProto req)
    throws ServiceException {
  try {
    Token<DelegationTokenIdentifier> token = server
        .getDelegationToken(new Text(req.getRenewer()));
    GetDelegationTokenResponseProto.Builder rspBuilder = 
        GetDelegationTokenResponseProto.newBuilder();
    if (token != null) {
      rspBuilder.setToken(PBHelper.convert(token));
    }
    return rspBuilder.build();
  } catch (IOException e) {
    throw new ServiceException(e);
  }
}
 

@Override
public GetDelegationTokenResponseProto getDelegationToken(
    RpcController controller, GetDelegationTokenRequestProto req)
    throws ServiceException {
  try {
    Token<DelegationTokenIdentifier> token = server
        .getDelegationToken(new Text(req.getRenewer()));
    GetDelegationTokenResponseProto.Builder rspBuilder = 
        GetDelegationTokenResponseProto.newBuilder();
    if (token != null) {
      rspBuilder.setToken(PBHelper.convert(token));
    }
    return rspBuilder.build();
  } catch (IOException e) {
    throw new ServiceException(e);
  }
}
 

@Override
public Token<DelegationTokenIdentifier> getDelegationToken(Text renewer)
    throws IOException {
  GetDelegationTokenRequestProto req = GetDelegationTokenRequestProto
      .newBuilder()
      .setRenewer(renewer.toString())
      .build();
  try {
    GetDelegationTokenResponseProto resp = rpcProxy.getDelegationToken(null, req);
    return resp.hasToken() ? PBHelper.convertDelegationToken(resp.getToken())
        : null;
  } catch (ServiceException e) {
    throw ProtobufHelper.getRemoteException(e);
  }
}
 

/**
 * Get a valid Delegation Token.
 *
 * @param renewer the designated renewer for the token
 * @return Token<OzoneDelegationTokenSelector>
 * @throws OMException
 */
@Override
public Token<OzoneTokenIdentifier> getDelegationToken(Text renewer)
    throws OMException {
  GetDelegationTokenRequestProto req = GetDelegationTokenRequestProto
      .newBuilder()
      .setRenewer(renewer == null ? "" : renewer.toString())
      .build();

  OMRequest omRequest = createOMRequest(Type.GetDelegationToken)
      .setGetDelegationTokenRequest(req)
      .build();

  final GetDelegationTokenResponseProto resp;
  try {
    resp =
        handleError(submitRequest(omRequest)).getGetDelegationTokenResponse();
    return resp.getResponse().hasToken() ?
        OMPBHelper.convertToDelegationToken(resp.getResponse().getToken())
        : null;
  } catch (IOException e) {
    if(e instanceof OMException) {
      throw (OMException)e;
    }
    throw new OMException("Get delegation token failed.", e,
        TOKEN_ERROR_OTHER);
  }
}
 

@Override
public GetDelegationTokenResponse getDelegationToken(
    GetDelegationTokenRequest request) throws IOException {
  GetDelegationTokenRequestProto requestProto = ((GetDelegationTokenRequestPBImpl)
      request).getProto();
  try {
    return new GetDelegationTokenResponsePBImpl(proxy.getDelegationToken(
        null, requestProto));
  } catch (ServiceException e) {
    throw unwrapAndThrowException(e);
  }
}
 

@Override
public GetDelegationTokenResponseProto getDelegationToken(
    RpcController controller, GetDelegationTokenRequestProto proto)
    throws ServiceException {
  GetDelegationTokenRequest request = new GetDelegationTokenRequestPBImpl(proto);
  try {
    GetDelegationTokenResponse response = real.getDelegationToken(request);
    return ((GetDelegationTokenResponsePBImpl)response).getProto();
  } catch (IOException e) {
    throw new ServiceException(e);
  }
}
 

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

@Override
public GetDelegationTokenResponse getDelegationToken(
    GetDelegationTokenRequest request) throws YarnException,
    IOException {
  GetDelegationTokenRequestProto requestProto =
      ((GetDelegationTokenRequestPBImpl) request).getProto();
  try {
    return new GetDelegationTokenResponsePBImpl(proxy.getDelegationToken(
      null, requestProto));
  } catch (ServiceException e) {
    RPCUtil.unwrapAndThrowException(e);
    return null;
  }
}
 

@Override
public GetDelegationTokenResponse getDelegationToken(
    GetDelegationTokenRequest request) throws YarnException, IOException {
  GetDelegationTokenRequestProto requestProto =
      ((GetDelegationTokenRequestPBImpl) request).getProto();
  try {
    return new GetDelegationTokenResponsePBImpl(proxy.getDelegationToken(
      null, requestProto));
  } catch (ServiceException e) {
    RPCUtil.unwrapAndThrowException(e);
    return null;
  }
}
 

@Override
public Token<DelegationTokenIdentifier> getDelegationToken(Text renewer)
    throws IOException {
  GetDelegationTokenRequestProto req = GetDelegationTokenRequestProto
      .newBuilder()
      .setRenewer(renewer.toString())
      .build();
  try {
    GetDelegationTokenResponseProto resp = rpcProxy.getDelegationToken(null, req);
    return resp.hasToken() ? PBHelper.convertDelegationToken(resp.getToken())
        : null;
  } catch (ServiceException e) {
    throw ProtobufHelper.getRemoteException(e);
  }
}
 

@Override
public GetDelegationTokenResponse getDelegationToken(
    GetDelegationTokenRequest request) throws IOException {
  GetDelegationTokenRequestProto requestProto = ((GetDelegationTokenRequestPBImpl)
      request).getProto();
  try {
    return new GetDelegationTokenResponsePBImpl(proxy.getDelegationToken(
        null, requestProto));
  } catch (ServiceException e) {
    throw unwrapAndThrowException(e);
  }
}
 

@Override
public GetDelegationTokenResponseProto getDelegationToken(
    RpcController controller, GetDelegationTokenRequestProto proto)
    throws ServiceException {
  GetDelegationTokenRequest request = new GetDelegationTokenRequestPBImpl(proto);
  try {
    GetDelegationTokenResponse response = real.getDelegationToken(request);
    return ((GetDelegationTokenResponsePBImpl)response).getProto();
  } catch (IOException e) {
    throw new ServiceException(e);
  }
}
 

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

@Override
public GetDelegationTokenResponse getDelegationToken(
    GetDelegationTokenRequest request) throws YarnException, IOException {
  GetDelegationTokenRequestProto requestProto =
      ((GetDelegationTokenRequestPBImpl) request).getProto();
  try {
    return new GetDelegationTokenResponsePBImpl(proxy.getDelegationToken(
      null, requestProto));
  } catch (ServiceException e) {
    RPCUtil.unwrapAndThrowException(e);
    return null;
  }
}
 

@Override
public GetDelegationTokenResponse getDelegationToken(
    GetDelegationTokenRequest request) throws YarnException,
    IOException {
  GetDelegationTokenRequestProto requestProto =
      ((GetDelegationTokenRequestPBImpl) request).getProto();
  try {
    return new GetDelegationTokenResponsePBImpl(proxy.getDelegationToken(
      null, requestProto));
  } catch (ServiceException e) {
    RPCUtil.unwrapAndThrowException(e);
    return null;
  }
}
 

public GetDelegationTokenRequestPBImpl() {
  builder = GetDelegationTokenRequestProto.newBuilder();
}
 

public GetDelegationTokenRequestPBImpl() {
  builder = GetDelegationTokenRequestProto.newBuilder();
}
 

public GetDelegationTokenRequestPBImpl (
    GetDelegationTokenRequestProto proto) {
  this.proto = proto;
  viaProto = true;
}
 

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

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

public GetDelegationTokenRequestPBImpl (
    GetDelegationTokenRequestProto proto) {
  this.proto = proto;
  viaProto = true;
}
 

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

public GetDelegationTokenRequestPBImpl() {
  builder = GetDelegationTokenRequestProto.newBuilder();
}
 

public GetDelegationTokenRequestPBImpl (
    GetDelegationTokenRequestProto proto) {
  this.proto = proto;
  viaProto = true;
}
 

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

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

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

public GetDelegationTokenRequestPBImpl (
    GetDelegationTokenRequestProto proto) {
  this.proto = proto;
  viaProto = true;
}