使用多进程连接数据库出现异常 TypeError: can‘t pickle sqlite3.Connection objects 的解决办法

定义了一个DbHelper类来完成数据库操作,如果使用普通的实例化对象没有问题,但如果讲这个实例化的对象用多进程调用就会报“TypeError: can’t pickle sqlite3.Connection objects”的异常

最后的解决办法就是把原来用进程直接调用的类方法封装到一个自定义函数里去

import sqlite3 as sq
import multiprocessing


class DbHelper():
    def __init__(self):
        self.conn = sq.connect('test.db')
        self.cur = self.conn.cursor()

    def select_table(self):
        print('开始查询')
        sql = '''SELECT * FROM stock'''
        self.cur.execute(sql)
        return self.cur.fetchall()

    def close_db(self):
        self.cur.close()
        self.conn.close()

def abc():
    a = DbHelper()
    print(a.select_table())
    a.close_db()


if __name__ == '__main__':
    p1 = multiprocessing.Process(target=abc)
    p2 = multiprocessing.Process(target=abc)
    p3 = multiprocessing.Process(target=abc)
    p1.start()
    p2.start()
    p3.start()

使用进程池时遇到的坑… TypeError: can‘t pickle _thread.lock objects

TypeError: can’t pickle _thread.RLock objects

你可能感兴趣的:(python)