start/stop/restart command for k8s

mgradm/cmd/restart/kubernetes.go

@@ -7,9 +7,8 @@
 package restart
 
 import (
-	"fmt"
-
 	"github.com/spf13/cobra"
+	"github.com/uyuni-project/uyuni-tools/shared/kubernetes"
 	"github.com/uyuni-project/uyuni-tools/shared/types"
 )
 
@@ -19,5 +18,5 @@ func kubernetesRestart(
 	cmd *cobra.Command,
 	args []string,
 ) error {
-	return fmt.Errorf("not implemented")
+	return kubernetes.RestartService(kubernetes.ServerFilter)
 }

mgradm/cmd/start/kubernetes.go

@@ -7,9 +7,8 @@
 package start
 
 import (
-	"fmt"
-
 	"github.com/spf13/cobra"
+	"github.com/uyuni-project/uyuni-tools/shared/kubernetes"
 	"github.com/uyuni-project/uyuni-tools/shared/types"
 )
 
@@ -19,5 +18,5 @@ func kubernetesStart(
 	cmd *cobra.Command,
 	args []string,
 ) error {
-	return fmt.Errorf("not implemented")
+	return kubernetes.StartService(kubernetes.ServerFilter)
 }

mgradm/cmd/stop/kubernetes.go

@@ -7,9 +7,8 @@
 package stop
 
 import (
-	"fmt"
-
 	"github.com/spf13/cobra"
+	"github.com/uyuni-project/uyuni-tools/shared/kubernetes"
 	"github.com/uyuni-project/uyuni-tools/shared/types"
 )
 
@@ -19,5 +18,5 @@ func kubernetesStop(
 	cmd *cobra.Command,
 	args []string,
 ) error {
-	return fmt.Errorf("not implemented")
+	return kubernetes.StopService(kubernetes.ServerFilter)
 }

shared/kubernetes/kubernetes.go

@@ -5,6 +5,7 @@
 package kubernetes
 
 import (
+	"fmt"
 	"os"
 	"strings"
 
@@ -87,3 +88,26 @@ func guessIngress() string {
 
 	return ingress
 }
+
+// RestartService restarts the pod.
+func RestartService(ServerFilter string) error {
+	if err := StopService(ServerFilter); err != nil {
+		return fmt.Errorf("cannot stop %s: %s", ServerFilter, err)
+	}
+	return StartService(ServerFilter)
+}
+
+// StartService starts the pod.
+func StartService(ServerFilter string) error {
+	// if something is running, we don't need to set replicas to 1
+	if _, err := GetNode("uyuni"); err != nil {
+		return ReplicasTo(ServerFilter, 1)
+	}
+	log.Debug().Msgf("Already running")
+	return nil
+}
+
+// StopService stop the pod.
+func StopService(ServerFilter string) error {
+	return ReplicasTo(ServerFilter, 0)
+}

shared/kubernetes/utils.go

@@ -221,7 +221,7 @@ func waitForReplica(podname string, replica uint) error {
 	if replica == 0 {
 		return waitForReplicaZero(podname)
 	}
-	cmdArgs := []string{"get", "rs", podname, "--output=custom-columns=DESIRED:.status.replicas", "--no-headers"}
+	cmdArgs := []string{"get", "pod", podname, "--output=custom-columns=STATUS:.status.phase", "--no-headers"}
 
 	var err error
 
@@ -231,7 +231,7 @@ func waitForReplica(podname string, replica uint) error {
 		if err != nil {
 			return fmt.Errorf("cannot execute %s: %s", strings.Join(cmdArgs, string(" ")), err)
 		}
-		if string(outStr) == fmt.Sprint(replica) {
+		if string(outStr) == "Running" {
 			log.Debug().Msgf("%s pod replica is now %d", podname, replica)
 			break
 		}

shared/utils/exec.go

@@ -53,7 +53,7 @@ func RunCmdOutput(logLevel zerolog.Level, command string, args ...string) ([]byt
 
 	output, err := exec.Command(command, args...).Output()
 	if err != nil {
-		log.Err(err).Msgf("Command returned Error: %s", output)
+		log.Warn().Msgf("Command returned with error: %s. Be sure to handle the error if required", err)
 	} else if len(output) > 0 {
 		log.Debug().Msgf("Command output: %s", output)
 	}