package k8sapi import ( "context" "fmt" "github.com/coreos/go-systemd/v22/dbus" ) func restartSystemdUnit(ctx context.Context, unit string) error { conn, err := dbus.NewSystemdConnectionContext(ctx) if err != nil { return fmt.Errorf("establishing systemd connection failed: %w", err) } restartChan := make(chan string) if _, err := conn.RestartUnitContext(ctx, unit, "replace", restartChan); err != nil { return fmt.Errorf("restarting systemd unit %q failed: %w", unit, err) } // Wait for the restart to finish and actually check if it was // successful or not. result := <-restartChan switch result { case "done": return nil default: return fmt.Errorf("restarting systemd unit %q failed: expected %v but received %v", unit, "done", result) } } func startSystemdUnit(ctx context.Context, unit string) error { conn, err := dbus.NewSystemdConnectionContext(ctx) if err != nil { return fmt.Errorf("establishing systemd connection failed: %w", err) } startChan := make(chan string) if _, err := conn.StartUnitContext(ctx, unit, "replace", startChan); err != nil { return fmt.Errorf("starting systemd unit %q failed: %w", unit, err) } // Wait for the enable to finish and actually check if it was // successful or not. result := <-startChan switch result { case "done": return nil default: return fmt.Errorf("starting systemd unit %q failed: expected %v but received %v", unit, "done", result) } } func enableSystemdUnit(ctx context.Context, unitPath string) error { conn, err := dbus.NewSystemdConnectionContext(ctx) if err != nil { return fmt.Errorf("establishing systemd connection failed: %w", err) } if _, _, err := conn.EnableUnitFilesContext(ctx, []string{unitPath}, true, true); err != nil { return fmt.Errorf("enabling systemd unit %q failed: %w", unitPath, err) } return nil }