我正在尝试将 Flask 应用程序转换为 Quart 应用程序以添加异步模块并获得一些性能,如本文所述。为此,我使用 acouchbase.Bucket 对象连接到 Couchbase 存储桶。问题是夸脱速度较慢,并且在负载测试期间崩溃。这是我正在尝试的代码:
import asyncio
from quart import Quart, jsonify, g
from quart_openapi import Pint, Resource
import couchbase.experimental
couchbase.experimental.enable()
from acouchbase.bucket import Bucket
app = Pint(__name__, title = 'SomeTitle')
async def get_db():
"""
Helper function to initialize and retrive the Bucket object if not
already present.
"""
if not hasattr(g, 'cb_bucket'):
g.cb_bucket = Bucket('couchbase://localhost/bucketname', 'username', 'password')
await g.cb_bucket.connect()
return g.cb_bucket
@app.route("/apiname/<string:xId>")
class apiname(Resource):
async def get(self, xId):
cb = await get_db()
pickle_in = open('mlmodel', 'rb')
model = pickle.load(pickle_in)
y_pred_proba = model.predict_proba(Member).tolist()
return jsonify(y_pred_proba)
if __name__ == '__main__':
app.run(port = 5000, debug = True)
应用程序编译没有问题,但在负载测试期间它表现不佳并在一段时间后崩溃。一个非常相似的烧瓶应用程序(没有任何异步模块)比 quart 应用程序快,但您会期望带有异步模块的 quart 比烧瓶应用程序快:
Flask 等效项如下所示:
from flask import Flask, jsonify
from flask_restplus import Api, Resource
from couchbase.cluster import Cluster
from couchbase.cluster import PasswordAuthenticator
# Coucbase connections
cluster = Cluster('couchbase://localhost/')
authenticator = PasswordAuthenticator('username', 'password')
cluster.authenticate(authenticator)
cb = cluster.open_bucket('bucketname')
app = Flask(__name__)
api = Api(app=app)
@api.route("/apiname/<string:xId>")
class apiname(Resource):
def get(self, xId):
Member = cb.get(xId)
pickle_in = open('mlmodel', 'rb')
model = pickle.load(pickle_in)
y_pred_proba = model.predict_proba(Member).tolist()
return jsonify(y_pred_proba)
if __name__ == '__main__':
app.run(port = 5000, debug = True)
这是 quart 应用程序与烧瓶应用程序的比较。Flask 看起来比 Quart 应用程序快 10 倍,后者在测试后停止并出现此错误_96930 分段错误_。
任何答案/建议表示赞赏。