Examples with ServiceException org.apache.hbase.thirdparty.com.google.protobuf.ServiceException used on opensource projects

Example 11 with ServiceException

use of org.apache.hbase.thirdparty.com.google.protobuf.ServiceException in project hbase by apache.

the class MasterRpcServices method decommissionRegionServers.

@Override
public DecommissionRegionServersResponse decommissionRegionServers(RpcController controller, DecommissionRegionServersRequest request) throws ServiceException {
    try {
        server.checkInitialized();
        List<ServerName> servers = request.getServerNameList().stream().map(pbServer -> ProtobufUtil.toServerName(pbServer)).collect(Collectors.toList());
        boolean offload = request.getOffload();
        if (server.cpHost != null) {
            server.cpHost.preDecommissionRegionServers(servers, offload);
        }
        server.decommissionRegionServers(servers, offload);
        if (server.cpHost != null) {
            server.cpHost.postDecommissionRegionServers(servers, offload);
        }
    } catch (IOException io) {
        throw new ServiceException(io);
    }
    return DecommissionRegionServersResponse.newBuilder().build();
}

Example 12 with ServiceException

use of org.apache.hbase.thirdparty.com.google.protobuf.ServiceException in project hbase by apache.

the class MasterRpcServices method execProcedure.

/**
 * Triggers an asynchronous attempt to run a distributed procedure.
 * {@inheritDoc}
 */
@Override
public ExecProcedureResponse execProcedure(RpcController controller, ExecProcedureRequest request) throws ServiceException {
    try {
        server.checkInitialized();
        ProcedureDescription desc = request.getProcedure();
        MasterProcedureManager mpm = server.getMasterProcedureManagerHost().getProcedureManager(desc.getSignature());
        if (mpm == null) {
            throw new ServiceException(new DoNotRetryIOException("The procedure is not registered: " + desc.getSignature()));
        }
        LOG.info(server.getClientIdAuditPrefix() + " procedure request for: " + desc.getSignature());
        mpm.checkPermissions(desc, getAccessChecker(), RpcServer.getRequestUser().orElse(null));
        mpm.execProcedure(desc);
        // send back the max amount of time the client should wait for the procedure
        // to complete
        long waitTime = SnapshotDescriptionUtils.DEFAULT_MAX_WAIT_TIME;
        return ExecProcedureResponse.newBuilder().setExpectedTimeout(waitTime).build();
    } catch (ForeignException e) {
        throw new ServiceException(e.getCause());
    } catch (IOException e) {
        throw new ServiceException(e);
    }
}

Example 13 with ServiceException

use of org.apache.hbase.thirdparty.com.google.protobuf.ServiceException in project hbase by apache.

the class MasterRpcServices method execProcedureWithRet.

/**
 * Triggers a synchronous attempt to run a distributed procedure and sets
 * return data in response.
 * {@inheritDoc}
 */
@Override
public ExecProcedureResponse execProcedureWithRet(RpcController controller, ExecProcedureRequest request) throws ServiceException {
    rpcPreCheck("execProcedureWithRet");
    try {
        ProcedureDescription desc = request.getProcedure();
        MasterProcedureManager mpm = server.getMasterProcedureManagerHost().getProcedureManager(desc.getSignature());
        if (mpm == null) {
            throw new ServiceException("The procedure is not registered: " + desc.getSignature());
        }
        LOG.info(server.getClientIdAuditPrefix() + " procedure request for: " + desc.getSignature());
        byte[] data = mpm.execProcedureWithRet(desc);
        ExecProcedureResponse.Builder builder = ExecProcedureResponse.newBuilder();
        // set return data if available
        if (data != null) {
            builder.setReturnData(UnsafeByteOperations.unsafeWrap(data));
        }
        return builder.build();
    } catch (IOException e) {
        throw new ServiceException(e);
    }
}

Example 14 with ServiceException

use of org.apache.hbase.thirdparty.com.google.protobuf.ServiceException in project hbase by apache.

the class MasterRpcServices method setTableStateInMeta.

/**
 * Update state of the table in meta only. This is required by hbck in some situations to cleanup
 * stuck assign/ unassign regions procedures for the table.
 *
 * @return previous state of the table
 */
@Override
public GetTableStateResponse setTableStateInMeta(RpcController controller, SetTableStateInMetaRequest request) throws ServiceException {
    rpcPreCheck("setTableStateInMeta");
    TableName tn = ProtobufUtil.toTableName(request.getTableName());
    try {
        TableState prevState = this.server.getTableStateManager().getTableState(tn);
        TableState newState = TableState.convert(tn, request.getTableState());
        LOG.info("{} set table={} state from {} to {}", server.getClientIdAuditPrefix(), tn, prevState.getState(), newState.getState());
        this.server.getTableStateManager().setTableState(tn, newState.getState());
        return GetTableStateResponse.newBuilder().setTableState(prevState.convert()).build();
    } catch (Exception e) {
        throw new ServiceException(e);
    }
}

Example 15 with ServiceException

use of org.apache.hbase.thirdparty.com.google.protobuf.ServiceException in project hbase by apache.

the class MasterRpcServices method execMasterService.

@Override
public ClientProtos.CoprocessorServiceResponse execMasterService(final RpcController controller, final ClientProtos.CoprocessorServiceRequest request) throws ServiceException {
    rpcPreCheck("execMasterService");
    try {
        ServerRpcController execController = new ServerRpcController();
        ClientProtos.CoprocessorServiceCall call = request.getCall();
        String serviceName = call.getServiceName();
        String methodName = call.getMethodName();
        if (!server.coprocessorServiceHandlers.containsKey(serviceName)) {
            throw new UnknownProtocolException(null, "No registered Master Coprocessor Endpoint found for " + serviceName + ". Has it been enabled?");
        }
        Service service = server.coprocessorServiceHandlers.get(serviceName);
        ServiceDescriptor serviceDesc = service.getDescriptorForType();
        MethodDescriptor methodDesc = CoprocessorRpcUtils.getMethodDescriptor(methodName, serviceDesc);
        Message execRequest = CoprocessorRpcUtils.getRequest(service, methodDesc, call.getRequest());
        final Message.Builder responseBuilder = service.getResponsePrototype(methodDesc).newBuilderForType();
        service.callMethod(methodDesc, execController, execRequest, (message) -> {
            if (message != null) {
                responseBuilder.mergeFrom(message);
            }
        });
        Message execResult = responseBuilder.build();
        if (execController.getFailedOn() != null) {
            throw execController.getFailedOn();
        }
        String remoteAddress = RpcServer.getRemoteAddress().map(InetAddress::toString).orElse("");
        User caller = RpcServer.getRequestUser().orElse(null);
        AUDITLOG.info("User {} (remote address: {}) master service request for {}.{}", caller, remoteAddress, serviceName, methodName);
        return CoprocessorRpcUtils.getResponse(execResult, HConstants.EMPTY_BYTE_ARRAY);
    } catch (IOException ie) {
        throw new ServiceException(ie);
    }
}