kafka-python 3.0.1__tar.gz → 3.0.2__tar.gz

{kafka_python-3.0.1 → kafka_python-3.0.2}/CHANGES.md

@@ -1,3 +1,9 @@
+# 3.0.2 (Jun 18, 2026)
+
+### Fixes
+
+* fix memory leak of cancelled timeout tasks (@azdobylak / #3077)
+
 # 3.0.1 (Jun 17, 2026)
 
 ### Fixes

{kafka_python-3.0.1 → kafka_python-3.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kafka-python
-Version: 3.0.1
+Version: 3.0.2
 Summary: Pure Python client for Apache Kafka
 Author-email: Dana Powers <dana.powers@gmail.com>
 License-Expression: Apache-2.0

{kafka_python-3.0.1 → kafka_python-3.0.2}/kafka/net/selector.py

@@ -122,6 +122,21 @@ class Task:
             raise RuntimeError('Task exception is already set!')
         self._exc = exc
 
+    def close(self):
+        stack = self._stack
+        while stack:
+            coro, stack = stack
+            if inspect.isgenerator(coro) or inspect.iscoroutine(coro):
+                try:
+                    coro.close()
+                except ValueError:
+                    # currently-executing coroutine -- can't close it from
+                    # within itself; bail without corrupting _stack.
+                    return
+                except Exception:
+                    log.exception('Error closing coroutine for cancelled task')
+        self._stack = None
+
     @property
     def is_done(self):
         return self._stack is None
@@ -363,13 +378,29 @@ class NetworkSelector:
             result = await result
         return result
 
-    def unschedule(self, task):
+    def _remove_scheduled(self, task):
         if task.scheduled_at is not None:
-            self._scheduled.remove((task.scheduled_at, task))
+            try:
+                self._scheduled.remove((task.scheduled_at, task))
+            except ValueError:
+                pass
+            else:
+                # re-heapify to ensure heap structure is valid
+                heapq.heapify(self._scheduled)
             task.scheduled_at = None
 
+    def _retire_task(self, task):
+        if task is self._current:
+            return
+        self._pending_tasks.discard(task)
+        task.close()
+
+    def unschedule(self, task):
+        self._remove_scheduled(task)
+        self._retire_task(task)
+
     def reschedule(self, when, task):
-        self.unschedule(task)
+        self._remove_scheduled(task)
         self.call_at(when, task)
         return task
 

kafka_python-3.0.2/kafka/version.py

@@ -0,0 +1 @@
+__version__ = '3.0.2'

{kafka_python-3.0.1 → kafka_python-3.0.2}/kafka_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kafka-python
-Version: 3.0.1
+Version: 3.0.2
 Summary: Pure Python client for Apache Kafka
 Author-email: Dana Powers <dana.powers@gmail.com>
 License-Expression: Apache-2.0

kafka_python-3.0.1/kafka/version.py

@@ -1 +0,0 @@
-__version__ = '3.0.1'