79 lines
2.0 KiB
Go
79 lines
2.0 KiB
Go
package models
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"log"
|
|
"main/configs"
|
|
"net/http"
|
|
"time"
|
|
)
|
|
|
|
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"`
|
|
}
|
|
|
|
func (server *Server) CheckStatus() (err error) {
|
|
// 不用類型的模型有不同的狀態檢查方式
|
|
if server.Type == "train" {
|
|
}
|
|
|
|
resp, err := http.Get(fmt.Sprintf("http://%s:%d/status", server.IP, server.Port))
|
|
if err != nil {
|
|
log.Println("服務器狀態異常", err)
|
|
server.Status = "異常"
|
|
return
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
// 解碼json
|
|
var data map[string]interface{}
|
|
if err = json.NewDecoder(resp.Body).Decode(&data); err != nil {
|
|
return
|
|
}
|
|
|
|
log.Println("data:", data)
|
|
|
|
// 檢查服務器是否正常
|
|
if data["status"] != "ok" {
|
|
log.Println("服務器狀態異常", err)
|
|
server.Status = "異常"
|
|
return
|
|
}
|
|
|
|
configs.ORMDB().Save(&server)
|
|
|
|
// 檢查服務器是否正常
|
|
return
|
|
}
|
|
|
|
func init() {
|
|
configs.ORMDB().AutoMigrate(&Server{})
|
|
|
|
// 添加一個預設的訓練機
|
|
configs.ORMDB().Create(&Server{
|
|
Name: "GPU T4",
|
|
Type: "train",
|
|
IP: "106.15.192.42",
|
|
Port: 7860,
|
|
Status: "閒置",
|
|
})
|
|
|
|
// 檢查所有服務器的狀態, 無效的服務器設置為異常
|
|
var servers []Server
|
|
configs.ORMDB().Find(&servers)
|
|
for _, server := range servers {
|
|
server.CheckStatus()
|
|
}
|
|
}
|