blob: 3076d7e7d456d2e44b4d36aa6bebd89a6ddcd690 (
plain)
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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
package com.orbekk.protobuf;
import java.io.IOException;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.ExecutorService;
import com.google.protobuf.Descriptors;
import com.google.protobuf.InvalidProtocolBufferException;
import com.google.protobuf.Message;
import com.google.protobuf.RpcCallback;
import com.google.protobuf.Service;
public class RequestDispatcher extends Thread {
public static int DEFAULT_QUEUE_SIZE = 5;
private volatile boolean isStopped = false;
private final BlockingQueue<Data.Response> output;
private final ServiceHolder services;
/** A pool that can be shared among all dispatchers. */
private final ExecutorService pool;
private static class RequestHandler implements Runnable {
private final Data.Request request;
private final Data.Response.Builder response =
Data.Response.newBuilder();
private final BlockingQueue<Data.Response> output;
private final ServiceHolder services;
private final Rpc rpc = new Rpc();
private final RpcCallback<Message> callback =
new RpcCallback<Message>() {
@Override public void run(Message responseMessage) {
if (responseMessage != null) {
response.setResponseProto(responseMessage.toByteString());
}
rpc.writeTo(response);
try {
output.put(response.build());
} catch (InterruptedException e) {
// Terminate callback.
return;
}
}
};
public RequestHandler(Data.Request request,
BlockingQueue<Data.Response> output,
ServiceHolder services) {
this.request = request;
this.output = output;
this.services = services;
}
public void internalRun() throws InterruptedException {
Service service = services.get(request.getFullServiceName());
if (service == null) {
response.setError(Data.Response.RpcError.UNKNOWN_SERVICE);
output.put(response.build());
return;
}
Descriptors.MethodDescriptor method =
service.getDescriptorForType()
.findMethodByName(request.getMethodName());
if (method == null) {
response.setError(Data.Response.RpcError.UNKNOWN_METHOD);
output.put(response.build());
return;
}
Message requestMessage = null;
try {
requestMessage = service.getRequestPrototype(method)
.toBuilder().mergeFrom(request.getRequestProto()).build();
} catch (InvalidProtocolBufferException e) {
response.setError(Data.Response.RpcError.INVALID_PROTOBUF);
output.put(response.build());
return;
}
response.setRequestId(request.getRequestId());
service.callMethod(method, rpc, requestMessage, callback);
}
@Override public void run() {
try {
internalRun();
} catch (InterruptedException e) {
// Terminate request.
return;
}
}
}
public RequestDispatcher(ExecutorService pool,
BlockingQueue<Data.Response> output,
ServiceHolder services) {
this.pool = pool;
this.output = output;
this.services = services;
}
public void handleRequest(Data.Request request) throws InterruptedException {
RequestHandler handler = new RequestHandler(request, output, services);
pool.execute(handler);
}
}
|