コード例 #1
0
  @Override
  public CreateResponseProto create(RpcController controller, CreateRequestProto req)
      throws ServiceException {
    try {
      HdfsFileStatus result = null;
      if (req.hasPolicy()) {
        result =
            server.create(
                req.getSrc(),
                PBHelper.convert(req.getMasked()),
                req.getClientName(),
                PBHelper.convert(req.getCreateFlag()),
                req.getCreateParent(),
                (short) req.getReplication(),
                req.getBlockSize(),
                PBHelper.convert(req.getPolicy()));
      } else {
        result =
            server.create(
                req.getSrc(),
                PBHelper.convert(req.getMasked()),
                req.getClientName(),
                PBHelper.convert(req.getCreateFlag()),
                req.getCreateParent(),
                (short) req.getReplication(),
                req.getBlockSize());
      }

      if (result != null) {
        return CreateResponseProto.newBuilder().setFs(PBHelper.convert(result)).build();
      }
    } catch (IOException e) {
      throw new ServiceException(e);
    }
    return VOID_CREATE_RESPONSE;
  }
/**
 * This class is used on the server side. Calls come across the wire for the for protocol {@link
 * ClientNamenodeProtocolPB}. This class translates the PB data types to the native data types used
 * inside the NN as specified in the generic ClientProtocol.
 */
@InterfaceAudience.Private
@InterfaceStability.Stable
public class ClientNamenodeProtocolServerSideTranslatorPB implements ClientNamenodeProtocolPB {
  private final ClientProtocol server;

  /**
   * Constructor
   *
   * @param server - the NN server
   * @throws IOException
   */
  public ClientNamenodeProtocolServerSideTranslatorPB(ClientProtocol server) throws IOException {
    this.server = server;
  }

  @Override
  public GetBlockLocationsResponseProto getBlockLocations(
      RpcController controller, GetBlockLocationsRequestProto req) throws ServiceException {
    try {
      LocatedBlocks b = server.getBlockLocations(req.getSrc(), req.getOffset(), req.getLength());
      Builder builder = GetBlockLocationsResponseProto.newBuilder();
      if (b != null) {
        builder.setLocations(PBHelper.convert(b)).build();
      }
      return builder.build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public GetServerDefaultsResponseProto getServerDefaults(
      RpcController controller, GetServerDefaultsRequestProto req) throws ServiceException {
    try {
      FsServerDefaults result = server.getServerDefaults();
      return GetServerDefaultsResponseProto.newBuilder()
          .setServerDefaults(PBHelper.convert(result))
          .build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  static final CreateResponseProto VOID_CREATE_RESPONSE = CreateResponseProto.newBuilder().build();

  @Override
  public CreateResponseProto create(RpcController controller, CreateRequestProto req)
      throws ServiceException {
    try {
      server.create(
          req.getSrc(),
          PBHelper.convert(req.getMasked()),
          req.getClientName(),
          PBHelper.convert(req.getCreateFlag()),
          req.getCreateParent(),
          (short) req.getReplication(),
          req.getBlockSize());
    } catch (IOException e) {
      throw new ServiceException(e);
    }
    return VOID_CREATE_RESPONSE;
  }

  static final AppendResponseProto NULL_APPEND_RESPONSE = AppendResponseProto.newBuilder().build();

  @Override
  public AppendResponseProto append(RpcController controller, AppendRequestProto req)
      throws ServiceException {
    try {
      LocatedBlock result = server.append(req.getSrc(), req.getClientName());
      if (result != null) {
        return AppendResponseProto.newBuilder().setBlock(PBHelper.convert(result)).build();
      }
      return NULL_APPEND_RESPONSE;
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public SetReplicationResponseProto setReplication(
      RpcController controller, SetReplicationRequestProto req) throws ServiceException {
    try {
      boolean result = server.setReplication(req.getSrc(), (short) req.getReplication());
      return SetReplicationResponseProto.newBuilder().setResult(result).build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  static final SetPermissionResponseProto VOID_SET_PERM_RESPONSE =
      SetPermissionResponseProto.newBuilder().build();

  @Override
  public SetPermissionResponseProto setPermission(
      RpcController controller, SetPermissionRequestProto req) throws ServiceException {
    try {
      server.setPermission(req.getSrc(), PBHelper.convert(req.getPermission()));
    } catch (IOException e) {
      throw new ServiceException(e);
    }
    return VOID_SET_PERM_RESPONSE;
  }

  static final SetOwnerResponseProto VOID_SET_OWNER_RESPONSE =
      SetOwnerResponseProto.newBuilder().build();

  @Override
  public SetOwnerResponseProto setOwner(RpcController controller, SetOwnerRequestProto req)
      throws ServiceException {
    try {
      server.setOwner(
          req.getSrc(),
          req.hasUsername() ? req.getUsername() : null,
          req.hasGroupname() ? req.getGroupname() : null);
    } catch (IOException e) {
      throw new ServiceException(e);
    }
    return VOID_SET_OWNER_RESPONSE;
  }

  static final AbandonBlockResponseProto VOID_ADD_BLOCK_RESPONSE =
      AbandonBlockResponseProto.newBuilder().build();

  @Override
  public AbandonBlockResponseProto abandonBlock(
      RpcController controller, AbandonBlockRequestProto req) throws ServiceException {
    try {
      server.abandonBlock(PBHelper.convert(req.getB()), req.getSrc(), req.getHolder());
    } catch (IOException e) {
      throw new ServiceException(e);
    }
    return VOID_ADD_BLOCK_RESPONSE;
  }

  @Override
  public AddBlockResponseProto addBlock(RpcController controller, AddBlockRequestProto req)
      throws ServiceException {

    try {
      List<DatanodeInfoProto> excl = req.getExcludeNodesList();
      LocatedBlock result =
          server.addBlock(
              req.getSrc(),
              req.getClientName(),
              req.hasPrevious() ? PBHelper.convert(req.getPrevious()) : null,
              (excl == null || excl.size() == 0)
                  ? null
                  : PBHelper.convert(excl.toArray(new DatanodeInfoProto[excl.size()])));
      return AddBlockResponseProto.newBuilder().setBlock(PBHelper.convert(result)).build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public GetAdditionalDatanodeResponseProto getAdditionalDatanode(
      RpcController controller, GetAdditionalDatanodeRequestProto req) throws ServiceException {
    try {
      List<DatanodeInfoProto> existingList = req.getExistingsList();
      List<DatanodeInfoProto> excludesList = req.getExcludesList();
      LocatedBlock result =
          server.getAdditionalDatanode(
              req.getSrc(),
              PBHelper.convert(req.getBlk()),
              PBHelper.convert(existingList.toArray(new DatanodeInfoProto[existingList.size()])),
              PBHelper.convert(excludesList.toArray(new DatanodeInfoProto[excludesList.size()])),
              req.getNumAdditionalNodes(),
              req.getClientName());
      return GetAdditionalDatanodeResponseProto.newBuilder()
          .setBlock(PBHelper.convert(result))
          .build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public CompleteResponseProto complete(RpcController controller, CompleteRequestProto req)
      throws ServiceException {
    try {
      boolean result =
          server.complete(
              req.getSrc(),
              req.getClientName(),
              req.hasLast() ? PBHelper.convert(req.getLast()) : null);
      return CompleteResponseProto.newBuilder().setResult(result).build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  static final ReportBadBlocksResponseProto VOID_REP_BAD_BLOCK_RESPONSE =
      ReportBadBlocksResponseProto.newBuilder().build();

  @Override
  public ReportBadBlocksResponseProto reportBadBlocks(
      RpcController controller, ReportBadBlocksRequestProto req) throws ServiceException {
    try {
      List<LocatedBlockProto> bl = req.getBlocksList();
      server.reportBadBlocks(
          PBHelper.convertLocatedBlock(bl.toArray(new LocatedBlockProto[bl.size()])));
    } catch (IOException e) {
      throw new ServiceException(e);
    }
    return VOID_REP_BAD_BLOCK_RESPONSE;
  }

  static final ConcatResponseProto VOID_CONCAT_RESPONSE = ConcatResponseProto.newBuilder().build();

  @Override
  public ConcatResponseProto concat(RpcController controller, ConcatRequestProto req)
      throws ServiceException {
    try {
      List<String> srcs = req.getSrcsList();
      server.concat(req.getTrg(), srcs.toArray(new String[srcs.size()]));
    } catch (IOException e) {
      throw new ServiceException(e);
    }
    return VOID_CONCAT_RESPONSE;
  }

  @Override
  public RenameResponseProto rename(RpcController controller, RenameRequestProto req)
      throws ServiceException {
    try {
      boolean result = server.rename(req.getSrc(), req.getDst());
      return RenameResponseProto.newBuilder().setResult(result).build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  static final Rename2ResponseProto VOID_RENAME2_RESPONSE =
      Rename2ResponseProto.newBuilder().build();

  @Override
  public Rename2ResponseProto rename2(RpcController controller, Rename2RequestProto req)
      throws ServiceException {

    try {
      server.rename2(
          req.getSrc(), req.getDst(), req.getOverwriteDest() ? Rename.OVERWRITE : Rename.NONE);
    } catch (IOException e) {
      throw new ServiceException(e);
    }
    return VOID_RENAME2_RESPONSE;
  }

  @Override
  public DeleteResponseProto delete(RpcController controller, DeleteRequestProto req)
      throws ServiceException {
    try {
      boolean result = server.delete(req.getSrc(), req.getRecursive());
      return DeleteResponseProto.newBuilder().setResult(result).build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public MkdirsResponseProto mkdirs(RpcController controller, MkdirsRequestProto req)
      throws ServiceException {
    try {
      boolean result =
          server.mkdirs(req.getSrc(), PBHelper.convert(req.getMasked()), req.getCreateParent());
      return MkdirsResponseProto.newBuilder().setResult(result).build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  static final GetListingResponseProto NULL_GETLISTING_RESPONSE =
      GetListingResponseProto.newBuilder().build();

  @Override
  public GetListingResponseProto getListing(RpcController controller, GetListingRequestProto req)
      throws ServiceException {
    try {
      DirectoryListing result =
          server.getListing(req.getSrc(), req.getStartAfter().toByteArray(), req.getNeedLocation());
      if (result != null) {
        return GetListingResponseProto.newBuilder().setDirList(PBHelper.convert(result)).build();
      } else {
        return NULL_GETLISTING_RESPONSE;
      }
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  static final RenewLeaseResponseProto VOID_RENEWLEASE_RESPONSE =
      RenewLeaseResponseProto.newBuilder().build();

  @Override
  public RenewLeaseResponseProto renewLease(RpcController controller, RenewLeaseRequestProto req)
      throws ServiceException {
    try {
      server.renewLease(req.getClientName());
      return VOID_RENEWLEASE_RESPONSE;
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public RecoverLeaseResponseProto recoverLease(
      RpcController controller, RecoverLeaseRequestProto req) throws ServiceException {
    try {
      boolean result = server.recoverLease(req.getSrc(), req.getClientName());
      return RecoverLeaseResponseProto.newBuilder().setResult(result).build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public RestoreFailedStorageResponseProto restoreFailedStorage(
      RpcController controller, RestoreFailedStorageRequestProto req) throws ServiceException {
    try {
      boolean result = server.restoreFailedStorage(req.getArg());
      return RestoreFailedStorageResponseProto.newBuilder().setResult(result).build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public GetFsStatsResponseProto getFsStats(RpcController controller, GetFsStatusRequestProto req)
      throws ServiceException {
    try {
      return PBHelper.convert(server.getStats());
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public GetDatanodeReportResponseProto getDatanodeReport(
      RpcController controller, GetDatanodeReportRequestProto req) throws ServiceException {
    try {
      DatanodeInfoProto[] result =
          PBHelper.convert(server.getDatanodeReport(PBHelper.convert(req.getType())));
      return GetDatanodeReportResponseProto.newBuilder().addAllDi(Arrays.asList(result)).build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public GetPreferredBlockSizeResponseProto getPreferredBlockSize(
      RpcController controller, GetPreferredBlockSizeRequestProto req) throws ServiceException {
    try {
      long result = server.getPreferredBlockSize(req.getFilename());
      return GetPreferredBlockSizeResponseProto.newBuilder().setBsize(result).build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public SetSafeModeResponseProto setSafeMode(RpcController controller, SetSafeModeRequestProto req)
      throws ServiceException {
    try {
      boolean result = server.setSafeMode(PBHelper.convert(req.getAction()));
      return SetSafeModeResponseProto.newBuilder().setResult(result).build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  static final SaveNamespaceResponseProto VOID_SAVENAMESPACE_RESPONSE =
      SaveNamespaceResponseProto.newBuilder().build();

  @Override
  public SaveNamespaceResponseProto saveNamespace(
      RpcController controller, SaveNamespaceRequestProto req) throws ServiceException {
    try {
      server.saveNamespace();
      return VOID_SAVENAMESPACE_RESPONSE;
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  static final RefreshNodesResponseProto VOID_REFRESHNODES_RESPONSE =
      RefreshNodesResponseProto.newBuilder().build();

  @Override
  public RefreshNodesResponseProto refreshNodes(
      RpcController controller, RefreshNodesRequestProto req) throws ServiceException {
    try {
      server.refreshNodes();
      return VOID_REFRESHNODES_RESPONSE;
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  static final FinalizeUpgradeResponseProto VOID_FINALIZEUPGRADE_RESPONSE =
      FinalizeUpgradeResponseProto.newBuilder().build();

  @Override
  public FinalizeUpgradeResponseProto finalizeUpgrade(
      RpcController controller, FinalizeUpgradeRequestProto req) throws ServiceException {
    try {
      server.finalizeUpgrade();
      return VOID_FINALIZEUPGRADE_RESPONSE;
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public DistributedUpgradeProgressResponseProto distributedUpgradeProgress(
      RpcController controller, DistributedUpgradeProgressRequestProto req)
      throws ServiceException {
    try {
      UpgradeStatusReport result =
          server.distributedUpgradeProgress(PBHelper.convert(req.getAction()));
      DistributedUpgradeProgressResponseProto.Builder builder =
          DistributedUpgradeProgressResponseProto.newBuilder();
      if (result != null) {
        builder.setReport(PBHelper.convert(result));
      }
      return builder.build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public ListCorruptFileBlocksResponseProto listCorruptFileBlocks(
      RpcController controller, ListCorruptFileBlocksRequestProto req) throws ServiceException {
    try {
      CorruptFileBlocks result =
          server.listCorruptFileBlocks(req.getPath(), req.hasCookie() ? req.getCookie() : null);
      return ListCorruptFileBlocksResponseProto.newBuilder()
          .setCorrupt(PBHelper.convert(result))
          .build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  static final MetaSaveResponseProto VOID_METASAVE_RESPONSE =
      MetaSaveResponseProto.newBuilder().build();

  @Override
  public MetaSaveResponseProto metaSave(RpcController controller, MetaSaveRequestProto req)
      throws ServiceException {
    try {
      server.metaSave(req.getFilename());
      return VOID_METASAVE_RESPONSE;
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  static final GetFileInfoResponseProto NULL_GETFILEINFO_RESPONSE =
      GetFileInfoResponseProto.newBuilder().build();

  @Override
  public GetFileInfoResponseProto getFileInfo(RpcController controller, GetFileInfoRequestProto req)
      throws ServiceException {
    try {
      HdfsFileStatus result = server.getFileInfo(req.getSrc());

      if (result != null) {
        return GetFileInfoResponseProto.newBuilder().setFs(PBHelper.convert(result)).build();
      }
      return NULL_GETFILEINFO_RESPONSE;
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  static final GetFileLinkInfoResponseProto NULL_GETFILELINKINFO_RESPONSE =
      GetFileLinkInfoResponseProto.newBuilder().build();

  @Override
  public GetFileLinkInfoResponseProto getFileLinkInfo(
      RpcController controller, GetFileLinkInfoRequestProto req) throws ServiceException {
    try {
      HdfsFileStatus result = server.getFileLinkInfo(req.getSrc());
      if (result != null) {
        System.out.println("got non null result for getFileLinkInfo for " + req.getSrc());
        return GetFileLinkInfoResponseProto.newBuilder().setFs(PBHelper.convert(result)).build();
      } else {
        System.out.println("got  null result for getFileLinkInfo for " + req.getSrc());
        return NULL_GETFILELINKINFO_RESPONSE;
      }

    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public GetContentSummaryResponseProto getContentSummary(
      RpcController controller, GetContentSummaryRequestProto req) throws ServiceException {
    try {
      ContentSummary result = server.getContentSummary(req.getPath());
      return GetContentSummaryResponseProto.newBuilder()
          .setSummary(PBHelper.convert(result))
          .build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  static final SetQuotaResponseProto VOID_SETQUOTA_RESPONSE =
      SetQuotaResponseProto.newBuilder().build();

  @Override
  public SetQuotaResponseProto setQuota(RpcController controller, SetQuotaRequestProto req)
      throws ServiceException {
    try {
      server.setQuota(req.getPath(), req.getNamespaceQuota(), req.getDiskspaceQuota());
      return VOID_SETQUOTA_RESPONSE;
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  static final FsyncResponseProto VOID_FSYNC_RESPONSE = FsyncResponseProto.newBuilder().build();

  @Override
  public FsyncResponseProto fsync(RpcController controller, FsyncRequestProto req)
      throws ServiceException {
    try {
      server.fsync(req.getSrc(), req.getClient());
      return VOID_FSYNC_RESPONSE;
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  static final SetTimesResponseProto VOID_SETTIMES_RESPONSE =
      SetTimesResponseProto.newBuilder().build();

  @Override
  public SetTimesResponseProto setTimes(RpcController controller, SetTimesRequestProto req)
      throws ServiceException {
    try {
      server.setTimes(req.getSrc(), req.getMtime(), req.getAtime());
      return VOID_SETTIMES_RESPONSE;
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  static final CreateSymlinkResponseProto VOID_CREATESYMLINK_RESPONSE =
      CreateSymlinkResponseProto.newBuilder().build();

  @Override
  public CreateSymlinkResponseProto createSymlink(
      RpcController controller, CreateSymlinkRequestProto req) throws ServiceException {
    try {
      server.createSymlink(
          req.getTarget(),
          req.getLink(),
          PBHelper.convert(req.getDirPerm()),
          req.getCreateParent());
      return VOID_CREATESYMLINK_RESPONSE;
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public GetLinkTargetResponseProto getLinkTarget(
      RpcController controller, GetLinkTargetRequestProto req) throws ServiceException {
    try {
      String result = server.getLinkTarget(req.getPath());
      return GetLinkTargetResponseProto.newBuilder().setTargetPath(result).build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public UpdateBlockForPipelineResponseProto updateBlockForPipeline(
      RpcController controller, UpdateBlockForPipelineRequestProto req) throws ServiceException {
    try {
      LocatedBlockProto result =
          PBHelper.convert(
              server.updateBlockForPipeline(PBHelper.convert(req.getBlock()), req.getClientName()));
      return UpdateBlockForPipelineResponseProto.newBuilder().setBlock(result).build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  static final UpdatePipelineResponseProto VOID_UPDATEPIPELINE_RESPONSE =
      UpdatePipelineResponseProto.newBuilder().build();

  @Override
  public UpdatePipelineResponseProto updatePipeline(
      RpcController controller, UpdatePipelineRequestProto req) throws ServiceException {
    try {
      List<DatanodeIDProto> newNodes = req.getNewNodesList();
      server.updatePipeline(
          req.getClientName(),
          PBHelper.convert(req.getOldBlock()),
          PBHelper.convert(req.getNewBlock()),
          PBHelper.convert(newNodes.toArray(new DatanodeIDProto[newNodes.size()])));
      return VOID_UPDATEPIPELINE_RESPONSE;
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public GetDelegationTokenResponseProto getDelegationToken(
      RpcController controller, GetDelegationTokenRequestProto req) throws ServiceException {
    try {
      BlockTokenIdentifierProto result =
          PBHelper.convert(server.getDelegationToken(new Text(req.getRenewer())));
      return GetDelegationTokenResponseProto.newBuilder().setToken(result).build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public RenewDelegationTokenResponseProto renewDelegationToken(
      RpcController controller, RenewDelegationTokenRequestProto req) throws ServiceException {
    try {
      long result = server.renewDelegationToken(PBHelper.convertDelegationToken(req.getToken()));
      return RenewDelegationTokenResponseProto.newBuilder().setNewExireTime(result).build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  static final CancelDelegationTokenResponseProto VOID_CANCELDELEGATIONTOKEN_RESPONSE =
      CancelDelegationTokenResponseProto.newBuilder().build();

  @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);
    }
  }

  static final SetBalancerBandwidthResponseProto VOID_SETBALANCERBANDWIDTH_RESPONSE =
      SetBalancerBandwidthResponseProto.newBuilder().build();

  @Override
  public SetBalancerBandwidthResponseProto setBalancerBandwidth(
      RpcController controller, SetBalancerBandwidthRequestProto req) throws ServiceException {
    try {
      server.setBalancerBandwidth(req.getBandwidth());
      return VOID_SETBALANCERBANDWIDTH_RESPONSE;
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public GetDataEncryptionKeyResponseProto getDataEncryptionKey(
      RpcController controller, GetDataEncryptionKeyRequestProto request) throws ServiceException {
    try {
      DataEncryptionKey encryptionKey = server.getDataEncryptionKey();
      return GetDataEncryptionKeyResponseProto.newBuilder()
          .setDataEncryptionKey(PBHelper.convert(encryptionKey))
          .build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }
}
コード例 #3
0
/**
 * This class is used on the server side. Calls come across the wire for the for protocol {@link
 * ClientNamenodeProtocolPB}. This class translates the PB data types to the native data types used
 * inside the NN as specified in the generic ClientProtocol.
 */
@InterfaceAudience.Private
@InterfaceStability.Stable
public class ClientNamenodeProtocolServerSideTranslatorPB implements ClientNamenodeProtocolPB {
  private final ClientProtocol server;

  private static final CreateResponseProto VOID_CREATE_RESPONSE =
      CreateResponseProto.newBuilder().build();

  private static final AppendResponseProto VOID_APPEND_RESPONSE =
      AppendResponseProto.newBuilder().build();

  private static final SetPermissionResponseProto VOID_SET_PERM_RESPONSE =
      SetPermissionResponseProto.newBuilder().build();

  private static final SetOwnerResponseProto VOID_SET_OWNER_RESPONSE =
      SetOwnerResponseProto.newBuilder().build();

  private static final AbandonBlockResponseProto VOID_ADD_BLOCK_RESPONSE =
      AbandonBlockResponseProto.newBuilder().build();

  private static final ReportBadBlocksResponseProto VOID_REP_BAD_BLOCK_RESPONSE =
      ReportBadBlocksResponseProto.newBuilder().build();

  private static final ConcatResponseProto VOID_CONCAT_RESPONSE =
      ConcatResponseProto.newBuilder().build();

  private static final Rename2ResponseProto VOID_RENAME2_RESPONSE =
      Rename2ResponseProto.newBuilder().build();

  private static final GetListingResponseProto VOID_GETLISTING_RESPONSE =
      GetListingResponseProto.newBuilder().build();

  private static final RenewLeaseResponseProto VOID_RENEWLEASE_RESPONSE =
      RenewLeaseResponseProto.newBuilder().build();

  private static final RefreshNodesResponseProto VOID_REFRESHNODES_RESPONSE =
      RefreshNodesResponseProto.newBuilder().build();

  private static final GetFileInfoResponseProto VOID_GETFILEINFO_RESPONSE =
      GetFileInfoResponseProto.newBuilder().build();

  private static final GetFileLinkInfoResponseProto VOID_GETFILELINKINFO_RESPONSE =
      GetFileLinkInfoResponseProto.newBuilder().build();

  private static final SetQuotaResponseProto VOID_SETQUOTA_RESPONSE =
      SetQuotaResponseProto.newBuilder().build();

  private static final FsyncResponseProto VOID_FSYNC_RESPONSE =
      FsyncResponseProto.newBuilder().build();

  private static final SetTimesResponseProto VOID_SETTIMES_RESPONSE =
      SetTimesResponseProto.newBuilder().build();

  private static final CreateSymlinkResponseProto VOID_CREATESYMLINK_RESPONSE =
      CreateSymlinkResponseProto.newBuilder().build();

  private static final ClientNamenodeProtocolProtos.AddBlockChecksumResponseProto
      VOID_ADDBLOCKCHECKSUM_RESPONSE =
          ClientNamenodeProtocolProtos.AddBlockChecksumResponseProto.newBuilder().build();

  private static final UpdatePipelineResponseProto VOID_UPDATEPIPELINE_RESPONSE =
      UpdatePipelineResponseProto.newBuilder().build();

  private static final CancelDelegationTokenResponseProto VOID_CANCELDELEGATIONTOKEN_RESPONSE =
      CancelDelegationTokenResponseProto.newBuilder().build();

  private static final SetBalancerBandwidthResponseProto VOID_SETBALANCERBANDWIDTH_RESPONSE =
      SetBalancerBandwidthResponseProto.newBuilder().build();

  private static final ClientNamenodeProtocolProtos.ChangeConfResponseProto
      VOID_CHANGECONF_RESPONSE =
          ClientNamenodeProtocolProtos.ChangeConfResponseProto.newBuilder().build();

  private static final ClientNamenodeProtocolProtos.SetMetaEnabledResponseProto
      VOID_SET_META_ENABLED_RESPONSE =
          ClientNamenodeProtocolProtos.SetMetaEnabledResponseProto.newBuilder().build();

  /**
   * Constructor
   *
   * @param server - the NN server
   * @throws IOException
   */
  public ClientNamenodeProtocolServerSideTranslatorPB(ClientProtocol server) throws IOException {
    this.server = server;
  }

  @Override
  public GetBlockLocationsResponseProto getBlockLocations(
      RpcController controller, GetBlockLocationsRequestProto req) throws ServiceException {
    try {
      LocatedBlocks b = server.getBlockLocations(req.getSrc(), req.getOffset(), req.getLength());
      Builder builder = GetBlockLocationsResponseProto.newBuilder();
      if (b != null) {
        builder.setLocations(PBHelper.convert(b)).build();
      }
      return builder.build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public ClientNamenodeProtocolProtos.GetMissingBlockLocationsResponseProto
      getMissingBlockLocations(
          RpcController controller,
          ClientNamenodeProtocolProtos.GetMissingBlockLocationsRequestProto req)
          throws ServiceException {
    try {
      LocatedBlocks b = server.getMissingBlockLocations(req.getFilePath());
      ClientNamenodeProtocolProtos.GetMissingBlockLocationsResponseProto.Builder builder =
          ClientNamenodeProtocolProtos.GetMissingBlockLocationsResponseProto.newBuilder();
      if (b != null) {
        builder.setLocations(PBHelper.convert(b)).build();
      }
      return builder.build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public ClientNamenodeProtocolProtos.AddBlockChecksumResponseProto addBlockChecksum(
      RpcController controller, ClientNamenodeProtocolProtos.AddBlockChecksumRequestProto req)
      throws ServiceException {
    try {
      server.addBlockChecksum(req.getSrc(), req.getBlockIndex(), req.getChecksum());
    } catch (IOException e) {
      throw new ServiceException(e);
    }
    return VOID_ADDBLOCKCHECKSUM_RESPONSE;
  }

  @Override
  public ClientNamenodeProtocolProtos.GetBlockChecksumResponseProto getBlockChecksum(
      RpcController controller, ClientNamenodeProtocolProtos.GetBlockChecksumRequestProto req)
      throws ServiceException {
    try {
      long checksum = server.getBlockChecksum(req.getSrc(), req.getBlockIndex());
      ClientNamenodeProtocolProtos.GetBlockChecksumResponseProto.Builder builder =
          ClientNamenodeProtocolProtos.GetBlockChecksumResponseProto.newBuilder()
              .setChecksum(checksum);
      return builder.build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public GetServerDefaultsResponseProto getServerDefaults(
      RpcController controller, GetServerDefaultsRequestProto req) throws ServiceException {
    try {
      FsServerDefaults result = server.getServerDefaults();
      return GetServerDefaultsResponseProto.newBuilder()
          .setServerDefaults(PBHelper.convert(result))
          .build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public CreateResponseProto create(RpcController controller, CreateRequestProto req)
      throws ServiceException {
    try {
      HdfsFileStatus result = null;
      if (req.hasPolicy()) {
        result =
            server.create(
                req.getSrc(),
                PBHelper.convert(req.getMasked()),
                req.getClientName(),
                PBHelper.convert(req.getCreateFlag()),
                req.getCreateParent(),
                (short) req.getReplication(),
                req.getBlockSize(),
                PBHelper.convert(req.getPolicy()));
      } else {
        result =
            server.create(
                req.getSrc(),
                PBHelper.convert(req.getMasked()),
                req.getClientName(),
                PBHelper.convert(req.getCreateFlag()),
                req.getCreateParent(),
                (short) req.getReplication(),
                req.getBlockSize());
      }

      if (result != null) {
        return CreateResponseProto.newBuilder().setFs(PBHelper.convert(result)).build();
      }
    } catch (IOException e) {
      throw new ServiceException(e);
    }
    return VOID_CREATE_RESPONSE;
  }

  @Override
  public AppendResponseProto append(RpcController controller, AppendRequestProto req)
      throws ServiceException {
    try {
      LocatedBlock result = server.append(req.getSrc(), req.getClientName());
      if (result != null) {
        return AppendResponseProto.newBuilder().setBlock(PBHelper.convert(result)).build();
      }
      return VOID_APPEND_RESPONSE;
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public SetReplicationResponseProto setReplication(
      RpcController controller, SetReplicationRequestProto req) throws ServiceException {
    try {
      boolean result = server.setReplication(req.getSrc(), (short) req.getReplication());
      return SetReplicationResponseProto.newBuilder().setResult(result).build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public ClientNamenodeProtocolProtos.SetMetaEnabledResponseProto setMetaEnabled(
      RpcController controller, ClientNamenodeProtocolProtos.SetMetaEnabledRequestProto req)
      throws ServiceException {
    try {
      server.setMetaEnabled(req.getSrc(), req.getMetaEnabled());
    } catch (IOException e) {
      throw new ServiceException(e);
    }
    return VOID_SET_META_ENABLED_RESPONSE;
  }

  @Override
  public SetPermissionResponseProto setPermission(
      RpcController controller, SetPermissionRequestProto req) throws ServiceException {
    try {
      server.setPermission(req.getSrc(), PBHelper.convert(req.getPermission()));
    } catch (IOException e) {
      throw new ServiceException(e);
    }
    return VOID_SET_PERM_RESPONSE;
  }

  @Override
  public SetOwnerResponseProto setOwner(RpcController controller, SetOwnerRequestProto req)
      throws ServiceException {
    try {
      server.setOwner(
          req.getSrc(),
          req.hasUsername() ? req.getUsername() : null,
          req.hasGroupname() ? req.getGroupname() : null);
    } catch (IOException e) {
      throw new ServiceException(e);
    }
    return VOID_SET_OWNER_RESPONSE;
  }

  @Override
  public AbandonBlockResponseProto abandonBlock(
      RpcController controller, AbandonBlockRequestProto req) throws ServiceException {
    try {
      server.abandonBlock(PBHelper.convert(req.getB()), req.getSrc(), req.getHolder());
    } catch (IOException e) {
      throw new ServiceException(e);
    }
    return VOID_ADD_BLOCK_RESPONSE;
  }

  @Override
  public AddBlockResponseProto addBlock(RpcController controller, AddBlockRequestProto req)
      throws ServiceException {

    try {
      List<DatanodeInfoProto> excl = req.getExcludeNodesList();
      LocatedBlock result =
          server.addBlock(
              req.getSrc(),
              req.getClientName(),
              req.hasPrevious() ? PBHelper.convert(req.getPrevious()) : null,
              (excl == null || excl.size() == 0)
                  ? null
                  : PBHelper.convert(excl.toArray(new DatanodeInfoProto[excl.size()])));
      return AddBlockResponseProto.newBuilder().setBlock(PBHelper.convert(result)).build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public GetAdditionalDatanodeResponseProto getAdditionalDatanode(
      RpcController controller, GetAdditionalDatanodeRequestProto req) throws ServiceException {
    try {
      List<DatanodeInfoProto> existingList = req.getExistingsList();
      List<DatanodeInfoProto> excludesList = req.getExcludesList();
      LocatedBlock result =
          server.getAdditionalDatanode(
              req.getSrc(),
              PBHelper.convert(req.getBlk()),
              PBHelper.convert(existingList.toArray(new DatanodeInfoProto[existingList.size()])),
              PBHelper.convert(excludesList.toArray(new DatanodeInfoProto[excludesList.size()])),
              req.getNumAdditionalNodes(),
              req.getClientName());
      return GetAdditionalDatanodeResponseProto.newBuilder()
          .setBlock(PBHelper.convert(result))
          .build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public CompleteResponseProto complete(RpcController controller, CompleteRequestProto req)
      throws ServiceException {
    try {
      boolean result =
          server.complete(
              req.getSrc(),
              req.getClientName(),
              req.hasLast() ? PBHelper.convert(req.getLast()) : null);
      return CompleteResponseProto.newBuilder().setResult(result).build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public ReportBadBlocksResponseProto reportBadBlocks(
      RpcController controller, ReportBadBlocksRequestProto req) throws ServiceException {
    try {
      List<LocatedBlockProto> bl = req.getBlocksList();
      server.reportBadBlocks(
          PBHelper.convertLocatedBlock(bl.toArray(new LocatedBlockProto[bl.size()])));
    } catch (IOException e) {
      throw new ServiceException(e);
    }
    return VOID_REP_BAD_BLOCK_RESPONSE;
  }

  @Override
  public ConcatResponseProto concat(RpcController controller, ConcatRequestProto req)
      throws ServiceException {
    try {
      List<String> srcs = req.getSrcsList();
      server.concat(req.getTrg(), srcs.toArray(new String[srcs.size()]));
    } catch (IOException e) {
      throw new ServiceException(e);
    }
    return VOID_CONCAT_RESPONSE;
  }

  @Override
  public RenameResponseProto rename(RpcController controller, RenameRequestProto req)
      throws ServiceException {
    try {
      boolean result = server.rename(req.getSrc(), req.getDst());
      return RenameResponseProto.newBuilder().setResult(result).build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public Rename2ResponseProto rename2(RpcController controller, Rename2RequestProto req)
      throws ServiceException {

    try {
      Options.Rename[] options;
      if (req.hasKeepEncodingStatus() && req.getKeepEncodingStatus()) {
        options = new Rename[2];
        options[1] = Rename.KEEP_ENCODING_STATUS;
      } else {
        options = new Rename[1];
      }
      options[0] = req.getOverwriteDest() ? Rename.OVERWRITE : Rename.NONE;
      server.rename2(req.getSrc(), req.getDst(), options);
    } catch (IOException e) {
      throw new ServiceException(e);
    }
    return VOID_RENAME2_RESPONSE;
  }

  @Override
  public DeleteResponseProto delete(RpcController controller, DeleteRequestProto req)
      throws ServiceException {
    try {
      boolean result = server.delete(req.getSrc(), req.getRecursive());
      return DeleteResponseProto.newBuilder().setResult(result).build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public MkdirsResponseProto mkdirs(RpcController controller, MkdirsRequestProto req)
      throws ServiceException {
    try {
      boolean result =
          server.mkdirs(req.getSrc(), PBHelper.convert(req.getMasked()), req.getCreateParent());
      return MkdirsResponseProto.newBuilder().setResult(result).build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public GetListingResponseProto getListing(RpcController controller, GetListingRequestProto req)
      throws ServiceException {
    try {
      DirectoryListing result =
          server.getListing(req.getSrc(), req.getStartAfter().toByteArray(), req.getNeedLocation());
      if (result != null) {
        return GetListingResponseProto.newBuilder().setDirList(PBHelper.convert(result)).build();
      } else {
        return VOID_GETLISTING_RESPONSE;
      }
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public RenewLeaseResponseProto renewLease(RpcController controller, RenewLeaseRequestProto req)
      throws ServiceException {
    try {
      server.renewLease(req.getClientName());
      return VOID_RENEWLEASE_RESPONSE;
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public RecoverLeaseResponseProto recoverLease(
      RpcController controller, RecoverLeaseRequestProto req) throws ServiceException {
    try {
      boolean result = server.recoverLease(req.getSrc(), req.getClientName());
      return RecoverLeaseResponseProto.newBuilder().setResult(result).build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public GetFsStatsResponseProto getFsStats(RpcController controller, GetFsStatusRequestProto req)
      throws ServiceException {
    try {
      return PBHelper.convert(server.getStats());
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public GetDatanodeReportResponseProto getDatanodeReport(
      RpcController controller, GetDatanodeReportRequestProto req) throws ServiceException {
    try {
      List<? extends DatanodeInfoProto> result =
          PBHelper.convert(server.getDatanodeReport(PBHelper.convert(req.getType())));
      return GetDatanodeReportResponseProto.newBuilder().addAllDi(result).build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public GetPreferredBlockSizeResponseProto getPreferredBlockSize(
      RpcController controller, GetPreferredBlockSizeRequestProto req) throws ServiceException {
    try {
      long result = server.getPreferredBlockSize(req.getFilename());
      return GetPreferredBlockSizeResponseProto.newBuilder().setBsize(result).build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public SetSafeModeResponseProto setSafeMode(RpcController controller, SetSafeModeRequestProto req)
      throws ServiceException {
    try {
      boolean result = server.setSafeMode(PBHelper.convert(req.getAction()), req.getChecked());
      return SetSafeModeResponseProto.newBuilder().setResult(result).build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public RefreshNodesResponseProto refreshNodes(
      RpcController controller, RefreshNodesRequestProto req) throws ServiceException {
    try {
      server.refreshNodes();
      return VOID_REFRESHNODES_RESPONSE;
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public ListCorruptFileBlocksResponseProto listCorruptFileBlocks(
      RpcController controller, ListCorruptFileBlocksRequestProto req) throws ServiceException {
    try {
      CorruptFileBlocks result =
          server.listCorruptFileBlocks(req.getPath(), req.hasCookie() ? req.getCookie() : null);
      return ListCorruptFileBlocksResponseProto.newBuilder()
          .setCorrupt(PBHelper.convert(result))
          .build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public GetFileInfoResponseProto getFileInfo(RpcController controller, GetFileInfoRequestProto req)
      throws ServiceException {
    try {
      HdfsFileStatus result = server.getFileInfo(req.getSrc());

      if (result != null) {
        return GetFileInfoResponseProto.newBuilder().setFs(PBHelper.convert(result)).build();
      }
      return VOID_GETFILEINFO_RESPONSE;
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public GetFileLinkInfoResponseProto getFileLinkInfo(
      RpcController controller, GetFileLinkInfoRequestProto req) throws ServiceException {
    try {
      HdfsFileStatus result = server.getFileLinkInfo(req.getSrc());
      if (result != null) {
        System.out.println("got non null result for getFileLinkInfo for " + req.getSrc());
        return GetFileLinkInfoResponseProto.newBuilder().setFs(PBHelper.convert(result)).build();
      } else {
        System.out.println("got  null result for getFileLinkInfo for " + req.getSrc());
        return VOID_GETFILELINKINFO_RESPONSE;
      }

    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public GetContentSummaryResponseProto getContentSummary(
      RpcController controller, GetContentSummaryRequestProto req) throws ServiceException {
    try {
      ContentSummary result = server.getContentSummary(req.getPath());
      return GetContentSummaryResponseProto.newBuilder()
          .setSummary(PBHelper.convert(result))
          .build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public SetQuotaResponseProto setQuota(RpcController controller, SetQuotaRequestProto req)
      throws ServiceException {
    try {
      server.setQuota(req.getPath(), req.getNamespaceQuota(), req.getDiskspaceQuota());
      return VOID_SETQUOTA_RESPONSE;
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public FsyncResponseProto fsync(RpcController controller, FsyncRequestProto req)
      throws ServiceException {
    try {
      server.fsync(req.getSrc(), req.getClient(), req.getLastBlockLength());
      return VOID_FSYNC_RESPONSE;
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public SetTimesResponseProto setTimes(RpcController controller, SetTimesRequestProto req)
      throws ServiceException {
    try {
      server.setTimes(req.getSrc(), req.getMtime(), req.getAtime());
      return VOID_SETTIMES_RESPONSE;
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public CreateSymlinkResponseProto createSymlink(
      RpcController controller, CreateSymlinkRequestProto req) throws ServiceException {
    try {
      server.createSymlink(
          req.getTarget(),
          req.getLink(),
          PBHelper.convert(req.getDirPerm()),
          req.getCreateParent());
      return VOID_CREATESYMLINK_RESPONSE;
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public GetLinkTargetResponseProto getLinkTarget(
      RpcController controller, GetLinkTargetRequestProto req) throws ServiceException {
    try {
      String result = server.getLinkTarget(req.getPath());
      GetLinkTargetResponseProto.Builder builder = GetLinkTargetResponseProto.newBuilder();
      if (result != null) {
        builder.setTargetPath(result);
      }
      return builder.build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public UpdateBlockForPipelineResponseProto updateBlockForPipeline(
      RpcController controller, UpdateBlockForPipelineRequestProto req) throws ServiceException {
    try {
      LocatedBlockProto result =
          PBHelper.convert(
              server.updateBlockForPipeline(PBHelper.convert(req.getBlock()), req.getClientName()));
      return UpdateBlockForPipelineResponseProto.newBuilder().setBlock(result).build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public UpdatePipelineResponseProto updatePipeline(
      RpcController controller, UpdatePipelineRequestProto req) throws ServiceException {
    try {
      List<DatanodeIDProto> newNodes = req.getNewNodesList();
      server.updatePipeline(
          req.getClientName(),
          PBHelper.convert(req.getOldBlock()),
          PBHelper.convert(req.getNewBlock()),
          PBHelper.convert(newNodes.toArray(new DatanodeIDProto[newNodes.size()])));
      return VOID_UPDATEPIPELINE_RESPONSE;
    } 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 RenewDelegationTokenResponseProto renewDelegationToken(
      RpcController controller, RenewDelegationTokenRequestProto req) throws ServiceException {
    try {
      long result = server.renewDelegationToken(PBHelper.convertDelegationToken(req.getToken()));
      return RenewDelegationTokenResponseProto.newBuilder().setNewExpiryTime(result).build();
    } catch (IOException e) {
      throw new ServiceException(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 SetBalancerBandwidthResponseProto setBalancerBandwidth(
      RpcController controller, SetBalancerBandwidthRequestProto req) throws ServiceException {
    try {
      server.setBalancerBandwidth(req.getBandwidth());
      return VOID_SETBALANCERBANDWIDTH_RESPONSE;
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public GetDataEncryptionKeyResponseProto getDataEncryptionKey(
      RpcController controller, GetDataEncryptionKeyRequestProto request) throws ServiceException {
    try {
      GetDataEncryptionKeyResponseProto.Builder builder =
          GetDataEncryptionKeyResponseProto.newBuilder();
      DataEncryptionKey encryptionKey = server.getDataEncryptionKey();
      if (encryptionKey != null) {
        builder.setDataEncryptionKey(PBHelper.convert(encryptionKey));
      }
      return builder.build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public ClientNamenodeProtocolProtos.PingResponseProto ping(
      RpcController controller, ClientNamenodeProtocolProtos.PingRequestProto request)
      throws ServiceException {
    try {
      server.ping();
      PingResponseProto.Builder builder = PingResponseProto.newBuilder();
      return builder.build();
    } catch (IOException ex) {
      throw new ServiceException(ex);
    }
  }

  @Override
  public ClientNamenodeProtocolProtos.GetEncodingStatusResponseProto getEncodingStatus(
      RpcController controller, ClientNamenodeProtocolProtos.GetEncodingStatusRequestProto request)
      throws ServiceException {
    try {
      EncodingStatus status = server.getEncodingStatus(request.getPath());
      ClientNamenodeProtocolProtos.GetEncodingStatusResponseProto.Builder builder =
          ClientNamenodeProtocolProtos.GetEncodingStatusResponseProto.newBuilder();
      builder.setEncodingStatus(PBHelper.convert(status));
      return builder.build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public ClientNamenodeProtocolProtos.EncodeFileResponseProto encodeFile(
      RpcController controller, ClientNamenodeProtocolProtos.EncodeFileRequestProto request)
      throws ServiceException {
    try {
      server.encodeFile(request.getPath(), PBHelper.convert(request.getPolicy()));
      ClientNamenodeProtocolProtos.EncodeFileResponseProto.Builder builder =
          ClientNamenodeProtocolProtos.EncodeFileResponseProto.newBuilder();
      return builder.build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public ClientNamenodeProtocolProtos.RevokeEncodingResponseProto revokeEncoding(
      RpcController controller, ClientNamenodeProtocolProtos.RevokeEncodingRequestProto request)
      throws ServiceException {
    try {
      server.revokeEncoding(request.getPath(), (short) request.getReplication());
      ClientNamenodeProtocolProtos.RevokeEncodingResponseProto.Builder builder =
          ClientNamenodeProtocolProtos.RevokeEncodingResponseProto.newBuilder();
      return builder.build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public ClientNamenodeProtocolProtos.GetRepairedBlockLocationsResponseProto
      getRepairedBlockLocations(
          RpcController controller,
          ClientNamenodeProtocolProtos.GetRepairedBlockLocationsRequsestProto request)
          throws ServiceException {
    try {
      LocatedBlock b =
          server.getRepairedBlockLocations(
              request.getSourcePath(),
              request.getParityPath(),
              PBHelper.convert(request.getBlock()),
              request.getIsParity());
      ClientNamenodeProtocolProtos.GetRepairedBlockLocationsResponseProto.Builder builder =
          ClientNamenodeProtocolProtos.GetRepairedBlockLocationsResponseProto.newBuilder();
      if (b != null) {
        builder.setLocatedBlocks(PBHelper.convert(b)).build();
      }
      return builder.build();
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  @Override
  public ClientNamenodeProtocolProtos.ActiveNamenodeListResponseProto getActiveNamenodesForClient(
      RpcController controller, ClientNamenodeProtocolProtos.ActiveNamenodeListRequestProto request)
      throws ServiceException {
    try {
      SortedActiveNodeList anl = server.getActiveNamenodesForClient();
      ClientNamenodeProtocolProtos.ActiveNamenodeListResponseProto response =
          convertANListToResponseProto(anl);
      return response;
    } catch (IOException e) {
      throw new ServiceException(e);
    }
  }

  private ClientNamenodeProtocolProtos.ActiveNamenodeListResponseProto convertANListToResponseProto(
      SortedActiveNodeList anlWrapper) {
    List<ActiveNode> anl = anlWrapper.getActiveNodes();
    ClientNamenodeProtocolProtos.ActiveNamenodeListResponseProto.Builder anlrpb =
        ClientNamenodeProtocolProtos.ActiveNamenodeListResponseProto.newBuilder();
    for (int i = 0; i < anl.size(); i++) {
      ActiveNodeProtos.ActiveNodeProto anp = convertANToResponseProto(anl.get(i));
      anlrpb.addNamenodes(anp);
    }
    return anlrpb.build();
  }

  private ActiveNodeProtos.ActiveNodeProto convertANToResponseProto(ActiveNode p) {
    ActiveNodeProtos.ActiveNodeProto.Builder anp = ActiveNodeProtos.ActiveNodeProto.newBuilder();
    anp.setId(p.getId());
    anp.setHostname(p.getHostname());
    anp.setIpAddress(p.getIpAddress());
    anp.setPort(p.getPort());
    anp.setHttpAddress(p.getHttpAddress());
    return anp.build();
  }

  @Override
  public ClientNamenodeProtocolProtos.ChangeConfResponseProto changeConf(
      RpcController controller, ClientNamenodeProtocolProtos.ChangeConfProto request)
      throws ServiceException {
    try {
      server.changeConf(request.getPropsList(), request.getNewValsList());
      return VOID_CHANGECONF_RESPONSE;
    } catch (IOException ex) {
      throw new ServiceException(ex);
    }
  }
}