b5d0021fb6
On BSD, the storagenode-updater falls back to renaming the storagenode without doing anything to restart the service. Like the approach we have for linux, this change finds the process ID of the storagenode using pgrep and sends an interrupt signal to the process. Closes https://github.com/storj/storj/issues/5333 Change-Id: Icced90ea3e831528804784c2170a3b8b14952e8c
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
|
|
}
|