SQLite3 请求在 Twisted 中挂起

SQLite3 Request Hangs in Twisted

我正在尝试编写一个脚本来自动更新数据库的架构。但是,由于某种原因,在我对 adbapi.ConnectionPool 发出的第二个请求期间,Twisted 挂起。这是代码:

update.py

import os
import glob
import imp

from twisted.internet import reactor
from twisted.enterprise import adbapi
from twisted.internet import defer

@defer.inlineCallbacks
def get_schema_version(conn):
    schema_exists = yield conn.runQuery("SELECT name FROM sqlite_master WHERE type='table' AND name='schema_meta';")
    defer.returnValue(0)

def add_schema_files(schemas):
    # Finds and imports all schema_*.py files into the list
    module_files = glob.glob(os.path.dirname(os.path.abspath(__file__)) + "/schema_*.py")
    for mod in module_files:
        module_name = os.path.basename(os.path.splitext(mod)[0])
        newmod = imp.load_source('%s'%module_name, mod)
        schemas.append( (module_name, newmod) )

@defer.inlineCallbacks
def update_schema(conn):
    # Update the database schema to the latest version
    schema_version = yield get_schema_version(conn)
    print "At schema version %d" % schema_version
    schemas = []
    add_schema_files(schemas)
    schemas = sorted(schemas, key=lambda tup: tup[0])
    for i in range(schema_version, len(schemas)):
        # schemas[0] is v1, schemas[1] is v2, etc
        print "Updating to version %d" % (i+1)
        yield schemas[i][1].update(conn)

if __name__ == '__main__':
    conn = adbapi.ConnectionPool("sqlite3", "data.db", check_same_thread=False)
    d = update_schema(conn)
    d.addCallback(exit)
    reactor.run()

schema_1.py

from twisted.internet import defer

update_query = """
CREATE TABLE schema_meta (
version INT NOT NULL
);
INSERT INTO schema_meta (version) VALUES (1);
"""

@defer.inlineCallbacks
def update(conn):
    yield conn.runQuery(update_query)

它挂在 schema_1.py 中的 yield conn.runQuery(update_query)

此外,当我中断脚本时,出现以下 sqlite 错误:

sqlite3.Warning: You can only execute one statement at a time.

SQLite 允许您一次 运行 编写一个查询。您的代码是异步的,可以在第一个查询完成之前启动一个新查询。

要处理这个问题,您必须序列化查询或使用数据库服务器而不是 SQLite。

事实证明,问题不在于扭曲,而在于 SQLite 查询。您一次只能执行一个查询,所以我更新了 schema_1.py 并且成功了。

schema_1.py

from twisted.internet import defer

update_query1 = """
CREATE TABLE schema_meta (
version INT NOT NULL
);
"""

update_query2 = """
INSERT INTO schema_meta (version) VALUES (1);
"""

@defer.inlineCallbacks
def update(dbpool):
    yield dbpool.runQuery(update_query1)
    yield dbpool.runQuery(update_query2)