diff options
author | Francis Charette-Migneault <francis.charette.migneault@gmail.com> | 2023-04-18 04:54:17 +0000 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-04-18 10:54:17 +0600 |
commit | f86f1fc6e1caab6bcd2351e3b95424cece6015e4 (patch) | |
tree | c9e1e1e4a50961c941ce6a22f117184309a0f4be /kombu | |
parent | 0dd813a81997a42a1a40477641a5349116662b70 (diff) | |
download | kombu-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.py | 9 |
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) |