add more params to cluster_create.go

openinfradev · Mar 17, 2022 · 9cb663f · 9cb663f
1 parent 12218a6
commit 9cb663f
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/cmd/cluster_create.go b/cmd/cluster_create.go
@@ -31,6 +31,7 @@ import (
 
 const (
   address = "tks-cluster-lcm.taco-cat.xyz:9110"
+  //address = "localhost:9110"
 )
 
 // clusterCreateCmd represents the create command
@@ -63,7 +64,7 @@ tks cluster create <CLUSTERNAME> --contract-id <CONTRACTID> --csp-id <CSPID>`,
     ContractId, _ := cmd.Flags().GetString("contract-id")
     CspId, _ := cmd.Flags().GetString("csp-id")
 
-    conf := &pb.ClusterRawConf{}
+    conf := pb.ClusterRawConf{}
     conf.SshKeyName, _ = cmd.Flags().GetString("ssh-key-name")
     conf.Region, _ = cmd.Flags().GetString("region")
     conf.MachineType, _ = cmd.Flags().GetString("machine-type")
@@ -79,7 +80,7 @@ tks cluster create <CLUSTERNAME> --contract-id <CONTRACTID> --csp-id <CSPID>`,
       Name: ClusterName,
       ContractId: ContractId,
       CspId: CspId,
-      Conf: conf,
+      Conf: &conf,
     }
 
     m := protojson.MarshalOptions{
@@ -110,4 +111,9 @@ func init() {
   clusterCreateCmd.MarkFlagRequired("contract-id")
   clusterCreateCmd.Flags().String("csp-id", "", "CSP ID")
   clusterCreateCmd.MarkFlagRequired("csp-id")
+  clusterCreateCmd.Flags().String("region", "", "AWS Region")
+  clusterCreateCmd.Flags().String("num-of-az", "", "Number of availability zones in selected region")
+  clusterCreateCmd.Flags().String("ssh-key-name", "", "SSH key name for EC2 instance connection")
+  clusterCreateCmd.Flags().String("machine-type", "", "machine type of worker node")
+  clusterCreateCmd.Flags().String("machine-replicas", "", "machine replicas of worker node")
 }