Update streaming_pull_manager.py to address NoneType

A fix for #1238
googleapis · Aug 27, 2024 · a33f49c · a33f49c
1 parent 7cf2d11
commit a33f49c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/google/cloud/pubsub_v1/subscriber/_protocol/streaming_pull_manager.py b/google/cloud/pubsub_v1/subscriber/_protocol/streaming_pull_manager.py
@@ -941,7 +941,12 @@ def _shutdown(self, reason: Any = None) -> None:
             assert self._leaser is not None
             self._leaser.stop()
 
-            total = len(dropped_messages) + len(
+            if dropped_messages is not None:
+                total = len(dropped_messages)
+            else:
+                total = 0
+
+            total = total + len(
                 self._messages_on_hold._messages_on_hold
             )
             _LOGGER.debug(f"NACK-ing all not-yet-dispatched messages (total: {total}).")