Fix Cadence CLI (#5005)

cadence-workflow · Sep 29, 2022 · 832f71a · 832f71a
1 parent 23abf96
commit 832f71a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/tools/cli/database.go b/tools/cli/database.go
@@ -77,6 +77,10 @@ func getDBFlags() []cli.Flag {
 			Name:  FlagDBRegion,
 			Usage: "persistence region",
 		},
+		cli.IntFlag{
+			Name:  FlagDBShard,
+			Usage: "number of db shards in a sharded SQL database",
+		},
 		cli.StringFlag{
 			Name:  FlagUsername,
 			Usage: "persistence username",
@@ -301,6 +305,9 @@ func overrideSQLDataStore(c *cli.Context, cfg *config.SQL) {
 	if c.IsSet(FlagDBType) || cfg.PluginName == "" {
 		cfg.PluginName = c.String(FlagDBType)
 	}
+	if c.IsSet(FlagDBShard) || cfg.NumShards == 0 {
+		cfg.NumShards = c.Int(FlagDBShard)
+	}
 	if c.IsSet(FlagUsername) || cfg.User == "" {
 		cfg.User = c.String(FlagUsername)
 	}

diff --git a/tools/cli/flags.go b/tools/cli/flags.go
@@ -37,6 +37,7 @@ const (
 	FlagDBAddress                         = "db_address"
 	FlagDBPort                            = "db_port"
 	FlagDBRegion                          = "db_region"
+	FlagDBShard                           = "db_shard"
 	FlagHistoryAddressWithAlias           = FlagHistoryAddress + ", had"
 	FlagProtoVersion                      = "protocol_version"
 	FlagDomainID                          = "domain_id"