Adding CLI

glustercli/cmd/rebalance.go

@@ -1,30 +1,29 @@
 package cmd
 
 import (
-	fmt
+	"fmt"
 
 	"github.com/gluster/glusterd2/pkg/restclient"
 	rebalance "github.com/gluster/glusterd2/plugins/rebalance/api"
 )
 
 const (
-	helpRebalanceCmd		= "Gluster Rebalance"
-	helpRebalanceStartCmd		= "Start rebalance session for gluster volume"
-	helpRebalanceStatusCmd		= "Status of rebalance seesion"
-	helpRebalanceStopCmd		= "Stop rebalance session"
+	helpRebalanceCmd       = "Gluster Rebalance"
+	helpRebalanceStartCmd  = "Start rebalance session for gluster volume"
+	helpRebalanceStatusCmd = "Status of rebalance seesion"
+	helpRebalanceStopCmd   = "Stop rebalance session"
 )
 
-(
-	flagRebalanceStartCmdForce         bool
-	flagRebalanceStartCmdFixLayout     bool
+var (
+	flagRebalanceStartCmdForce     bool
+	flagRebalanceStartCmdFixLayout bool
 )
 
 var rebalanceCmd = &cobra.Command{
-	Use:	"volume rebalance",
-	Short:	helpRebalanceCmd
+	Use:   "volume rebalance",
+	Short: helpRebalanceCmd,
 }
 
-
 func init() {
 
 	// Rebalance Start
@@ -36,20 +35,31 @@ func init() {
 }
 
 var rebalaceStartCmd = &cobra.Command{
-	Use:	"<VOLNAME> start [flags]",
-	Short:	helpRebalanceStartCmd,
-	Args:	cobra.ExactArgs(1),
-	Run:	func(cmd *cobra.Command, args []string) {
-			volname := cmd.Flags().Args()[0]
-			err := client.VolumeStart(volname, "")
-			if err != nil {
-				if verbose {
-					log.WithError(err.Error()).WithFields(log.Fields{
-						"volume": volname,
-					}).Error("rebalance start failed")
+	Use:   "<VOLNAME> start [flags]",
+	Short: helpRebalanceStartCmd,
+	Args:  cobra.ExactArgs(1),
+	Run: func(cmd *cobra.Command, args []string) {
+		volname := cmd.Flags().Args()[0]
+		var err error
+		if flagRebalanceStartCmdForce && flagRebalanceStartCmdFixLayout {
+			err := errors.New("Conflicting options found")
+			failure("Please provide only 1 option", err, 1)
+		}
+		if flagRebalanceStartCmdForce {
+			err = client.VolumeStart(volname, "force")
+		} else if flagRebalanceStartCmdFixLayout {
+			err = client.VolumeStart(volname, "fix-layout")
+		} else {
+			err = client.VolumeStart(volname, "")
+		}
+		if err != nil {
+			if verbose {
+				log.WithError(err.Error()).WithFields(log.Fields{
+					"volume": volname,
+				}).Error("rebalance start failed")
 			}
 			failure("rebalance start failed", err, 1)
 		}
 		fmt.Printf("Rebalance for %s started successfully\n", volname)
-		},
+	},
 }