fix: 修复禁用所有轮询配置重启后再启用无法工作的问题
All checks were successful
构建并部署到测试环境(无 SSH) / build-and-deploy (push) Successful in 7m23s

- 新增 RedisPollingConfigChangedChannel 常量,用于跨进程配置变更通知
- ConfigService 注入 Redis,Create/Delete/UpdateStatus 后发布 Pub/Sub 事件
- PollingConfigManager 新增 WatchChanges() 方法,收到通知立即刷新内存缓存
- PollingInitializer 新增 Restart() 方法,支持初始化完成后安全重启(CAS 防并发)
- Worker 订阅配置变更事件,configs 从空变为非空时触发 Initializer.Restart()

根因:启动时所有配置禁用导致分片队列为空,Initializer 是一次性的,
之后启用配置只更新 DB 但不重建队列,调度器无卡可处理
This commit is contained in:
2026-04-16 17:32:43 +08:00
parent 0ec16d4afa
commit aef20d2ab1
6 changed files with 86 additions and 3 deletions

View File

@@ -63,6 +63,45 @@ func (m *PollingConfigManager) Load(ctx context.Context) error {
return nil
}
// WatchChanges 订阅 Redis 配置变更频道,收到通知后立即刷新内存缓存
// onRefreshed 回调参数:(变更前是否有配置, 变更后是否有配置)
// 当从无配置变为有配置时,调用方可据此触发 Initializer 重启
func (m *PollingConfigManager) WatchChanges(ctx context.Context, onRefreshed func(hadConfigs, hasConfigs bool)) {
go func() {
pubsub := m.redis.Subscribe(ctx, constants.RedisPollingConfigChangedChannel())
defer pubsub.Close()
ch := pubsub.Channel()
for {
select {
case <-ctx.Done():
return
case msg, ok := <-ch:
if !ok {
return
}
m.mu.RLock()
hadConfigs := len(m.configs) > 0
m.mu.RUnlock()
m.logger.Info("收到配置变更通知,立即刷新内存缓存", zap.String("event", msg.Payload))
if err := m.Load(ctx); err != nil {
m.logger.Warn("配置变更后刷新缓存失败,等待下次定时刷新", zap.Error(err))
continue
}
m.mu.RLock()
hasConfigs := len(m.configs) > 0
m.mu.RUnlock()
if onRefreshed != nil {
onRefreshed(hadConfigs, hasConfigs)
}
}
}
}()
}
// Start 启动定时刷新(每 5 分钟自动 Load 一次)
// 使用 sync.Once 确保只启动一个刷新 goroutine
func (m *PollingConfigManager) Start(ctx context.Context) {

View File

@@ -94,6 +94,19 @@ func (p *PollingInitializer) IsCompleted() bool {
return p.initCompleted.Load()
}
// Restart 重新执行初始化(当轮询配置从空变为非空时调用)
// 使用 CAS 确保只有初始化已完成时才能重启,避免并发重入
func (p *PollingInitializer) Restart(ctx context.Context) {
if !p.initCompleted.CompareAndSwap(true, false) {
p.logger.Info("轮询初始化仍在进行中,跳过重启")
return
}
p.setStatus("pending", "")
p.wg.Add(1)
go p.run(ctx)
p.logger.Info("轮询初始化已重新启动(配置变更触发)")
}
// GetProgress 返回当前初始化进度快照(加锁读取,返回值拷贝)
func (p *PollingInitializer) GetProgress() InitProgress {
p.progress.mu.RLock()