celery/funtests/benchmarks/reqi.py

from celery import current_app, task, uuid
from celery.worker.consumer import Consumer
#from celery.worker.job import Request
from celery.app.task import Context
from celery.concurrency.solo import TaskPool
from celery.app.amqp import TASK_BARE
from time import time
from Queue import Queue
from librabbitmq import Message
from celery.utils.functional import noop
from celery.worker.job import NEEDS_KWDICT
from celery.datastructures import AttributeDict
import socket
import sys

@task(accept_magic_kwargs=False)
def T():
    pass

class Request(object):
    #__slots__ = ('app', 'name', 'id', 'args', 'kwargs',
    #             'on_ack', 'delivery_info', 'hostname',
    #             'eventer', 'connection_errors',
    #             'task', 'eta', 'expires', 'flags',
    #             'request_dict', 'acknowledged',
    #             'worker_pid', 'started',
    #             '_already_revoked', '_terminate_on_ack', '_tzlocal')
    eta = None
    started = False
    acknowledged = _already_revoked = False
    worker_pid = _terminate_on_ack = None
    _tzlocal = None
    expires = None
    delivery_info = {}
    flags = 0
    args = ()

    def __init__(self, body, on_ack=noop,
            hostname=None, eventer=None, app=None,
            connection_errors=None, request_dict=None,
            delivery_info=None, task=None, Context=Context, **opts):
        self.app = app
        self.name = body['task']
        self.id = body['id']
        self.args = body['args']
        try:
            self.kwargs = body['kwargs']
            if NEEDS_KWDICT:
                self.kwargs = kwdict(self.kwargs)
        except KeyError:
            self.kwargs = {}
        try:
            self.flags = body['flags']
        except KeyError:
            pass
        self.on_ack = on_ack
        self.hostname = hostname
        self.eventer = eventer
        self.connection_errors = connection_errors or ()
        self.task = task or self.app._tasks[self.name]
        if 'eta' in body:
            eta = body['eta']
            tz = tz_utc if utc else self.tzlocal
            self.eta = tz_to_local(maybe_iso8601(eta), self.tzlocal, tz)
        if 'expires' in body:
            expires = body['expires']
            tz = tz_utc if utc else self.tzlocal
            self.expires = tz_to_local(maybe_iso8601(expires),
                                       self.tzlocal, tz)
        if delivery_info:
            self.delivery_info = {
                'exchange': delivery_info.get('exchange'),
                'routing_key': delivery_info.get('routing_key'),
            }

        self.request_dict = AttributeDict(
                {'called_directly': False,
                 'callbacks': [],
                 'errbacks': [],
                 'chord': None}, **body)




tid = uuid()
hostname = socket.gethostname()
task = {'task': T.name, 'args': (), 'kwargs': {}, 'id': tid, 'flags': 0}
app = current_app._get_current_object()

m = Message(None, {}, {}, task)

ts = time()
for i in xrange(1000000):
    x = Request(task, hostname=hostname, app=app, task=task)
print(time() - ts)