Examples with AgentShutdownRequest alluxio.hub.proto.AgentShutdownRequest used on opensource projects

Example 1 with AgentShutdownRequest

use of alluxio.hub.proto.AgentShutdownRequest in project alluxio by Alluxio.

the class ManagerProcessContext method handleStatusRuntimeException.

private void handleStatusRuntimeException(String message, Throwable t) {
    LogUtils.warnWithException(LOG, message, t);
    if (Status.fromThrowable(t).getCode() == Status.UNAUTHENTICATED.getCode()) {
        if (mLock.tryLock()) {
            try {
                // shut down the Hub Manager and Agents
                String msg = String.format("Shutting down the Hub Agent because the Hub Manager is " + "unauthenticated. Check %s, %s properties in the Hub Manager's " + "alluxio-site.properties.", PropertyKey.Name.HUB_AUTHENTICATION_API_KEY, PropertyKey.Name.HUB_AUTHENTICATION_SECRET_KEY);
                AgentShutdownRequest req = AgentShutdownRequest.newBuilder().setLogMessage(msg).setExitCode(401).build();
                Function<AgentManagerServiceGrpc.AgentManagerServiceBlockingStub, AgentShutdownResponse> x = (client) -> {
                    AgentShutdownResponse resp = client.shutdown(req);
                    return resp;
                };
                try {
                    execOnHub(x);
                } catch (Exception e) {
                    LogUtils.warnWithException(LOG, "Failed to call shutdown() on Hub Agent. " + "Continuing with shutting down Hub Manager.", e);
                }
            } finally {
                mLock.unlock();
            }
        }
        LOG.error(String.format("Shutting down the Hub manager because it is unauthenticated. " + "Check %s, %s properties in alluxio-site.properties.", PropertyKey.Name.HUB_AUTHENTICATION_API_KEY, PropertyKey.Name.HUB_AUTHENTICATION_SECRET_KEY));
        System.exit(401);
    }
}