瀏覽代碼

bench_worker: Use amqplib if running under PyPy

Ask Solem 13 年之前
父節點
當前提交
7938a95135
共有 1 個文件被更改,包括 5 次插入1 次删除
  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,