Question

我是一个基于defaultdict的类这样的类，它还有一个默认的成员变量（我的猜测：这就是问题所在？）：

from collections import defaultdict
class A(defaultdict):
  def __init__(self):
    super(A, self).__init__(int)
    self.B = defaultdict(int)

  def methodA(self, id):
    # update values in self.B and self[key], for example:
    # self[id] = 1000
    # self[1] = 1
    # self.B[id] = 2000
    # self.B[2] = 2

  def get_B(self):
    return self.B.iteritems()
  def get_dict(self):
    return self.iteritems()

我使用concurrent.futures创建A的不同实例，如下所示：

from concurrent.futures import ProcessPoolExecutor, as_completed

futures = []
num_processes = 2 #some integer value

def process_parallel(id):
  A_1 = A()
  A_1.methodA(id)
  return A_1

with ProcessPoolExecutor() as exec:
  for p in range(num_processes):
    futures.append(exec.submit(process_parallel, p))

for f in as_completed(futures):
  A_instance = f.result()
  # iterate over A_instance.get_dict() and A_instance.get_B
  # A_instance[1] can be accessed, but A_instance.B is empty

问题是在期货完成并返回结果后，B成员变量不包含任何内容。要检查，我确保B非空，甚至可以在methodA的末尾打印其元素，但一旦回到主流程中，突然B似乎是走了。

B应单独初始化吗？这样做的正确方法是什么？

Answer 1

我认为这里发生的事情是ProcessPoolExecutor使用pickle来序列化和反序列化您的类实例，并且不保留defaultdict的内容，即实例的'B'属性。

您可以通过在解释器中创建和更新类的实例，对它们进行pickle和unpickling以及检查结果来验证这一点。

如果你创建一个容器类来保存两个默认值，它们都能保持酸洗并且内容完好无损，所以这可能是最好的方法。例如（假设python3）：

class A:

    def __init__(self):
        self.foo = defaultdict(int)
        self.bar = defaultdict(int)

    def baz(self, value):
        # update defaultdicts

ProcessPoolExecutor：返回的对象中丢失的成员变量

1 个答案: