TRM-coding | d1cbf67 | 2025-06-18 15:15:08 +0800 | [diff] [blame] | 1 | from app import create_app |
| 2 | from flask_cors import CORS |
TRM-coding | c4b4f3d | 2025-06-18 19:02:46 +0800 | [diff] [blame] | 3 | import os |
| 4 | import psutil |
| 5 | from flask import Flask,g,request |
| 6 | import time |
| 7 | from sqlalchemy import create_engine |
| 8 | from sqlalchemy.orm import sessionmaker |
| 9 | from config import Config |
| 10 | from app.functions.Fpost import Fpost; |
TRM-coding | d1cbf67 | 2025-06-18 15:15:08 +0800 | [diff] [blame] | 11 | |
| 12 | app = create_app() |
TRM-coding | 2a8fd60 | 2025-06-19 19:33:16 +0800 | [diff] [blame] | 13 | CORS(app, |
| 14 | resources={r"/*": {"origins": "*"}}, |
| 15 | supports_credentials=True, |
| 16 | allow_headers=["Content-Type"] |
| 17 | ) |
TRM-coding | d1cbf67 | 2025-06-18 15:15:08 +0800 | [diff] [blame] | 18 | |
TRM-coding | c4b4f3d | 2025-06-18 19:02:46 +0800 | [diff] [blame] | 19 | proc=psutil.Process(os.getpid()) |
| 20 | @app.before_request |
| 21 | def before_request(): |
| 22 | g.start_time=time.time() |
| 23 | g.start_cpu=proc.cpu_times() |
| 24 | g.start_mem=proc.memory_info() |
| 25 | |
| 26 | @app.after_request |
| 27 | def after_request(response): |
| 28 | end_time = time.time() |
| 29 | end_cpu = proc.cpu_times() |
| 30 | end_mem = proc.memory_info() |
| 31 | |
| 32 | elapsed = end_time - g.start_time |
| 33 | cpu_user = end_cpu.user - g.start_cpu.user |
| 34 | cpu_sys = end_cpu.system - g.start_cpu.system |
| 35 | mem_rss = end_mem.rss - g.start_mem.rss |
| 36 | |
| 37 | #写入性能消耗 |
| 38 | engine=create_engine(Config.SQLURL) |
| 39 | SessionLocal = sessionmaker(bind=engine) |
| 40 | session = SessionLocal() |
| 41 | f=Fpost(session) |
| 42 | f.recordsyscost( |
| 43 | request.path, |
| 44 | elapsed, |
| 45 | cpu_user, |
| 46 | cpu_sys, |
| 47 | mem_rss |
| 48 | ) |
| 49 | return response |
TRM-coding | d1cbf67 | 2025-06-18 15:15:08 +0800 | [diff] [blame] | 50 | if __name__ == "__main__": |
| 51 | app.run(debug=True,port=5713,host='0.0.0.0') |