生成图像(默认参数)

models/server.go

@@ -1,6 +1,7 @@
 package models
 
 import (
+	"database/sql/driver"
 	"encoding/json"
 	"fmt"
 	"main/configs"
@@ -8,23 +9,40 @@ import (
 	"time"
 )
 
-type Server struct {
-	ID        string                   `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 ModelList []string
+
+func (list *ModelList) Scan(value interface{}) error {
+	return json.Unmarshal(value.([]byte), list)
 }
 
+func (list ModelList) Value() (driver.Value, error) {
+	return json.Marshal(list)
+}
+
+type Server struct {
+	ID        string    `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    ModelList `json:"models"`
+	CreatedAt time.Time `json:"created_at" gorm:"autoCreateTime"`
+	UpdatedAt time.Time `json:"updated_at" gorm:"autoUpdateTime"`
+}
+
+// 获取所有服务器
+func GetServers() (servers []Server, err error) {
+	err = configs.ORMDB().Find(&servers).Error
+	return
+}
+
+// 檢查服務器是否正常
 func (server *Server) CheckStatus() error {
 	switch server.Type {
-	case "訓練":
+	case "训练":
 		resp, err := http.Get(fmt.Sprintf("http://%s:%d/dreambooth/status", server.IP, server.Port))
 		if err != nil {
 			server.Status = "異常"