99 lines
2.7 KiB
Go
99 lines
2.7 KiB
Go
// Copyright (C) 2020 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
// +build windows,!unittest
|
|
|
|
package main
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"os/exec"
|
|
|
|
"github.com/zeebo/errs"
|
|
"go.uber.org/zap"
|
|
|
|
"storj.io/private/version"
|
|
"storj.io/storj/private/version/checker"
|
|
)
|
|
|
|
// loopFunc is func that is run by the update cycle.
|
|
func loopFunc(ctx context.Context) error {
|
|
zap.L().Info("Downloading versions.", zap.String("Server Address", runCfg.Version.ServerAddress))
|
|
|
|
all, err := checker.New(runCfg.Version.ClientConfig).All(ctx)
|
|
if err != nil {
|
|
zap.L().Error("Error retrieving version info.", zap.Error(err))
|
|
return nil
|
|
}
|
|
|
|
if err := update(ctx, runCfg.ServiceName, runCfg.BinaryLocation, all.Processes.Storagenode); err != nil {
|
|
// don't finish loop in case of error just wait for another execution
|
|
zap.L().Error("Error updating service.", zap.String("Service", runCfg.ServiceName), zap.Error(err))
|
|
}
|
|
|
|
if err := updateSelf(ctx, updaterBinaryPath, all.Processes.StoragenodeUpdater); err != nil {
|
|
// don't finish loop in case of error just wait for another execution
|
|
zap.L().Error("Error updating service.", zap.String("Service", updaterServiceName), zap.Error(err))
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func updateSelf(ctx context.Context, binaryLocation string, ver version.Process) error {
|
|
currentVersion, err := binaryVersion(binaryLocation)
|
|
if err != nil {
|
|
return errs.Wrap(err)
|
|
}
|
|
|
|
zap.L().Info("Current binary version",
|
|
zap.String("Service", updaterServiceName),
|
|
zap.String("Version", currentVersion.String()),
|
|
)
|
|
|
|
// should update
|
|
newVersion, reason, err := version.ShouldUpdateVersion(currentVersion, nodeID, ver)
|
|
if err != nil {
|
|
return errs.Wrap(err)
|
|
}
|
|
if newVersion.IsZero() {
|
|
zap.L().Info(reason, zap.String("Service", updaterServiceName))
|
|
return nil
|
|
}
|
|
|
|
newVersionPath := prependExtension(binaryLocation, newVersion.Version)
|
|
|
|
if err = downloadBinary(ctx, parseDownloadURL(newVersion.URL), newVersionPath); err != nil {
|
|
return errs.Wrap(err)
|
|
}
|
|
|
|
downloadedVersion, err := binaryVersion(newVersionPath)
|
|
if err != nil {
|
|
return errs.Combine(errs.Wrap(err), os.Remove(newVersionPath))
|
|
}
|
|
|
|
newSemVer, err := newVersion.SemVer()
|
|
if err != nil {
|
|
return errs.Combine(err, os.Remove(newVersionPath))
|
|
}
|
|
|
|
if newSemVer.Compare(downloadedVersion) != 0 {
|
|
err := errs.New("invalid version downloaded: wants %s got %s", newVersion.Version, downloadedVersion)
|
|
return errs.Combine(err, os.Remove(newVersionPath))
|
|
}
|
|
|
|
zap.L().Info("Restarting service.", zap.String("Service", updaterServiceName))
|
|
return restartSelf(binaryLocation, newVersionPath)
|
|
}
|
|
|
|
func restartSelf(bin, newbin string) error {
|
|
args := []string{
|
|
"restart-service",
|
|
"--binary-location", bin,
|
|
"--service-name", updaterServiceName,
|
|
newbin,
|
|
}
|
|
|
|
return exec.Command(bin, args...).Start()
|
|
}
|