整理代碼

routers/servers.go

@@ -24,15 +24,9 @@ func ServersGet(w http.ResponseWriter, r *http.Request) {
 	// 獲取服務器列表
 	db.Offset((listview.Page - 1) * listview.PageSize).Limit(listview.PageSize).Find(&server_list)
 	for _, server := range server_list {
-		// 驗證服務器狀態
-		resp, err := http.Get(fmt.Sprintf("http://%s:%d/docs", server.IP, server.Port))
-		if err != nil || resp.StatusCode != http.StatusOK {
-			server.Status = "異常"
-		} else {
-			server.Status = "正常"
-		}
-		// 讀取模型信息
-		resp, err = http.Get(fmt.Sprintf("http://%s:%d/sdapi/v1/sd-models", server.IP, server.Port))
+		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 {