騰訊雲對接

2023-06-05 15:50:47 +08:00
parent baa020f6ab
commit da829dc7da
4 changed files with 259 additions and 80 deletions
--- 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"`
 }
--- 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 {
+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
 }
--- 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)
+//	}
+//}
--- 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))