jobserv/src/main/java/JobServ/JobServServer.java

/*
 * JobServServer
 *
 * v1.0
 *
 * May 18, 2019
 */

package JobServ;

import io.grpc.Server;
import io.grpc.ServerBuilder;
import io.grpc.stub.StreamObserver;
import io.grpc.netty.GrpcSslContexts;
import io.grpc.netty.NettyServerBuilder;
import io.netty.handler.ssl.ClientAuth;
import io.netty.handler.ssl.SslContextBuilder;
import io.netty.handler.ssl.SslProvider;

import java.io.IOException;
import java.util.logging.Logger;

/*
 * The JobServServer class implements the JobServ protobuf API
 * It does this by extending the gRPC stub code.
 * Additionally, JobServServer starts and manages a daemon
 * Which accepts incoming connections from client.
 */
public class JobServServer {
    private static final Logger logger = Logger.getLogger(JobServServer.class.getName());

    private Server server;
    private final int port;
    private final String certChainFilePath;
    private final String privateKeyFilePath;
    private final String trustCertCollectionFilePath;

    public JobServServer(int port,
                         String certChainFilePath,
                         String privateKeyFilePath,
                         String trustCertCollectionFilePath) {
        this.port = port;
        this.certChainFilePath = certChainFilePath;
        this.privateKeyFilePath = privateKeyFilePath;
        this.trustCertCollectionFilePath = trustCertCollectionFilePath;
    }

    private SslContextBuilder getSslContextBuilder() {
        SslContextBuilder sslClientContextBuilder = SslContextBuilder.forServer(new File(certChainFilePath),
                                                                                new File(privateKeyFilePath));

            sslClientContextBuilder.trustManager(new File(trustCertCollectionFilePath));
            sslClientContextBuilder.clientAuth(ClientAuth.REQUIRE);

        return GrpcSslContexts.configure(sslClientContextBuilder);

    // starts the GRPC API Server
    private void start() throws IOException {
        // TODO: this should be passed in from a configuration manager
        server = ServerBuilder.forPort(this.port)
            .addService(new ShellServerImpl())
            .sslContext(getSslContextBuilder().build())
            .build()
            .start();
        logger.info("Server initialized!");

        Runtime.getRuntime().addShutdownHook(new Thread() {
                @Override
                public void run() {
                    // JVM shutdown might break logger functionality
                    // so investigate this....
                    logger.info("Shutting down server");
                    JobServServer.this.stop();
                }
            });
    }

    private void stop() {
        if (server != null) {
            server.shutdown();
        }
    }

    // spinlock for main()
    private void blockUntilShutdown() throws InterruptedException {
        if (server != null) {
            server.awaitTermination();
        }
    }

    // Main function. starts GRPC server and spins until server is shutdown
    public static void main(String[] args) throws IOException, InterruptedException {
        int port = 8448; // TODO: port and key/cert files should be handled by a config manager 

        if(args.length < 3) {
            System.out.println("Usage: ./jobserv-server certchain privatekey truststore");
            return;
        }

        final JobServServer server = new JobServServer(8448, args[0], args[1], args[2]);
        server.start();
        server.blockUntilShutdown();
    }

    // wrap around stub code generated by GRPC
    static class ShellServerImpl extends ShellServerGrpc.ShellServerImplBase {

        @Override
        public void getStatus(PIDMessage request,
                              StreamObserver<StatusMessage> responder) {
            // TODO: Implement job controller, check request type, do the thing
            StatusMessage reply = StatusMessage.newBuilder()
                .setIsRunning(true)
                .build();
            responder.onNext(reply);
            responder.onCompleted();
        }

        @Override
        public void getOutput(PIDMessage request,
                              StreamObserver<OutputMessage> responder) {
            // TODO: Implement job controller, get output, do the thing
            OutputMessage reply = OutputMessage.newBuilder()
                .setOutput("test data")
                .build();
            responder.onNext(reply);
            responder.onCompleted();
        }

        @Override
        public void newJob(NewJobMessage request,
                           StreamObserver<PIDMessage> responder) {
            // TODO: Implement job controller, Start Job, do the thing
            PIDMessage reply = PIDMessage.newBuilder()
                .setPid(-1)
                .build();
            responder.onNext(reply);
            responder.onCompleted();
        }

        @Override
        public void getReturn(PIDMessage request,
                              StreamObserver<ReturnMessage> responder) {
            // TODO: Implement job controller, get return code
            ReturnMessage reply = ReturnMessage.newBuilder()
                .setProcessReturnCode(277)
                .build();
            responder.onNext(reply);
            responder.onCompleted();
        }

        @Override
        public void killJob(PIDMessage request,
                            StreamObserver<StatusMessage> responder) {
            // TODO: implement job controller, do the thing
            // TODO: kill job here
            StatusMessage reply = StatusMessage.newBuilder()
                .setIsRunning(false)
                .build();
            responder.onNext(reply);
            responder.onCompleted();
        }
    }
}