mirrors/nginx-ui
1
0
Fork 0
mirror of https://github.com/0xJacky/nginx-ui.git synced 2025-05-12 10:55:51 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 7

fix: websocket of upstream availability test not close

Browse source
This commit is contained in:
0xJacky 2023-12-15 22:34:29 +08:00
parent dfee30e9d9
commit 53e1c32444
No known key found for this signature in database
GPG key ID: B6E4A6E4A561BAF0
1 changed files with 33 additions and 32 deletions
Download patch file Download diff file Expand all files Collapse all files

63
api/upstream/upstream.go
View file

@ -1,45 +1,46 @@
package upstream package upstream
import ( import (
"github.com/0xJacky/Nginx-UI/internal/logger" "github.com/0xJacky/Nginx-UI/internal/logger"
"github.com/0xJacky/Nginx-UI/internal/upstream" "github.com/0xJacky/Nginx-UI/internal/upstream"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
"github.com/gorilla/websocket" "github.com/gorilla/websocket"
"net/http" "net/http"
"time" "time"
) )
func AvailabilityTest(c *gin.Context) { func AvailabilityTest(c *gin.Context) {
var upGrader = websocket.Upgrader{ var upGrader = websocket.Upgrader{
CheckOrigin: func(r *http.Request) bool { CheckOrigin: func(r *http.Request) bool {
return true return true
}, },
} }
// upgrade http to websocket // upgrade http to websocket
ws, err := upGrader.Upgrade(c.Writer, c.Request, nil) ws, err := upGrader.Upgrade(c.Writer, c.Request, nil)
if err != nil { if err != nil {
logger.Error(err) logger.Error(err)
return return
} }
defer ws.Close() defer ws.Close()
var body []string var body []string
err = ws.ReadJSON(&body) err = ws.ReadJSON(&body)
if err != nil { if err != nil {
logger.Error(err) logger.Error(err)
return return
} }
for { for {
err = ws.WriteJSON(upstream.AvailabilityTest(body)) err = ws.WriteJSON(upstream.AvailabilityTest(body))
if err != nil { if err != nil {
logger.Error(err) logger.Error(err)
} return
}
time.Sleep(10 * time.Second) time.Sleep(10 * time.Second)
} }
} }