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, } }