aboutsummaryrefslogtreecommitdiff
path: root/stub
diff options
context:
space:
mode:
authorCarl Mastrangelo <notcarl@google.com>2017-07-13 14:33:25 -0700
committerGitHub <noreply@github.com>2017-07-13 14:33:25 -0700
commitd387bfe72fff7b89f2cf7ce09bca418feb322085 (patch)
treee5df7fe727e43b89da21223dd3ddb7c2c274f11f /stub
parent26caa488a5a5ead1d9dac17bc0facf975bed5e35 (diff)
downloadgrpc-grpc-java-d387bfe72fff7b89f2cf7ce09bca418feb322085.tar.gz
stub: Use named classes rather than anonymous in ServerCalls
This has a number of small benefits. First, it makes stack traces easier to read. For example: Old class names: ``` ServerCalls$1$1.class ServerCalls$1.class ServerCalls$2$1.class ServerCalls$2.class ``` New class names: ``` ServerCalls$StreamingServerCallHandler.class ServerCalls$StreamingServerCallHandler$StreamingServerCallListener.class ServerCalls$UnaryServerCallHandler.class ServerCalls$UnaryServerCallHandler$UnaryServerCallListener.class ``` This is much easier to read quickly, espcially if line numbers don't match between HEAD and the code that prints the stack trace. Another benefit of this is that it drops a class file from the jar (`EmptyServerCallListener`). Lastly, it makes it easier in the future to test, since the specific class can be referenced from a test. Traditionally this class hasn't been easy to test.
Diffstat (limited to 'stub')
-rw-r--r--stub/src/main/java/io/grpc/stub/ServerCalls.java312
1 files changed, 171 insertions, 141 deletions
diff --git a/stub/src/main/java/io/grpc/stub/ServerCalls.java b/stub/src/main/java/io/grpc/stub/ServerCalls.java
index f9a84085c..4d0f8c14a 100644
--- a/stub/src/main/java/io/grpc/stub/ServerCalls.java
+++ b/stub/src/main/java/io/grpc/stub/ServerCalls.java
@@ -104,85 +104,182 @@ public final class ServerCalls {
extends StreamingRequestMethod<ReqT, RespT> {
}
+ private static final class UnaryServerCallHandler<ReqT, RespT>
+ implements ServerCallHandler<ReqT, RespT> {
+
+ private final UnaryRequestMethod<ReqT, RespT> method;
+
+ // Non private to avoid synthetic class
+ UnaryServerCallHandler(UnaryRequestMethod<ReqT, RespT> method) {
+ this.method = method;
+ }
+
+ @Override
+ public ServerCall.Listener<ReqT> startCall(ServerCall<ReqT, RespT> call, Metadata headers) {
+ Preconditions.checkArgument(
+ call.getMethodDescriptor().getType().clientSendsOneMessage(),
+ "asyncUnaryRequestCall is only for clientSendsOneMessage methods");
+ ServerCallStreamObserverImpl<ReqT, RespT> responseObserver =
+ new ServerCallStreamObserverImpl<ReqT, RespT>(call);
+ // We expect only 1 request, but we ask for 2 requests here so that if a misbehaving client
+ // sends more than 1 requests, ServerCall will catch it. Note that disabling auto
+ // inbound flow control has no effect on unary calls.
+ call.request(2);
+ return new UnaryServerCallListener(responseObserver, call);
+ }
+
+ private final class UnaryServerCallListener extends ServerCall.Listener<ReqT> {
+ private final ServerCall<ReqT, RespT> call;
+ private final ServerCallStreamObserverImpl<ReqT, RespT> responseObserver;
+ private boolean canInvoke = true;
+ private ReqT request;
+
+ // Non private to avoid synthetic class
+ UnaryServerCallListener(
+ ServerCallStreamObserverImpl<ReqT, RespT> responseObserver,
+ ServerCall<ReqT, RespT> call) {
+ this.call = call;
+ this.responseObserver = responseObserver;
+ }
+
+ @Override
+ public void onMessage(ReqT request) {
+ if (this.request != null) {
+ // Safe to close the call, because the application has not yet been invoked
+ call.close(
+ Status.INTERNAL.withDescription(TOO_MANY_REQUESTS),
+ new Metadata());
+ canInvoke = false;
+ return;
+ }
+
+ // We delay calling method.invoke() until onHalfClose() to make sure the client
+ // half-closes.
+ this.request = request;
+ }
+
+ @Override
+ public void onHalfClose() {
+ if (!canInvoke) {
+ return;
+ }
+ if (request == null) {
+ // Safe to close the call, because the application has not yet been invoked
+ call.close(
+ Status.INTERNAL.withDescription(MISSING_REQUEST),
+ new Metadata());
+ return;
+ }
+
+ method.invoke(request, responseObserver);
+ responseObserver.freeze();
+ if (call.isReady()) {
+ // Since we are calling invoke in halfClose we have missed the onReady
+ // event from the transport so recover it here.
+ onReady();
+ }
+ }
+
+ @Override
+ public void onCancel() {
+ responseObserver.cancelled = true;
+ if (responseObserver.onCancelHandler != null) {
+ responseObserver.onCancelHandler.run();
+ }
+ }
+
+ @Override
+ public void onReady() {
+ if (responseObserver.onReadyHandler != null) {
+ responseObserver.onReadyHandler.run();
+ }
+ }
+ }
+ }
+
/**
* Creates a {@code ServerCallHandler} for a unary request call method of the service.
*
* @param method an adaptor to the actual method on the service implementation.
*/
private static <ReqT, RespT> ServerCallHandler<ReqT, RespT> asyncUnaryRequestCall(
- final UnaryRequestMethod<ReqT, RespT> method) {
- return new ServerCallHandler<ReqT, RespT>() {
+ UnaryRequestMethod<ReqT, RespT> method) {
+ return new UnaryServerCallHandler<ReqT, RespT>(method);
+ }
+
+ private static final class StreamingServerCallHandler<ReqT, RespT>
+ implements ServerCallHandler<ReqT, RespT> {
+
+ private final StreamingRequestMethod<ReqT, RespT> method;
+
+ // Non private to avoid synthetic class
+ StreamingServerCallHandler(StreamingRequestMethod<ReqT, RespT> method) {
+ this.method = method;
+ }
+
+ @Override
+ public ServerCall.Listener<ReqT> startCall(ServerCall<ReqT, RespT> call, Metadata headers) {
+ ServerCallStreamObserverImpl<ReqT, RespT> responseObserver =
+ new ServerCallStreamObserverImpl<ReqT, RespT>(call);
+ StreamObserver<ReqT> requestObserver = method.invoke(responseObserver);
+ responseObserver.freeze();
+ if (responseObserver.autoFlowControlEnabled) {
+ call.request(1);
+ }
+ return new StreamingServerCallListener(requestObserver, responseObserver, call);
+ }
+
+ private final class StreamingServerCallListener extends ServerCall.Listener<ReqT> {
+
+ private final StreamObserver<ReqT> requestObserver;
+ private final ServerCallStreamObserverImpl<ReqT, RespT> responseObserver;
+ private final ServerCall<ReqT, RespT> call;
+ private boolean halfClosed = false;
+
+ // Non private to avoid synthetic class
+ StreamingServerCallListener(
+ StreamObserver<ReqT> requestObserver,
+ ServerCallStreamObserverImpl<ReqT, RespT> responseObserver,
+ ServerCall<ReqT, RespT> call) {
+ this.requestObserver = requestObserver;
+ this.responseObserver = responseObserver;
+ this.call = call;
+ }
+
+ @Override
+ public void onMessage(ReqT request) {
+ requestObserver.onNext(request);
+
+ // Request delivery of the next inbound message.
+ if (responseObserver.autoFlowControlEnabled) {
+ call.request(1);
+ }
+ }
+
+ @Override
+ public void onHalfClose() {
+ halfClosed = true;
+ requestObserver.onCompleted();
+ }
+
+ @Override
+ public void onCancel() {
+ responseObserver.cancelled = true;
+ if (responseObserver.onCancelHandler != null) {
+ responseObserver.onCancelHandler.run();
+ }
+ if (!halfClosed) {
+ requestObserver.onError(Status.CANCELLED.asException());
+ }
+ }
+
@Override
- public ServerCall.Listener<ReqT> startCall(
- final ServerCall<ReqT, RespT> call,
- Metadata headers) {
- Preconditions.checkArgument(
- call.getMethodDescriptor().getType().clientSendsOneMessage(),
- "asyncUnaryRequestCall is only for clientSendsOneMessage methods");
- final ServerCallStreamObserverImpl<ReqT, RespT> responseObserver =
- new ServerCallStreamObserverImpl<ReqT, RespT>(call);
- // We expect only 1 request, but we ask for 2 requests here so that if a misbehaving client
- // sends more than 1 requests, ServerCall will catch it. Note that disabling auto
- // inbound flow control has no effect on unary calls.
- call.request(2);
- return new EmptyServerCallListener<ReqT>() {
- boolean canInvoke = true;
- ReqT request;
- @Override
- public void onMessage(ReqT request) {
- if (this.request != null) {
- // Safe to close the call, because the application has not yet been invoked
- call.close(
- Status.INTERNAL.withDescription(TOO_MANY_REQUESTS),
- new Metadata());
- canInvoke = false;
- return;
- }
-
- // We delay calling method.invoke() until onHalfClose() to make sure the client
- // half-closes.
- this.request = request;
- }
-
- @Override
- public void onHalfClose() {
- if (!canInvoke) {
- return;
- }
- if (request == null) {
- // Safe to close the call, because the application has not yet been invoked
- call.close(
- Status.INTERNAL.withDescription(MISSING_REQUEST),
- new Metadata());
- return;
- }
-
- method.invoke(request, responseObserver);
- responseObserver.freeze();
- if (call.isReady()) {
- // Since we are calling invoke in halfClose we have missed the onReady
- // event from the transport so recover it here.
- onReady();
- }
- }
-
- @Override
- public void onCancel() {
- responseObserver.cancelled = true;
- if (responseObserver.onCancelHandler != null) {
- responseObserver.onCancelHandler.run();
- }
- }
-
- @Override
- public void onReady() {
- if (responseObserver.onReadyHandler != null) {
- responseObserver.onReadyHandler.run();
- }
- }
- };
+ public void onReady() {
+ if (responseObserver.onReadyHandler != null) {
+ responseObserver.onReadyHandler.run();
+ }
}
- };
+ }
}
/**
@@ -191,58 +288,8 @@ public final class ServerCalls {
* @param method an adaptor to the actual method on the service implementation.
*/
private static <ReqT, RespT> ServerCallHandler<ReqT, RespT> asyncStreamingRequestCall(
- final StreamingRequestMethod<ReqT, RespT> method) {
- return new ServerCallHandler<ReqT, RespT>() {
- @Override
- public ServerCall.Listener<ReqT> startCall(
- final ServerCall<ReqT, RespT> call,
- Metadata headers) {
- final ServerCallStreamObserverImpl<ReqT, RespT> responseObserver =
- new ServerCallStreamObserverImpl<ReqT, RespT>(call);
- final StreamObserver<ReqT> requestObserver = method.invoke(responseObserver);
- responseObserver.freeze();
- if (responseObserver.autoFlowControlEnabled) {
- call.request(1);
- }
- return new EmptyServerCallListener<ReqT>() {
- boolean halfClosed = false;
-
- @Override
- public void onMessage(ReqT request) {
- requestObserver.onNext(request);
-
- // Request delivery of the next inbound message.
- if (responseObserver.autoFlowControlEnabled) {
- call.request(1);
- }
- }
-
- @Override
- public void onHalfClose() {
- halfClosed = true;
- requestObserver.onCompleted();
- }
-
- @Override
- public void onCancel() {
- responseObserver.cancelled = true;
- if (responseObserver.onCancelHandler != null) {
- responseObserver.onCancelHandler.run();
- }
- if (!halfClosed) {
- requestObserver.onError(Status.CANCELLED.asException());
- }
- }
-
- @Override
- public void onReady() {
- if (responseObserver.onReadyHandler != null) {
- responseObserver.onReadyHandler.run();
- }
- }
- };
- }
- };
+ StreamingRequestMethod<ReqT, RespT> method) {
+ return new StreamingServerCallHandler<ReqT, RespT>(method);
}
private static interface UnaryRequestMethod<ReqT, RespT> {
@@ -263,6 +310,7 @@ public final class ServerCalls {
private Runnable onReadyHandler;
private Runnable onCancelHandler;
+ // Non private to avoid synthetic class
ServerCallStreamObserverImpl(ServerCall<ReqT, RespT> call) {
this.call = call;
}
@@ -352,24 +400,6 @@ public final class ServerCalls {
}
}
- private static class EmptyServerCallListener<ReqT> extends ServerCall.Listener<ReqT> {
- @Override
- public void onMessage(ReqT request) {
- }
-
- @Override
- public void onHalfClose() {
- }
-
- @Override
- public void onCancel() {
- }
-
- @Override
- public void onComplete() {
- }
- }
-
/**
* Sets unimplemented status for method on given response stream for unary call.
*