From cc9e86b8a45e9b82d9b3bd2759da37f3a0ee7937 Mon Sep 17 00:00:00 2001 From: CW <840459546@qq.com> Date: Tue, 11 Feb 2025 11:11:20 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B3=A8=E9=87=8A=E8=B6=85=E6=97=B6=E9=99=90?= =?UTF-8?q?=E5=88=B6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- flask_app/routes/upload.py | 2 +- flask_app/run_serve.py | 6 ++++-- flask_app/start_up.py | 9 +++++++-- 3 files changed, 12 insertions(+), 5 deletions(-) diff --git a/flask_app/routes/upload.py b/flask_app/routes/upload.py index e01337d..93e551e 100644 --- a/flask_app/routes/upload.py +++ b/flask_app/routes/upload.py @@ -14,7 +14,7 @@ from flask_app.ConnectionLimiter import require_connection_limit upload_bp = Blueprint('upload', __name__) @upload_bp.route('/upload', methods=['POST']) @validate_and_setup_logger -@require_connection_limit(timeout=720) +# @require_connection_limit(timeout=720) def zbparse(): #大解析 logger = g.logger try: diff --git a/flask_app/run_serve.py b/flask_app/run_serve.py index 100de81..14a137a 100644 --- a/flask_app/run_serve.py +++ b/flask_app/run_serve.py @@ -1,6 +1,8 @@ #flask_app/run_serve.py from waitress import serve from flask_app.start_up import create_app + + if __name__ == "__main__": - app=create_app() - serve(app, host='0.0.0.0', port=5000) \ No newline at end of file + app = create_app() + serve(app, host='0.0.0.0', port=5000) diff --git a/flask_app/start_up.py b/flask_app/start_up.py index 1695695..48b7014 100644 --- a/flask_app/start_up.py +++ b/flask_app/start_up.py @@ -9,11 +9,15 @@ from flask_app.routes.upload import upload_bp from flask_app.routes.test_zbparse import test_zbparse_bp from flask_app.general.llm.清除file_id import delete_file_by_ids,read_file_ids from flask_app.routes.judge_zbfile import judge_zbfile_bp + + class FlaskAppWithLimiter(Flask): def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) # 初始化一个字典来存储每个蓝图的限流器 self.connection_limiters = {} + + def create_app(): app = FlaskAppWithLimiter(__name__) # 创建全局日志记录器 @@ -27,11 +31,11 @@ def create_app(): app.connection_limiters['upload'] = ConnectionLimiter(max_connections=10) app.connection_limiters['get_deviation'] = ConnectionLimiter(max_connections=10) app.connection_limiters['default'] = ConnectionLimiter(max_connections=10) - app.connection_limiters['judge_zbfile']=ConnectionLimiter(max_connections=30) + app.connection_limiters['judge_zbfile'] = ConnectionLimiter(max_connections=30) @app.teardown_request def teardown_request(exception): - #接口请求之后都会执行该代码,做一些清理工作 + # 接口请求之后都会执行该代码,做一些清理工作 output_folder = getattr(g, 'output_folder', None) if output_folder: # 执行与output_folder相关的清理操作(例如删除临时文件) @@ -45,6 +49,7 @@ def create_app(): logger.info("清理完毕!") return app + if __name__ == '__main__': app = create_app() app.run(debug=True, host='0.0.0.0', port=5000)