Merge pull request #771 from Parsely/bugfix/delete_nonode

catch nonodeerror when relinquishing partition
Parsely · Feb 27, 2018 · e9dd6d2 · e9dd6d2
2 parents a4b6793 + db76558
commit e9dd6d2
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/pykafka/balancedconsumer.py b/pykafka/balancedconsumer.py
@@ -617,8 +617,11 @@ def _remove_partitions(self, partitions):
         :type partitions: Iterable of :class:`pykafka.partition.Partition`
         """
         for p in partitions:
-            # TODO pass zk node version to make sure we still own this node
-            self._zookeeper.delete(self._path_from_partition(p))
+            try:
+                # TODO pass zk node version to make sure we still own this node
+                self._zookeeper.delete(self._path_from_partition(p))
+            except NoNodeException:
+                pass
 
     def _add_partitions(self, partitions):
         """Add partitions to the zookeeper registry for this consumer.