feat: health check for provider

MetaCubeX · Aug 31, 2023 · 03dda3c · 03dda3c
1 parent b4fc1b4
commit 03dda3c
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 20 deletions.
diff --git a/src/pages/Proxies.tsx b/src/pages/Proxies.tsx
@@ -13,7 +13,8 @@ export default () => {
     updateProxy,
     setProxyGroupByProxyName,
     delayTestByProxyGroupName,
-    updateProxyProviderByProviderName,
+    updateProviderByProviderName,
+    healthCheckByProviderName,
   } = useProxies()
 
   const [collapsedMap, setCollapsedMap] = createSignal<Record<string, boolean>>(
@@ -33,9 +34,14 @@ export default () => {
     delayTestByProxyGroupName(name)
   }
 
+  const onHealthCheckClick = (e: MouseEvent, name: string) => {
+    e.stopPropagation()
+    healthCheckByProviderName(name)
+  }
+
   const onUpdateProviderClick = (e: MouseEvent, name: string) => {
     e.stopPropagation()
-    updateProxyProviderByProviderName(name)
+    updateProviderByProviderName(name)
   }
 
   return (
@@ -112,14 +118,22 @@ export default () => {
                     </div>
                   </div>
 
-                  <button
-                    class="btn btn-circle btn-sm"
-                    onClick={(e) =>
-                      onUpdateProviderClick(e, proxyProvider.name)
-                    }
-                  >
-                    <IconReload />
-                  </button>
+                  <div>
+                    <button
+                      class="btn btn-circle btn-sm mr-2"
+                      onClick={(e) =>
+                        onUpdateProviderClick(e, proxyProvider.name)
+                      }
+                    >
+                      <IconReload />
+                    </button>
+                    <button
+                      class="btn btn-circle btn-sm"
+                      onClick={(e) => onHealthCheckClick(e, proxyProvider.name)}
+                    >
+                      <IconBrandSpeedtest />
+                    </button>
+                  </div>
                 </div>
               )
               const content = (

diff --git a/src/signals/proxies.ts b/src/signals/proxies.ts
@@ -65,19 +65,19 @@ export function useProxies() {
         `group/${proxyGroupName}/delay?url=https%3A%2F%2Fwww.gstatic.com%2Fgenerate_204&timeout=2000`,
       )
       .json()
-    const dMap = delayMap()
 
-    Object.entries(data).forEach(([name, time]) => {
-      dMap[name] = time
-    })
+    setDelayMap({ ...delayMap(), ...data })
+  }
 
-    setDelayMap({ ...dMap })
+  const updateProviderByProviderName = async (proxyProviderName: string) => {
+    await request.put(`providers/proxies/${proxyProviderName}`)
+    await updateProxy()
   }
 
-  const updateProxyProviderByProviderName = async (
-    proxyProviderName: string,
-  ) => {
-    await request.put(`/providers/proxies/${proxyProviderName}`)
+  const healthCheckByProviderName = async (providerName: string) => {
+    await request.get(`providers/proxies/${providerName}/healthcheck`, {
+      timeout: 30 * 1000, // thie api was a little bit slow sometimes...
+    })
     await updateProxy()
   }
 
@@ -89,6 +89,7 @@ export function useProxies() {
     delayMap,
     updateProxy,
     setProxyGroupByProxyName,
-    updateProxyProviderByProviderName,
+    updateProviderByProviderName,
+    healthCheckByProviderName,
   }
 }