我正在尝试使用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。有什么想法吗?
答案 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()
函数说明:
要解决此问题,您只需将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)