summaryrefslogtreecommitdiff
path: root/kombu
diff options
context:
space:
mode:
authorFrancis Charette-Migneault <francis.charette.migneault@gmail.com>2023-04-18 04:54:17 +0000
committerGitHub <noreply@github.com>2023-04-18 10:54:17 +0600
commitf86f1fc6e1caab6bcd2351e3b95424cece6015e4 (patch)
treec9e1e1e4a50961c941ce6a22f117184309a0f4be /kombu
parent0dd813a81997a42a1a40477641a5349116662b70 (diff)
downloadkombu-f86f1fc6e1caab6bcd2351e3b95424cece6015e4.tar.gz
fix mongodb transport obsolete calls (#1694)
* fix mongodb transport obsolete calls + add test mock specs to check valid methods * fix linting
Diffstat (limited to 'kombu')
-rw-r--r--kombu/transport/mongodb.py9
1 files changed, 5 insertions, 4 deletions
diff --git a/kombu/transport/mongodb.py b/kombu/transport/mongodb.py
index b923f5f4..9eef6b57 100644
--- a/kombu/transport/mongodb.py
+++ b/kombu/transport/mongodb.py
@@ -65,6 +65,7 @@ class BroadcastCursor:
def __init__(self, cursor):
self._cursor = cursor
+ self._offset = 0
self.purge(rewind=False)
def get_size(self):
@@ -77,7 +78,7 @@ class BroadcastCursor:
if rewind:
self._cursor.rewind()
- # Fast forward the cursor past old events
+ # Fast-forward the cursor past old events
self._offset = self._cursor.collection.count_documents({})
self._cursor = self._cursor.skip(self._offset)
@@ -221,7 +222,7 @@ class Channel(virtual.Channel):
if queue in self._fanout_queues:
self._get_broadcast_cursor(queue).purge()
else:
- self.messages.remove({'queue': queue})
+ self.messages.delete_many({'queue': queue})
return size
@@ -257,10 +258,10 @@ class Channel(virtual.Channel):
self.routing.update_one(lookup, {'$set': data}, upsert=True)
def queue_delete(self, queue, **kwargs):
- self.routing.remove({'queue': queue})
+ self.routing.delete_many({'queue': queue})
if self.ttl:
- self.queues.remove({'_id': queue})
+ self.queues.delete_one({'_id': queue})
super().queue_delete(queue, **kwargs)