From 7ddbea8bcb86be8a69ffd75ce1fdd48a4482aa14 Mon Sep 17 00:00:00 2001
From: LittleSheep <littlesheep.code@hotmail.com>
Date: Sun, 23 Jun 2024 16:11:49 +0800
Subject: [PATCH] :ambulance: Fix query services too much 429

---
 .idea/workspace.xml |  9 ++++-----
 pkg/hyper/conn.go   | 23 ++++++++++++++++++++++-
 2 files changed, 26 insertions(+), 6 deletions(-)

diff --git a/.idea/workspace.xml b/.idea/workspace.xml
index 1eed986..096ec55 100644
--- a/.idea/workspace.xml
+++ b/.idea/workspace.xml
@@ -4,10 +4,9 @@
     <option name="autoReloadType" value="ALL" />
   </component>
   <component name="ChangeListManager">
-    <list default="true" id="3fefb2c4-b6f9-466b-a523-53352e8d6f95" name="更改" comment=":bug: Fix registration service issue">
+    <list default="true" id="3fefb2c4-b6f9-466b-a523-53352e8d6f95" name="更改" comment=":bug: Fix avatar url missing endpoint prefix">
       <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/pkg/internal/models/accounts.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/internal/models/accounts.go" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/settings.toml" beforeDir="false" afterPath="$PROJECT_DIR$/settings.toml" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/pkg/hyper/conn.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/hyper/conn.go" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -144,7 +143,6 @@
     </option>
   </component>
   <component name="VcsManagerConfiguration">
-    <MESSAGE value=":sparkles: Basis perm nodes feature" />
     <MESSAGE value=":sparkles: Permission check" />
     <MESSAGE value=":zap: In memory auth context cache" />
     <MESSAGE value=":sparkles: Bug fixes of permission check" />
@@ -169,7 +167,8 @@
     <MESSAGE value=":bug: FIx cannot resolve service" />
     <MESSAGE value=":sparkles: Accepts token in querystring" />
     <MESSAGE value=":bug: Fix registration service issue" />
-    <option name="LAST_COMMIT_MESSAGE" value=":bug: Fix registration service issue" />
+    <MESSAGE value=":bug: Fix avatar url missing endpoint prefix" />
+    <option name="LAST_COMMIT_MESSAGE" value=":bug: Fix avatar url missing endpoint prefix" />
   </component>
   <component name="VgoProject">
     <settings-migrated>true</settings-migrated>
diff --git a/pkg/hyper/conn.go b/pkg/hyper/conn.go
index c660d0a..391e9a2 100644
--- a/pkg/hyper/conn.go
+++ b/pkg/hyper/conn.go
@@ -1,15 +1,20 @@
 package hyper
 
 import (
+	"context"
 	"fmt"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/credentials/insecure"
+	health "google.golang.org/grpc/health/grpc_health_v1"
+	"time"
 
 	_ "github.com/mbobakov/grpc-consul-resolver"
 )
 
 type HyperConn struct {
 	Addr string
+
+	cacheGrpcConn map[string]*grpc.ClientConn
 }
 
 func NewHyperConn(addr string) *HyperConn {
@@ -17,10 +22,26 @@ func NewHyperConn(addr string) *HyperConn {
 }
 
 func (v *HyperConn) DiscoverServiceGRPC(name string) (*grpc.ClientConn, error) {
+	if val, ok := v.cacheGrpcConn[name]; ok {
+		ctx, cancel := context.WithTimeout(context.Background(), time.Second*10)
+		defer cancel()
+		if _, err := health.NewHealthClient(val).Check(ctx, &health.HealthCheckRequest{
+			Service: name,
+		}); err == nil {
+			return val, nil
+		} else {
+			delete(v.cacheGrpcConn, name)
+		}
+	}
+
 	target := fmt.Sprintf("consul://%s/%s", v.Addr, name)
-	return grpc.NewClient(
+	conn, err := grpc.NewClient(
 		target,
 		grpc.WithTransportCredentials(insecure.NewCredentials()),
 		grpc.WithDefaultServiceConfig(`{"loadBalancingPolicy": "round_robin"}`),
 	)
+	if err == nil {
+		v.cacheGrpcConn[name] = conn
+	}
+	return conn, err
 }