uhost clone&reset-password&create-image and udisk snapshost manage

Author: lixiaojun629

CHANGELOG.md

@@ -1,4 +1,11 @@
 ## Change Log
+v0.1.5
+* support batch operation.
+
+v0.1.4
+* add udisk.
+* polling udisk and uhost long time operation
+* async complete resource-id
 
 v0.1.3
 * integrate auto completion.

base/client.go

@@ -1,26 +1,33 @@
 package base
 
 import (
+	pudisk "github.com/ucloud/ucloud-sdk-go/private/services/udisk"
+	puhost "github.com/ucloud/ucloud-sdk-go/private/services/uhost"
 	"github.com/ucloud/ucloud-sdk-go/services/pathx"
 	"github.com/ucloud/ucloud-sdk-go/services/uaccount"
 	"github.com/ucloud/ucloud-sdk-go/services/udisk"
 	"github.com/ucloud/ucloud-sdk-go/services/uhost"
-	"github.com/ucloud/ucloud-sdk-go/services/ulb"
 	"github.com/ucloud/ucloud-sdk-go/services/unet"
 	"github.com/ucloud/ucloud-sdk-go/services/vpc"
 	"github.com/ucloud/ucloud-sdk-go/ucloud"
 	"github.com/ucloud/ucloud-sdk-go/ucloud/auth"
 )
 
+//PrivateUDiskClient 私有模块的udisk client 即未在官网开放的接口
+type PrivateUDiskClient = pudisk.UDiskClient
+
+//PrivateUHostClient 私有模块的udisk client 即未在官网开放的接口
+type PrivateUHostClient = puhost.UHostClient
+
 //Client aggregate client for business
 type Client struct {
 	uaccount.UAccountClient
 	uhost.UHostClient
 	unet.UNetClient
-	ulb.ULBClient
 	vpc.VPCClient
 	pathx.PathXClient
 	udisk.UDiskClient
+	PrivateUHostClient
 }
 
 // NewClient will return a aggregate client
@@ -29,9 +36,9 @@ func NewClient(config *ucloud.Config, credential *auth.Credential) *Client {
 		*uaccount.NewClient(config, credential),
 		*uhost.NewClient(config, credential),
 		*unet.NewClient(config, credential),
-		*ulb.NewClient(config, credential),
 		*vpc.NewClient(config, credential),
 		*pathx.NewClient(config, credential),
 		*udisk.NewClient(config, credential),
+		*puhost.NewClient(config, credential),
 	}
 }

base/config.go

@@ -187,7 +187,6 @@ func init() {
 	ConfigInstance.LoadConfig()
 	timeout, _ := time.ParseDuration("15s")
 	ClientConfig = &sdk.Config{
-		ProjectId: ConfigInstance.ProjectID,
 		BaseUrl:   "https://api.ucloud.cn/",
 		Timeout:   timeout,
 		UserAgent: fmt.Sprintf("UCloud CLI v%s", Version),

base/util.go

@@ -20,6 +20,7 @@ import (
 	"github.com/ucloud/ucloud-sdk-go/ucloud/response"
 
 	"github.com/ucloud/ucloud-cli/model"
+	"github.com/ucloud/ucloud-cli/ux"
 )
 
 //ConfigPath 配置文件路径
@@ -263,59 +264,162 @@ var RegionLabel = map[string]string{
 	"afr-nigeria":  "Lagos",
 }
 
-//Poll 轮询
-func Poll(describeFunc func(string, string, string, string) (interface{}, error)) func(string, string, string, string, []string) chan bool {
-	stateFields := []string{"State", "Status"}
-	return func(resourceID, projectID, region, zone string, targetState []string) chan bool {
-		w := waiter.StateWaiter{
-			Pending: []string{"pending"},
-			Target:  []string{"avaliable"},
-			Refresh: func() (interface{}, string, error) {
-				inst, err := describeFunc(resourceID, projectID, region, zone)
-				if err != nil {
-					return nil, "", err
-				}
+//Poller 轮询器
+type Poller struct {
+	stateFields   []string
+	DescribeFunc  func(string, string, string, string) (interface{}, error)
+	Out           io.Writer
+	Timeout       time.Duration
+	SdescribeFunc func(string) (interface{}, error)
+}
+
+//Spoll 简化版
+func (p *Poller) Spoll(resourceID, pollText string, targetStates []string) {
+	w := waiter.StateWaiter{
+		Pending: []string{"pending"},
+		Target:  []string{"avaliable"},
+		Refresh: func() (interface{}, string, error) {
+			inst, err := p.SdescribeFunc(resourceID)
+			if err != nil {
+				return nil, "", err
+			}
 
-				if inst == nil {
-					return nil, "pending", nil
+			if inst == nil {
+				return nil, "pending", nil
+			}
+			instValue := reflect.ValueOf(inst)
+			instValue = reflect.Indirect(instValue)
+			instType := instValue.Type()
+			if instValue.Kind() != reflect.Struct {
+				return nil, "", fmt.Errorf("Instance is not struct")
+			}
+			state := ""
+			for i := 0; i < instValue.NumField(); i++ {
+				for _, sf := range p.stateFields {
+					if instType.Field(i).Name == sf {
+						state = instValue.Field(i).String()
+					}
 				}
-				instValue := reflect.ValueOf(inst)
-				instValue = reflect.Indirect(instValue)
-				instType := instValue.Type()
-				if instValue.Kind() != reflect.Struct {
-					return nil, "", fmt.Errorf("Instance is not struct")
+			}
+			if state != "" {
+				for _, t := range targetStates {
+					if t == state {
+						return inst, "avaliable", nil
+					}
 				}
-				state := ""
-				for i := 0; i < instValue.NumField(); i++ {
-					for _, sf := range stateFields {
-						if instType.Field(i).Name == sf {
-							state = instValue.Field(i).String()
-						}
+			}
+			return nil, "pending", nil
+
+		},
+		Timeout: p.Timeout,
+	}
+
+	done := make(chan bool)
+	go func() {
+		if resp, err := w.Wait(); err != nil {
+			log.Error(err)
+			if _, ok := err.(*waiter.TimeoutError); ok {
+				done <- false
+				return
+			}
+		} else {
+			log.Infof("%#v", resp)
+		}
+		done <- true
+	}()
+
+	spinner := ux.NewDotSpinner(p.Out)
+	spinner.Start(pollText)
+	ret := <-done
+	if ret {
+		spinner.Stop()
+	} else {
+		spinner.Timeout()
+	}
+}
+
+//Poll function
+func (p *Poller) Poll(resourceID, projectID, region, zone, pollText string, targetState []string) {
+	w := waiter.StateWaiter{
+		Pending: []string{"pending"},
+		Target:  []string{"avaliable"},
+		Refresh: func() (interface{}, string, error) {
+			inst, err := p.DescribeFunc(resourceID, projectID, region, zone)
+			if err != nil {
+				return nil, "", err
+			}
+
+			if inst == nil {
+				return nil, "pending", nil
+			}
+			instValue := reflect.ValueOf(inst)
+			instValue = reflect.Indirect(instValue)
+			instType := instValue.Type()
+			if instValue.Kind() != reflect.Struct {
+				return nil, "", fmt.Errorf("Instance is not struct")
+			}
+			state := ""
+			for i := 0; i < instValue.NumField(); i++ {
+				for _, sf := range p.stateFields {
+					if instType.Field(i).Name == sf {
+						state = instValue.Field(i).String()
 					}
 				}
-				if state != "" {
-					for _, t := range targetState {
-						if t == state {
-							return inst, "avaliable", nil
-						}
+			}
+			if state != "" {
+				for _, t := range targetState {
+					if t == state {
+						return inst, "avaliable", nil
 					}
 				}
-				return nil, "pending", nil
+			}
+			return nil, "pending", nil
 
-			},
-			Timeout: 5 * time.Minute,
-		}
+		},
+		Timeout: p.Timeout,
+	}
 
-		done := make(chan bool)
-		go func() {
-			if resp, err := w.Wait(); err != nil {
-				log.Error(err)
-			} else {
-				log.Infof("%#v", resp)
+	done := make(chan bool)
+	go func() {
+		if resp, err := w.Wait(); err != nil {
+			log.Error(err)
+			if _, ok := err.(*waiter.TimeoutError); ok {
+				done <- false
+				return
 			}
-			done <- true
-		}()
-		return done
+		} else {
+			log.Infof("%#v", resp)
+		}
+		done <- true
+	}()
+
+	spinner := ux.NewDotSpinner(p.Out)
+	spinner.Start(pollText)
+	ret := <-done
+	if ret {
+		spinner.Stop()
+	} else {
+		spinner.Timeout()
+	}
+}
+
+//NewSpoller simple
+func NewSpoller(describeFunc func(string) (interface{}, error), out io.Writer) *Poller {
+	return &Poller{
+		SdescribeFunc: describeFunc,
+		Out:           out,
+		stateFields:   []string{"State", "Status"},
+		Timeout:       10 * time.Minute,
+	}
+}
+
+//NewPoller 轮询
+func NewPoller(describeFunc func(string, string, string, string) (interface{}, error), out io.Writer) *Poller {
+	return &Poller{
+		DescribeFunc: describeFunc,
+		Out:          out,
+		stateFields:  []string{"State", "Status"},
+		Timeout:      10 * time.Minute,
 	}
 }