gameui/ai
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

允许ws跨域

Browse Source
This commit is contained in:
桜華
2023-06-20 19:57:39 +08:00
parent e795ac5d9a
commit 4fe977eebb
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
routers/images.go
View File

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