53 lines
1.8 KiB
Python
53 lines
1.8 KiB
Python
"""Ip_sound_column Worker 插件
|
||
|
||
统一风格模板 — 基于 chongming_worker.plugin.Plugin 基类。
|
||
|
||
工作流:
|
||
1. 编辑 config/workers/ip_sound_column/*.toml 定义 RPC 端点
|
||
2. uv run gen_models config/workers/ip_sound_column src/plugins/ip_sound_column
|
||
3. 在 router/<操作>.py 中实现业务逻辑
|
||
4. 在 on_init / on_cleanup 中初始化 / 清理资源
|
||
"""
|
||
import logging
|
||
|
||
from chongming_worker.plugin import Plugin
|
||
|
||
from . import router # noqa: F401 触发 @rpc_method 注册
|
||
# from .routes import router # 可选: 自定义 FastAPI 路由（网关自动发现）
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
class IpSoundColumnPlugin(Plugin):
|
||
"""ip_sound_column 插件 — 统一生命周期。"""
|
||
|
||
type = "ip_sound_column"
|
||
db_name = "ip_sound_column.db" # 自动在 data_dir 下建 SQLite 库（不需要则置 None）
|
||
auto_bind = True # 首次初始化时自动 auto_bind_models(self.type)
|
||
|
||
async def on_init(self):
|
||
"""首次 create_workers 时调用一次：初始化资源、启动后台任务。
|
||
|
||
数据库已由基类 _bootstrap 自动初始化（self.db）。
|
||
"""
|
||
config = self.config
|
||
logger.info(
|
||
"[ip_sound_column] 初始化开始 (count=%s, ip=%s)",
|
||
config.get("count", 1),
|
||
config.get("ip", "unknown")
|
||
)
|
||
ip = config.get("ip", "unknown")
|
||
self.set_var("ip", ip)
|
||
|
||
logger.info("[ip_sound_column] 初始化完成")
|
||
|
||
async def on_cleanup(self):
|
||
"""cleanup 时调用：关闭自定义客户端等（后台任务与数据库由基类处理）。"""
|
||
logger.info("[ip_sound_column] 清理完成")
|
||
|
||
|
||
# 暴露给 app lifespan 的模块级入口（bound method，签名与旧写法完全一致）
|
||
_plugin = IpSoundColumnPlugin()
|
||
create_workers = _plugin.create_workers
|
||
cleanup = _plugin.cleanup
|