生成图像(默认参数)

routers/models.go

@@ -15,11 +15,8 @@ import (
 	"strconv"
 
 	"github.com/gorilla/mux"
-	"github.com/gorilla/websocket"
 )
 
-var manager = models.NewWebSocketManager()
-
 func init() {
 	// 初始化模型路由: 检查本地模型目录是否存在, 不存在则创建
 	if _, err := os.Stat("data/models"); err != nil {
@@ -190,39 +187,6 @@ func ModelsPost(w http.ResponseWriter, r *http.Request) {
 
 // 獲取模型詳情
 func ModelItemGet(w http.ResponseWriter, r *http.Request) {
-	if r.Header.Get("Upgrade") == "websocket" {
-		vars := mux.Vars(r)
-		id, _ := strconv.Atoi(vars["id"])
-
-		var model = models.Model{ID: id}
-		if err := configs.ORMDB().Take(&model, id).Error; err != nil {
-			w.WriteHeader(http.StatusNotFound)
-			return
-		}
-
-		upgrader := websocket.Upgrader{}
-		conn, err := upgrader.Upgrade(w, r, nil)
-		if err != nil {
-			log.Println(err)
-			return
-		}
-		defer conn.Close()
-		wsid := manager.AddConnection(conn)
-		defer manager.RemoveConnection(wsid)
-		for {
-			_, msg, err := conn.ReadMessage()
-			if err != nil {
-				log.Println(err)
-				return
-			}
-			log.Println(string(msg))
-			if string(msg) == "close" {
-				break
-			}
-		}
-		return
-	}
-
 	var model = models.Model{ID: utils.ParamInt(mux.Vars(r)["id"], 0)}
 	if err := configs.ORMDB().Take(&model, utils.ParamInt(mux.Vars(r)["id"], 0)).Error; err != nil {
 		w.WriteHeader(http.StatusNotFound)