数据库连接 - Python多线程使用同一个MySQL连接，如何在所有线程工作完成后关闭游标、连接？

Question

def getAllItems(startNum, endNum, tcount = 10):
    for torder in range(tcount):
        t = MyThread(torder, tcount, startNum, endNum)
        t.start()
    dbUtils.closeCur()
    dbUtils.closeConn()
    print "INFO: cursor and connection closed, work done!"

如上代码，MyThread定义了工作内容，其中会调用一个全局的数据库连接对象（dbUtils的成员变量），我想让最后三行可以在所有线程工作完成后做收尾工作，但目前的写法是错误的，一旦开启进程，最后三行就运行了，把数据库连接关闭了，然后前面的所有线程就死锁了。

求问怎么达成我需要的目标呢？谢谢。

Answer 1

用join来等待线程结束:

def getAllItems(startNum, endNum, tcount = 10):
    threads = []
    for torder in range(tcount):
        t = MyThread(torder, tcount, startNum, endNum)
        t.start()
        threads.append(t)
    for t in threads:
        t.join()
    dbUtils.closeCur()
    dbUtils.closeConn()
    print "INFO: cursor and connection closed, work done!"

Answer 2

import threading
import time

def worker():
    time.sleep(1)
    print('done!')

if __name__ == '__main__':
    threads = [threading.Thread(target=worker) for _ in range(5)]
    for t in threads:
        t.start()
    for t in threads:
        t.join()
    print('all done!')

调用Thread.join()就可以了