More work on fix for #424 - TypeError: can't pickle _thread.lock objects.

AWS and Mongo brockers should be pickeled correctly now.
This commit is contained in:
ihuk
2020-10-16 18:40:26 +02:00
parent 8b3d6b7d2a
commit 5076367168
2 changed files with 9 additions and 0 deletions
+5
View File
@@ -10,6 +10,11 @@ class Sqs(Broker):
super(Sqs, self).__init__(list_key)
self.queue = self.get_queue()
def __setstate__(self, state):
super(Sqs, self).__setstate__(state)
self.sqs = None
self.queue = self.get_queue()
def enqueue(self, task):
response = self.queue.send_message(MessageBody=task)
return response.get("MessageId")
+4
View File
@@ -19,6 +19,10 @@ class Mongo(Broker):
super(Mongo, self).__init__(list_key)
self.collection = self.get_collection()
def __setstate__(self, state):
super(Mongo, self).__setstate__(state)
self.collection = self.get_collection()
@staticmethod
def get_connection(list_key: str = Conf.PREFIX) -> MongoClient:
return MongoClient(**Conf.MONGO)