python：如何在具有队列的线程之间共享sqlite连接？

Question

我在 Arch Linux x86_64 上使用 Python 3.2.1。
我正在尝试使用类似于以下内容的一些代码在线程定时循环中更新 sqlite 数据库：

import sqlite3
from threading import Timer
from queue import Queue

class DBQueue(Queue):
    def give(self, item):
        self.task_done()
        self.join()
        self.put(item)
        return True


def timer():
    print('A')
    Timer(3, add).start()


def add():
    print('B')
    db = qdb.get()
    cur = db.cursor()
    cur.execute('INSERT INTO Foo (id) VALUES (NULL)')
    qdb.give(db)
    timer()

qdb = DBQueue()
# SOLUTION #1:
# qdb.put(sqlite3.connect(':memory:', check_same_thread=False))
# SOLUTION #2: see Eli Bendersky's answer
qdb.put(sqlite3.connect(':memory:'))
db = qdb.get()
cur = db.cursor()
cur.execute('CREATE TABLE Foo (id INTEGER PRIMARY KEY)')
qdb.give(db)
timer()

不幸的是返回：

A
B
Exception in thread Thread-1:
Traceback (most recent call last):
  File "/usr/lib/python3.2/threading.py", line 736, in _bootstrap_inner
    self.run()
  File "/usr/lib/python3.2/threading.py", line 942, in run
    self.function(*self.args, **self.kwargs)
  File "/home/dario/dev/python/prova/src/prova4.py", line 27, in add
    cursor = db.cursor()
sqlite3.ProgrammingError: SQLite objects created in a thread can only be used in that same thread.The object was created in thread id 140037302638336 and this is thread id 140037262886656

仅共享游标不会给出更好的结果：

conn = sqlite3.connect(':memory:')
qdb.put(conn.cursor())

我很确定我根本不理解如何使用队列在线程之间共享数据库，有人可以帮助我吗？ 谢谢你！

Answer 1

您不需要队列 - 只需使用两个线程到同一数据库的单独连接即可。请记住，当单独的连接将数据提交到数据库时，您不应该对顺序抱有太多期望。就好像您的程序有两个不同的实例同时访问数据库一样。

---

如果由于某种原因你觉得你绝对必须共享一个连接，那么试试这个：既然你遇到了从一个线程创建 SQLite 对象并在另一个线程中使用它们的问题，为什么不委派处理任务数据库/连接在单个线程中，并让它通过队列与其他线程通信。更具体地说：

• 线程 DB_thread：“拥有”连接。从其他线程获取队列中的命令，执行它们，将结果放入“结果队列”中
• 线程 A、B、C：将“命令”传递到队列中并从“结果队列”中获取结果。

请注意，这些命令不是 SQLite 对象。