Flask RESTful API:数据库连接池问题

时间:2015-11-20 05:40:00

标签: python azure flask flask-sqlalchemy flask-restful

我正在尝试使用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时,我得到了结果,之后我停止获取结果,除非我重新启动我的脚本(有时甚至是我的VM),之后我再次获得结果。我认为数据库连接池存在一些问题,但我关闭了连接并将其处理掉了。我不知道为什么API只给我一些时间的结果,因为我必须每天重新启动我的VM。有什么想法吗?

1 个答案:

答案 0 :(得分:0)

根据我的经验,该问题是由编码create_engine(conn_string)在类Report内创建数据库池引起的,因此始终执行create&每个restful请求的db pool的destory操作。这不是使用SQLAlchemy ORM的正确方法,并且导致与数据库连接相关的IO资源冲突,请参阅http://docs.sqlalchemy.org/en/rel_1_0/core/connections.html#sqlalchemy.engine.Engine下面的engine.dispose()函数说明:

enter image description here

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

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)