feat: upstream health check #166

Author: 0xJacky

api/upstream/router.go

@@ -0,0 +1,7 @@
+package upstream
+
+import "github.com/gin-gonic/gin"
+
+func InitRouter(r *gin.RouterGroup) {
+	r.GET("/availability_test", AvailabilityTest)
+}

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)
+    }
+}

app/package.json

@@ -34,7 +34,7 @@
     "vue-router": "^4.2.5",
     "vue3-ace-editor": "2.2.4",
     "vue3-apexcharts": "^1.4.4",
-    "vue3-gettext": "3.0.0-beta.2",
+    "vue3-gettext": "^3.0.0-beta.3",
     "vuedraggable": "^4.1.0",
     "xterm": "^5.3.0",
     "xterm-addon-attach": "^0.9.0",
@@ -67,8 +67,7 @@
     "unplugin-auto-import": "^0.17.1",
     "unplugin-vue-components": "^0.25.2",
     "unplugin-vue-define-options": "^1.4.0",
-    "vite": "^5.0.8",
-    "vite-plugin-html": "^3.2.0",
+    "vite": "^5.0.9",
     "vite-svg-loader": "^5.1.0",
     "vue-tsc": "^1.8.22"
   }