psycopg2.ThreadConnectionPool、uWSGI和Flas的并发问题

Question

我用的是烧瓶，psycopg2和uWSGI。我使用psycopg2.ThreadConnectionPool作为数据库连接池，只有cursor.callproc用于查询数据库。在

问题：有时，在并发请求期间，过程调用结果会被混淆，代码正在查询procedure_1，但却得到procedure_2的结果（对于另一个并发客户端，则相反）。即使在uwsgi.ini中有threads=1，并且只有processes用于并发。^psycopg2.ThreadConnectionPool的{}设置为1。如果processes在uwsgi.ini中设置为1，则问题将消失

什么会导致问题？在

以下是重现问题的简化代码：

烧瓶应用：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import contextlib

import flask
import psycopg2
import psycopg2.pool


app = flask.Flask(__name__)


class ThreadConnectionPool(psycopg2.pool.ThreadedConnectionPool):

    @contextlib.contextmanager
    def item(self):
    close = True
    connection = self.getconn()
    connection.autocommit = True
    try:
        yield connection
        close = False
    finally:
        self.putconn(connection, close=close or connection.closed)


pool = ThreadConnectionPool(maxconn=1, minconn=1, **kwargs)  # kwargs are omitted here 

# sp_1 always returns 1, sp_2 always returns 2
EXPECTED = {'sp_1': 1, 'sp_2': 2}


@app.route('/main/', methods=['GET'])
def main_handler():
    procname = flask.request.args['procname']

    with pool.item() as connection:
        cursor = connection.cursor()
        cursor.callproc(procname)

        rows = cursor.fetchall()

        if rows[0][0] == EXPECTED[procname]:
            return 'OK\n'
        else:
            # procedure returned something that it shouldn't return
            return 'ERROR\n'


if __name__ == '__main__':
    app.run()

在uwsgi.ini文件以下内容：

^{pr2}$

程序：

-- sp_1
BEGIN
    Return 1;
END;


-- sp_2
BEGIN
    Return 2;
END;

复制（不是每次请求都会发生，而是经常发生）：

curl 'http://mymodule.local/main/?procname=sp_1' & curl 'http://mymodule.local/main/?procname=sp_2'
[1] 10636
ERROR
ERROR
[1]+  Done                    curl 'http://mymodule.local/main/?procname=sp_1'

提前谢谢！在

Answer 1

您正在初始化主线程池，在它调用fork（）（以生成worker）之后，所有的线程池都将被弄乱（除非您对其进行管理）。确保为每个工作线程初始化一次池，或者在uWSGI中使用lazy apps=true为每个工作线程加载一次应用程序。在