added oncancel to handle exception

flipkart-incubator · Jun 26, 2024 · c267fd0 · c267fd0
1 parent 878d515
commit c267fd0
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/guice/src/main/java/com/flipkart/gjex/grpc/interceptor/FilterInterceptor.java b/guice/src/main/java/com/flipkart/gjex/grpc/interceptor/FilterInterceptor.java
@@ -180,6 +180,18 @@ public void onMessage(ReqT request) {
                     detachContext(contextWithHeaders, previous);    // detach headers from gRPC context
                 }
             }
+
+            @Override
+            public void onCancel() {
+                Context previous = attachContext(contextWithHeaders);   // attaching headers to gRPC context
+                try {
+                    super.onCancel();
+                } catch (RuntimeException ex) {
+                    handleException(call, ex);
+                } finally {
+                    detachContext(contextWithHeaders, previous);    // detach headers from gRPC context
+                }
+            }
         };
     }