python - Flask RESTful API : Issue with db connection pool

Question

我正在尝试使用 Flask 框架创建 REST API 端点。这是我的完整工作脚本:

from flask import Flask, jsonify
from flask_restful import Resource, Api
from flask_restful import reqparse
from sqlalchemy import create_engine
from flask.ext.httpauth import HTTPBasicAuth
from flask.ext.cors import CORS

conn_string = "mssql+pyodbc://x:x@x:1433/x?driver=SQL Server"

auth = HTTPBasicAuth()

@auth.get_password
def get_password(username):
    if username == 'x':
        return 'x'
    return None

app = Flask(__name__)
cors = CORS(app)
api = Api(app)

class Report(Resource):
    decorators = [auth.login_required]

    def get(self):
        parser = reqparse.RequestParser()
        parser.add_argument('start', type = str)
        parser.add_argument('end', type = str)
        args = parser.parse_args()
        e = create_engine(conn_string)
        conn = e.connect()

        stat = """
        select x from report 
        """

        query = conn.execute(stat)

        json_dict = []


        for i in query.cursor.fetchall():
                res = {'x': i[0], 'xx': i[1]}
                json_dict.append(res)


        conn.close()
        e.dispose()
        return jsonify(results=json_dict)


api.add_resource(Report, '/report')

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

问题是，当我仅调用此 API 一天左右时，我会得到结果，此后我将停止获取结果，除非重新启动我的脚本(有时甚至是我的虚拟机)，然后再次得到结果。我认为数据库连接池或其他问题存在一些问题，但我正在关闭连接并处理它。我不知道为什么 API 只在一段时间内给我结果，因为我必须每天重新启动我的虚拟机。有什么想法吗？

Answer 1

根据我的经验，该问题是由编写 create_engine(conn_string) 代码在类 Report 中创建数据库池引起的，以便始终执行数据库的创建和销毁操作适合每个安静要求的游泳池。这不是使用 SQLAlchemy ORM 的正确方法，并且会导致与数据库连接相关的 IO 资源冲突，请参阅下面的 engine.dispose() 函数说明 http://docs.sqlalchemy.org/en/rel_1_0/core/connections.html#sqlalchemy.engine.Engine :

要解决此问题，您只需将 e = create_engine(conn_string) 移至代码下面的 conn_string = "mssql+pyodbc://x:x@x: 1433/x?driver=SQL Server" 并删除类 Report 中的代码 e.dispose()，请参见下文。

conn_string = "mssql+pyodbc://x:x@x:1433/x?driver=SQL Server"
e = create_engine(conn_string) # To here

在def get(delf)函数中:

args = parser.parse_args()
# Move: e = create_engine(conn_string)
conn = e.connect()

和

conn.close()
# Remove: e.dispose()
return jsonify(results=json_dict)