Spring/spring-grpc
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add GrpcExceptionHandledServerCall to catch exceptions

Browse Source 
Signed-off-by: Thomas McKernan <tmeaglei@gmail.com>
This commit is contained in:
Thomas McKernan
2025-05-14 19:06:10 -05:00
committed by Dave Syer
parent 3a7bcdf4e8
commit bd70fcd69f
7 changed files with 85 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
spring-grpc-core/src/main/java/org/springframework/grpc/server/exception/GrpcExceptionHandledServerCall.java Normal file
View File

@@ -0,0 +1,46 @@
/*
* Copyright 2024-2024 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.springframework.grpc.server.exception;
import io.grpc.ForwardingServerCall;
import io.grpc.Metadata;
import io.grpc.ServerCall;
import io.grpc.Status;
import io.grpc.StatusException;
public class GrpcExceptionHandledServerCall<ReqT, RespT>
extends ForwardingServerCall.SimpleForwardingServerCall<ReqT, RespT> {
private final GrpcExceptionHandler exceptionHandler;
protected GrpcExceptionHandledServerCall(ServerCall<ReqT, RespT> delegate, GrpcExceptionHandler handler) {
super(delegate);
this.exceptionHandler = handler;
}
@Override
public void close(Status status, Metadata trailers) {
if (status.getCode() == Status.Code.UNKNOWN && status.getCause() != null) {
final Throwable cause = status.getCause();
final StatusException statusException = this.exceptionHandler.handleException(cause);
super.close(statusException.getStatus(), trailers);
}
else {
super.close(status, trailers);
}
}
}

8
spring-grpc-core/src/main/java/org/springframework/grpc/server/exception/GrpcExceptionHandlerInterceptor.java
View File

@@ -65,16 +65,18 @@ public class GrpcExceptionHandlerInterceptor implements ServerInterceptor {
ServerCallHandler<ReqT, RespT> next) {
Listener<ReqT> listener;
FallbackHandler handler = new FallbackHandler(this.exceptionHandler);
final GrpcExceptionHandledServerCall<ReqT, RespT> exceptionHandledServerCall = new GrpcExceptionHandledServerCall<>(
call, handler);
try {
listener = next.startCall(call, headers);
listener = next.startCall(exceptionHandledServerCall, headers);
}
catch (Throwable t) {
call.close(handler.handleException(t).getStatus(), headers(t));
exceptionHandledServerCall.close(handler.handleException(t).getStatus(), headers(t));
listener = new Listener<ReqT>() {
};
return listener;
}
return new ExceptionHandlerListener<>(listener, call, handler);
return new ExceptionHandlerListener<>(listener, exceptionHandledServerCall, handler);
}
private static Metadata headers(Throwable t) {