gRPC 和 protobuf 开发框架

syntax = "proto3";

option java_multiple_files = true;
option java_package = "com.rocka";
option java_outer_classname = "HelloWorldProto";
option objc_class_prefix = "HLW";

package com.rocka;

service Greeter{
    //定义一个方法，输入的是
    rpc SayHello(HelloRequest) returns(HelloReply){}
}

message HelloRequest{
    string name = 1;
}

message HelloReply{
    string message = 1;
}

public class TestServer {

    private int port = 8883;
    private Server server;

    public static void main(String[] args) throws IOException, InterruptedException {
        final TestServer testServer = new TestServer();
        testServer.start();
        testServer.blockUntilShtudown();
    }

    private void start() throws IOException {
        server = ServerBuilder.forPort(port)
                .addService(new GreeterImpl())
                .build()
                .start();
        System.out.println("Sever started , listening on " + port);
        Runtime.getRuntime().addShutdownHook(new Thread() {
            @Override
            public void run() {
                // Use stderr here since the logger may have been reset by its JVM shutdown hook.
                System.err.println("*** shutting down gRPC server since JVM is shutting down");
                TestServer.this.stop();
                System.err.println("*** server shut down");
            }
        });
    }

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

    private void blockUntilShtudown() throws InterruptedException {
        if (server != null) {
            server.awaitTermination();
        }
    }


    private class GreeterImpl extends GreeterGrpc.GreeterImplBase {
        public AtomicInteger count = new AtomicInteger(0);

        @Override
        public void sayHello(HelloRequest request, StreamObserver<HelloReply> responseObserver) {
            System.out.println("SayHello。。");
            HelloReply reply = HelloReply.newBuilder().setMessage("Hello " + request.getName()).build();
            responseObserver.onNext(reply);
            responseObserver.onCompleted();
            System.out.println("TT:" + count.incrementAndGet() + Thread.currentThread().getName());
        }
    }

}

public class TestClient {

    private final ManagedChannel channel;

    private final GreeterGrpc.GreeterBlockingStub blockingStub;

    public TestClient(String host, int port) {
        channel = ManagedChannelBuilder.forAddress(host, port).usePlaintext(true).build();
        blockingStub = GreeterGrpc.newBlockingStub(channel);
    }

    public void shutdown() throws InterruptedException {
        channel.shutdown().awaitTermination(5, TimeUnit.SECONDS);
    }

    public void greet(String word){
        HelloRequest request = HelloRequest.newBuilder().setName(word).build();
        HelloReply response;

        try{
            response = blockingStub.sayHello(request);
        } catch (StatusRuntimeException e){
            e.printStackTrace();
            return;
        }

        System.out.println(response.getMessage());
    }

    public static void main(String[] args) throws Exception{
        TestClient client = new TestClient("localhost", 8883);
        try {
            String user = "Rocka";
            if(args.length > 0){
                user = args[0];
            }
            client.greet(user);
        }finally {
            client.shutdown();
        }
    }
}