shuguang/nezha
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

💩 fix self update, go-update could not restart it-self

Browse Source
This commit is contained in:
naiba 2020-11-30 17:05:36 +08:00
parent 286e174943
commit fe3b9629f5
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
cmd/agent/main.go
View File

@ -48,6 +48,8 @@ var (
updateCh = make(chan struct{}, 0) updateCh = make(chan struct{}, 0)
) )
const AGENT_UPGRADE = 55
func doSelfUpdate() { func doSelfUpdate() {
defer func() { defer func() {
time.Sleep(time.Minute * 20) time.Sleep(time.Minute * 20)
@ -64,10 +66,8 @@ func doSelfUpdate() {
// latest version is the same as current version. It means current binary is up to date. // latest version is the same as current version. It means current binary is up to date.
log.Println("Current binary is the latest version", version) log.Println("Current binary is the latest version", version)
} else { } else {
log.Println("Prev version", dao.Version)
dao.Version = latest.Version.String()
client.Register(ctx, monitor.GetHost().PB())
log.Println("Successfully updated to version", latest.Version) log.Println("Successfully updated to version", latest.Version)
os.Exit(AGENT_UPGRADE)
} }
} }