From a63e85aa3700bd408d6e50c36a7e307d40db9442 Mon Sep 17 00:00:00 2001 From: 0xJacky Date: Fri, 15 Dec 2023 23:22:38 +0800 Subject: [PATCH] chore: update code styles for upstream.go --- api/upstream/upstream.go | 70 ++++++++++++++++++++-------------------- 1 file changed, 35 insertions(+), 35 deletions(-) diff --git a/api/upstream/upstream.go b/api/upstream/upstream.go index 7f742a9a..e6874213 100644 --- a/api/upstream/upstream.go +++ b/api/upstream/upstream.go @@ -1,49 +1,49 @@ package upstream import ( - "github.com/0xJacky/Nginx-UI/internal/logger" - "github.com/0xJacky/Nginx-UI/internal/upstream" - "github.com/gin-gonic/gin" - "github.com/gorilla/websocket" - "net/http" - "time" + "github.com/0xJacky/Nginx-UI/internal/logger" + "github.com/0xJacky/Nginx-UI/internal/upstream" + "github.com/gin-gonic/gin" + "github.com/gorilla/websocket" + "net/http" + "time" ) func AvailabilityTest(c *gin.Context) { - var upGrader = websocket.Upgrader{ - CheckOrigin: func(r *http.Request) bool { - return true - }, - } - // upgrade http to websocket - ws, err := upGrader.Upgrade(c.Writer, c.Request, nil) - if err != nil { - logger.Error(err) - return - } + var upGrader = websocket.Upgrader{ + CheckOrigin: func(r *http.Request) bool { + return true + }, + } + // upgrade http to websocket + ws, err := upGrader.Upgrade(c.Writer, c.Request, nil) + if err != nil { + logger.Error(err) + return + } - defer ws.Close() + defer ws.Close() - var body []string + var body []string - err = ws.ReadJSON(&body) + err = ws.ReadJSON(&body) - if err != nil { - logger.Error(err) - return - } + if err != nil { + logger.Error(err) + return + } - for { - err = ws.WriteJSON(upstream.AvailabilityTest(body)) + for { + err = ws.WriteJSON(upstream.AvailabilityTest(body)) - if err != nil || websocket.IsUnexpectedCloseError(err, - websocket.CloseGoingAway, - websocket.CloseNoStatusReceived, - websocket.CloseNormalClosure) { - logger.Error(err) - break - } + if err != nil || websocket.IsUnexpectedCloseError(err, + websocket.CloseGoingAway, + websocket.CloseNoStatusReceived, + websocket.CloseNormalClosure) { + logger.Error(err) + break + } - time.Sleep(10 * time.Second) - } + time.Sleep(10 * time.Second) + } }