当前位置：文江博客话题详情

Python 多处理：自定义进程池

发布于 2024-07-16 07:58:26 字数 130 浏览 1 评论 0原文

我将 Process 类子类化为一个名为 EdgeRenderer 的类。我想使用 multiprocessing.Pool，但我希望它们成为 EdgeRenderer 的实例，而不是常规进程。可能的？如何？

原文

分享到QQ

分享到微博

如果你对这篇内容有疑问，欢迎到本站社区发帖提问参与讨论，获取更多帮助，或者扫码二维码加入 Web 技术交流群。

发布评论

需要登录才能够评论，你可以免费注册一个本站的账号。

有木有妳兜一样 2024-07-23 07:58:26

这似乎有效：

import multiprocessing as mp

ctx = mp.get_context()  # get the default context

class MyProcess(ctx.Process):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        print("Hi, I'm custom a process")

ctx.Process = MyProcess  # override the context's Process

def worker(x):
    print(x**2)

p = ctx.Pool(4)
nums = range(10)
p.map(worker, nums)

This seems to work:

import multiprocessing as mp

ctx = mp.get_context()  # get the default context

class MyProcess(ctx.Process):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        print("Hi, I'm custom a process")

ctx.Process = MyProcess  # override the context's Process

def worker(x):
    print(x**2)

p = ctx.Pool(4)
nums = range(10)
p.map(worker, nums)

回复收藏 0 原文

◇流星雨 2024-07-23 07:58:26

来自杰西·诺勒：

目前不支持
API，但不会是一个糟糕的补充。
我会考虑将其添加到
本周 python2.7/2.6.3 3.1

回复收藏 0 原文

他不在意 2024-07-23 07:58:26

我在 API 中没有看到它的任何钩子。您也许可以使用 initializer 和 initargs 参数来复制所需的功能。或者，您可以将功能构建到用于映射的可调用对象中：

class EdgeRenderTask(object):
    def op1(self,*args):
        ...
    def op2(self,*args):
        ...
p = Pool(processes = 10)
e = EdgeRenderTask()
p.apply_async(e.op1,arg_list)
p.map(e.op2,arg_list)

I don't see any hook for it in the API. You might be able to get away with replicating your desired functionality by using initializer and initargs argument. Alternately, you can build the functionality into the callable object that you use for mapping:

class EdgeRenderTask(object):
    def op1(self,*args):
        ...
    def op2(self,*args):
        ...
p = Pool(processes = 10)
e = EdgeRenderTask()
p.apply_async(e.op1,arg_list)
p.map(e.op2,arg_list)

回复收藏 0 原文

~没有更多了~