Question

我正在编写一个使用multiprocessing.managers.DictProxy的程序。程序遍历一个目录，并创建一个dict，其中username是pwd.getpwuid(os.stat(file)[4])[0]获得的密钥，密钥的相应值将是包含该用户拥有的文件的列表。例如，对于假定的数据结构：

{'root': ["/boot/vmlinuz", "/boot/grub"], 'testuser': ["/home/testuser", "/home/testuser/.bashrc"]}

我写的代码是

#!/usr/bin/python
import os
import multiprocessing
import sys
import pwd
import grp
manager_attributes = multiprocessing.Manager()
file_stats_user = manager_attributes.dict()    
def filestat(file):
  try:
      stat = os.stat(file)
      user = pwd.getpwuid(stat[4])[0]
      group = grp.getgrgid(stat[5])[0]
      if user not in file_stats_user:
          file_stats_user[user] = []
      file_stats_user[user].append(file)
  except OSError, e:
      print e

try:
    cores = (multiprocessing.cpu_count()*2)
except:
    cores = 8
print "Starting parallel execution with ", cores, "concurrency"
pool_get_attributes = multiprocessing.Pool(cores)
pool_get_attributes.map(filestat, files)
pool_get_attributes.close()
pool_get_attributes.join()

其中files是获得的所有文件的列表。

但是，当我打印file_stats_user这是一个multiprocessing.managers.DictProxy对象时，我得到了键，但列表值为空，如

{'root': [], 'testuser': []}

file_stats_user[user].append(file)未将文件名附加到相应的密钥。我在这做错了什么？

谢谢。

Answer 1

编辑如果两个进程同时更新同一用户的dict，则上一个解决方案会出现竞争条件

您的解决方案不起作用，因为默认情况下，列表是当前进程的本地列表，并且对其进行的更新未被代理。

为了使工作，您应该使用一个代理列表，以便可以在所有过程中更新

替换

file_stats_user[user] = []

通过

file_stats_user[user] = manager_attributes.list()

现在当你附加到列表时，它在所有进程中都是相同的

但是使用共享对象是一种做多进程的坏方法

更好的方法是收集每次通话的结果

from collections import defaultdict
def filestat(file):
  try:
      stat = os.stat(file)
      user = pwd.getpwuid(stat[4])[0]
      group = grp.getgrgid(stat[5])[0]
      return user, file
  except OSError, e:
      print e
      return None
try:
    cores = (multiprocessing.cpu_count()*2)
except:
    cores = 8
print "Starting parallel execution with ", cores, "concurrency"
pool_get_attributes = multiprocessing.Pool(cores)
result_map = pool_get_attributes.map(filestat, files)
result = defaultdict(list)
for user,file in (r for r in result_map if r is not None):
    result[user].append(file)
pool_get_attributes.close()
pool_get_attributes.join()

Python多处理DictProxy附加到列表的dict不起作用

1 个答案: