From da829dc7da3bad608792705e9a9c6274f358a73b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=A1=9C=E8=8F=AF?= Date: Mon, 5 Jun 2023 15:50:47 +0800 Subject: [PATCH] =?UTF-8?q?=E9=A8=B0=E8=A8=8A=E9=9B=B2=E5=B0=8D=E6=8E=A5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- models/Model.go | 2 +- models/server.go | 170 +++++++++++++++++++++++++++++++----------- models/server_test.go | 161 +++++++++++++++++++++++++++++++-------- routers/servers.go | 6 +- 4 files changed, 259 insertions(+), 80 deletions(-) diff --git a/models/Model.go b/models/Model.go index 846c8c1..047b936 100644 --- a/models/Model.go +++ b/models/Model.go @@ -27,7 +27,7 @@ type Model struct { Tags TagList `json:"tags"` UserID int `json:"user_id"` DatasetID int `json:"dataset_id"` - ServerID int `json:"server_id"` + ServerID string `json:"server_id"` CreatedAt time.Time `json:"created_at" gorm:"autoCreateTime"` UpdatedAt time.Time `json:"updated_at" gorm:"autoUpdateTime"` } diff --git a/models/server.go b/models/server.go index a8dd9eb..dec20d2 100644 --- a/models/server.go +++ b/models/server.go @@ -14,7 +14,7 @@ import ( ) type Server struct { - ID int `json:"id" gorm:"primary_key"` + ID string `json:"id" gorm:"primary_key"` Name string `json:"name"` Type string `json:"type"` // (訓練|推理) IP string `json:"ip"` @@ -27,32 +27,27 @@ type Server struct { UpdatedAt time.Time `json:"updated_at" gorm:"autoUpdateTime"` } -type Config struct { - TencentCloud struct { - SecretId string `yaml:"SecretId"` - SecretKey string `yaml:"SecretKey"` - } `yaml:"TencentCloud"` +type ServerList struct { + Total int `json:"total"` + List []Server `json:"list"` + Config struct { + TencentCloud struct { + SecretId string `yaml:"SecretId"` + SecretKey string `yaml:"SecretKey"` + Region string `yaml:"Region"` + } `yaml:"TencentCloud"` + } `json:"-"` } // 通過腾讯云API創建服務器 -func CreateServerByTencentCloud(SecretId, SecretKey string) error { - credential := common.NewCredential(SecretId, SecretKey) - cpf := profile.NewClientProfile() - cpf.HttpProfile.Endpoint = "cvm.tencentcloudapi.com" - client, err := cvm.NewClient(credential, "ap-shanghai", cpf) +func (servers *ServerList) Create() 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 fmt.Errorf("初始化騰訊雲SDK客戶端失敗: %v", err) } - - // 实例化一个请求对象,每个接口都会对应一个request对象 + // 实例化一个请求对象, 指定啓動模板, 以創建指定規格的服務器 request := cvm.NewRunInstancesRequest() - - // 指定啓動模板, 以創建指定規格的服務器 - request.LaunchTemplate = &cvm.LaunchTemplate{ - LaunchTemplateId: common.StringPtr("lt-ks6y5evh"), - } - - // 返回的resp是一个RunInstancesResponse的实例，与请求对象对应 + request.LaunchTemplate = &cvm.LaunchTemplate{LaunchTemplateId: common.StringPtr("lt-ks6y5evh")} response, err := client.RunInstances(request) if _, ok := err.(*errors.TencentCloudSDKError); ok { return fmt.Errorf("已返回 API 错误: %v", err) @@ -60,17 +55,78 @@ func CreateServerByTencentCloud(SecretId, SecretKey string) error { if err != nil { return fmt.Errorf("运行实例失败: %v", err) } - // 输出json格式的字符串回包 - fmt.Printf("%s", response.ToJsonString()) + + // 使用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) + } + + // 直接打印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 + } + } + return nil } // 通過腾讯云API註銷指定的服務器 -func TerminateServerByTencentCloud(SecretId, SecretKey string) error { - credential := common.NewCredential(SecretId, SecretKey) - cpf := profile.NewClientProfile() - cpf.HttpProfile.Endpoint = "cvm.tencentcloudapi.com" - client, err := cvm.NewClient(credential, "ap-shanghai", cpf) +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()) if err != nil { return fmt.Errorf("初始化騰訊雲SDK客戶端失敗: %v", err) } @@ -79,7 +135,7 @@ func TerminateServerByTencentCloud(SecretId, SecretKey string) error { request := cvm.NewTerminateInstancesRequest() // 指定要註銷的服務器ID - request.InstanceIds = []*string{common.StringPtr("ins-q0z8ev39")} + request.InstanceIds = []*string{common.StringPtr(InstanceId)} // 返回的resp是一个TerminateInstancesResponse的实例，与请求对象对应 response, err := client.TerminateInstances(request) @@ -89,17 +145,37 @@ func TerminateServerByTencentCloud(SecretId, SecretKey string) error { if err != nil { return fmt.Errorf("註銷實例失敗: %v", err) } - // 输出json格式的字符串回包 - fmt.Printf("%s", response.ToJsonString()) + + fmt.Println("註銷服務器成功:", response.Response) + + // 從列表中刪除服務器 + for i, server := range servers.List { + if server.ID == InstanceId { + servers.List = append(servers.List[:i], servers.List[i+1:]...) + break + } + } + return nil } -// 檢查騰訊雲中的服務器狀態 -func CheckServerStatusByTencentCloud(SecretId, SecretKey string) error { - credential := common.NewCredential(SecretId, SecretKey) - cpf := profile.NewClientProfile() - cpf.HttpProfile.Endpoint = "cvm.tencentcloudapi.com" - client, err := cvm.NewClient(credential, "ap-shanghai", cpf) +// 獲取服務器列表 +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) } @@ -115,17 +191,21 @@ func CheckServerStatusByTencentCloud(SecretId, SecretKey string) error { if err != nil { return fmt.Errorf("运行实例失败: %v", err) } - // 输出json格式的字符串回包 - fmt.Printf("%s", response.ToJsonString()) - // 解析JSON - var data map[string]interface{} - if err := json.Unmarshal([]byte(response.ToJsonString()), &data); err != nil { - return 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 } - fmt.Println(data["TotalCount"]) - fmt.Println(data["InstanceSet"]) - // TODO: 判斷每臺服務器狀態 return nil } diff --git a/models/server_test.go b/models/server_test.go index e0f556c..4acd56e 100644 --- a/models/server_test.go +++ b/models/server_test.go @@ -1,6 +1,7 @@ package models import ( + "fmt" "io/ioutil" "path/filepath" "testing" @@ -8,50 +9,148 @@ import ( "gopkg.in/yaml.v2" ) +// 單元測試:服務器列表 +func TestGetServerList(t *testing.T) { + absPath, _ := filepath.Abs("../data/config.yaml") + configFile, err := ioutil.ReadFile(absPath) + if err != nil { + t.Errorf("讀取配置文件失敗: %v", err) + } + var servers ServerList + if err := yaml.Unmarshal(configFile, &servers.Config); err != nil { + t.Errorf("格式化配置文件失敗: %v", err) + } + + fmt.Println(servers.Config) + + if err := servers.Read(); err != nil { + t.Errorf("獲取服務器列表失敗: %v", err) + } + + // 打印現有的服務器列表 + for _, server := range servers.List { + fmt.Println(server) + } +} + +// 單元測試: 刪除所有服務器 +func TestDeleteAllServer(t *testing.T) { + absPath, _ := filepath.Abs("../data/config.yaml") + configFile, err := ioutil.ReadFile(absPath) + if err != nil { + t.Errorf("讀取配置文件失敗: %v", err) + } + var servers ServerList + if err := yaml.Unmarshal(configFile, &servers.Config); err != nil { + t.Errorf("格式化配置文件失敗: %v", err) + } + + fmt.Println(servers.Config) + + if err := servers.Read(); err != nil { + t.Errorf("獲取服務器列表失敗: %v", err) + } + + for _, server := range servers.List { + if err := servers.Delete(server.ID); err != nil { + t.Errorf("註銷服務器失敗: %v", err) + } + } +} + // 單元測試:創建服務器 +func TestCreateServer(t *testing.T) { + absPath, _ := filepath.Abs("../data/config.yaml") + configFile, err := ioutil.ReadFile(absPath) + if err != nil { + t.Errorf("讀取配置文件失敗: %v", err) + } + var servers ServerList + if err := yaml.Unmarshal(configFile, &servers.Config); err != nil { + t.Errorf("格式化配置文件失敗: %v", err) + } + + fmt.Println(servers.Config) + + if err := servers.Create(); err != nil { + t.Errorf("創建服務器失敗: %v", err) + } +} + +// 單元測試:刪除指定服務器 +func TestDeleteServer(t *testing.T) { + absPath, _ := filepath.Abs("../data/config.yaml") + configFile, err := ioutil.ReadFile(absPath) + if err != nil { + t.Errorf("讀取配置文件失敗: %v", err) + } + var servers ServerList + if err := yaml.Unmarshal(configFile, &servers.Config); err != nil { + t.Errorf("格式化配置文件失敗: %v", err) + } + + fmt.Println(servers.Config) + + if err := servers.Delete("ins-jfpq52jr"); err != nil { + t.Errorf("註銷服務器失敗: %v", err) + } +} + +// 單元測試:全流程: 創建服務器 -> 註銷服務器 func TestCreateServerByTencentCloud(t *testing.T) { absPath, _ := filepath.Abs("../data/config.yaml") configFile, err := ioutil.ReadFile(absPath) if err != nil { t.Errorf("讀取配置文件失敗: %v", err) } - var config Config - if err := yaml.Unmarshal(configFile, &config); err != nil { + var servers ServerList + if err := yaml.Unmarshal(configFile, &servers.Config); err != nil { t.Errorf("格式化配置文件失敗: %v", err) } - if err := CreateServerByTencentCloud(config.TencentCloud.SecretId, config.TencentCloud.SecretKey); err != nil { - t.Errorf("創建服務器失敗: %v", err) - } -} -// 單元測試:獲取服務器列表 -func TestGetServerListByTencentCloud(t *testing.T) { - absPath, _ := filepath.Abs("../data/config.yaml") - configFile, err := ioutil.ReadFile(absPath) - if err != nil { - t.Errorf("讀取配置文件失敗: %v", err) - } - var config Config - if err := yaml.Unmarshal(configFile, &config); err != nil { - t.Errorf("格式化配置文件失敗: %v", err) - } - if err := CheckServerStatusByTencentCloud(config.TencentCloud.SecretId, config.TencentCloud.SecretKey); err != nil { + fmt.Println(servers.Config) + + if err := servers.Read(); err != nil { t.Errorf("獲取服務器列表失敗: %v", err) } -} -// 單元測試:註銷服務器 -func TestTerminateServerByTencentCloud(t *testing.T) { - absPath, _ := filepath.Abs("../data/config.yaml") - configFile, err := ioutil.ReadFile(absPath) - if err != nil { - t.Errorf("讀取配置文件失敗: %v", err) - } - var config Config - if err := yaml.Unmarshal(configFile, &config); err != nil { - t.Errorf("格式化配置文件失敗: %v", err) - } - if err := TerminateServerByTencentCloud(config.TencentCloud.SecretId, config.TencentCloud.SecretKey); err != nil { + //if err := servers.Create(); err != nil { + // t.Errorf("創建服務器失敗: %v", err) + //} + + if err := servers.Delete("ins-pglpixe1"); err != nil { t.Errorf("註銷服務器失敗: %v", err) } } + +//// 單元測試:獲取服務器列表 +//func TestGetServerListByTencentCloud(t *testing.T) { +// absPath, _ := filepath.Abs("../data/config.yaml") +// configFile, err := ioutil.ReadFile(absPath) +// if err != nil { +// t.Errorf("讀取配置文件失敗: %v", err) +// } +// var config Config +// if err := yaml.Unmarshal(configFile, &config); err != nil { +// t.Errorf("格式化配置文件失敗: %v", err) +// } +// if err := CheckServerStatusByTencentCloud(config.TencentCloud.SecretId, config.TencentCloud.SecretKey); err != nil { +// t.Errorf("獲取服務器列表失敗: %v", err) +// } +//} +//// 單元測試:註銷服務器 +//func TestTerminateServerByTencentCloud(t *testing.T) { +// absPath, _ := filepath.Abs("../data/config.yaml") +// configFile, err := ioutil.ReadFile(absPath) +// if err != nil { +// t.Errorf("讀取配置文件失敗: %v", err) +// } +// var config Config +// if err := yaml.Unmarshal(configFile, &config); err != nil { +// t.Errorf("格式化配置文件失敗: %v", err) +// } +// id := "ins-5ht4x6g5" +// if err := TerminateServerByTencentCloud(config.TencentCloud.SecretId, config.TencentCloud.SecretKey, id); err != nil { +// t.Errorf("註銷服務器失敗: %v", err) +// } +//} diff --git a/routers/servers.go b/routers/servers.go index 8f7c667..c684e89 100644 --- a/routers/servers.go +++ b/routers/servers.go @@ -114,14 +114,14 @@ func ServersPost(w http.ResponseWriter, r *http.Request) { } func ServersItemGet(w http.ResponseWriter, r *http.Request) { - server := models.Server{ID: utils.ParamInt(mux.Vars(r)["id"], 0)} + server := models.Server{ID: mux.Vars(r)["id"]} configs.ORMDB().First(&server) w.Header().Set("Content-Type", "application/json; charset=utf-8") w.Write(utils.ToJSON(server)) } func ServersItemPatch(w http.ResponseWriter, r *http.Request) { - server := models.Server{ID: utils.ParamInt(mux.Vars(r)["id"], 0)} + server := models.Server{ID: mux.Vars(r)["id"]} configs.ORMDB().First(&server) // TODO: update server configs.ORMDB().Save(&server) @@ -130,7 +130,7 @@ func ServersItemPatch(w http.ResponseWriter, r *http.Request) { } func ServersItemDelete(w http.ResponseWriter, r *http.Request) { - server := models.Server{ID: utils.ParamInt(mux.Vars(r)["id"], 0)} + server := models.Server{ID: mux.Vars(r)["id"]} configs.ORMDB().Delete(&server) w.Header().Set("Content-Type", "application/json; charset=utf-8") w.Write(utils.ToJSON(server))