feat(deisctl): specify # of routers to install

Author: ljalbertsimard

deisctl/client/client.go

@@ -4,6 +4,7 @@ import (
 	"errors"
 	"fmt"
 	"os"
+	"strconv"
 
 	"github.com/deis/deis/deisctl/backend"
 	"github.com/deis/deis/deisctl/backend/fleet"
@@ -109,7 +110,7 @@ Examples:
 // Install loads the definitions of components from local unit files.
 // After Install, the components will be available to Start.
 func (c *Client) Install(argv []string) error {
-	usage := `Loads the definitions of components from local unit files.
+	usage := fmt.Sprintf(`Loads the definitions of components from local unit files.
 
 After install, the components will be available to start.
 
@@ -120,13 +121,24 @@ After install, the components will be available to start.
 
 Usage:
   deisctl install [<target>...] [options]
-`
+
+Options:
+  --router-mesh-size=<num>  Number of routers to be loaded when installing the platform [default: %d].
+`, cmd.DefaultRouterMeshSize)
 	// parse command-line arguments
 	args, err := docopt.Parse(usage, argv, true, "", false)
 	if err != nil {
 		return err
 	}
 
+	meshSizeArg, _ := args["--router-mesh-size"].(string)
+	parsedValue, err := strconv.ParseUint(meshSizeArg, 0, 8)
+	if err != nil || parsedValue < 1 {
+		fmt.Print("Error: argument --router-mesh-size: invalid value, make sure the value is an integer between 1 and 255.\n")
+		return err
+	}
+	cmd.RouterMeshSize = uint8(parsedValue)
+
 	return cmd.Install(args["<target>"].([]string), c.Backend, cmd.CheckRequiredKeys)
 }
 

deisctl/cmd/cmd.go

@@ -27,6 +27,8 @@ const (
 	StatelessPlatformCommand string = "stateless-platform"
 	swarm                    string = "swarm"
 	mesos                    string = "mesos"
+	// DefaultRouterMeshSize defines the default number of routers to be loaded when installing the platform.
+	DefaultRouterMeshSize uint8 = 3
 )
 
 // ListUnits prints a list of installed units.
@@ -45,6 +47,9 @@ var Stdout io.Writer = os.Stderr
 // Location to write standard error information. By default, this is the os.Stderr.
 var Stderr io.Writer = os.Stdout
 
+// Number of routers to be installed. By default, it's DefaultRouterMeshSize.
+var RouterMeshSize = DefaultRouterMeshSize
+
 // Scale grows or shrinks the number of running components.
 // Currently "router", "registry" and "store-gateway" are the only types that can be scaled.
 func Scale(targets []string, b backend.Backend) error {
@@ -316,11 +321,19 @@ func installDefaultServices(b backend.Backend, stateless bool, wg *sync.WaitGrou
 	wg.Wait()
 
 	fmt.Fprintln(out, "Routing mesh...")
-	b.Create([]string{"router@1", "router@2", "router@3"}, wg, out, err)
+	b.Create(getRouters(), wg, out, err)
 	wg.Wait()
 
 }
 
+func getRouters() []string {
+	routers := make([]string, RouterMeshSize)
+	for i := uint8(0); i < RouterMeshSize; i++ {
+		routers[i] = fmt.Sprintf("router@%d", i+1)
+	}
+	return routers
+}
+
 // Uninstall unloads the definitions of the specified components.
 // After Uninstall, the components will be unavailable until Install is called.
 func Uninstall(targets []string, b backend.Backend) error {

deisctl/cmd/cmd_test.go

@@ -423,6 +423,23 @@ func TestInstallPlatform(t *testing.T) {
 	}
 }
 
+func TestInstallPlatformWithCustomRouterMeshSize(t *testing.T) {
+	t.Parallel()
+
+	b := backendStub{}
+	expected := []string{"store-daemon", "store-monitor", "store-metadata", "store-volume",
+		"store-gateway@1", "logger", "logspout", "database", "registry@1",
+		"controller", "builder", "publisher", "router@1", "router@2", "router@3", "router@4", "router@5"}
+	RouterMeshSize = 5
+
+	Install([]string{"platform"}, &b, fakeCheckKeys)
+	RouterMeshSize = DefaultRouterMeshSize
+
+	if !reflect.DeepEqual(b.installedUnits, expected) {
+		t.Error(fmt.Errorf("Expected %v, Got %v", expected, b.installedUnits))
+	}
+}
+
 func TestInstallStatelessPlatform(t *testing.T) {
 	t.Parallel()