merge: 同步 main (辐射本地读 + 传输压缩/网络设置) 到全量分钟分支

冲突两处均为 minute-refresh status 字段两侧各自新增, 两者全保留
(healthy + provider/provider_effective/repair_only)。
This commit is contained in:
shy3130
2026-09-01 13:05:21 +08:00
12 changed files with 513 additions and 27 deletions
+49 -8
View File
@@ -1,6 +1,8 @@
"""K 线 / 同步 API。"""
from __future__ import annotations
import gzip
import json
import logging
import math
from datetime import date, timedelta
@@ -9,7 +11,7 @@ from zoneinfo import ZoneInfo
from functools import lru_cache
from typing import Optional
from fastapi import APIRouter, HTTPException, Query, Request
from fastapi import APIRouter, HTTPException, Query, Request, Response
from app.indicators.pipeline import compute_enriched, compute_enriched_single
from app.market_time import cn_now, cn_today, in_continuous_session
@@ -22,6 +24,41 @@ logger = logging.getLogger(__name__)
router = APIRouter(prefix="/api/kline", tags=["kline"])
def _gzip_payload(request: Request, payload: dict, *, pref_key: str) -> dict | Response:
"""大 JSON 响应的传输压缩: 偏好开启 + 客户端接受 gzip + 响应超阈值才压。
分时/日K批量各自独立偏好键 (网络设置里大开关批量、子开关单独控制)。
level 6 实测 13MB ≈ 290ms CPU 压掉 87%; level 9 要 2.5s 不可用。
datetime → isoformat, 与 FastAPI jsonable_encoder 输出一致
(前端 since 增量按字符串字典序比较, 格式必须与非压缩路径相同)。
"""
from app.services import preferences as _prefs
_getters = {
"minute_batch_compress": _prefs.get_minute_batch_compress,
"daily_batch_compress": _prefs.get_daily_batch_compress,
}
getter = _getters.get(pref_key)
compress_on = False
if getter is not None:
try:
compress_on = bool(getter())
except Exception: # 偏好读取异常按不压缩返回原样
compress_on = False
headers = getattr(request, "headers", None) or {}
if compress_on and "gzip" in (headers.get("accept-encoding") or ""):
raw = json.dumps(
payload, ensure_ascii=False, separators=(",", ":"), allow_nan=True,
default=lambda o: o.isoformat() if hasattr(o, "isoformat") else str(o),
).encode()
if len(raw) > 1024:
return Response(
content=gzip.compress(raw, 6),
media_type="application/json",
headers={"Content-Encoding": "gzip", "Vary": "Accept-Encoding"},
)
return payload
def _minute_allowed(capset) -> bool:
"""是否有分钟K权限 (TickFlow Pro+ 或 custom minute 源)。"""
from app.tickflow.capabilities import Cap
@@ -567,7 +604,8 @@ def get_daily_batch(request: Request, body: dict):
if not sub.is_empty():
result[sub["symbol"][0]] = sub.to_dicts()
return {"data": result}
# 日K批量同为大响应端点 (千只自选 MB 级), 与分时各自独立压缩开关
return _gzip_payload(request, {"data": result}, pref_key="daily_batch_compress")
@router.post("/minute-batch")
@@ -779,12 +817,15 @@ def get_minute_batch(request: Request, body: dict):
}
result = {sym: rows for sym, rows in result.items() if rows}
# full_minute_local: 本轮 prefer_local 生效 (本地分区由全量分钟服务供给, 股票未做补拉)
return {
"data": result,
"full_minute_local": full_minute_healthy,
"incremental": since_dt is not None,
}
return _gzip_payload(
request,
{
"data": result,
"full_minute_local": full_minute_healthy,
"incremental": since_dt is not None,
},
pref_key="minute_batch_compress",
)
@router.get("/minute-range")
+26
View File
@@ -428,6 +428,14 @@ class DataSourceJobTimeoutPrefs(BaseModel):
data_source_long_job_timeout_s: int = Field(ge=60)
class MinuteBatchCompressPrefs(BaseModel):
minute_batch_compress: bool
class DailyBatchCompressPrefs(BaseModel):
daily_batch_compress: bool
class DatasetFieldMapItem(BaseModel):
source: str
target: str
@@ -506,6 +514,8 @@ def get_preferences() -> dict:
"financial_data_provider": preferences.get_financial_provider(),
"data_source_job_timeout_s": preferences.get_data_source_job_timeout_s(),
"data_source_long_job_timeout_s": preferences.get_data_source_long_job_timeout_s(),
"minute_batch_compress": preferences.get_minute_batch_compress(),
"daily_batch_compress": preferences.get_daily_batch_compress(),
**preferences.get_realtime_quote_scope(),
"pipeline_pull_a_share": preferences.get_pipeline_pull_a_share(),
"pipeline_pull_etf": preferences.get_pipeline_pull_etf(),
@@ -793,6 +803,22 @@ def update_data_source_job_timeouts(req: DataSourceJobTimeoutPrefs) -> dict:
return req.model_dump()
@router.put("/preferences/minute-batch-compress")
def update_minute_batch_compress(req: MinuteBatchCompressPrefs) -> dict:
"""保存分时批量响应的 gzip 传输压缩开关。逐请求即时读取, 保存后立即生效。"""
from app.services import preferences
preferences.save({"minute_batch_compress": req.minute_batch_compress})
return {"minute_batch_compress": preferences.get_minute_batch_compress()}
@router.put("/preferences/daily-batch-compress")
def update_daily_batch_compress(req: DailyBatchCompressPrefs) -> dict:
"""保存日K批量响应的 gzip 传输压缩开关 (与分时独立)。逐请求即时读取。"""
from app.services import preferences
preferences.save({"daily_batch_compress": req.daily_batch_compress})
return {"daily_batch_compress": preferences.get_daily_batch_compress()}
@router.put("/preferences/mining-schedule")
def update_mining_schedule(req: MiningSchedulePrefs) -> dict:
"""一次更新周度自动 mining 配置。"""
+2 -1
View File
@@ -358,7 +358,7 @@ class MinuteRefreshService:
def is_healthy(self) -> bool:
"""读侧 freshness 判定: 本地 kline_minute 分区是否正被服务持续写入。
条件: 偏好开启 + 轮询线程存活 + 最近一轮距现在 ≤ max(2×间隔, 30s)。
条件: 偏好开启 + 轮询线程存活 + 最近一轮距现在 ≤ max(2*间隔, 30s)。
健康时消费方 (自选分时等) 可信任本地分区、跳过批量补拉;
连续失败轮不更新 last_round_at → 自动超时判不健康, 无需单独盯 last_error。
"""
@@ -385,6 +385,7 @@ class MinuteRefreshService:
return {
"enabled": enabled,
"running": running,
"healthy": self.is_healthy(),
"provider": self.active_provider(),
"provider_effective": self._resolve_custom()[1],
"repair_only": self.repair_only(),
+30 -7
View File
@@ -9,6 +9,7 @@ import copy
import json
import logging
import re
import threading
from pathlib import Path
logger = logging.getLogger(__name__)
@@ -54,14 +55,22 @@ def load() -> dict:
return copy.deepcopy(_cache)
_SAVE_LOCK = threading.Lock()
def save(updates: dict) -> dict:
"""合并写入。返回新内容。"""
current = load()
current.update(updates)
_path().write_text(
json.dumps(current, indent=2, ensure_ascii=False), encoding="utf-8",
)
_invalidate_cache()
"""合并写入。返回新内容。
锁内 read-modify-write: FastAPI 同步端点跑线程池, 并行 PUT 各自基于旧快照
写盘会互相覆盖 (实测: 压缩总开关并行写分时/日K两键, 后写者把先写者覆盖)。
"""
with _SAVE_LOCK:
current = load()
current.update(updates)
_path().write_text(
json.dumps(current, indent=2, ensure_ascii=False), encoding="utf-8",
)
_invalidate_cache()
return current
@@ -238,6 +247,20 @@ def get_data_source_long_job_timeout_s() -> int:
return max(DATA_SOURCE_JOB_TIMEOUT_MIN_S, timeout_s)
def get_minute_batch_compress() -> bool:
"""分时批量响应是否启用 gzip 传输压缩。默认开启 (公网部署传输是大头);
本机/内网可关闭省服务端 CPU。每次请求即时读取, 开关保存后立即生效。
"""
raw = load().get("minute_batch_compress", True)
return bool(raw)
def get_daily_batch_compress() -> bool:
"""日K批量响应是否启用 gzip 传输压缩 (与分时各自独立配置)。默认开启。"""
raw = load().get("daily_batch_compress", True)
return bool(raw)
def _allowed_data_providers() -> set[str]:
try:
from app.data_providers import custom as custom_sources
+17 -4
View File
@@ -1241,11 +1241,24 @@ class QuoteService:
)
capset = getattr(self._app_state, "capabilities", None)
support = intraday_monitor_support(capset)
if not support["available"] or len(symbols) > int(support["max_symbols"]):
return self._intraday_signal_evaluator.inject(enriched, [])
minute_df = fetch_intraday_monitor_batch(sorted(symbols), capset, now=now)
# 全量分钟健康时股票读本地分区 (服务按间隔持续落盘, 与 API 同一列契约),
# 免去每分钟 bucket 一次的全量 API 拉取; ETF 不在服务 universe 内,
# 本地读空/异常回落原 API 路径 (含能力与上限检查)
minute_df = pl.DataFrame()
if asset_type == "stock":
svc = getattr(self._app_state, "minute_refresh", None) if self._app_state else None
if svc is not None and svc.is_healthy() and self._repo is not None:
try:
minute_df = self._repo.get_minute_batch(sorted(symbols), cn_today())
except Exception as e: # 本地读异常回落 API
logger.warning("分时信号本地读失败, 回退 API 路径: %s", e)
minute_df = pl.DataFrame()
if minute_df.is_empty():
support = intraday_monitor_support(capset)
if not support["available"] or len(symbols) > int(support["max_symbols"]):
return self._intraday_signal_evaluator.inject(enriched, [])
minute_df = fetch_intraday_monitor_batch(sorted(symbols), capset, now=now)
prev_close: dict[str, float] = {}
available_cols = set(enriched.columns)
for row in enriched.filter(pl.col("symbol").is_in(sorted(symbols))).iter_rows(named=True):
@@ -11,6 +11,8 @@ from __future__ import annotations
from types import SimpleNamespace
from typing import ClassVar
import polars as pl
from app.services import quote_service as qs
from app.services.index_const import CORE_INDEX_SYMBOLS
@@ -114,3 +116,76 @@ def test_custom_provider_index_fetch_error_is_soft(monkeypatch):
service, captured = _service_with_provider(monkeypatch, _Boom())
service._fetch_full_market_quotes()
assert len(captured) == 1 and captured[0][0]["symbol"] == "600519.SH"
# ---- 监控分时注入: 全量分钟健康时股票读本地分区 ----
def _injection_env(monkeypatch, *, healthy, local_df, asset_type="stock", symbols=None):
"""构造 _inject_intraday_signals 最小环境, 捕获传入 evaluator 的 minute_df。"""
symbols = symbols or {"600519.SH"}
service = qs.QuoteService()
service._repo = SimpleNamespace(
get_minute_batch=lambda syms, d: local_df,
)
engine = SimpleNamespace(
intraday_signal_symbols=lambda at: set(symbols) if at == asset_type else set(),
)
minute_svc = SimpleNamespace(is_healthy=lambda: healthy)
service._app_state = SimpleNamespace(minute_refresh=minute_svc)
import app.services.quote_service as qsm
api_calls: list[list[str]] = []
monkeypatch.setattr(
qsm, "_noop", qsm.__dict__.get("_noop", None), raising=False) # 占位无操作
from app.services.kline_sync import intraday_monitor_support
monkeypatch.setattr(
"app.services.quote_service.logger", qsm.logger, raising=False)
# 打桩 API 拉取路径 (健康时不应被调)
import app.services.kline_sync as ks
monkeypatch.setattr(
ks, "fetch_intraday_monitor_batch",
lambda symbols, capset, *, now=None: (api_calls.append(list(symbols)), local_df)[1])
monkeypatch.setattr(
ks, "intraday_monitor_support",
lambda capset: {"available": True, "max_symbols": 200, "source": "minute_batch"})
evaluator = SimpleNamespace(
evaluate=lambda minute_df, **kw: (captured.append(minute_df), [])[1],
inject=lambda enriched, signals: enriched,
)
captured: list = []
service._intraday_signal_evaluator = evaluator
service._intraday_signal_bucket = {}
return service, engine, api_calls, captured
def test_intraday_signals_read_local_when_healthy(monkeypatch):
"""健康时股票读本地分区, 不触发 API 拉取。"""
local = pl.DataFrame({
"symbol": ["600519.SH"], "datetime": ["2026-01-15 09:31:00"],
"open": [100.0], "high": [101.0], "low": [99.0], "close": [100.5],
"volume": [1000.0], "amount": [100500.0],
})
service, engine, api_calls, captured = _injection_env(monkeypatch, healthy=True, local_df=local)
enriched = pl.DataFrame({"symbol": ["600519.SH"], "close": [100.5]})
service._inject_intraday_signals(enriched, engine, asset_type="stock")
assert api_calls == [] # 未走 API
assert captured and captured[0].height == 1 # evaluator 拿到本地数据
def test_intraday_signals_fall_back_to_api_when_unhealthy(monkeypatch):
"""不健康 (服务关/挂) 时回落原 API 拉取路径。"""
service, engine, api_calls, captured = _injection_env(monkeypatch, healthy=False, local_df=pl.DataFrame())
enriched = pl.DataFrame({"symbol": ["600519.SH"], "close": [100.5]})
service._inject_intraday_signals(enriched, engine, asset_type="stock")
assert api_calls == [["600519.SH"]] # 走了 API
def test_intraday_signals_etf_never_reads_local(monkeypatch):
"""ETF 不在全量分钟 universe: 即使健康也走 API 路径。"""
service, engine, api_calls, captured = _injection_env(
monkeypatch, healthy=True, local_df=pl.DataFrame(), asset_type="etf", symbols={"510300.SH"})
enriched = pl.DataFrame({"symbol": ["510300.SH"], "close": [4.0]})
service._inject_intraday_signals(enriched, engine, asset_type="etf")
assert api_calls == [["510300.SH"]]
+170
View File
@@ -1011,3 +1011,173 @@ def test_minute_refresh_is_healthy_requires_recent_round(monkeypatch):
monkeypatch.setattr(mr.preferences, "get_minute_refresh_enabled", lambda: False)
svc._state.last_round_at = time_mod.time() - 10
assert svc.is_healthy() is False
# ---------- 测试: 分时批量传输压缩 (网络设置开关) ----------
def test_minute_batch_compress_preference_default_and_toggle(monkeypatch):
"""偏好默认开启; 关闭后 getter 立即反映 (逐请求读取, 无缓存)。"""
from app.services import preferences as prefs
monkeypatch.setattr(prefs, "load", lambda: {})
assert prefs.get_minute_batch_compress() is True
monkeypatch.setattr(prefs, "load", lambda: {"minute_batch_compress": False})
assert prefs.get_minute_batch_compress() is False
def _compress_mock_env(monkeypatch, *, compress_on, accept="gzip, deflate"):
"""构造 get_minute_batch 压缩路径的最小 mock 环境, 返回 (mock_request, sync_spy)。"""
from app.api import kline as kline_api
from app.services import preferences as prefs
monkeypatch.setattr(prefs, "get_minute_batch_compress", lambda: compress_on)
sync_spy = MagicMock(return_value=_mock_minute_df())
monkeypatch.setattr(kline_api.kline_sync, "sync_minute_batch", sync_spy)
mock_repo = MagicMock()
mock_repo.get_etf_symbol_set.return_value = set()
mock_repo.get_minute_batch.return_value = _mock_minute_rows("600519.SH", 100)
mock_capset = MagicMock()
mock_capset.has.return_value = True
mock_capset.limits.return_value = None
mock_request = MagicMock()
mock_request.app.state.repo = mock_repo
mock_request.app.state.capabilities = mock_capset
mock_request.headers = {"accept-encoding": accept} if accept else {}
return mock_request, sync_spy
def test_get_minute_batch_gzip_response_when_enabled(monkeypatch):
"""开关开 + 客户端接受 gzip + 响应超阈值 → 返回 gzip Response, 解压后 JSON 完整。"""
import gzip as gzip_mod
from app.api import kline as kline_api
from fastapi import Response
mock_request, _ = _compress_mock_env(monkeypatch, compress_on=True)
result = kline_api.get_minute_batch(
mock_request, {"symbols": ["600519.SH"], "date": "2026-01-15"}
)
assert isinstance(result, Response)
assert result.headers["content-encoding"] == "gzip"
import json as json_mod
payload = json_mod.loads(gzip_mod.decompress(result.body))
assert payload["full_minute_local"] is False
assert len(payload["data"]["600519.SH"]) > 0
def test_get_minute_batch_plain_when_disabled(monkeypatch):
"""开关关 → 恒返回普通 dict, 不做压缩。"""
from app.api import kline as kline_api
mock_request, _ = _compress_mock_env(monkeypatch, compress_on=False)
result = kline_api.get_minute_batch(
mock_request, {"symbols": ["600519.SH"], "date": "2026-01-15"}
)
assert isinstance(result, dict) and "600519.SH" in result["data"]
def test_get_minute_batch_plain_without_accept_encoding(monkeypatch):
"""开关开但客户端未声明 gzip (如裸 curl) → 尊重协商, 原样返回。"""
from app.api import kline as kline_api
mock_request, _ = _compress_mock_env(monkeypatch, compress_on=True, accept=None)
result = kline_api.get_minute_batch(
mock_request, {"symbols": ["600519.SH"], "date": "2026-01-15"}
)
assert isinstance(result, dict)
# ---------- 测试: 日K批量传输压缩 (与分时独立开关) ----------
def _daily_mock_env(monkeypatch, *, compress_on, accept="gzip, deflate"):
"""构造 get_daily_batch 压缩路径的最小 mock。"""
from app.api import kline as kline_api
from app.services import preferences as prefs
monkeypatch.setattr(prefs, "get_daily_batch_compress", lambda: compress_on)
# 20 根日K (date 列), 足以过 1KB 阈值
n = 20
daily_df = pl.DataFrame({
"symbol": ["600519.SH"] * n,
"date": [date(2026, 1, 1) + timedelta(days=i) for i in range(n)],
"open": [100.0] * n, "high": [101.0] * n,
"low": [99.0] * n, "close": [100.5] * n, "volume": [1000.0] * n,
})
mock_repo = MagicMock()
mock_repo.resolve_asset_type.return_value = "stock"
mock_repo.get_daily_batch.return_value = daily_df
mock_request = MagicMock()
mock_request.app.state.repo = mock_repo
mock_request.headers = {"accept-encoding": accept} if accept else {}
return mock_request
def test_daily_batch_gzip_when_enabled(monkeypatch):
"""日K压缩开 + 接受 gzip → 压缩 Response, 解压 JSON 完整。"""
import gzip as gzip_mod
import json as json_mod
from app.api import kline as kline_api
from fastapi import Response
req = _daily_mock_env(monkeypatch, compress_on=True)
result = kline_api.get_daily_batch(req, {"symbols": ["600519.SH"], "days": 20})
assert isinstance(result, Response)
assert result.headers["content-encoding"] == "gzip"
payload = json_mod.loads(gzip_mod.decompress(result.body))
assert len(payload["data"]["600519.SH"]) == 20
def test_daily_batch_plain_when_disabled(monkeypatch):
from app.api import kline as kline_api
req = _daily_mock_env(monkeypatch, compress_on=False)
result = kline_api.get_daily_batch(req, {"symbols": ["600519.SH"], "days": 20})
assert isinstance(result, dict) and "600519.SH" in result["data"]
def test_daily_batch_independent_from_minute_switch(monkeypatch):
"""日K与分时独立: 分时关、日K开 → 日K仍压缩 (helper 按 pref_key 走各自 getter)。"""
import gzip as gzip_mod
from app.api import kline as kline_api
from app.services import preferences as prefs
from fastapi import Response
monkeypatch.setattr(prefs, "get_minute_batch_compress", lambda: False)
req = _daily_mock_env(monkeypatch, compress_on=True)
result = kline_api.get_daily_batch(req, {"symbols": ["600519.SH"], "days": 20})
assert isinstance(result, Response) and gzip_mod.decompress(result.body)
def test_preferences_parallel_saves_do_not_lose_each_other(tmp_path, monkeypatch):
"""回归: 并行 save 不同键不得互相覆盖 (压缩总开关并行 PUT 两键的竞态)。
save 是 read-modify-write, 无锁时两线程同时基于旧快照写盘,
后写者会把先写者的更新覆盖掉。
"""
import threading
from app.services import preferences as prefs
monkeypatch.setattr(prefs, "_path", lambda: tmp_path / "preferences.json")
prefs._invalidate_cache()
prefs.save({"minute_batch_compress": True})
barrier = threading.Barrier(2)
def write_key(key: str) -> None:
barrier.wait() # 尽量同时进入 save
prefs.save({key: False})
t1 = threading.Thread(target=write_key, args=("minute_batch_compress",))
t2 = threading.Thread(target=write_key, args=("daily_batch_compress",))
t1.start(); t2.start(); t1.join(); t2.join()
final = prefs.load()
assert final["minute_batch_compress"] is False
assert final["daily_batch_compress"] is False