創建騰訊雲服務器

go.mod

@@ -10,13 +10,21 @@ require (
 	gorm.io/gorm v1.25.1
 )
 
-require github.com/mattn/go-sqlite3 v1.14.16 // indirect
+require (
+	github.com/davecgh/go-spew v1.1.1 // indirect
+	github.com/mattn/go-sqlite3 v1.14.16 // indirect
+	github.com/pmezard/go-difflib v1.0.0 // indirect
+	github.com/stretchr/objx v0.5.0 // indirect
+	gopkg.in/yaml.v3 v3.0.1 // indirect
+)
 
 require (
 	github.com/jinzhu/inflection v1.0.0 // indirect
 	github.com/jinzhu/now v1.1.5 // indirect
+	github.com/stretchr/testify v1.8.4
 	github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common v1.0.669
 	github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/cvm v1.0.669
+	google.golang.org/protobuf v1.30.0
 	gopkg.in/yaml.v2 v2.4.0
 	gorm.io/driver/sqlite v1.5.0
 )

go.sum

@@ -1,3 +1,8 @@
+github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
+github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=
+github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
 github.com/google/uuid v1.3.0 h1:t6JiXgmwXMjEs8VusXIJk2BXHsn+wx8BZdTaoZ5fu7I=
 github.com/google/uuid v1.3.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
 github.com/gorilla/mux v1.8.0 h1:i40aqfkR1h2SlN9hojwV5ZA91wcXFOvkdNIeFDP5koI=
@@ -11,15 +16,32 @@ github.com/jinzhu/now v1.1.5/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/
 github.com/mattn/go-sqlite3 v1.14.15/go.mod h1:2eHXhiwb8IkHr+BDWZGa96P6+rkvnG63S2DGjv9HUNg=
 github.com/mattn/go-sqlite3 v1.14.16 h1:yOQRA0RpS5PFz/oikGwBEqvAWhWg5ufRz4ETLjwpU1Y=
 github.com/mattn/go-sqlite3 v1.14.16/go.mod h1:2eHXhiwb8IkHr+BDWZGa96P6+rkvnG63S2DGjv9HUNg=
+github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
 github.com/russross/blackfriday v1.6.0 h1:KqfZb0pUVN2lYqZUYRddxF4OR8ZMURnJIG5Y3VRLtww=
 github.com/russross/blackfriday v1.6.0/go.mod h1:ti0ldHuxg49ri4ksnFxlkCfN+hvslNlmVHqNRXXJNAY=
+github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
+github.com/stretchr/objx v0.5.0 h1:1zr/of2m5FGMsad5YfcqgdqdWrIhu+EBEJRhR1U7z/c=
+github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
+github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
+github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
+github.com/stretchr/testify v1.8.4 h1:CcVxjf3Q8PM0mHUKJCdn+eZZtm5yQwehR5yeSVQQcUk=
+github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
 github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common v1.0.669 h1:5KKJBcemqKONBFxMdMyLMvk+TrqXaEPhqe9TrZqB3r0=
 github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common v1.0.669/go.mod h1:7sCQWVkxcsR38nffDW057DRGk8mUjK1Ing/EFOK8s8Y=
 github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/cvm v1.0.669 h1:gc1bPO/YVfuXEIs+HbQ/gFlFjdkJjOsjm8xWqF7hPww=
 github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/cvm v1.0.669/go.mod h1:hhy13j6NKKxt/g62JZEDekJNQx3EAevnHopmwlt2tRc=
+golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
+google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw=
+google.golang.org/protobuf v1.30.0 h1:kPPoIgf3TsEvrm0PFe15JQ+570QVxYzEvvHqChK+cng=
+google.golang.org/protobuf v1.30.0/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=
 gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=
+gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
+gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
+gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
 gorm.io/driver/sqlite v1.5.0 h1:zKYbzRCpBrT1bNijRnxLDJWPjVfImGEn0lSnUY5gZ+c=
 gorm.io/driver/sqlite v1.5.0/go.mod h1:kDMDfntV9u/vuMmz8APHtHF0b4nyBB7sfCieC6G8k8I=
 gorm.io/gorm v1.24.7-0.20230306060331-85eaf9eeda11/go.mod h1:L4uxeKpfBml98NYqVqwAdmV1a2nBtAec/cf3fpucW/k=

models/server.go

@@ -3,8 +3,6 @@ package models
 import (
 	"encoding/json"
 	"fmt"
-	"io/ioutil"
-	"log"
 	"main/configs"
 	"net/http"
 	"time"
@@ -13,8 +11,6 @@ import (
 	"github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common/errors"
 	"github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common/profile"
 	cvm "github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/cvm/v20170312"
-
-	"gopkg.in/yaml.v2"
 )
 
 type Server struct {
@@ -38,46 +34,100 @@ type Config struct {
 	} `yaml:"TencentCloud"`
 }
 
-func CreateServerByTencentCloud() {
+// 通過腾讯云API創建服務器
-	// 從 data/config.yaml 中獲取配置
+func CreateServerByTencentCloud(SecretId, SecretKey string) error {
-	configFile, err := ioutil.ReadFile("data/config.yaml")
+	credential := common.NewCredential(SecretId, SecretKey)
-	if err != nil {
-		log.Fatalf("Failed to read config file: %v", err)
-	}
-
-	var config Config
-	err = yaml.Unmarshal(configFile, &config)
-	if err != nil {
-		log.Fatalf("Failed to unmarshal config file: %v", err)
-	}
-
-	fmt.Println(config.TencentCloud.SecretId)
-	fmt.Println(config.TencentCloud.SecretKey)
-
-	// 实例化一个认证对象，入参需要传入腾讯云账户 SecretId 和 SecretKey，此处还需注意密钥对的保密
-	// 代码泄露可能会导致 SecretId 和 SecretKey 泄露，并威胁账号下所有资源的安全性。以下代码示例仅供参考，建议采用更安全的方式来使用密钥，请参见：https://cloud.tencent.com/document/product/1278/85305
-	// 密钥可前往官网控制台 https://console.cloud.tencent.com/cam/capi 进行获取
-	// 实例化一个client选项，可选的，没有特殊需求可以跳过
-	// 实例化要请求产品的client对象,clientProfile是可选的
-	credential := common.NewCredential(config.TencentCloud.SecretId, config.TencentCloud.SecretKey)
 	cpf := profile.NewClientProfile()
 	cpf.HttpProfile.Endpoint = "cvm.tencentcloudapi.com"
-	client, _ := cvm.NewClient(credential, "", cpf)
+	client, err := cvm.NewClient(credential, "ap-shanghai", cpf)
+	if err != nil {
+		return fmt.Errorf("初始化騰訊雲SDK客戶端失敗: %v", err)
+	}
 
 	// 实例化一个请求对象,每个接口都会对应一个request对象
 	request := cvm.NewRunInstancesRequest()
 
+	// 指定啓動模板, 以創建指定規格的服務器
+	request.LaunchTemplate = &cvm.LaunchTemplate{
+		LaunchTemplateId: common.StringPtr("lt-ks6y5evh"),
+	}
+
 	// 返回的resp是一个RunInstancesResponse的实例，与请求对象对应
 	response, err := client.RunInstances(request)
 	if _, ok := err.(*errors.TencentCloudSDKError); ok {
-		fmt.Printf("An API error has returned: %s", err)
+		return fmt.Errorf("已返回 API 错误: %v", err)
-		return
 	}
 	if err != nil {
-		panic(err)
+		return fmt.Errorf("运行实例失败: %v", err)
 	}
 	// 输出json格式的字符串回包
 	fmt.Printf("%s", response.ToJsonString())
+	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)
+	if err != nil {
+		return fmt.Errorf("初始化騰訊雲SDK客戶端失敗: %v", err)
+	}
+
+	// 实例化一个请求对象,每个接口都会对应一个request对象
+	request := cvm.NewTerminateInstancesRequest()
+
+	// 指定要註銷的服務器ID
+	request.InstanceIds = []*string{common.StringPtr("ins-q0z8ev39")}
+
+	// 返回的resp是一个TerminateInstancesResponse的实例，与请求对象对应
+	response, err := client.TerminateInstances(request)
+	if _, ok := err.(*errors.TencentCloudSDKError); ok {
+		return fmt.Errorf("已返回 API 错误: %v", err)
+	}
+	if err != nil {
+		return fmt.Errorf("註銷實例失敗: %v", err)
+	}
+	// 输出json格式的字符串回包
+	fmt.Printf("%s", response.ToJsonString())
+	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)
+	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)
+	}
+	// 输出json格式的字符串回包
+	fmt.Printf("%s", response.ToJsonString())
+
+	// 解析JSON
+	var data map[string]interface{}
+	if err := json.Unmarshal([]byte(response.ToJsonString()), &data); err != nil {
+		return err
+	}
+	fmt.Println(data["TotalCount"])
+	fmt.Println(data["InstanceSet"])
+	// TODO: 判斷每臺服務器狀態
+
+	return nil
 }
 
 func (server *Server) CheckStatus() error {

models/server_test.go

@@ -0,0 +1,57 @@
+package models
+
+import (
+	"io/ioutil"
+	"path/filepath"
+	"testing"
+
+	"gopkg.in/yaml.v2"
+)
+
+// 單元測試:創建服務器
+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 {
+		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 {
+		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 {
+		t.Errorf("註銷服務器失敗: %v", err)
+	}
+}