python - 使用 psycopg2 在 python 上进行多进程

Question

我正在使用 python 将许多记录添加到 Postgres 数据库中，因为我必须插入数百万条记录，我需要并行化 2 个函数:

insert_A(cur)  
insert_B(cur) 

哪里:

conn=psycopg2.connect(.....)
cur = conn.cursor() 

我尝试了一个解决方案，我在此处的另一篇文章中找到了堆栈溢出，例如:

result1= pool.apply_async(insert_A, cur)  
result2= pool.apply_async(insert_B, cur)  
answer1=result1.get(timeout=15)  
answer2=result2.get(timeout=15)  

但是我得到了这个错误:回溯(最近调用最后):

中的文件“test.py”，第 387 行 answer1=result1.get(timeout=15)
文件“/Library/Frameworks/Python.framework/Versions/2.7/lib/python2.7/multiprocessing/pool.py”，第 558 行，在 get
提高 self 值(value)
psycopg2.InterfaceError: 游标没有连接

有人可以帮助我吗？ :(

Answer 1

最后我是这样解决的:

if __name__ == '__main__':
    p = Process(target=insert_A, args=(ARG,))
    q = Process(target=insert_measurement_real, args=(ARG,))
    p.start()
    q.start()
    p.join()
    q.join()

并在函数内部创建连接而不是在函数外部创建它(并将连接作为参数传递)...