Parcourir la source

bench_worker: Use amqplib if running under PyPy

Ask Solem il y a 13 ans
Parent
commit
7938a95135
1 fichiers modifiés avec 5 ajouts et 1 suppressions
  1. 5 1
      funtests/benchmarks/bench_worker.py

+ 5 - 1
funtests/benchmarks/bench_worker.py

@@ -15,8 +15,12 @@ from celery import Celery
 
 DEFAULT_ITS = 20000
 
+BROKER_TRANSPORT = "librabbitmq"
+if hasattr(sys, "pypy_version_info"):
+    BROKER_TRANSPORT = "amqplib"
+
 celery = Celery(__name__)
-celery.conf.update(BROKER_TRANSPORT="librabbitmq",
+celery.conf.update(BROKER_TRANSPORT=BROKER_TRANSPORT,
                    BROKER_POOL_LIMIT=10,
                    CELERYD_POOL="solo",
                    CELERY_PREFETCH_MULTIPLIER=0,