from gevent import monkey

monkey.patch_all()
import asyncio

import time
from celery import Celery
from celery.result import AsyncResult

app = Celery(broker='amqp://xxx', backend='redis://:xxx')

app.config_from_object('config')


@app.task(name='tasks.add')
def add(x, y):
    return x + y


def on_result_ready(result: AsyncResult):
    print('Received result for id %r: %r' % (result.id, result.result,))
    result.forget()
    print("forget end")


if __name__ == "__main__":
    for _ in range(10):
        async def test(my_nym):
            res: AsyncResult = add.apply_async(args=[my_nym, my_nym])
            res.then(on_result_ready)


        asyncio.ensure_future(test(1))
        asyncio.ensure_future(test(2))
        asyncio.ensure_future(test(3))
        asyncio.ensure_future(test(4))

    import asyncio

    asyncio.get_event_loop().run_forever()

__all__ = [
    'patch_all',
    'patch_builtins',
    'patch_dns',
    'patch_os',
    'patch_select',
    'patch_signal',
    'patch_socket',
    'patch_ssl',
    'patch_subprocess',
    'patch_sys',
    'patch_thread',
    'patch_time',
    # query functions
    'get_original',
    'is_module_patched',
    'is_object_patched',
    # module functions
    'main',
]

from gevent import monkey
monkey.patch_all()