mirror of
https://github.com/kubereboot/kured.git
synced 2026-05-11 19:06:41 +00:00
94e73465ad
Without this, we are loosing features based on previous logrus implementation. Now, we will log the stdout and stderr for each call. Next to this, we ensure the call of the log. methods will be ready for the switch to get rid of logrus in the future. Signed-off-by: Jean-Philippe Evrard <open-source@a.spamming.party>
49 lines
1.6 KiB
Go
49 lines
1.6 KiB
Go
package reboot
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"github.com/google/shlex"
|
|
log "github.com/sirupsen/logrus"
|
|
"os/exec"
|
|
"strings"
|
|
)
|
|
|
|
// CommandRebooter holds context-information for a reboot with command
|
|
type CommandRebooter struct {
|
|
RebootCommand []string
|
|
}
|
|
|
|
// Reboot triggers the reboot command
|
|
func (c CommandRebooter) Reboot() error {
|
|
log.Infof("Invoking command: %s", c.RebootCommand)
|
|
|
|
bufStdout := new(bytes.Buffer)
|
|
bufStderr := new(bytes.Buffer)
|
|
cmd := exec.Command(c.RebootCommand[0], c.RebootCommand[1:]...)
|
|
cmd.Stdout = bufStdout
|
|
cmd.Stderr = bufStderr
|
|
|
|
if err := cmd.Run(); err != nil {
|
|
return fmt.Errorf("error invoking reboot command %s: %v (stdout: %v, stderr: %v)", c.RebootCommand, err, bufStdout.String(), bufStderr.String())
|
|
}
|
|
log.Info("Invoked reboot command", "cmd", strings.Join(cmd.Args, " "), "stdout", bufStdout.String(), "stderr", bufStderr.String())
|
|
return nil
|
|
}
|
|
|
|
// NewCommandRebooter is the constructor to create a CommandRebooter from a string not
|
|
// yet shell lexed. You can skip this constructor if you parse the data correctly first
|
|
// when instantiating a CommandRebooter instance.
|
|
func NewCommandRebooter(rebootCommand string) (*CommandRebooter, error) {
|
|
if rebootCommand == "" {
|
|
return nil, fmt.Errorf("no reboot command specified")
|
|
}
|
|
cmd := []string{"/usr/bin/nsenter", fmt.Sprintf("-m/proc/%d/ns/mnt", 1), "--"}
|
|
parsedCommand, err := shlex.Split(rebootCommand)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error %v when parsing reboot command %s", err, rebootCommand)
|
|
}
|
|
cmd = append(cmd, parsedCommand...)
|
|
return &CommandRebooter{RebootCommand: cmd}, nil
|
|
}
|