Exit codes for -upgrade and -upgrade-check (fixes #194)

This commit is contained in:
Jakob Borg 2014-07-31 10:26:45 +02:00
parent 86b3ff3099
commit 7b8c126aa1
3 changed files with 41 additions and 9 deletions

View File

@ -131,10 +131,12 @@ func main() {
var reset bool var reset bool
var showVersion bool var showVersion bool
var doUpgrade bool var doUpgrade bool
var doUpgradeCheck bool
flag.StringVar(&confDir, "home", getDefaultConfDir(), "Set configuration directory") flag.StringVar(&confDir, "home", getDefaultConfDir(), "Set configuration directory")
flag.BoolVar(&reset, "reset", false, "Prepare to resync from cluster") flag.BoolVar(&reset, "reset", false, "Prepare to resync from cluster")
flag.BoolVar(&showVersion, "version", false, "Show version") flag.BoolVar(&showVersion, "version", false, "Show version")
flag.BoolVar(&doUpgrade, "upgrade", false, "Perform upgrade") flag.BoolVar(&doUpgrade, "upgrade", false, "Perform upgrade")
flag.BoolVar(&doUpgradeCheck, "upgrade-check", false, "Check for available upgrade")
flag.IntVar(&logFlags, "logflags", logFlags, "Set log flags") flag.IntVar(&logFlags, "logflags", logFlags, "Set log flags")
flag.Usage = usageFor(flag.CommandLine, usage, extraUsage) flag.Usage = usageFor(flag.CommandLine, usage, extraUsage)
flag.Parse() flag.Parse()
@ -148,12 +150,34 @@ func main() {
if doUpgrade { if doUpgrade {
err := upgrade() err := upgrade()
if err == errVersionUpToDate {
os.Exit(2)
}
if err != nil { if err != nil {
l.Fatalln(err) l.Fatalln(err) // exits 1
} }
return return
} }
if doUpgradeCheck {
rel, err := currentRelease()
if err != nil {
log.Fatalln(err) // exits 1
}
switch compareVersions(rel.Tag, Version) {
case -1:
l.Okf("Current version %s is newer than latest release %s.", Version, rel.Tag)
os.Exit(2)
case 0:
l.Okf("Already running the latest version, %s. Not upgrading.", Version)
os.Exit(2)
default:
l.Infof("An upgrade to %s is available.", rel.Tag)
os.Exit(0)
}
}
if len(os.Getenv("GOGC")) == 0 { if len(os.Getenv("GOGC")) == 0 {
debug.SetGCPercent(25) debug.SetGCPercent(25)
} }

View File

@ -26,6 +26,11 @@ import (
var GoArchExtra string // "", "v5", "v6", "v7" var GoArchExtra string // "", "v5", "v6", "v7"
var (
errVersionUpToDate = errors.New("current version is up to date")
errVersionUnknown = errors.New("couldn't fetch release information")
)
func upgrade() error { func upgrade() error {
if runtime.GOOS == "windows" { if runtime.GOOS == "windows" {
return errors.New("Upgrade currently unsupported on Windows") return errors.New("Upgrade currently unsupported on Windows")
@ -44,10 +49,10 @@ func upgrade() error {
switch compareVersions(rel.Tag, Version) { switch compareVersions(rel.Tag, Version) {
case -1: case -1:
l.Okf("Current version %s is newer than latest release %s. Not upgrading.", Version, rel.Tag) l.Okf("Current version %s is newer than latest release %s. Not upgrading.", Version, rel.Tag)
return nil return errVersionUpToDate
case 0: case 0:
l.Okf("Already running the latest version, %s. Not upgrading.", Version) l.Okf("Already running the latest version, %s. Not upgrading.", Version)
return nil return errVersionUpToDate
default: default:
l.Infof("Attempting upgrade to %s...", rel.Tag) l.Infof("Attempting upgrade to %s...", rel.Tag)
} }
@ -80,7 +85,7 @@ func upgrade() error {
} }
} }
return fmt.Errorf("Found no asset for %q", expectedRelease) return errVersionUnknown
} }
func currentRelease() (githubRelease, error) { func currentRelease() (githubRelease, error) {
@ -88,13 +93,16 @@ func currentRelease() (githubRelease, error) {
if err != nil { if err != nil {
return githubRelease{}, err return githubRelease{}, err
} }
if resp.StatusCode > 299 {
return githubRelease{}, fmt.Errorf("API call returned HTTP error: %s", resp.Status)
}
var rels []githubRelease var rels []githubRelease
json.NewDecoder(resp.Body).Decode(&rels) json.NewDecoder(resp.Body).Decode(&rels)
resp.Body.Close() resp.Body.Close()
if len(rels) == 0 { if len(rels) == 0 {
return githubRelease{}, errors.New("no releases found") return githubRelease{}, errVersionUnknown
} }
if strings.Contains(Version, "-beta") { if strings.Contains(Version, "-beta") {
@ -113,7 +121,7 @@ func currentRelease() (githubRelease, error) {
return rel, nil return rel, nil
} }
} }
return githubRelease{}, errors.New("no suitable release found") return githubRelease{}, errVersionUnknown
} }
} }

View File

@ -128,7 +128,7 @@ func (l *Logger) Fatalln(vals ...interface{}) {
s := fmt.Sprintln(vals...) s := fmt.Sprintln(vals...)
l.logger.Output(2, "FATAL: "+s) l.logger.Output(2, "FATAL: "+s)
l.callHandlers(LevelFatal, s) l.callHandlers(LevelFatal, s)
os.Exit(3) os.Exit(1)
} }
func (l *Logger) Fatalf(format string, vals ...interface{}) { func (l *Logger) Fatalf(format string, vals ...interface{}) {
@ -137,7 +137,7 @@ func (l *Logger) Fatalf(format string, vals ...interface{}) {
s := fmt.Sprintf(format, vals...) s := fmt.Sprintf(format, vals...)
l.logger.Output(2, "FATAL: "+s) l.logger.Output(2, "FATAL: "+s)
l.callHandlers(LevelFatal, s) l.callHandlers(LevelFatal, s)
os.Exit(3) os.Exit(1)
} }
func (l *Logger) FatalErr(err error) { func (l *Logger) FatalErr(err error) {
@ -147,6 +147,6 @@ func (l *Logger) FatalErr(err error) {
l.logger.SetFlags(l.logger.Flags() | log.Lshortfile) l.logger.SetFlags(l.logger.Flags() | log.Lshortfile)
l.logger.Output(2, "FATAL: "+err.Error()) l.logger.Output(2, "FATAL: "+err.Error())
l.callHandlers(LevelFatal, err.Error()) l.callHandlers(LevelFatal, err.Error())
os.Exit(3) os.Exit(1)
} }
} }