Fix: first double run check version

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

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

Loading…
Cancel
Save