81 lines
1.8 KiB
Go
81 lines
1.8 KiB
Go
|
// Copyright (C) 2020 Storj Labs, Inc.
|
||
|
|
// See LICENSE for copying information.
|
||
|
|
|
||
|
|
//go:build service && (darwin || freebsd || dragonfly || netbsd || openbsd || solaris)
|
||
|
|
// +build service
|
||
|
|
// +build darwin freebsd dragonfly netbsd openbsd solaris
|
||
|
|
|
||
|
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"os"
|
||
|
|
"os/exec"
|
||
|
|
"strconv"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/spf13/cobra"
|
||
|
|
"github.com/zeebo/errs"
|
||
|
|
)
|
||
|
|
|
||
|
|
func cmdRestart(cmd *cobra.Command, args []string) error {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func restartService(ctx context.Context, service, binaryLocation, newVersionPath, backupPath string) error {
|
||
|
|
if err := os.Rename(binaryLocation, backupPath); err != nil {
|
||
|
|
return errs.Wrap(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := os.Rename(newVersionPath, binaryLocation); err != nil {
|
||
|
|
return errs.Combine(err, os.Rename(backupPath, binaryLocation), os.Remove(newVersionPath))
|
||
|
|
}
|
||
|
|
|
||
|
|
if service == updaterServiceName {
|
||
|
|
os.Exit(1)
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := stopProcess(service); err != nil {
|
||
|
|
err = errs.New("error stopping %s service: %v", service, err)
|
||
|
|
return errs.Combine(err, os.Rename(backupPath, binaryLocation))
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func stopProcess(service string) (err error) {
|
||
|
|
pid, err := getServicePID(service)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
p, err := os.FindProcess(pid)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
return p.Signal(os.Interrupt)
|
||
|
|
}
|
||
|
|
|
||
|
|
func getServicePID(service string) (int, error) {
|
||
|
|
args := []string{
|
||
|
|
"-n", // only return the newest process (though we expect only one to be running)
|
||
|
|
"-x", // match the whole name
|
||
|
|
service,
|
||
|
|
}
|
||
|
|
|
||
|
|
cmd := exec.Command("pgrep", args...)
|
||
|
|
|
||
|
|
out, err := cmd.CombinedOutput()
|
||
|
|
if err != nil {
|
||
|
|
return 0, errs.New("Error retrieving service pid: pgrep: %s %v", string(out), err)
|
||
|
|
}
|
||
|
|
|
||
|
|
trimmed := strings.TrimSuffix(string(out), "\n")
|
||
|
|
|
||
|
|
pid, err := strconv.Atoi(trimmed)
|
||
|
|
if err != nil {
|
||
|
|
return 0, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return pid, nil
|
||
|
|
}
|