suspend.go

package cmd

import (
	"fmt"

	"github.com/spf13/cobra"
	"k8s.io/client-go/util/retry"
)

var suspendCmd = &cobra.Command{
	Use:   "suspend [cronjob name]",
	Short: "Suspend a cronjob",
	Long:  "Suspend a cronjob",
	Run:   suspend,
}

var unSuspendCmd = &cobra.Command{
	Use:   "unsuspend [cronjob name]",
	Short: "Unsuspend a cronjob",
	Long:  "Unsuspend a cronjob",
	Run:   unsuspend,
}

func suspend(_ *cobra.Command, args []string) {
	toggleSuspend(args, true)
}

func unsuspend(_ *cobra.Command, args []string) {
	toggleSuspend(args, false)
}

func toggleSuspend(args []string, status bool) {

	cronjobName := args[0]

	clientset := mustGetClientset()

	retryErr := retry.RetryOnConflict(retry.DefaultRetry, func() error {
		// RetryOnConflict uses exponential backoff to avoid exhausting the apiserver
		cronjob := getCronjob(namespace, cronjobName)

		cronjob.Spec.Suspend = &status
		_, updateErr := clientset.BatchV1beta1().CronJobs(namespace).Update(cronjob)
		return updateErr
	})

	if retryErr != nil {
		panic(fmt.Errorf("Update failed: %v", retryErr))
	}
	fmt.Printf("Updated cronjob %q.\n", cronjobName)
}