From 6abd1ebfa5090655a549efd954c865766afd94a8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=A1=9C=E8=8F=AF?= Date: Mon, 5 Jun 2023 16:48:22 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9C=8D=E5=8B=99=E5=99=A8=E5=88=9D=E5=A7=8B?= =?UTF-8?q?=E7=8B=80=E6=85=8B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- models/server.go | 200 +++++++++++++++++------------------------- models/server_test.go | 5 +- 2 files changed, 85 insertions(+), 120 deletions(-) diff --git a/models/server.go b/models/server.go index dec20d2..c311ff1 100644 --- a/models/server.go +++ b/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.LaunchTemplate = &cvm.LaunchTemplate{LaunchTemplateId: common.StringPtr("lt-ks6y5evh")} - response, err := client.RunInstances(request) + + // 实例化一个请求对象,每个接口都会对应一个request对象 + request := cvm.NewDescribeInstancesRequest() + 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 - - // 格式化打印JSON - var data struct { - Response struct { - InstanceIdSet []string `json:"InstanceIdSet"` - } `json:"Response"` - } - 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) + 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) } - // 直接打印JSON - fmt.Println(response2.ToJsonString()) - - // 格式化打印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 - } + if len(response.Response.InstanceSet) == 0 { + fmt.Println("服務器列表為空") + return nil } 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 "訓練": diff --git a/models/server_test.go b/models/server_test.go index 4acd56e..187f6ec 100644 --- a/models/server_test.go +++ b/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) } // 單元測試:刪除指定服務器