需求,一个异步脚本里大量使用run_in_executor()
封装同步调用,然后脚本本身用了 os.fork()搞出很多子进程。理论上最好所有子进程共享同一个线程池而不是每个进程单独拥有一个线程池。
import asyncio, time, os
from concurrent.futures import ThreadPoolExecutor
executor = ThreadPoolExecutor()
def sync_call(x):
time.sleep(0.1)
return x+1
async def main():
loop = asyncio.get_running_loop()
while True:
res = await asyncio.gather(*(loop.run_in_executor(executor, sync_call, x) for x in range(100)))
assert res == [_ for _ in range(1,101)]
print(True)
if __name__ == '__main__':
for _ in range(16):
pid = os.fork()
if pid == 0:
break
asyncio.run(main())
简化下来大概是这样的代码,其中线程池是自己几年前写的,当时确保了在单进程内使用多线程的情况下是绝对安全的,现在已经看不懂代码了,怎样才能知道多进程的情况下是否能确保安全呢?
上文这个代码拿来跑当测试,倒是也没报错,但是我怎么知道它是不是刚好没遇到特殊情况所以没坑。。
这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。
V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。
V2EX is a community of developers, designers and creative people.