add deploy type

Signed-off-by: WhereAreBugs <[email protected]>

cli/cli/cli.go

@@ -69,6 +69,7 @@ type CurveAdm struct {
 	clusterName         string // current cluster name
 	clusterTopologyData string // cluster topology
 	clusterPoolData     string // cluster pool
+	clusterType         string // cluster type like develop, production, etc.
 	monitor             storage.Monitor
 }
 
@@ -195,7 +196,7 @@ func (curveadm *CurveAdm) init() error {
 	curveadm.clusterTopologyData = cluster.Topology
 	curveadm.clusterPoolData = cluster.Pool
 	curveadm.monitor = monitor
-
+	curveadm.clusterType = cluster.Type
 	return nil
 }
 
@@ -276,6 +277,7 @@ func (curveadm *CurveAdm) ClusterUUId() string               { return curveadm.c
 func (curveadm *CurveAdm) ClusterName() string               { return curveadm.clusterName }
 func (curveadm *CurveAdm) ClusterTopologyData() string       { return curveadm.clusterTopologyData }
 func (curveadm *CurveAdm) ClusterPoolData() string           { return curveadm.clusterPoolData }
+func (curveadm *CurveAdm) ClusterType() string               { return curveadm.clusterType }
 func (curveadm *CurveAdm) Monitor() storage.Monitor          { return curveadm.monitor }
 
 func (curveadm *CurveAdm) GetHost(host string) (*hosts.HostConfig, error) {

cli/command/cluster/add.go

@@ -47,12 +47,18 @@ var (
 	CHECK_TOPOLOGY_PLAYBOOK_STEPS = []int{
 		playbook.CHECK_TOPOLOGY,
 	}
+	SUPPORTED_DEPLOY_TYPES = []string{
+		"production",
+		"test",
+		"develop",
+	}
 )
 
 type addOptions struct {
 	name        string
 	descriotion string
 	filename    string
+	deployType  string
 }
 
 func NewAddCommand(curveadm *cli.CurveAdm) *cobra.Command {
@@ -73,7 +79,7 @@ func NewAddCommand(curveadm *cli.CurveAdm) *cobra.Command {
 	flags := cmd.Flags()
 	flags.StringVarP(&options.descriotion, "description", "m", "", "Description for cluster")
 	flags.StringVarP(&options.filename, "topology", "f", "", "Specify the path of topology file")
-
+	flags.StringVarP(&options.deployType, "deploy-type", "t", "develop", "Specify the deploy type of cluster")
 	return cmd
 }
 
@@ -134,6 +140,15 @@ func checkTopology(curveadm *cli.CurveAdm, data string, options addOptions) erro
 	return pb.Run()
 }
 
+func checkDeployType(deployType string) error {
+	for _, t := range SUPPORTED_DEPLOY_TYPES {
+		if t == deployType {
+			return nil
+		}
+	}
+	return errno.ERR_UNSUPPORT_DEPLOY_TYPE.F("deploy type: %s", deployType)
+}
+
 func runAdd(curveadm *cli.CurveAdm, options addOptions) error {
 	// 1) check wether cluster already exist
 	name := options.name
@@ -161,14 +176,20 @@ func runAdd(curveadm *cli.CurveAdm, options addOptions) error {
 		return err
 	}
 
-	// 4) insert cluster (with topology) into database
+	// 4) check deploy type
+	err = checkDeployType(options.deployType)
+	if err != nil {
+		return err
+	}
+
+	// 5) insert cluster (with topology) into database
 	uuid := uuid.NewString()
-	err = storage.InsertCluster(name, uuid, options.descriotion, data)
+	err = storage.InsertCluster(name, uuid, options.descriotion, data, options.deployType)
 	if err != nil {
 		return errno.ERR_INSERT_CLUSTER_FAILED.E(err)
 	}
 
-	// 5) print success prompt
+	// 6) print success prompt
 	curveadm.WriteOutln("Added cluster '%s'", name)
 	return nil
 }

cli/command/cluster/import.go

@@ -100,7 +100,7 @@ func importCluster(storage *storage.Storage, dbfile, name string) error {
 	}
 
 	// insert cluster
-	err = storage.InsertCluster(name, cluster.UUId, cluster.Description, cluster.Topology)
+	err = storage.InsertCluster(name, cluster.UUId, cluster.Description, cluster.Topology, "")
 	if err != nil {
 		return err
 	}

cli/command/deploy.go

@@ -302,6 +302,7 @@ func displayDeployTitle(curveadm *cli.CurveAdm, dcs []*topology.DeployConfig) {
 	curveadm.WriteOutln("Cluster Name    : %s", curveadm.ClusterName())
 	curveadm.WriteOutln("Cluster Kind    : %s", dcs[0].GetKind())
 	curveadm.WriteOutln("Cluster Services: %s", serviceStats(dcs))
+	curveadm.WriteOutln("Cluster Type    : %s", curveadm.ClusterType())
 	curveadm.WriteOutln("")
 }
 

internal/errno/errno.go

@@ -254,7 +254,7 @@ var (
 	ERR_NO_SERVICES_MATCHED            = EC(210006, "no services matched")
 	// TODO: please check pool set disk type
 	ERR_INVALID_DISK_TYPE = EC(210007, "poolset disk type must be lowercase and can only be one of ssd, hdd and nvme")
-
+	ERR_UNSUPPORT_DEPLOY_TYPE          = EC(210008, "unknown deploy type")
 	// 220: commad options (client common)
 	ERR_UNSUPPORT_CLIENT_KIND = EC(220000, "unsupport client kind")
 	// 221: command options (client/bs)

internal/storage/sql.go

@@ -103,6 +103,7 @@ type Cluster struct {
 	Topology    string
 	Pool        string
 	Current     bool
+	Type        string
 }
 
 var (
@@ -116,14 +117,15 @@ var (
 			topology TEXT NULL,
 			pool TEXT NULL,
 			create_time DATE NOT NULL,
-			current INTEGER DEFAULT 0
+			current INTEGER DEFAULT 0,
+			type TEXT NULL
 		)
 	`
 
 	// insert cluster
 	InsertCluster = `
-		INSERT INTO clusters(uuid, name, description, topology, pool, create_time)
-		VALUES(?, ?, ?, ?, "", datetime('now','localtime'))
+		INSERT INTO clusters(uuid, name, description, topology, type, pool, create_time)
+		VALUES(?, ?, ?, ?, ?, "", datetime('now','localtime'))
 	`
 
 	// delete cluster

internal/storage/storage.go

@@ -160,8 +160,8 @@ func (s *Storage) GetHostses() ([]Hosts, error) {
 }
 
 // cluster
-func (s *Storage) InsertCluster(name, uuid, description, topology string) error {
-	return s.write(InsertCluster, uuid, name, description, topology)
+func (s *Storage) InsertCluster(name, uuid, description, topology string, deployType string) error {
+	return s.write(InsertCluster, uuid, name, description, topology,deployType)
 }
 
 func (s *Storage) DeleteCluster(name string) error {
@@ -187,6 +187,7 @@ func (s *Storage) getClusters(query string, args ...interface{}) ([]Cluster, err
 			&cluster.Pool,
 			&cluster.CreateTime,
 			&cluster.Current,
+			&cluster.Type,
 		)
 		if err != nil {
 			return nil, err