package app import ( "context" "os" "sync" "time" "github.com/golang/freetype/truetype" "github.com/wailsapp/wails/v2/pkg/runtime" "douyu_go/internal/activity" "douyu_go/internal/client" "douyu_go/internal/config" "douyu_go/internal/constants" imgpkg "douyu_go/internal/image" "douyu_go/internal/logger" "douyu_go/internal/seckill" "douyu_go/internal/trace" "douyu_go/internal/types" ) // App 应用主结构体,所有导出方法可被前端通过 Wails 绑定调用。 // 绑定方法按领域拆分到 app_*.go 文件。 type App struct { ctx context.Context clientMu sync.RWMutex client *client.DouyuClient scheduler *seckill.Scheduler logStore *logger.MemoryLogStore traceMu sync.RWMutex trace *trace.Recorder paths types.AppPaths font *truetype.Font cfgMu sync.RWMutex cfg types.Config productsMu sync.RWMutex products []types.Product productsAt time.Time // accountsMu 串行化账号文件修改,避免并发导入、编辑或删除互相覆盖。 accountsMu sync.Mutex closeLog func() closeTrace func() } // getClient 返回当前 client 指针快照(读锁保护)。 func (a *App) getClient() *client.DouyuClient { a.clientMu.RLock() defer a.clientMu.RUnlock() return a.client } // getTrace 返回当前 trace 记录器快照(读锁保护)。 func (a *App) getTrace() *trace.Recorder { a.traceMu.RLock() defer a.traceMu.RUnlock() return a.trace } // getCfg 返回当前配置的深拷贝(读锁保护)。 func (a *App) getCfg() types.Config { a.cfgMu.RLock() defer a.cfgMu.RUnlock() return cloneConfig(a.cfg) } // setClient 替换 client(写锁保护)。 func (a *App) setClient(c *client.DouyuClient) { a.clientMu.Lock() a.client = c a.clientMu.Unlock() } // setCfg 替换配置(写锁保护)。 func (a *App) setCfg(c types.Config) { a.cfgMu.Lock() a.cfg = c a.cfgMu.Unlock() } // replaceTrace 根据配置替换 trace 记录器,旧记录器在锁外关闭,避免阻塞读路径。 func (a *App) replaceTrace(cfg types.Config) { traceCfg := types.TraceConfig{ Enabled: cfg.TraceEnabled, MaxBodyBytes: cfg.TraceMaxBody, } tr, closeTrace := trace.InitTrace(a.paths.LogDir, traceCfg) a.traceMu.Lock() oldClose := a.closeTrace a.trace = tr a.closeTrace = closeTrace a.traceMu.Unlock() if oldClose != nil { oldClose() } } // cloneConfig 深拷贝配置,避免 RetryStages 切片共享引用。 func cloneConfig(c types.Config) types.Config { cp := c if c.RetryStages != nil { cp.RetryStages = make([]types.RetryStage, len(c.RetryStages)) copy(cp.RetryStages, c.RetryStages) } return cp } // New 创建 App 实例。 func New() *App { return &App{} } // Startup Wails 应用启动时调用。 func (a *App) Startup(ctx context.Context) { a.ctx = ctx a.paths = config.DefaultAppPaths() a.logStore = logger.NewMemoryLogStore(800) a.closeLog = logger.InitLogging(a.logStore, a.paths.LogDir) // 加载配置并应用运行时设置 a.cfg = config.LoadConfig(a.paths.ConfigFile) logger.SetMinLevel(a.cfg.LogLevel) a.replaceTrace(a.cfg) // 解析当前活动:配置中的 activity ID 决定使用哪套接口参数 act := activity.Get(a.cfg.Activity) if act == nil { act = activity.Default() logger.LogEvent("活动", "warn", "配置中的活动 %q 未识别,已回退到默认活动 %s", a.cfg.Activity, act.DisplayName) } a.client = client.New(act, a.cfg.TimeAPIURL, a.getTrace()) a.font = imgpkg.LoadFont(a.paths.FontFiles) a.scheduler = seckill.New(a.client, a.paths, a.font, a.cfg.Concurrency, a.cfg.Attempts, a.cfg.RetryStages, a.cfg.SeckillMode) a.scheduler.SetStateChangeCallback(func(state string) { runtime.EventsEmit(a.ctx, seckillStateEvent, state) }) a.scheduler.LoadAccounts() if err := os.MkdirAll(a.paths.SuccessDir, 0755); err != nil { logger.LogEvent("图片", "warn", "成功图片目录创建失败,图片保存功能可能不可用: %v", err) } if sample, err := a.client.ServerTimeSample(); err == nil { logger.SetDouyuOffset(sample.Offset) logger.LogEvent("时间", "success", "服务端时间获取成功: %s", sample.EstimatedServerNow().In(constants.CST).Format("2006-01-02 15:04:05.000")) } else { logger.LogEvent("时间", "error", "服务端时间获取失败: %v", err) } logger.LogEvent("系统", "success", "应用启动完成 | 账号: %d", a.scheduler.AccountCount()) } // Shutdown Wails 应用关闭时调用。 func (a *App) Shutdown(ctx context.Context) { if a.scheduler != nil { a.scheduler.Close() } a.traceMu.Lock() closeTrace := a.closeTrace a.closeTrace = nil a.trace = nil a.traceMu.Unlock() if closeTrace != nil { closeTrace() } if a.closeLog != nil { a.closeLog() } } // GetVersion 返回应用版本号。 func (a *App) GetVersion() string { if constants.BuildTime != "unknown" { return constants.Version + " (" + constants.BuildTime + ")" } return constants.Version }