Examples with FamilyPath org.apache.hadoop.hbase.shaded.protobuf.generated.ClientProtos.BulkLoadHFileRequest.FamilyPath used on opensource projects

Example 1 with FamilyPath

use of org.apache.hadoop.hbase.shaded.protobuf.generated.ClientProtos.BulkLoadHFileRequest.FamilyPath in project hbase by apache.

the class RSRpcServices method bulkLoadHFile.

/**
   * Atomically bulk load several HFiles into an open region
   * @return true if successful, false is failed but recoverably (no action)
   * @throws ServiceException if failed unrecoverably
   */
@Override
public BulkLoadHFileResponse bulkLoadHFile(final RpcController controller, final BulkLoadHFileRequest request) throws ServiceException {
    long start = EnvironmentEdgeManager.currentTime();
    try {
        checkOpen();
        requestCount.increment();
        Region region = getRegion(request.getRegion());
        boolean bypass = false;
        boolean loaded = false;
        Map<byte[], List<Path>> map = null;
        if (!request.hasBulkToken()) {
            // Old style bulk load. This will not be supported in future releases
            List<Pair<byte[], String>> familyPaths = new ArrayList<>(request.getFamilyPathCount());
            for (FamilyPath familyPath : request.getFamilyPathList()) {
                familyPaths.add(new Pair<>(familyPath.getFamily().toByteArray(), familyPath.getPath()));
            }
            if (region.getCoprocessorHost() != null) {
                bypass = region.getCoprocessorHost().preBulkLoadHFile(familyPaths);
            }
            try {
                if (!bypass) {
                    map = region.bulkLoadHFiles(familyPaths, request.getAssignSeqNum(), null, request.getCopyFile());
                    if (map != null) {
                        loaded = true;
                    }
                }
            } finally {
                if (region.getCoprocessorHost() != null) {
                    loaded = region.getCoprocessorHost().postBulkLoadHFile(familyPaths, map, loaded);
                }
            }
        } else {
            // secure bulk load
            map = regionServer.secureBulkLoadManager.secureBulkLoadHFiles(region, request);
        }
        BulkLoadHFileResponse.Builder builder = BulkLoadHFileResponse.newBuilder();
        if (map != null) {
            loaded = true;
        }
        builder.setLoaded(loaded);
        return builder.build();
    } catch (IOException ie) {
        throw new ServiceException(ie);
    } finally {
        if (regionServer.metricsRegionServer != null) {
            regionServer.metricsRegionServer.updateBulkLoad(EnvironmentEdgeManager.currentTime() - start);
        }
    }
}