Вы можете реализовать декоратор, чтобы сделать ваши функции асинхронными, хотя это немного сложно. multiprocessing
Модуль полна маленьких причуд и , казалось бы , произвольных ограничений - все больше оснований , чтобы инкапсулировать его за дружественный интерфейс, хотя.
from inspect import getmodule
from multiprocessing import Pool
def async(decorated):
r'''Wraps a top-level function around an asynchronous dispatcher.
when the decorated function is called, a task is submitted to a
process pool, and a future object is returned, providing access to an
eventual return value.
The future object has a blocking get() method to access the task
result: it will return immediately if the job is already done, or block
until it completes.
This decorator won't work on methods, due to limitations in Python's
pickling machinery (in principle methods could be made pickleable, but
good luck on that).
'''
# Keeps the original function visible from the module global namespace,
# under a name consistent to its __name__ attribute. This is necessary for
# the multiprocessing pickling machinery to work properly.
module = getmodule(decorated)
decorated.__name__ += '_original'
setattr(module, decorated.__name__, decorated)
def send(*args, **opts):
return async.pool.apply_async(decorated, args, opts)
return send
Код ниже иллюстрирует использование декоратора:
@async
def printsum(uid, values):
summed = 0
for value in values:
summed += value
print("Worker %i: sum value is %i" % (uid, summed))
return (uid, summed)
if __name__ == '__main__':
from random import sample
# The process pool must be created inside __main__.
async.pool = Pool(4)
p = range(0, 1000)
results = []
for i in range(4):
result = printsum(i, sample(p, 100))
results.append(result)
for result in results:
print("Worker %i: sum value is %i" % result.get())
В реальном случае я бы немного подробнее остановился на декораторе, предоставив какой-то способ отключить его для отладки (при сохранении будущего интерфейса на месте) или, возможно, средство для работы с исключениями; но я думаю, что это достаточно хорошо демонстрирует принцип.
async
иawait
синтаксис от 3.5).