如何在 django 视图中使用 python 多处理模块

Question

我有一个简单的函数来遍历 URL 列表，使用 GET 检索一些信息并相应地更新数据库 (PostgresSQL)。该功能完美运行。但是，一次一个地浏览每个 URL 会占用太多时间。

使用 python，我可以执行以下操作来并行执行这些任务：

from multiprocessing import Pool

def updateDB(ip):
     code goes here...

if __name__ == '__main__':
    pool = Pool(processes=4)              # process per core
    pool.map(updateDB, ip)

这工作得很好。但是，我试图找到如何在 django 项目上做同样的事情。目前我有一个函数（视图）可以遍历每个 URL 以获取信息并更新数据库。

我唯一能找到的就是使用 Celery，但这对于我想要执行的简单任务来说似乎有点过于强大了。

有什么简单的我可以做或者我必须使用 Celery 吗？

Answer 1

目前我有一个函数（视图）可以遍历每个 URL 以获取 信息，并更新数据库。

这意味着响应时间对您来说并不重要，而不是在后台（异步）执行，如果您的响应时间减少 4（使用 4 个子进程/线程），您可以在前台执行.如果是这种情况，您可以简单地将示例代码放在您的视图中。喜欢

from multiprocessing import Pool

def updateDB(ip):
     code goes here...

def my_view(request):
    pool = Pool(processes=4)              # process per core
    pool.map(updateDB, ip)
    return HttpResponse("SUCCESS")

但是，如果您想在后台异步执行此操作，那么您应该使用 Celery 或遵循@BasicWolf 的建议之一。

Answer 2

虽然使用 Celery 可能看起来有点过头了，但它是一种众所周知的异步任务方式。本质上，Django 服务于 WSGI 请求-响应循环，它对多处理或后台任务一无所知。

以下是替代选项：

• Django background tasks - 可能更适合您的情况。
• Redis queue

Answer 3

我会推荐使用gevent 进行多线程解决方案而不是多处理。在限制生成新进程的生产环境中，多处理可能会导致问题。

示例代码：

from django.shortcuts import HttpResponse
from gevent.pool import Pool

def square(number):
    return number * number

def home(request):
    pool = Pool(50)
    numbers = [1, 3, 5]
    results = pool.map(square, numbers)
    return HttpResponse(results)