Prechádzať zdrojové kódy

Merge pull request #2580 from fatihsucu/hotfix_for_pymongo

Mongodb default options update in 3.0
PMickael 10 rokov pred
rodič
commit
580ce0efd1
1 zmenil súbory, kde vykonal 8 pridanie a 3 odobranie
  1. 8 3
      celery/backends/mongodb.py

+ 8 - 3
celery/backends/mongodb.py

@@ -78,9 +78,14 @@ class MongoBackend(BaseBackend):
 
         self.url = url
 
-        # default options
-        self.options.setdefault('max_pool_size', self.max_pool_size)
-        self.options.setdefault('auto_start_request', False)
+
+        # default options according to pymongo version
+        if pymongo.version_tuple >= (3,):
+            self.options.setdefault('maxPoolSize', self.max_pool_size)
+        else:
+            self.options.setdefault('max_pool_size', self.max_pool_size)
+            self.options.setdefault('auto_start_request', False)
+
 
         # update conf with mongo uri data, only if uri was given
         if self.url: