feat: upstream health check #166

api/upstream/upstream.go

@@ -0,0 +1,45 @@
+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"
+)
+
+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
+    }
+
+    defer ws.Close()
+
+    var body []string
+
+    err = ws.ReadJSON(&body)
+
+    if err != nil {
+        logger.Error(err)
+        return
+    }
+    
+    for {
+        err = ws.WriteJSON(upstream.AvailabilityTest(body))
+
+        if err != nil {
+            logger.Error(err)
+        }
+
+        time.Sleep(10 * time.Second)
+    }
+}