GET类型通道创建端点；修改完善返回格式处理逻辑；动态刷新remote令牌

pkg/remote/remote.go

@@ -1,6 +1,7 @@
 package remote
 
 import (
+	"context"
 	"encoding/json"
 	"errors"
 	"fmt"
@@ -9,8 +10,10 @@ import (
 	"net/http/httputil"
 	"net/url"
 	"platform/pkg/env"
+	"platform/pkg/rds"
 	"strconv"
 	"strings"
+	"time"
 )
 
 // CloudClient 定义云服务接口
@@ -28,16 +31,14 @@ type GatewayClient interface {
 }
 
 type cloud struct {
-	url   string
-	token string
+	url string
 }
 
 var Cloud CloudClient
 
 func Init() {
 	Cloud = &cloud{
-		url:   env.RemoteAddr,
-		token: env.RemoteToken,
+		url: env.RemoteAddr,
 	}
 }
 
@@ -260,34 +261,98 @@ func (c *cloud) requestCloud(method string, url string, data string) (*http.Resp
 	}
 
 	req.Header.Set("Content-Type", "application/json")
-	req.Header.Set("token", c.token)
 
-	if env.DebugHttpDump {
-		str, err := httputil.DumpRequest(req, true)
+	var resp *http.Response
+	for range 2 {
+		token, err := c.token(false)
 		if err != nil {
 			return nil, err
 		}
-		fmt.Println("==============================")
-		fmt.Println(string(str))
-	}
+		req.Header.Set("token", token)
 
-	resp, err := http.DefaultClient.Do(req)
-	if err != nil {
-		return nil, err
-	}
+		if env.DebugHttpDump {
+			str, err := httputil.DumpRequest(req, true)
+			if err != nil {
+				return nil, err
+			}
+			fmt.Println("==============================")
+			fmt.Println(string(str))
+		}
 
-	if env.DebugHttpDump {
-		str, err := httputil.DumpResponse(resp, true)
+		resp, err = http.DefaultClient.Do(req)
 		if err != nil {
 			return nil, err
 		}
-		fmt.Println("==============================")
-		fmt.Println(string(str))
+
+		if env.DebugHttpDump {
+			str, err := httputil.DumpResponse(resp, true)
+			if err != nil {
+				return nil, err
+			}
+			fmt.Println("==============================")
+			fmt.Println(string(str))
+		}
+
+		if resp.StatusCode != 401 {
+			break
+		}
 	}
 
 	return resp, nil
 }
 
+func (c *cloud) token(refresh bool) (string, error) {
+	// redis 获取令牌
+	if !refresh {
+		token, err := rds.Client.Get(context.Background(), "remote:token").Result()
+		if err == nil && token != "" {
+			return token, nil
+		}
+	}
+
+	// redis 获取失败，重新获取
+	resp, err := http.Get(env.RemoteToken)
+	if err != nil {
+		return "", err
+	}
+	defer func(Body io.ReadCloser) {
+		_ = Body.Close()
+	}(resp.Body)
+
+	dump, err := httputil.DumpResponse(resp, true)
+	if err != nil {
+		return "", err
+	}
+	fmt.Println(string(dump))
+
+	if resp.StatusCode != http.StatusOK {
+		return "", errors.New("failed to get token")
+	}
+
+	body, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return "", err
+	}
+	var result map[string]any
+	err = json.Unmarshal(body, &result)
+	if err != nil {
+		return "", err
+	}
+
+	if result["code"].(float64) != 1 {
+		return "", errors.New("failed to get cloud token")
+	}
+
+	// redis 设置令牌
+	token := result["token"].(string)
+	err = rds.Client.Set(context.Background(), "remote:token", token, 1*time.Hour).Err()
+	if err != nil {
+		return "", err
+	}
+
+	return token, nil
+}
+
 type gateway struct {
 	url      string
 	username string