feat(reboot): reboot after 10s by default (#670)

To inform the control plane that the reboot request has been processed,
reboot after 10 seconds.

Signed-off-by: Tao Li <[email protected]>

Author: eahydra

pkg/host/reboot.go

@@ -45,7 +45,7 @@ func WithSystemctl(b bool) OpOption {
 
 var ErrNotRoot = errors.New("must be run as sudo/root")
 
-// Reboots the system.
+// Reboot reboots the system.
 func Reboot(ctx context.Context, opts ...OpOption) error {
 	options := &Op{}
 	if err := options.applyOpts(opts); err != nil {

pkg/session/serve.go

@@ -74,21 +74,19 @@ func (s *Session) serve() {
 		}
 
 		ctx, cancel := context.WithTimeout(context.Background(), time.Minute)
-		if payload.Method == "reboot" {
-			rerr := pkghost.Reboot(ctx, pkghost.WithDelaySeconds(0))
-
-			if rerr != nil {
-				log.Logger.Errorf("failed to trigger reboot machine: %v", rerr)
-			}
-
-			cancel()
-			continue
-		}
 
 		needExit := -1
 		response := &Response{}
 
 		switch payload.Method {
+		case "reboot":
+			// To inform the control plane that the reboot request has been processed, reboot after 10 seconds.
+			err := pkghost.Reboot(s.ctx, pkghost.WithDelaySeconds(10))
+			if err != nil {
+				log.Logger.Errorf("failed to trigger reboot machine: %v", err)
+				response.Error = err.Error()
+			}
+
 		case "metrics":
 			metrics, err := s.getMetrics(ctx, payload)
 			if err != nil {