refactor(machine): ssh service support ctx

Author: ihexon

cmd/bauklotze/machine/start.go

@@ -155,7 +155,7 @@ func start(cmd *cobra.Command, args []string) error {
 
 		// Start Sleep Notifier and dispatch tasks
 		logrus.Infof("Start Sleep Notifier and dispatch tasks")
-		go shim.SleepNotifier(mc)
+		go shim.SleepNotifier(ctx, mc)
 
 		if err = mc.UpdateLastBoot(); err != nil {
 			logrus.Errorf("failed to update last boot time: %v", err)

pkg/machine/shim/host.go

@@ -209,23 +209,23 @@ func Start(ctx context.Context, mc *vmconfig.MachineConfig) (context.Context, er
 }
 
 // SleepNotifier start Sleep Notifier and dispatch tasks
-func SleepNotifier(mc *vmconfig.MachineConfig) {
+func SleepNotifier(ctx context.Context, mc *vmconfig.MachineConfig) {
 	notifierCh := notifier.GetInstance().Start()
 	for { //nolint:gosimple
 		select {
 		case activity := <-notifierCh:
 			if activity.Type == notifier.Awake {
 				logrus.Infof("machine awake, dispatch tasks")
-				if err := TimeSync(mc); err != nil {
+				if err := TimeSync(ctx, mc); err != nil {
 					logrus.Errorf("Failed to sync timestamp: %v", err)
 				}
 			}
 		}
 	}
 }
 
-func TimeSync(mc *vmconfig.MachineConfig) error {
-	return sshService.DoTimeSync(mc) //nolint:wrapcheck
+func TimeSync(ctx context.Context, mc *vmconfig.MachineConfig) error {
+	return sshService.DoTimeSync(ctx, mc) //nolint:wrapcheck
 }
 
 func DiskSync(mc *vmconfig.MachineConfig) error {

pkg/machine/shim/networking.go

@@ -32,7 +32,7 @@ func ConductVMReadinessCheck(ctx context.Context, mc *vmconfig.MachineConfig) bo
 			time.Sleep(defaultBackoff)
 		}
 
-		if err := sshService.GetKernelInfo(mc); err != nil {
+		if err := sshService.GetKernelInfo(ctx, mc); err != nil {
 			logrus.Warnf("SSH readiness check for machine failed: %v, try again", err)
 			continue
 		}

pkg/machine/ssh/service/service.go

@@ -4,20 +4,21 @@
 package service
 
 import (
+	"context"
 	"fmt"
 	"time"
 
 	"bauklotze/pkg/machine/vmconfig"
 )
 
-func GetKernelInfo(mc *vmconfig.MachineConfig) error {
-	return run(mc, "uname", []string{
+func GetKernelInfo(ctx context.Context, mc *vmconfig.MachineConfig) error {
+	return runCtx(ctx, mc, "uname", []string{
 		"-a",
 	})
 }
 
-func DoTimeSync(mc *vmconfig.MachineConfig) error {
-	return run(mc, "date", []string{
+func DoTimeSync(ctx context.Context, mc *vmconfig.MachineConfig) error {
+	return runCtx(ctx, mc, "date", []string{
 		"-s",
 		fmt.Sprintf("@%d", time.Now().Unix()),
 	})