Added ability to manage cluster auto-failover settings.

Author: brett19

cmd/cluster-settings-disable-autofailover.go

@@ -0,0 +1,30 @@
+package cmd
+
+import (
+	"github.com/spf13/cobra"
+	"go.uber.org/zap"
+)
+
+var clusterSettingsDisableAutoFailoverCmd = &cobra.Command{
+	Use:   "disable-autofailover <cluster-id>",
+	Short: "Disables auto-failover for a cluster",
+	Args:  cobra.ExactArgs(1),
+	Run: func(cmd *cobra.Command, args []string) {
+		helper := CmdHelper{}
+		logger := helper.GetLogger()
+		ctx := helper.GetContext()
+
+		clusterId := args[0]
+
+		_, deployer, cluster := helper.IdentifyCluster(ctx, clusterId)
+
+		err := deployer.SetAutoFailover(ctx, cluster.GetID(), false, 0)
+		if err != nil {
+			logger.Fatal("failed to disable auto-failover", zap.Error(err))
+		}
+	},
+}
+
+func init() {
+	clusterSettingsCmd.AddCommand(clusterSettingsDisableAutoFailoverCmd)
+}

cmd/cluster-settings-enable-autofailover.go

@@ -0,0 +1,33 @@
+package cmd
+
+import (
+	"github.com/spf13/cobra"
+	"go.uber.org/zap"
+)
+
+var clusterSettingsEnableAutoFailoverCmd = &cobra.Command{
+	Use:   "enable-autofailover <cluster-id>",
+	Short: "Enables auto-failover for a cluster",
+	Args:  cobra.ExactArgs(1),
+	Run: func(cmd *cobra.Command, args []string) {
+		helper := CmdHelper{}
+		logger := helper.GetLogger()
+		ctx := helper.GetContext()
+
+		clusterId := args[0]
+		timeout, _ := cmd.Flags().GetInt("timeout")
+
+		_, deployer, cluster := helper.IdentifyCluster(ctx, clusterId)
+
+		err := deployer.SetAutoFailover(ctx, cluster.GetID(), true, timeout)
+		if err != nil {
+			logger.Fatal("failed to enable auto-failover", zap.Error(err))
+		}
+	},
+}
+
+func init() {
+	clusterSettingsCmd.AddCommand(clusterSettingsEnableAutoFailoverCmd)
+
+	clusterSettingsEnableAutoFailoverCmd.Flags().Int("timeout", 120, "Auto-failover timeout in seconds")
+}

cmd/cluster-settings.go

@@ -0,0 +1,15 @@
+package cmd
+
+import (
+	"github.com/spf13/cobra"
+)
+
+var clusterSettingsCmd = &cobra.Command{
+	Use:   "cluster-settings",
+	Short: "Provides cluster settings management tools",
+	Run:   nil,
+}
+
+func init() {
+	rootCmd.AddCommand(clusterSettingsCmd)
+}

deployment/caodeploy/deployer.go

@@ -788,3 +788,7 @@ func (d *Deployer) RebalanceCluster(ctx context.Context, clusterID string, nodes
 func (d *Deployer) KillCouchbase(ctx context.Context, clusterID string, nodes []string) error {
 	return errors.New("caodeploy does not support killing couchbase process")
 }
+
+func (d *Deployer) SetAutoFailover(ctx context.Context, clusterID string, enabled bool, timeout int) error {
+	return errors.New("caodeploy does not support setting auto-failover")
+}

deployment/clouddeploy/deployer.go

@@ -2503,3 +2503,7 @@ func (d *Deployer) RebalanceCluster(ctx context.Context, clusterID string, nodes
 func (d *Deployer) KillCouchbase(ctx context.Context, clusterID string, nodeIDs []string) error {
 	return errors.New("clouddeploy does not support killing couchbase process")
 }
+
+func (d *Deployer) SetAutoFailover(ctx context.Context, clusterID string, enabled bool, timeout int) error {
+	return errors.New("clouddeploy does not support setting auto-failover")
+}

deployment/deployer.go

@@ -152,4 +152,5 @@ type Deployer interface {
 	DropLink(ctx context.Context, columnarID, linkName string) error
 	EnableDataApi(ctx context.Context, clusterID string) error
 	KillCouchbase(ctx context.Context, clusterID string, nodes []string) error
+	SetAutoFailover(ctx context.Context, clusterID string, enabled bool, timeout int) error
 }

deployment/dockerdeploy/deployer.go

@@ -1399,6 +1399,18 @@ func (d *Deployer) RedeployCluster(ctx context.Context, clusterID string) error
 	return errors.New("docker deploy does not support redeploy cluster")
 }
 
+func (d *Deployer) SetAutoFailover(ctx context.Context, clusterID string, enabled bool, timeout int) error {
+	controller, err := d.getController(ctx, clusterID)
+	if err != nil {
+		return errors.Wrap(err, "failed to get controller for cluster")
+	}
+
+	return controller.Controller().SetAutoFailover(ctx, &clustercontrol.SetAutoFailoverOptions{
+		Enabled: enabled,
+		Timeout: timeout,
+	})
+}
+
 func (d *Deployer) CreateCapellaLink(ctx context.Context, columnarID, linkName, clusterId, directID string) error {
 	return errors.New("docker deploy does not support create capella link")
 }

deployment/localdeploy/deployer.go

@@ -279,3 +279,7 @@ func (d *Deployer) RebalanceCluster(ctx context.Context, clusterID string, nodes
 func (d *Deployer) KillCouchbase(ctx context.Context, clusterID string, nodes []string) error {
 	return errors.New("localdeploy does not support killing couchbase process")
 }
+
+func (d *Deployer) SetAutoFailover(ctx context.Context, clusterID string, enabled bool, timeout int) error {
+	return errors.New("localdeploy does not support setting auto-failover")
+}

utils/clustercontrol/controller.go

@@ -963,6 +963,20 @@ func (c *Controller) SetAppTelemetry(ctx context.Context, opts *AppTelemetryOpti
 	return c.doFormPost(ctx, "/settings/appTelemetry", form, true, nil)
 }
 
+type SetAutoFailoverOptions struct {
+	Enabled bool
+	Timeout int
+}
+
+func (c *Controller) SetAutoFailover(ctx context.Context, opts *SetAutoFailoverOptions) error {
+	form := make(url.Values)
+	form.Add("enabled", strconv.FormatBool(opts.Enabled))
+	if opts.Timeout > 0 {
+		form.Add("timeout", strconv.Itoa(opts.Timeout))
+	}
+	return c.doFormPost(ctx, "/settings/autoFailover", form, true, nil)
+}
+
 func (c *Controller) GetMetrics(ctx context.Context) (string, error) {
 	var resp []byte
 	err := c.doGet(ctx, "/metrics", &resp)