Add back releasing limit in gRPC's onCancel

Netflix · Jul 17, 2019 · b3c5f2b · b3c5f2b
1 parent 9414389
commit b3c5f2b
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/...in/java/com/netflix/concurrency/limits/grpc/server/ConcurrencyLimitServerInterceptor.java b/...in/java/com/netflix/concurrency/limits/grpc/server/ConcurrencyLimitServerInterceptor.java
@@ -202,6 +202,16 @@ public void onHalfClose() {
                                 throw t;
                             }
                         }
+
+                        @Override
+                        public void onCancel() {
+                            try {
+                                super.onCancel();
+                            } finally {
+                                safeComplete(listener::onDropped);
+                            }
+                        }
+
                     };
                 }
             })