131 lines
4.2 KiB
Go
131 lines
4.2 KiB
Go
package models
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
"io/ioutil"
|
||
"log"
|
||
"main/configs"
|
||
"net/http"
|
||
"time"
|
||
|
||
"github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common"
|
||
"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 {
|
||
ID int `json:"id" gorm:"primary_key"`
|
||
Name string `json:"name"`
|
||
Type string `json:"type"` // (訓練|推理)
|
||
IP string `json:"ip"`
|
||
Port int `json:"port"`
|
||
Status string `json:"status"` // (異常|初始化|閒置|就緒|工作中|關閉中)
|
||
UserName string `json:"username"`
|
||
Password string `json:"password"`
|
||
Models []map[string]interface{} `json:"models" gorm:"-"` // 數據庫不必保存
|
||
CreatedAt time.Time `json:"created_at" gorm:"autoCreateTime"`
|
||
UpdatedAt time.Time `json:"updated_at" gorm:"autoUpdateTime"`
|
||
}
|
||
|
||
type Config struct {
|
||
TencentCloud struct {
|
||
SecretId string `yaml:"SecretId"`
|
||
SecretKey string `yaml:"SecretKey"`
|
||
} `yaml:"TencentCloud"`
|
||
}
|
||
|
||
func CreateServerByTencentCloud() {
|
||
// 從 data/config.yaml 中獲取配置
|
||
configFile, err := ioutil.ReadFile("data/config.yaml")
|
||
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)
|
||
|
||
// 实例化一个请求对象,每个接口都会对应一个request对象
|
||
request := cvm.NewRunInstancesRequest()
|
||
|
||
// 返回的resp是一个RunInstancesResponse的实例，与请求对象对应
|
||
response, err := client.RunInstances(request)
|
||
if _, ok := err.(*errors.TencentCloudSDKError); ok {
|
||
fmt.Printf("An API error has returned: %s", err)
|
||
return
|
||
}
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
// 输出json格式的字符串回包
|
||
fmt.Printf("%s", response.ToJsonString())
|
||
}
|
||
|
||
func (server *Server) CheckStatus() error {
|
||
switch server.Type {
|
||
case "訓練":
|
||
resp, err := http.Get(fmt.Sprintf("http://%s:%d/dreambooth/status", server.IP, server.Port))
|
||
if err != nil {
|
||
server.Status = "異常"
|
||
return err
|
||
}
|
||
defer resp.Body.Close()
|
||
|
||
// 解碼JSON
|
||
var data map[string]interface{}
|
||
if err := json.NewDecoder(resp.Body).Decode(&data); err != nil {
|
||
return err
|
||
}
|
||
|
||
// 解碼JSON
|
||
var current_state map[string]interface{}
|
||
if err := json.Unmarshal([]byte(data["current_state"].(string)), ¤t_state); err != nil {
|
||
return err
|
||
}
|
||
//log.Println("current_state:", current_state)
|
||
|
||
// 檢查服務器是否正常
|
||
if !current_state["active"].(bool) {
|
||
server.Status = "異常"
|
||
return fmt.Errorf("服務器狀態異常: active=false")
|
||
}
|
||
server.Status = "正常"
|
||
case "推理":
|
||
server.Status = "異常"
|
||
default:
|
||
server.Status = "異常"
|
||
}
|
||
|
||
// 檢查服務器是否正常
|
||
return nil
|
||
}
|
||
|
||
func init() {
|
||
configs.ORMDB().AutoMigrate(&Server{})
|
||
// 檢查所有服務器的狀態, 無效的服務器設置為異常
|
||
var servers []Server
|
||
configs.ORMDB().Find(&servers)
|
||
for _, server := range servers {
|
||
server.CheckStatus()
|
||
}
|
||
}
|