补充Mock规则命中模式启动参数

This commit is contained in:
wkc
2026-03-22 12:59:12 +08:00
parent 07ed6c76c5
commit e6809c67fe
4 changed files with 67 additions and 0 deletions

View File

@@ -3,6 +3,9 @@
基于 FastAPI 实现的 Mock 服务器,用于模拟流水分析平台的 7 个核心接口
"""
import argparse
import os
from fastapi import FastAPI
from routers import api
from config.settings import settings
@@ -68,9 +71,23 @@ async def health_check():
}
def parse_args(argv=None):
parser = argparse.ArgumentParser()
parser.add_argument("--rule-hit-mode", choices=["subset", "all"], default="subset")
return parser.parse_args(argv)
def apply_rule_hit_mode(rule_hit_mode: str) -> None:
os.environ["RULE_HIT_MODE"] = rule_hit_mode
settings.RULE_HIT_MODE = rule_hit_mode
if __name__ == "__main__":
import uvicorn
args = parse_args()
apply_rule_hit_mode(args.rule_hit_mode)
# 启动服务器
uvicorn.run(
app,