close zk client after unregistering seesion

zalando · Jun 21, 2019 · cda8d12 · cda8d12
1 parent 8a11a83
commit cda8d12
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/main/java/org/zalando/nakadi/service/subscription/state/CleanupState.java b/src/main/java/org/zalando/nakadi/service/subscription/state/CleanupState.java
@@ -30,12 +30,15 @@ public void onEnter() {
         } finally {
             try {
                 getContext().unregisterSession();
-                getContext().getZkClient().close();
-            } catch (IOException e) {
-                getLog().error("Unexpected fail to release zk connection", e);
             } finally {
                 switchState(StreamingContext.DEAD_STATE);
             }
+
+            try {
+                getContext().getZkClient().close();
+            } catch (final IOException e) {
+                getLog().error("Unexpected fail to release zk connection", e);
+            }
         }
     }
 }