fix: 修复禁用所有轮询配置重启后再启用无法工作的问题
All checks were successful
构建并部署到测试环境(无 SSH) / build-and-deploy (push) Successful in 7m23s
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:
@@ -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) {
|
||||
|
||||
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user