add put_many to PeekableQueue

CHANGELOG.md

@@ -2,3 +2,4 @@
 
 * added `synchronize_on_attribute` for `Monitor`
 * added `FutureCollection`
+* added `put_many` to `PeekableQueue`

satella/__init__.py

-__version__ = '2.17.6a3'
+__version__ = '2.17.6a4'

satella/coding/concurrent/queue.py

@@ -32,6 +32,19 @@ class PeekableQueue(tp.Generic[T]):
             self.queue.append(item)
         self.inserted_condition.notify()
 
+    def put_many(self, items: tp.Sequence[T]) -> None:
+        """
+        Put multiple items
+
+        :param items: sequence of items to put
+        """
+        with self.lock:
+            items_count = 0
+            for item in items:
+                items_count += 1
+                self.queue.append(item)
+        self.inserted_condition.notify(items_count)
+
     @rethrow_as(WouldWaitMore, Empty)
     def __get(self, timeout, item_getter) -> T:
         self.lock.acquire()

tests/test_coding/test_concurrent.py

@@ -169,6 +169,14 @@ class TestConcurrent(unittest.TestCase):
         d = si.issue()
         self.assertGreater(d, c)
 
+    def test_peekable_queue_put_many(self):
+        pkb = PeekableQueue()
+
+        pkb.put_many([1, 2])
+
+        self.assertEqual(pkb.get(), 1)
+        self.assertEqual(pkb.get(), 2)
+
     def test_peekable_queue(self):
         pkb = PeekableQueue()