constraints_set.go

package cmd

import (
	"context"
	"errors"
	"strconv"

	"github.com/cybozu-go/cke"
	"github.com/cybozu-go/well"
	"github.com/spf13/cobra"
)

var cstrSet func(*cke.Constraints)

// constraintsSetCmd represents the "constraints set" command
var constraintsSetCmd = &cobra.Command{
	Use:   "set NAME VALUE",
	Short: "set a constraint for cluster configuration",
	Long: `Set a constraint for cluster configuration.

NAME is one of:
    control-plane-count
    minimum-workers
    maximum-workers

VALUE is an integer.`,

	Args: func(cmd *cobra.Command, args []string) error {
		if len(args) != 2 {
			return errors.New("wrong number of arguments")
		}

		val, err := strconv.Atoi(args[1])
		if err != nil {
			return err
		}

		switch args[0] {
		case "control-plane-count":
			cstrSet = func(cstr *cke.Constraints) {
				cstr.ControlPlaneCount = val
			}
		case "minimum-workers":
			cstrSet = func(cstr *cke.Constraints) {
				cstr.MinimumWorkers = val
			}
		case "maximum-workers":
			cstrSet = func(cstr *cke.Constraints) {
				cstr.MaximumWorkers = val
			}
		default:
			return errors.New("no such constraint: " + args[0])
		}

		return nil
	},
	RunE: func(cmd *cobra.Command, args []string) error {
		well.Go(func(ctx context.Context) error {
			cstr, err := storage.GetConstraints(ctx)
			switch err {
			case cke.ErrNotFound:
				cstr = cke.DefaultConstraints()
			case nil:
			default:
				return err
			}

			cstrSet(cstr)
			return storage.PutConstraints(ctx, cstr)
		})
		well.Stop()
		return well.Wait()
	},
}

func init() {
	constraintsCmd.AddCommand(constraintsSetCmd)
}