138 lines
3.9 KiB
Go
138 lines
3.9 KiB
Go
package routers
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
"io/ioutil"
|
||
"main/configs"
|
||
"main/models"
|
||
"main/utils"
|
||
"net/http"
|
||
|
||
"github.com/google/uuid"
|
||
"github.com/gorilla/mux"
|
||
)
|
||
|
||
func ServersGet(w http.ResponseWriter, r *http.Request) {
|
||
var listview models.ListView
|
||
listview.Page = utils.ParamInt(r.URL.Query().Get("page"), 1)
|
||
listview.PageSize = utils.ParamInt(r.URL.Query().Get("pageSize"), 10)
|
||
var server_list []models.Server
|
||
db := configs.ORMDB()
|
||
// 獲取服務器總數
|
||
db.Model(&models.Server{}).Count(&listview.Total)
|
||
// 獲取服務器列表
|
||
db.Offset((listview.Page - 1) * listview.PageSize).Limit(listview.PageSize).Find(&server_list)
|
||
for _, server := range server_list {
|
||
server.CheckStatus() // 驗證服務器狀態
|
||
//// 讀取模型信息
|
||
resp, err := http.Get(fmt.Sprintf("http://%s:%d/sdapi/v1/sd-models", server.IP, server.Port))
|
||
if err != nil || resp.StatusCode != http.StatusOK {
|
||
server.Models = []map[string]interface{}{}
|
||
} else {
|
||
var models []map[string]interface{}
|
||
body, _ := ioutil.ReadAll(resp.Body)
|
||
defer resp.Body.Close()
|
||
if err := json.Unmarshal(body, &models); err != nil {
|
||
server.Models = []map[string]interface{}{}
|
||
}
|
||
server.Models = models
|
||
}
|
||
listview.List = append(listview.List, server)
|
||
}
|
||
listview.Next = listview.Page*listview.PageSize < int(listview.Total)
|
||
listview.WriteJSON(w)
|
||
}
|
||
|
||
func ServersPost(w http.ResponseWriter, r *http.Request) {
|
||
var server models.Server
|
||
|
||
// 獲取參數
|
||
body, err := ioutil.ReadAll(r.Body)
|
||
if err != nil {
|
||
fmt.Println("獲取數據失敗:", err)
|
||
w.WriteHeader(http.StatusBadRequest)
|
||
w.Write([]byte(err.Error()))
|
||
return
|
||
}
|
||
defer r.Body.Close()
|
||
|
||
// 解碼JSON
|
||
if err := json.Unmarshal(body, &server); err != nil {
|
||
fmt.Println("解碼JSON失敗:", err)
|
||
w.WriteHeader(http.StatusBadRequest)
|
||
w.Write([]byte(err.Error()))
|
||
return
|
||
}
|
||
|
||
// 如果不指定類型，禁止創建服務器, 必須指定類型:訓練|推理
|
||
if server.Type != "訓練" && server.Type != "推理" {
|
||
w.WriteHeader(http.StatusBadRequest)
|
||
w.Write([]byte("必須指定類型:訓練|推理"))
|
||
return
|
||
}
|
||
|
||
// 如果不指定名稱，則使用uuid生成隨機名稱
|
||
if server.Name == "" {
|
||
server.Name = uuid.New().String()
|
||
}
|
||
|
||
// 如果不指定 port，則使用默認 port
|
||
if server.Port <= 0 {
|
||
server.Port = 7860
|
||
}
|
||
|
||
// 如果不指定IP，則自動創建新服務器
|
||
if server.IP == "" {
|
||
// TODO: 創建新服務器
|
||
w.WriteHeader(http.StatusBadRequest)
|
||
w.Write([]byte("必須指定IP, 因爲當前禁止自動創建服務器"))
|
||
return
|
||
}
|
||
|
||
// 檢查服務器是否已經存在
|
||
var count int64
|
||
configs.ORMDB().Model(&models.Server{}).Where("ip = ?", server.IP).Count(&count)
|
||
if count > 0 {
|
||
w.WriteHeader(http.StatusBadRequest)
|
||
w.Write([]byte("服務器已經存在"))
|
||
return
|
||
}
|
||
|
||
// 檢查服務器狀態是否正常
|
||
err = server.CheckStatus()
|
||
if err != nil {
|
||
w.WriteHeader(http.StatusBadRequest)
|
||
w.Write([]byte("服務器狀態錯誤:" + err.Error()))
|
||
return
|
||
}
|
||
|
||
// 創建服務器
|
||
configs.ORMDB().Create(&server)
|
||
w.Header().Set("Content-Type", "application/json; charset=utf-8")
|
||
w.Write(utils.ToJSON(server))
|
||
}
|
||
|
||
func ServersItemGet(w http.ResponseWriter, r *http.Request) {
|
||
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: mux.Vars(r)["id"]}
|
||
configs.ORMDB().First(&server)
|
||
// TODO: update server
|
||
configs.ORMDB().Save(&server)
|
||
w.Header().Set("Content-Type", "application/json; charset=utf-8")
|
||
w.Write(utils.ToJSON(server))
|
||
}
|
||
|
||
func ServersItemDelete(w http.ResponseWriter, r *http.Request) {
|
||
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))
|
||
}
|