From 6292443f69b4772b4cec03f48e6a25287eab8d1a Mon Sep 17 00:00:00 2001 From: huang Date: Wed, 15 Apr 2026 12:22:16 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20PollingBase=20=E6=96=B0=E5=A2=9E=20verb?= =?UTF-8?q?oseLog=20=E6=94=AF=E6=8C=81=EF=BC=8CWorker=20=E4=BC=A0=E5=85=A5?= =?UTF-8?q?=E9=85=8D=E7=BD=AE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit polling_base.go: PollingBase.verboseLog bool 字段，NewPollingBase 新增参数 main.go: cfg.Polling.VerboseLog 作为第6个参数传入 NewPollingBase Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent) Co-authored-by: Sisyphus --- cmd/worker/main.go | 2 +- internal/task/polling_base.go | 5 ++++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/cmd/worker/main.go b/cmd/worker/main.go index f7d71be..da226b4 100644 --- a/cmd/worker/main.go +++ b/cmd/worker/main.go @@ -151,7 +151,7 @@ func main() { pollingQueueMgr := polling.NewPollingQueueManager(redisClient, constants.PollingShardCount, appLogger) pollingIotCardStore := postgres.NewIotCardStore(db, redisClient) - pollingBase := task.NewPollingBase(redisClient, pollingQueueMgr, pollingConfigMgr, pollingIotCardStore, appLogger) + pollingBase := task.NewPollingBase(redisClient, pollingQueueMgr, pollingConfigMgr, pollingIotCardStore, appLogger, cfg.Polling.VerboseLog) // 后台渐进式初始化（将全量卡数据写入分片 Sorted Set） pollingInitializer := polling.NewPollingInitializer(pollingIotCardStore, redisClient, pollingConfigMgr, pollingQueueMgr, appLogger) diff --git a/internal/task/polling_base.go b/internal/task/polling_base.go index 7fbfb4e..243948f 100644 --- a/internal/task/polling_base.go +++ b/internal/task/polling_base.go @@ -29,13 +29,14 @@ var acquireConcurrencyScript = redis.NewScript(` `) // PollingBase 轮询共享基类 -// 封装并发控制、卡缓存、重入队、配置间隔查询等公共方法，4 个 Handler 共享 +// 封装并发控制、卡缓存、重入队、配置间隔查询等公共方法，所有 Handler 共享 type PollingBase struct { redis *redis.Client queueMgr *polling.PollingQueueManager configMgr *polling.PollingConfigManager iotCardStore *postgres.IotCardStore logger *zap.Logger + verboseLog bool } // NewPollingBase 创建轮询共享基类 @@ -45,6 +46,7 @@ func NewPollingBase( configMgr *polling.PollingConfigManager, iotCardStore *postgres.IotCardStore, logger *zap.Logger, + verboseLog bool, ) *PollingBase { return &PollingBase{ redis: redisClient, @@ -52,6 +54,7 @@ func NewPollingBase( configMgr: configMgr, iotCardStore: iotCardStore, logger: logger, + verboseLog: verboseLog, } }