584cfc3db2
- 将高密度服务与处理器日志迁移到新日志系统（LegacyPrintf/结构化日志） - 增加 stdlog bridge 与兼容测试，保留旧日志捕获能力 - 将 OpenAI 断流告警改为结构化 Warn 并改造对应测试为 sink 捕获 - 补齐后端相关文件 logger 引用并通过全量 go test
73 lines
2.1 KiB
Go
73 lines
2.1 KiB
Go
package service
|
|
|
|
import (
|
|
"fmt"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/Wei-Shaw/sub2api/internal/pkg/logger"
|
|
"github.com/zeromicro/go-zero/core/collection"
|
|
)
|
|
|
|
var newTimingWheel = collection.NewTimingWheel
|
|
|
|
// TimingWheelService wraps go-zero's TimingWheel for task scheduling
|
|
type TimingWheelService struct {
|
|
tw *collection.TimingWheel
|
|
stopOnce sync.Once
|
|
}
|
|
|
|
// NewTimingWheelService creates a new TimingWheelService instance
|
|
func NewTimingWheelService() (*TimingWheelService, error) {
|
|
// 1 second tick, 3600 slots = supports up to 1 hour delay
|
|
// execute function: runs func() type tasks
|
|
tw, err := newTimingWheel(1*time.Second, 3600, func(key, value any) {
|
|
if fn, ok := value.(func()); ok {
|
|
fn()
|
|
}
|
|
})
|
|
if err != nil {
|
|
return nil, fmt.Errorf("创建 timing wheel 失败: %w", err)
|
|
}
|
|
return &TimingWheelService{tw: tw}, nil
|
|
}
|
|
|
|
// Start starts the timing wheel
|
|
func (s *TimingWheelService) Start() {
|
|
logger.LegacyPrintf("service.timing_wheel", "%s", "[TimingWheel] Started (auto-start by go-zero)")
|
|
}
|
|
|
|
// Stop stops the timing wheel
|
|
func (s *TimingWheelService) Stop() {
|
|
s.stopOnce.Do(func() {
|
|
s.tw.Stop()
|
|
logger.LegacyPrintf("service.timing_wheel", "%s", "[TimingWheel] Stopped")
|
|
})
|
|
}
|
|
|
|
// Schedule schedules a one-time task
|
|
func (s *TimingWheelService) Schedule(name string, delay time.Duration, fn func()) {
|
|
if err := s.tw.SetTimer(name, fn, delay); err != nil {
|
|
logger.LegacyPrintf("service.timing_wheel", "[TimingWheel] SetTimer failed for %q: %v", name, err)
|
|
}
|
|
}
|
|
|
|
// ScheduleRecurring schedules a recurring task
|
|
func (s *TimingWheelService) ScheduleRecurring(name string, interval time.Duration, fn func()) {
|
|
var schedule func()
|
|
schedule = func() {
|
|
fn()
|
|
if err := s.tw.SetTimer(name, schedule, interval); err != nil {
|
|
logger.LegacyPrintf("service.timing_wheel", "[TimingWheel] recurring SetTimer failed for %q: %v", name, err)
|
|
}
|
|
}
|
|
if err := s.tw.SetTimer(name, schedule, interval); err != nil {
|
|
logger.LegacyPrintf("service.timing_wheel", "[TimingWheel] initial SetTimer failed for %q: %v", name, err)
|
|
}
|
|
}
|
|
|
|
// Cancel cancels a scheduled task
|
|
func (s *TimingWheelService) Cancel(name string) {
|
|
_ = s.tw.RemoveTimer(name)
|
|
}
|