mirror of
https://ghfast.top/https://github.com/aeroxw/easy-tdx.git
synced 2026-09-12 13:24:15 +08:00
Justin Gu
afcc100e35
EventBus 原本是纯发布/订阅管道，库内没有任何代码调用 publish()，导致 RealtimeStrategy 订阅后无数据可收（issue #34 根因）。 新增 RealtimeDataFeed：轮询 get_stock_quotes 快照 → 封装 MarketEvent → bus.publish，让 RealtimeStrategy 拿来就能跑。 - feed.py: async(run_async) + sync(run_sync, executor 线程避免卡 loop) 支持 interval/dedup(去重)/sessions(交易时段过滤，默认 9:15-11:30/13:00-15:00) symbol key 自动拼 f"{market}{code}" 匹配 EventBus.publish - test_realtime_feed.py: 20 个单测覆盖 publish 路径/symbol key 匹配/ 去重开关/空 df/客户端异常不崩溃/盘外不拉取/stop 退出 - engine.py + __init__.py docstring: 「API 骨架待实现」→「需配合 feed 使用」 - README: 新增实时行情轮询 Python API 段落（含 issue #34 三个坑说明）， WebSocket 段落标注「当前未联动数据源」现状 - examples/11_realtime/: 可直接运行的异步示例 通达信协议无服务端推送，本模块是约 3 秒延迟的快照轮询近似，非逐笔 tick。 验证：ruff + mypy(strict) + pytest 全量 956 passed（+20 新增），无回归。
74 lines
2.6 KiB
Python
74 lines
2.6 KiB
Python
"""演示：RealtimeDataFeed + RealtimeStrategy 实时行情轮询。
|
|
|
|
通达信协议没有服务端推送，只有请求/响应。本示例用 RealtimeDataFeed 轮询
|
|
``get_stock_quotes`` 的五档快照，封装成 MarketEvent 喂给 EventBus，驱动
|
|
RealtimeStrategy 的 on_tick 回调。
|
|
|
|
注意事项：
|
|
- 「实时」是轮询快照近似（默认约 3 秒延迟），不是逐笔 tick。
|
|
- 订阅 key 必须与 publish 内部拼的 f"{market}{code}" 一致，即 "SZ000001"。
|
|
- 传给 subscribe 的必须是实例的绑定方法 strategy.on_tick。
|
|
- 默认仅在 A 股交易时段（9:15-11:30 / 13:00-15:00）轮询；演示用 sessions=()
|
|
全天轮询。
|
|
|
|
使用客户端：AsyncMacClient（异步）
|
|
关键参数：
|
|
- symbols: [(Market.SZ, code), ...]，单批最多 80 只（协议上限）
|
|
- interval: 轮询间隔（秒），默认 3.0
|
|
- dedup: (price, volume) 未变的标的跳过发布，默认 True
|
|
"""
|
|
|
|
import asyncio
|
|
|
|
from easy_tdx import Market
|
|
from easy_tdx.mac.client import AsyncMacClient
|
|
from easy_tdx.realtime import EventBus, MarketEvent, RealtimeDataFeed, RealtimeStrategy
|
|
|
|
|
|
class MyStrategy(RealtimeStrategy):
|
|
"""示例策略：价格超过阈值时打印提醒。"""
|
|
|
|
def __init__(self, threshold: float = 10.0) -> None:
|
|
super().__init__()
|
|
self.threshold = threshold
|
|
|
|
def on_tick(self, event: MarketEvent) -> None:
|
|
flag = " ⚡ 超阈值!" if event.price > self.threshold else ""
|
|
print(
|
|
f"{event.market}{event.code} price={event.price:.2f} "
|
|
f"vol={int(event.volume)} name={event.data.get('name', '')}{flag}"
|
|
)
|
|
|
|
|
|
async def main() -> None:
|
|
bus = EventBus()
|
|
strategy = MyStrategy(threshold=10.5)
|
|
# 注意：key 带市场前缀，且传实例绑定方法
|
|
bus.subscribe("SZ000001", strategy.on_tick)
|
|
bus.subscribe("SH600519", strategy.on_tick)
|
|
|
|
feed = RealtimeDataFeed(
|
|
bus=bus,
|
|
symbols=[(Market.SZ, "000001"), (Market.SH, "600519")],
|
|
interval=3.0,
|
|
dedup=True,
|
|
sessions=(), # 演示用，全天轮询；实盘删掉此行用默认交易时段
|
|
)
|
|
|
|
print("开始轮询（Ctrl+C 退出）...")
|
|
async with AsyncMacClient.from_best_host() as client:
|
|
await feed.run_async(client)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
asyncio.run(main())
|
|
except KeyboardInterrupt:
|
|
print("\n已停止")
|
|
|
|
# 运行结果（示例）:
|
|
# 开始轮询（Ctrl+C 退出）...
|
|
# SZ000001 price=10.52 vol=102345 name=平安银行 ⚡ 超阈值!
|
|
# SH600519 price=1798.00 vol=5234 name=贵州茅台
|
|
# SZ000001 price=10.53 vol=102580 name=平安银行 ⚡ 超阈值!
|