ref #175

LogNet · Oct 27, 2021 · c0994d8 · c0994d8
1 parent 1d6295c
commit c0994d8
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/...g-boot-starter/src/main/java/org/lognet/springboot/grpc/security/SecurityInterceptor.java b/...g-boot-starter/src/main/java/org/lognet/springboot/grpc/security/SecurityInterceptor.java
@@ -215,23 +215,20 @@ public void onMessage(ReqT message) {
                                         METHOD_INVOCATION.get().setArguments(new Object[]{message});
                                         break;
                                     default:
-                                        fail(next,call,headers, new AuthenticationException("Unsupported call type "+call.getMethodDescriptor().getType()) {
-                                        });
+                                        throw  new AuthenticationException("Unsupported call type "+call.getMethodDescriptor().getType()) {};
                                 }
 
                                 beforeInvocation(METHOD_INVOCATION.get());
+                                super.onMessage(message);
                             } catch (AccessDeniedException | AuthenticationException e) {
-                                fail(next, call, headers, e);
-                                return;
+                                failureHandlingSupport.closeCall(e,call,headers);
                             } catch (Exception e) {
-                                fail(next, call, headers, new AuthenticationException("", e) {
-                                });
-                                return;
+                                failureHandlingSupport.closeCall( new AuthenticationException("", e) {},call, headers);
                             } finally {
                                 METHOD_INVOCATION.get().setArguments(null);
                             }
 
-                            super.onMessage(message);
+
                         }
                 );
             }