Skip to content

Commit ed52c0c

Browse files
authored
lock increment of buffer length (Azure#25685)
1 parent b9d4e81 commit ed52c0c

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

sdk/eventhub/azure-eventhub/azure/eventhub/_buffered_producer/_buffered_producer.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,8 @@ def put_events(self, events, timeout_time=None):
118118
self._buffered_queue.put(self._cur_batch)
119119
self._cur_batch = EventDataBatch(self._max_message_size_on_link)
120120
self._cur_batch.add(events)
121-
self._cur_buffered_len += new_events_len
121+
with self._lock:
122+
self._cur_buffered_len += new_events_len
122123

123124
def failsafe_callback(self, callback):
124125
def wrapper_callback(*args, **kwargs):

sdk/eventhub/azure-eventhub/azure/eventhub/aio/_buffered_producer/_buffered_producer_async.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,8 @@ async def put_events(self, events, timeout_time=None):
118118
self._buffered_queue.put(self._cur_batch)
119119
self._cur_batch = EventDataBatch(self._max_message_size_on_link)
120120
self._cur_batch.add(events)
121-
self._cur_buffered_len += new_events_len
121+
async with self._lock:
122+
self._cur_buffered_len += new_events_len
122123

123124
def failsafe_callback(self, callback):
124125
async def wrapper_callback(*args, **kwargs):

0 commit comments

Comments
 (0)