def test_uninstrument_new_brokers(self):
new_broker = LocalBroker()
remoulade.set_broker(new_broker)
new_broker.declare_actor(actor_div)
actor_div.send(1, 1)
spans = self.sorted_spans(self.memory_exporter.get_finished_spans())
self.assertEqual(len(spans), 0)
def local_broker():
broker = LocalBroker()
broker.emit_after("process_boot")
remoulade.set_broker(broker)
yield broker
broker.flush_all()
broker.emit_before("process_stop")
broker.close()
def setUp(self):
super().setUp()
RemouladeInstrumentor().instrument()
broker = LocalBroker()
remoulade.set_broker(broker)
broker.declare_actor(actor_div)
def rabbitmq_broker():
broker = RabbitmqBroker(max_priority=10)
check_rabbitmq(broker)
broker.emit_after("process_boot")
remoulade.set_broker(broker)
yield broker
broker.flush_all()
broker.emit_before("process_stop")
broker.close()
def rabbitmq_broker():
rmq_url = os.getenv("REMOULADE_TEST_RABBITMQ_URL") or "amqp://*****:*****@localhost:5784"
broker = RabbitmqBroker(max_priority=10, url=rmq_url)
check_rabbitmq(broker)
broker.emit_after("process_boot")
remoulade.set_broker(broker)
yield broker
broker.flush_all()
broker.emit_before("process_stop")
broker.close()
import celery
import pylibmc
import remoulade
from remoulade.brokers.rabbitmq import RabbitmqBroker
from remoulade.brokers.redis import RedisBroker
logger = logging.getLogger("example")
counter_key = "latench-bench-counter"
memcache_client = pylibmc.Client(["localhost"], binary=True)
memcache_pool = pylibmc.ClientPool(memcache_client, 8)
random.seed(1337)
if os.getenv("REDIS") == "1":
broker = RedisBroker()
remoulade.set_broker(broker)
celery_app = celery.Celery(broker="redis:///")
else:
broker = RabbitmqBroker(host="127.0.0.1")
remoulade.set_broker(broker)
celery_app = celery.Celery(broker="amqp:///")
def fib_bench(n):
p, q = 0, 1
while n > 0:
p, q = q, p + q
n -= 1
with memcache_pool.reserve() as client:
