Adding install.RestartService

Author: ycombinator

internal/pkg/agent/application/upgrade/rollback.go

@@ -145,10 +145,9 @@ func restartAgent(ctx context.Context) error {
 
 	restartViaServiceFn := func(ctx context.Context) error {
 		topPath := paths.Top()
-		install.StopService(topPath)
-		err := install.StartService(topPath)
+		err := install.RestartService(topPath)
 		if err != nil {
-			return err
+			return fmt.Errorf("failed to restart agent via service: %w", err)
 		}
 
 		return nil

internal/pkg/agent/install/install.go

@@ -172,6 +172,16 @@ func StopService(topPath string) error {
 	return nil
 }
 
+// RestartService stops and starts the installed service.
+func RestartService(topPath string) error {
+	err := StopService(topPath)
+	if err != nil {
+		return err
+	}
+
+	return StartService(topPath)
+}
+
 // FixPermissions fixes the permissions on the installed system.
 func FixPermissions(topPath string) error {
 	return fixPermissions(topPath)