moved set_flow_control_status out of lock

googleapis · Dec 12, 2023 · d167267 · d167267
1 parent 918b5b9
commit d167267
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/google/cloud/bigtable/batcher.py b/google/cloud/bigtable/batcher.py
@@ -132,7 +132,7 @@ def control_flow(self, batch_info):
         with self._lock:
             self.inflight_mutations += batch_info.mutations_count
             self.inflight_size += batch_info.mutations_size
-            self.set_flow_control_status()
+        self.set_flow_control_status()
 
     def wait(self):
         """
@@ -159,7 +159,7 @@ def release(self, batch_info):
         with self._lock:
             self.inflight_mutations -= batch_info.mutations_count
             self.inflight_size -= batch_info.mutations_size
-            self.set_flow_control_status()
+        self.set_flow_control_status()
 
 
 class MutationsBatcher(object):