ec.balance: use specific collection names

This commit is contained in:
Chris Lu 2019-06-01 02:00:18 -07:00
parent 133b772fb5
commit d6e2c877fa

View file

@ -26,7 +26,7 @@ func (c *commandEcBalance) Name() string {
func (c *commandEcBalance) Help() string {
return `balance all ec shards among volume servers
ec.balance [-c ALL|EACH_COLLECTION|<collection_name>] [-f]
ec.balance [-c EACH_COLLECTION|<collection_name>] [-f] [-dataCenter <data_center>]
Algorithm:
@ -56,9 +56,9 @@ func (c *commandEcBalance) Help() string {
func (c *commandEcBalance) Do(args []string, commandEnv *commandEnv, writer io.Writer) (err error) {
balanceCommand := flag.NewFlagSet(c.Name(), flag.ContinueOnError)
collection := balanceCommand.String("c", "EACH_COLLECTION", "collection name, or use \"ALL_COLLECTIONS\" across collections, \"EACH_COLLECTION\" for each collection")
collection := balanceCommand.String("c", "EACH_COLLECTION", "collection name, or use \"EACH_COLLECTION\" for each collection")
dc := balanceCommand.String("dataCenter", "", "only apply the balancing for this dataCenter")
applyBalancing := balanceCommand.Bool("f", false, "apply the balancing plan.")
applyBalancing := balanceCommand.Bool("f", false, "apply the balancing plan")
if err = balanceCommand.Parse(args); err != nil {
return nil
}
@ -94,10 +94,6 @@ func (c *commandEcBalance) Do(args []string, commandEnv *commandEnv, writer io.W
return err
}
}
} else if *collection == "ALL" {
if err = balanceEcVolumes(commandEnv, "ALL", *applyBalancing); err != nil {
return err
}
} else {
if err = balanceEcVolumes(commandEnv, *collection, *applyBalancing); err != nil {
return err