允许ws跨域

routers/images.go

@@ -30,6 +30,9 @@ func ImagesGet(w http.ResponseWriter, r *http.Request) {
 	// websocket 推理图像
 	if r.Header.Get("Upgrade") == "websocket" {
 		upgrader := websocket.Upgrader{}
+		upgrader.CheckOrigin = func(r *http.Request) bool {
+			return true
+		}
 		conn, err := upgrader.Upgrade(w, r, nil)
 		if err != nil {
 			log.Println(err)
@@ -147,7 +150,6 @@ func ImagesPost(w http.ResponseWriter, r *http.Request) {
 
 			w.Header().Set("Content-Type", "application/json; charset=utf-8")
 			json.NewEncoder(w).Encode(image_list)
-			//w.Write(utils.ToJSON({"task": task, "list": image_list}))
 			return
 		}