Fix: first double run check version

pull/21/head
zijiren233 2 years ago
parent cd0e4d6d46
commit 7b8f0cdb6f

@ -2,7 +2,6 @@ package bootstrap
import (
"context"
"sync"
"time"
log "github.com/sirupsen/logrus"
@ -15,55 +14,70 @@ func InitCheckUpdate(ctx context.Context) error {
if err != nil {
log.Fatalf("get version info error: %v", err)
}
go func() {
t := time.NewTicker(time.Second)
defer t.Stop()
execFile, err := version.ExecutableFile()
if err != nil {
execFile = "synctv"
}
var (
need bool
latest string
url string
once sync.Once
)
need, latest, url, err = check(ctx, v)
if err != nil {
log.Errorf("check update error: %v", err)
} else {
if need {
log.Infof("new version (%s) available: %s", latest, url)
log.Infof("run '%s self-update' to auto update", execFile)
}
}
SysNotify.RegisterSysNotifyTask(0, sysnotify.NewSysNotifyTask(
"check-update",
sysnotify.NotifyTypeEXIT,
func() error {
if need {
log.Infof("new version (%s) available: %s", latest, url)
log.Infof("run 'synctv self-update' to auto update")
log.Infof("run '%s self-update' to auto update", execFile)
}
return nil
},
))
t := time.NewTicker(time.Hour * 6)
defer t.Stop()
for range t.C {
l, err := v.CheckLatest(ctx)
need, latest, url, err = check(ctx, v)
if err != nil {
log.Errorf("check update error: %v", err)
continue
}
}
}()
return nil
}
func check(ctx context.Context, v *version.VersionInfo) (need bool, latest string, url string, err error) {
l, err := v.CheckLatest(ctx)
if err != nil {
return false, "", "", err
}
latest = l
b, err := v.NeedUpdate(ctx)
if err != nil {
log.Errorf("check update error: %v", err)
continue
return false, "", "", err
}
need = b
if b {
u, err := v.LatestBinaryURL(ctx)
if err != nil {
log.Errorf("check update error: %v", err)
continue
return false, "", "", err
}
url = u
}
once.Do(func() {
if b {
log.Infof("new version (%s) available: %s", latest, url)
log.Infof("run 'synctv self-update' to auto update")
}
t.Reset(time.Hour * 6)
})
}
}()
return nil
return need, latest, url, nil
}

Loading…
Cancel
Save