same/src/main/java/com/orbekk/same/benchmark/ExampleServer.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

package com.orbekk.same.benchmark;

import org.apache.log4j.Logger;

import com.google.protobuf.RpcCallback;
import com.google.protobuf.RpcController;
import com.orbekk.protobuf.SimpleProtobufServer;
import com.orbekk.same.benchmark.Example.Data;

public class ExampleServer {
    private final static Logger logger =
            Logger.getLogger(ExampleServer.class);
    private volatile SimpleProtobufServer server;
    
    class ServiceImpl extends Example.Service {
        @Override
        public void methodA(RpcController controller, Data request, RpcCallback<Data> done) {
            Data response = Data.newBuilder()
                .setMessage(request.getMessage())
                .setArg1(request.getArg1())
                .setArg2(request.getArg2())
                .build();
            done.run(response);
        }
    }
    
    public void runServer(int port) {
        server = SimpleProtobufServer.create(port);
        server.registerService(new ServiceImpl());
        server.start();
        logger.info("Running SimpleProtobufServer on port " + server.getPort());
    }
    
    public void stopServer() throws InterruptedException {
        server.interrupt();
    }
    
    public static void main(String[] args) {
        ExampleServer server = new ExampleServer();
        server.runServer(12000);
    }
}