服務器初始狀態

models/server.go

@@ -39,16 +39,30 @@ type ServerList struct {
 	} `json:"-"`
 }
 
-// 通過腾讯云API創建服務器
+// 獲取服務器列表
-func (servers *ServerList) Create() error {
+func (servers *ServerList) Read() error {
+	// 從數據庫中獲取全部服務器列表
+	configs.ORMDB().Find(&servers.List)
+
+	// 打印現有的服務器列表
+	for _, server := range servers.List {
+		fmt.Println(server)
+	}
+
+	// 如果一個服務器都沒有, 提示
+	if len(servers.List) == 0 {
+		fmt.Println("記錄的服務器列表為空")
+	}
+
+	// 從騰訊雲API獲取服務器列表
 	client, err := cvm.NewClient(common.NewCredential(servers.Config.TencentCloud.SecretId, servers.Config.TencentCloud.SecretKey), servers.Config.TencentCloud.Region, profile.NewClientProfile())
 	if err != nil {
 		return fmt.Errorf("初始化騰訊雲SDK客戶端失敗: %v", err)
 	}
-	// 实例化一个请求对象, 指定啓動模板, 以創建指定規格的服務器
+
-	request := cvm.NewRunInstancesRequest()
+	// 实例化一个请求对象,每个接口都会对应一个request对象
-	request.LaunchTemplate = &cvm.LaunchTemplate{LaunchTemplateId: common.StringPtr("lt-ks6y5evh")}
+	request := cvm.NewDescribeInstancesRequest()
-	response, err := client.RunInstances(request)
+	response, err := client.DescribeInstances(request)
 	if _, ok := err.(*errors.TencentCloudSDKError); ok {
 		return fmt.Errorf("已返回 API 错误: %v", err)
 	}
@@ -56,74 +70,73 @@ func (servers *ServerList) Create() error {
 		return fmt.Errorf("运行实例失败: %v", err)
 	}
 
-	// 使用SDK提供的封裝函數打印JSON
+	for _, instance := range response.Response.InstanceSet {
-
+		var server Server
-	// 格式化打印JSON
+		server.ID = *instance.InstanceId
-	var data struct {
+		server.Name = *instance.InstanceName
-		Response struct {
+		server.IP = *instance.PublicIpAddresses[0]
-			InstanceIdSet []string `json:"InstanceIdSet"`
+		server.Port = 7890
-		} `json:"Response"`
+		server.Status = "初始化"
-	}
+		servers.List = append(servers.List, server)
-	if err := json.Unmarshal([]byte(response.ToJsonString()), &data); err != nil {
-		return err
-	}
-	b, _ := json.MarshalIndent(data, "", "  ")
-	fmt.Println(string(b))
-
-	var server Server
-	server.ID = data.Response.InstanceIdSet[0]
-	server.Name = server.ID
-
-	// 從騰訊雲獲取服務器詳情(用InstanceId)
-	request2 := cvm.NewDescribeInstancesRequest()
-	response2, err := client.DescribeInstances(request2)
-	if _, ok := err.(*errors.TencentCloudSDKError); ok {
-		return fmt.Errorf("已返回 API 错误: %v", err)
-	}
-	if err != nil {
-		return fmt.Errorf("獲取實例詳情失敗: %v", err)
 	}
 
-	// 直接打印JSON
+	if len(response.Response.InstanceSet) == 0 {
-	fmt.Println(response2.ToJsonString())
+		fmt.Println("服務器列表為空")
-
+		return nil
-	// 格式化打印JSON
-	var data2 struct {
-		Response struct {
-			InstanceSet []struct {
-				CPU               int    `json:"CPU"`
-				Memory            int    `json:"Memory"`
-				InstanceId        string `json:"InstanceId"`
-				InstanceName      string `json:"InstanceName"`
-				CreatedTime       string `json:"CreatedTime"`
-				PublicIpAddresses []struct {
-					Ip string `json:"Ip"`
-				} `json:"PublicIpAddresses"`
-				SystemDisk struct {
-					DiskType string `json:"DiskType"`
-					DiskSize int    `json:"DiskSize"`
-				} `json:"SystemDisk"`
-			} `json:"InstanceSet"`
-		} `json:"Response"`
-	}
-	if err := json.Unmarshal([]byte(response2.ToJsonString()), &data2); err != nil {
-		return err
-	}
-	b2, _ := json.MarshalIndent(data2, "", "  ")
-	fmt.Println(string(b2))
-
-	for _, instance := range data2.Response.InstanceSet {
-		if instance.InstanceId == server.ID {
-			server.IP = instance.PublicIpAddresses[0].Ip
-			server.Port = 7890
-			server.Status = "初始化"
-			break
-		}
 	}
 
 	return nil
 }
 
+// 通過腾讯云API創建服務器
+func (servers *ServerList) Create() (server Server, err error) {
+	client, err := cvm.NewClient(common.NewCredential(servers.Config.TencentCloud.SecretId, servers.Config.TencentCloud.SecretKey), servers.Config.TencentCloud.Region, profile.NewClientProfile())
+	if err != nil {
+		return server, fmt.Errorf("初始化騰訊雲SDK客戶端失敗: %v", err)
+	}
+	// 实例化一个请求对象, 指定啓動模板, 以創建指定規格的服務器
+	request := cvm.NewRunInstancesRequest()
+	request.LaunchTemplate = &cvm.LaunchTemplate{LaunchTemplateId: common.StringPtr("lt-ks6y5evh")}
+	response, err := client.RunInstances(request)
+	if _, ok := err.(*errors.TencentCloudSDKError); ok {
+		return server, fmt.Errorf("已返回 API 错误: %v", err)
+	}
+	if err != nil {
+		return server, fmt.Errorf("运行实例失败: %v", err)
+	}
+
+	fmt.Println("創建服務器成功:", response.Response.InstanceIdSet[0])
+
+	request2 := cvm.NewDescribeInstancesRequest()
+	//request2.InstanceIds = []*string{response.Response.InstanceIdSet[0]}
+	response2, err := client.DescribeInstances(request2)
+	if _, ok := err.(*errors.TencentCloudSDKError); ok {
+		return server, fmt.Errorf("已返回 API 错误: %v", err)
+	}
+	if err != nil {
+		return server, fmt.Errorf("獲取實例詳情失敗: %v", err)
+	}
+
+	// 縮進格式化打印
+	b, _ := json.MarshalIndent(response2.Response.InstanceSet, "", "  ")
+	fmt.Println(string(b))
+
+	for _, instance := range response2.Response.InstanceSet {
+		if *instance.InstanceId != *response.Response.InstanceIdSet[0] {
+			var server Server
+			server.ID = *instance.InstanceId
+			server.Name = *instance.InstanceName
+			server.IP = *instance.PublicIpAddresses[0]
+			server.Port = 7890
+			server.Status = *instance.InstanceState
+			servers.List = append(servers.List, server)
+			configs.ORMDB().Create(&server)
+			break
+		}
+	}
+	return server, nil
+}
+
 // 通過腾讯云API註銷指定的服務器
 func (servers *ServerList) Delete(InstanceId string) error {
 	client, err := cvm.NewClient(common.NewCredential(servers.Config.TencentCloud.SecretId, servers.Config.TencentCloud.SecretKey), servers.Config.TencentCloud.Region, profile.NewClientProfile())
@@ -146,7 +159,7 @@ func (servers *ServerList) Delete(InstanceId string) error {
 		return fmt.Errorf("註銷實例失敗: %v", err)
 	}
 
-	fmt.Println("註銷服務器成功:", response.Response)
+	fmt.Println("註銷服務器成功:", InstanceId, response.Response)
 
 	// 從列表中刪除服務器
 	for i, server := range servers.List {
@@ -159,57 +172,6 @@ func (servers *ServerList) Delete(InstanceId string) error {
 	return nil
 }
 
-// 獲取服務器列表
-func (servers *ServerList) Read() error {
-	// 從數據庫中獲取全部服務器列表
-	configs.ORMDB().Find(&servers.List)
-
-	// 打印現有的服務器列表
-	for _, server := range servers.List {
-		fmt.Println(server)
-	}
-
-	// 如果一個服務器都沒有, 提示
-	if len(servers.List) == 0 {
-		fmt.Println("記錄的服務器列表為空")
-	}
-
-	// 從騰訊雲API獲取服務器列表
-	client, err := cvm.NewClient(common.NewCredential(servers.Config.TencentCloud.SecretId, servers.Config.TencentCloud.SecretKey), servers.Config.TencentCloud.Region, profile.NewClientProfile())
-	if err != nil {
-		return fmt.Errorf("初始化騰訊雲SDK客戶端失敗: %v", err)
-	}
-
-	// 实例化一个请求对象,每个接口都会对应一个request对象
-	request := cvm.NewDescribeInstancesRequest()
-
-	// 返回的resp是一个DescribeInstancesResponse的实例，与请求对象对应
-	response, err := client.DescribeInstances(request)
-	if _, ok := err.(*errors.TencentCloudSDKError); ok {
-		return fmt.Errorf("已返回 API 错误: %v", err)
-	}
-	if err != nil {
-		return fmt.Errorf("运行实例失败: %v", err)
-	}
-
-	for _, instance := range response.Response.InstanceSet {
-		var server Server
-		server.ID = *instance.InstanceId
-		server.Name = *instance.InstanceName
-		server.IP = *instance.PublicIpAddresses[0]
-		server.Port = 7890
-		server.Status = "初始化"
-		servers.List = append(servers.List, server)
-	}
-
-	if len(response.Response.InstanceSet) == 0 {
-		fmt.Println("服務器列表為空")
-		return nil
-	}
-
-	return nil
-}
-
 func (server *Server) CheckStatus() error {
 	switch server.Type {
 	case "訓練":

models/server_test.go

@@ -72,9 +72,12 @@ func TestCreateServer(t *testing.T) {
 
 	fmt.Println(servers.Config)
 
-	if err := servers.Create(); err != nil {
+	server, err := servers.Create()
+	if err != nil {
 		t.Errorf("創建服務器失敗: %v", err)
 	}
+
+	fmt.Println(server)
 }
 
 // 單元測試:刪除指定服務器