feat: 添加异步数据入库任务系统,支持全量/增量更新与实时进度展示

新增完整的异步任务管理体系:
1. 新增Redis缓存层实现任务状态实时存储与降级机制
2. 实现并发拉取+批量写入的高性能入库流程
3. 添加任务管理HTTP接口与路由
4. 重构行情控制器使用共享SQLite存储
5. 配套添加Docker Compose Redis启动配置
6. 前端新增任务管理面板支持操作与进度查看
This commit is contained in:
2026-08-18 10:09:52 +08:00
parent e28bfec957
commit 273fc90a95
18 changed files with 1682 additions and 80 deletions

187
internal/job/job.go Normal file
View File

@@ -0,0 +1,187 @@
package job
import (
"context"
"encoding/json"
"log"
"sync"
"time"
"github.com/google/uuid"
"stock-analysis/internal/cache"
"stock-analysis/internal/datasource"
)
// Config Manager 配置。
type Config struct {
Workers int // 并发 worker 数,默认 8
BatchSize int // 批量写入阈值,默认 20
RateInterval time.Duration // 全局限速间隔,默认 400ms(≈2.5 req/s)
RateBurst int // 限速突发额度,默认 4
}
// runningTask 记录运行中任务的取消句柄和状态快照。
type runningTask struct {
cancel context.CancelFunc
task *cache.Task
}
// Manager 管理任务注册表和全局限速器,允许多任务并发运行。
type Manager struct {
store Store
cache cache.Cache
ds datasource.MarketDataSource
lister BondLister
gate <-chan struct{} // 全局限速令牌(跨任务共享)
workers int
batchSize int
mu sync.RWMutex
running map[string]*runningTask
}
// NewManager 创建任务管理器,内部启动全局限速器。
func NewManager(store Store, c cache.Cache, ds datasource.MarketDataSource, lister BondLister, cfg Config) *Manager {
if cfg.Workers <= 0 {
cfg.Workers = 8
}
if cfg.BatchSize <= 0 {
cfg.BatchSize = 20
}
if cfg.RateInterval <= 0 {
cfg.RateInterval = 400 * time.Millisecond
}
if cfg.RateBurst <= 0 {
cfg.RateBurst = 4
}
return &Manager{
store: store,
cache: c,
ds: ds,
lister: lister,
gate: newRateLimiter(cfg.RateInterval, cfg.RateBurst),
workers: cfg.Workers,
batchSize: cfg.BatchSize,
running: make(map[string]*runningTask),
}
}
// Start 创建并异步启动一个任务,立即返回任务快照(pending 状态)。
func (m *Manager) Start(ctx context.Context, typ cache.TaskType, since string, limit int) (*cache.Task, error) {
params, _ := json.Marshal(map[string]any{"since": since, "limit": limit})
task := &cache.Task{
ID: uuid.New().String(),
Type: typ,
Status: cache.TaskStatusPending,
StartedAt: time.Now().Format(time.RFC3339),
Params: string(params),
}
// 持久化到 SQLite + Redis
if err := m.store.SaveTask(ctx, task); err != nil {
return nil, err
}
_ = m.cache.SetTask(ctx, task, 24*time.Hour)
_ = m.cache.AddActiveTask(ctx, task.ID)
// 创建可取消 context
taskCtx, cancel := context.WithCancel(context.Background())
m.mu.Lock()
m.running[task.ID] = &runningTask{cancel: cancel, task: task}
m.mu.Unlock()
// 异步执行
runner := NewRunner(m.store, m.cache, m.ds, m.gate, m.workers, m.batchSize)
go func() {
defer cancel()
runner.run(taskCtx, task, m.lister, since, limit)
// 任务结束后从 running map 移除
m.mu.Lock()
delete(m.running, task.ID)
m.mu.Unlock()
}()
log.Printf("[job] task %s created: type=%s, since=%s, limit=%d", task.ID, typ, since, limit)
return task, nil
}
// Cancel 取消运行中任务,返回是否成功取消。
func (m *Manager) Cancel(id string) bool {
m.mu.RLock()
rt, ok := m.running[id]
m.mu.RUnlock()
if !ok {
return false
}
rt.cancel()
log.Printf("[job] task %s cancelled", id)
return true
}
// Get 获取任务状态:Redis 优先(实时),回退 SQLite。
func (m *Manager) Get(ctx context.Context, id string) (*cache.Task, error) {
// 1. 先查 Redis(实时状态)
t, err := m.cache.GetTask(ctx, id)
if err == nil && t != nil {
return t, nil
}
// 2. 回退 SQLite
return m.store.GetTask(ctx, id)
}
// List 返回任务列表:SQLite 历史记录 + Redis 实时覆盖运行中任务。
func (m *Manager) List(ctx context.Context, limit int) ([]*cache.Task, error) {
tasks, err := m.store.ListTasks(ctx, limit)
if err != nil {
return nil, err
}
// 对运行中任务用 Redis 实时数据覆盖
for _, t := range tasks {
if t.Status == cache.TaskStatusRunning || t.Status == cache.TaskStatusPending {
if rt, _ := m.cache.GetTask(ctx, t.ID); rt != nil {
t.Done = rt.Done
t.Success = rt.Success
t.Failed = rt.Failed
t.Status = rt.Status
t.FailedCodes = rt.FailedCodes
}
}
}
return tasks, nil
}
// StopAll 优雅停止所有运行中任务(服务关闭时调用)。
func (m *Manager) StopAll() {
m.mu.Lock()
defer m.mu.Unlock()
for id, rt := range m.running {
rt.cancel()
log.Printf("[job] stopping task %s on shutdown", id)
}
}
// newRateLimiter 创建全局限速器(令牌桶模式)。
// 返回一个只读 channel,每次接收表示获得一个请求令牌。
// 稳态速率 = 1 token / interval,初始突发 = burst。
func newRateLimiter(interval time.Duration, burst int) <-chan struct{} {
ch := make(chan struct{}, burst)
// 预填突发额度
for i := 0; i < burst; i++ {
ch <- struct{}{}
}
go func() {
ticker := time.NewTicker(interval)
defer ticker.Stop()
for range ticker.C {
select {
case ch <- struct{}{}:
default: // 满了丢弃,保持稳态速率
}
}
}()
return ch
}