ygxbnet
d4cc335fbf
- 将全局变量 isRun 移动到 App 结构体内部作为实例字段 - 在 config.go 中定义配置键名为常量，提高代码可维护性 - 使用结构体实例字段替代全局变量管理上传状态 - 修改 StartLooking 函数中的上下文取消处理逻辑 - 移除上传程序退出日志的重复记录
79 lines
1.6 KiB
Go
79 lines
1.6 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"dypid-client/internal/config"
|
|
"dypid-client/internal/uploader"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/wailsapp/wails/v2/pkg/runtime"
|
|
)
|
|
|
|
// App struct
|
|
type App struct {
|
|
ctx context.Context
|
|
logChan chan string
|
|
uploaderCTX context.Context
|
|
uploaderCancel context.CancelFunc
|
|
isRun bool
|
|
}
|
|
|
|
func NewApp() *App {
|
|
return &App{}
|
|
}
|
|
|
|
// startup 程序初始化
|
|
func (a *App) startup(ctx context.Context) {
|
|
a.ctx = ctx
|
|
|
|
// 后台 goroutine 持续推送日志
|
|
a.logChan = make(chan string, 100)
|
|
go func() {
|
|
for log := range a.logChan {
|
|
runtime.EventsEmit(a.ctx, "log", log)
|
|
}
|
|
}()
|
|
|
|
//在程序启动时运行上传程序
|
|
a.uploaderCTX, a.uploaderCancel = context.WithCancel(a.ctx)
|
|
if config.APPConfig.IsRunOnStart {
|
|
time.Sleep(time.Second)
|
|
go uploader.StartLooking(a.uploaderCTX, &a.logChan, config.APPConfig.CheckDir)
|
|
a.isRun = true
|
|
}
|
|
}
|
|
|
|
// SelectPath 打开选择路径弹框
|
|
func (a *App) SelectPath() string {
|
|
dialog, _ := runtime.OpenDirectoryDialog(a.ctx, runtime.OpenDialogOptions{})
|
|
fmt.Println("选择路径：", dialog)
|
|
return dialog
|
|
}
|
|
|
|
func (a *App) GetConfig() config.Config {
|
|
return config.APPConfig
|
|
}
|
|
|
|
func (a *App) WriteConfig(key string, value any) {
|
|
fmt.Println(key, value)
|
|
config.WriteConfig(key, value)
|
|
}
|
|
|
|
func (a *App) StartUpload() {
|
|
if a.isRun {
|
|
return
|
|
}
|
|
a.uploaderCTX, a.uploaderCancel = context.WithCancel(a.ctx)
|
|
go uploader.StartLooking(a.uploaderCTX, &a.logChan, config.APPConfig.CheckDir)
|
|
a.isRun = true
|
|
}
|
|
|
|
func (a *App) StopUpload() {
|
|
if a.isRun {
|
|
a.uploaderCancel()
|
|
}
|
|
a.isRun = false
|
|
uploader.AddLog(&a.logChan, "上传程序已退出")
|
|
}
|