Make GrpcExceptionHandlerListener inner class

Author: DanielLiu1123

grpc-boot-autoconfigure/grpc-server-boot-autoconfigure/src/main/java/grpcstarter/server/feature/exceptionhandling/ExceptionHandlingServerInterceptor.java

@@ -1,10 +1,13 @@
 package grpcstarter.server.feature.exceptionhandling;
 
+import io.grpc.ForwardingServerCallListener;
 import io.grpc.Metadata;
 import io.grpc.ServerCall;
 import io.grpc.ServerCallHandler;
 import io.grpc.ServerInterceptor;
+import io.grpc.StatusRuntimeException;
 import java.util.List;
+import java.util.Optional;
 import java.util.stream.Collectors;
 import org.springframework.beans.factory.ObjectProvider;
 import org.springframework.core.Ordered;
@@ -37,4 +40,61 @@ public <I, O> ServerCall.Listener<I> interceptCall(
     public int getOrder() {
         return ORDER;
     }
+
+    private static final class GrpcExceptionHandlerListener<I, O>
+            extends ForwardingServerCallListener.SimpleForwardingServerCallListener<I> {
+        private final ServerCall<I, O> call;
+        private final Metadata headers;
+        private final List<GrpcExceptionResolver> grpcExceptionResolvers;
+        private final List<GrpcUnhandledExceptionProcessor> grpcUnhandledExceptionProcessors;
+
+        private GrpcExceptionHandlerListener(
+                ServerCall.Listener<I> delegate,
+                ServerCall<I, O> call,
+                Metadata headers,
+                List<GrpcExceptionResolver> grpcExceptionResolvers,
+                List<GrpcUnhandledExceptionProcessor> grpcUnhandledExceptionProcessors) {
+            super(delegate);
+            this.call = call;
+            this.headers = headers;
+            this.grpcExceptionResolvers = grpcExceptionResolvers;
+            this.grpcUnhandledExceptionProcessors = grpcUnhandledExceptionProcessors;
+        }
+
+        @Override
+        public void onMessage(I message) {
+            try {
+                super.onMessage(message);
+            } catch (Exception e) {
+                if (!handle(e)) {
+                    throw e;
+                }
+            }
+        }
+
+        @Override
+        public void onHalfClose() {
+            try {
+                super.onHalfClose();
+            } catch (Exception e) {
+                if (!handle(e)) {
+                    throw e;
+                }
+            }
+        }
+
+        private boolean handle(Exception e) {
+            for (GrpcExceptionResolver resolver : grpcExceptionResolvers) {
+                StatusRuntimeException sre = resolver.resolve(e, call, headers);
+                if (sre != null) {
+                    call.close(
+                            sre.getStatus(),
+                            Optional.ofNullable(sre.getTrailers()).orElseGet(Metadata::new));
+                    return true;
+                }
+            }
+            grpcUnhandledExceptionProcessors.forEach(processor -> processor.process(e, call, headers));
+            return false;
+        }
+    }
 }

grpc-boot-autoconfigure/grpc-server-boot-autoconfigure/src/main/java/grpcstarter/server/feature/exceptionhandling/GrpcExceptionHandlerListener.java

@@ -23,7 +23,7 @@
  * @author Freeman
  */
 @Getter
-public class GrpcExceptionHandlerMethod {
+class GrpcExceptionHandlerMethod {
 
     private final Object bean;
     private final Integer beanOrder;

grpc-boot-autoconfigure/grpc-server-boot-autoconfigure/src/main/java/grpcstarter/server/feature/exceptionhandling/annotation/GrpcExceptionHandlerMethod.java

@@ -139,80 +139,80 @@ This page was generated by [spring-configuration-property-documenter](https://gi
 
 |Key|Type|Description|Default value|Deprecation|
 |---|----|-----------|-------------|-----------|
-| cert-chain| org.springframework.core.io.Resource| @see TlsServerCredentials.Builder#getCertificateChain()| | | 
-| private-key| org.springframework.core.io.Resource| @see TlsServerCredentials.Builder#getPrivateKey()| | | 
-| private-key-password| java.lang.String| @see TlsServerCredentials.Builder#getPrivateKeyPassword()| | | 
+| cert-chain| org.springframework.core.io.Resource| | | | 
+| private-key| org.springframework.core.io.Resource| | | | 
+| private-key-password| java.lang.String| | | | 
 ### grpc.server.tls.trust-manager
 **Class:** `grpcstarter.server.GrpcServerProperties$Tls$TrustManager`
 
 |Key|Type|Description|Default value|Deprecation|
 |---|----|-----------|-------------|-----------|
-| root-certs| org.springframework.core.io.Resource| @see TlsServerCredentials.Builder#getRootCertificates()| | | 
+| root-certs| org.springframework.core.io.Resource| | | | 
 ### grpc.server.health.datasource
 **Class:** `grpcstarter.server.GrpcServerProperties$Health$DataSource`
 
 |Key|Type|Description|Default value|Deprecation|
 |---|----|-----------|-------------|-----------|
-| enabled| java.lang.Boolean| Whether to enable datasource health check, default true.| true| | 
-| service| java.lang.String| The service name that will be used for datasource health check, default value is 'datasource'.| datasource| | 
-| timeout| java.lang.Integer| \{@link #validationQuery} timeout, unit seconds.| | | 
-| validation-query| java.lang.String| The SQL query that will be used to validate datasource connection, default value is 'SELECT 1'.| SELECT 1| | 
+| enabled| java.lang.Boolean| | | | 
+| service| java.lang.String| | | | 
+| timeout| java.lang.Integer| | | | 
+| validation-query| java.lang.String| | | | 
 ### grpc.server.health.redis
 **Class:** `grpcstarter.server.GrpcServerProperties$Health$Redis`
 
 |Key|Type|Description|Default value|Deprecation|
 |---|----|-----------|-------------|-----------|
-| enabled| java.lang.Boolean| Whether to enable redis health check, default true.| true| | 
-| service| java.lang.String| The service name that will be used for redis health check, default value is 'redis'.| redis| | 
+| enabled| java.lang.Boolean| | | | 
+| service| java.lang.String| | | | 
 ### grpc.server
 **Class:** `grpcstarter.server.GrpcServerProperties`
 
 |Key|Type|Description|Default value|Deprecation|
 |---|----|-----------|-------------|-----------|
-| enable-empty-server| java.lang.Boolean| Whether to start a gRPC server when no service found, default true.| true| | 
-| enabled| java.lang.Boolean| Whether to enable gRPC server autoconfiguration, default true.| true| | 
-| max-inbound-message-size| org.springframework.util.unit.DataSize| The maximum message size allowed to be received on the server, default 4MB. @see GrpcUtil#DEFAULT_MAX_MESSAGE_SIZE| | | 
-| max-inbound-metadata-size| org.springframework.util.unit.DataSize| The maximum size of metadata allowed to be received, default 8KB. @see GrpcUtil#DEFAULT_MAX_HEADER_LIST_SIZE| | | 
-| port| java.lang.Integer| gRPC server port, default 9090, 0 or negative numbers will use random port.| 9090| | 
-| shutdown-timeout| java.lang.Long| Graceful shutdown timeout, default 30s, if 0 will wait forever util all active calls finished.| 30000| | 
+| enable-empty-server| java.lang.Boolean| | | | 
+| enabled| java.lang.Boolean| | | | 
+| max-inbound-message-size| org.springframework.util.unit.DataSize| | | | 
+| max-inbound-metadata-size| org.springframework.util.unit.DataSize| | | | 
+| port| java.lang.Integer| | | | 
+| shutdown-timeout| java.lang.Long| | | | 
 ### grpc.server.channelz
 **Class:** `grpcstarter.server.GrpcServerProperties$Channelz`
 
 |Key|Type|Description|Default value|Deprecation|
 |---|----|-----------|-------------|-----------|
-| enabled| java.lang.Boolean| Whether to register \{@link ChannelzService}, default false.| false| | 
-| max-page-size| java.lang.Integer| The maximum page size to return, default 100. @see AdminInterface| 100| | 
+| enabled| java.lang.Boolean| | | | 
+| max-page-size| java.lang.Integer| | | | 
 ### grpc.server.exception-handling
 **Class:** `grpcstarter.server.GrpcServerProperties$ExceptionHandling`
 
 |Key|Type|Description|Default value|Deprecation|
 |---|----|-----------|-------------|-----------|
-| default-exception-advice-enabled| java.lang.Boolean| Whether to enable \{@link DefaultGrpcExceptionAdvice}, default true. <p> \{@link DefaultGrpcExceptionAdvice} will handle exceptions recognized by gRPC, including: </p> <ul> <li>\{@link StatusRuntimeException}</li> <li>\{@link StatusException}</li> </ul> <p> When enabled, you can directly throw \{@link StatusRuntimeException} or \{@link StatusException} in service implementation, and the exception will be handled by \{@link DefaultGrpcExceptionAdvice}. </p> <pre>\{@code @GrpcService public class SimpleService extends SimpleServiceGrpc.SimpleServiceImplBase \{ @Override public void unaryRpc(SimpleRequest request, StreamObserver<SimpleResponse> responseObserver) \{ throw new StatusRuntimeException(Status.INVALID_ARGUMENT.withDescription("Invalid request")); } } }</pre> @see DefaultGrpcExceptionAdvice @since 3.2.3| true| | 
-| enabled| java.lang.Boolean| Whether to enable exception handling, default true.| true| | 
+| default-exception-advice-enabled| java.lang.Boolean| | | | 
+| enabled| java.lang.Boolean| | | | 
 ### grpc.server.health
 **Class:** `grpcstarter.server.GrpcServerProperties$Health`
 
 |Key|Type|Description|Default value|Deprecation|
 |---|----|-----------|-------------|-----------|
-| enabled| java.lang.Boolean| Whether to enable health check, default false.| true| | 
+| enabled| java.lang.Boolean| | | | 
 ### grpc.server.in-process
 **Class:** `grpcstarter.server.GrpcServerProperties$InProcess`
 
 |Key|Type|Description|Default value|Deprecation|
 |---|----|-----------|-------------|-----------|
-| name| java.lang.String| In-process server name, if configured, will create an in-process server, usually for testing.| | | 
+| name| java.lang.String| | | | 
 ### grpc.server.reflection
 **Class:** `grpcstarter.server.GrpcServerProperties$Reflection`
 
 |Key|Type|Description|Default value|Deprecation|
 |---|----|-----------|-------------|-----------|
-| enabled| java.lang.Boolean| Whether to register reflection service, default false.| false| | 
+| enabled| java.lang.Boolean| | | | 
 ### grpc.server.response
 **Class:** `grpcstarter.server.GrpcServerProperties$Response`
 
 |Key|Type|Description|Default value|Deprecation|
 |---|----|-----------|-------------|-----------|
-| max-description-length| java.lang.Integer| The maximum length of response description. <p> When the length of the description exceeds this value, it will be truncated. </p> @since 3.2.3| 2048| | 
+| max-description-length| java.lang.Integer| | | | 
 ### grpc.server.tls
 **Class:** `grpcstarter.server.GrpcServerProperties$Tls`
 
@@ -318,4 +318,4 @@ This page was generated by [spring-configuration-property-documenter](https://gi
 | enabled| java.lang.Boolean| Whether to enable validation, default is \{@code true}.| true| | 
 | order| java.lang.Integer| Validating interceptor order, default is \{@code 0}.| 0| | 
 
-This is a generated file, generated at: **2024-06-21T21:02:52.512970**
+This is a generated file, generated at: **2024-06-21T21:12:14.230036**