rpc缓存增加缓存的能力

client.go

@@ -8,9 +8,12 @@
 package httpclient
 
 import (
+	"git.zhangdeman.cn/gateway/httpclient/cache"
 	"git.zhangdeman.cn/gateway/httpclient/define"
 	"git.zhangdeman.cn/gateway/httpclient/validate"
+	"git.zhangdeman.cn/zhangdeman/serialize"
 	"github.com/go-resty/resty/v2"
+	"strings"
 	"time"
 )
 
@@ -19,7 +22,7 @@ import (
 // Author : go_developer@163.com<白茶清欢>
 //
 // Date : 15:27 2024/5/31
-func NewHttpClient(reqConfig *define.Request) (*HttpClient, error) {
+func NewHttpClient(reqConfig *define.Request, cacheInstance cache.ICache) (*HttpClient, error) {
 	// 验证配置正确性以及初始化默认值
 	if err := validate.RequestConfig(reqConfig); nil != err {
 		return nil, err
@@ -33,6 +36,7 @@ func NewHttpClient(reqConfig *define.Request) (*HttpClient, error) {
 		http5xxHandler:           make([]define.Http5xxHandler, 0),
 		httpBusinessErrorHandler: make([]define.HttpBusinessErrorHandler, 0),
 		requestFinishHandler:     make([]define.RequestFinishHandler, 0),
+		cacheInstance:            cacheInstance,
 	}
 	return hc, nil
 }
@@ -51,6 +55,7 @@ type HttpClient struct {
 	httpBusinessErrorHandler []define.HttpBusinessErrorHandler
 	requestSendErrorHandler  []define.RequestSendErrorHandler
 	requestFinishHandler     []define.RequestFinishHandler
+	cacheInstance            cache.ICache
 }
 
 // OnResponse4xx 4xx处理逻辑
@@ -135,6 +140,14 @@ func (hc *HttpClient) GetRestyClient() *resty.Client {
 //
 // Date : 15:52 2024/5/31
 func (hc *HttpClient) Request() *define.Response {
+	var (
+		cacheResult *define.Response
+	)
+
+	if cacheResult = hc.getCacheResult(); nil != cacheResult {
+		return cacheResult
+	}
+
 	hc.Client.OnBeforeRequest(hc.getRequestValidateMiddleware())  // 请求参数验证中间件必注册
 	hc.Client.OnAfterResponse(hc.getResponseValidateMiddleware()) // 响应验证中间件必注册
 	var (
@@ -169,7 +182,7 @@ func (hc *HttpClient) newResponse(seq int, requestCount int) *define.Response {
 		ResponseDataRule:  nil,
 		Seq:               seq,
 		RequestStartTime:  0,
-		FinishRequestTime: 0,
+		RequestFinishTime: 0,
 		UsedTime:          0,
 		RestyResponse:     nil,
 		IsSuccess:         false,
@@ -177,3 +190,51 @@ func (hc *HttpClient) newResponse(seq int, requestCount int) *define.Response {
 		FailInfo:          nil,
 	}
 }
+
+// getCacheResult 获取缓存结果
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 16:04 2024/6/3
+func (hc *HttpClient) getCacheResult() *define.Response {
+	if nil == hc.cacheInstance || !hc.cacheInstance.Enable() {
+		return nil
+	}
+	startTime := time.Now().UnixMilli()
+	cacheKey := hc.cacheInstance.GetKey(hc.reqConfig)
+	cacheValue := strings.TrimSpace(hc.cacheInstance.GetValue(cacheKey))
+	if len(cacheValue) == 0 {
+		return nil
+	}
+	response := hc.newResponse(-1, -1)
+	if err := serialize.JSON.UnmarshalWithNumber([]byte(cacheValue), response); nil != err {
+		return nil
+	}
+	response.IsCache = true                                                    // 设置缓存标记
+	response.RequestStartTime = startTime                                      // 开始时间
+	response.RequestFinishTime = time.Now().UnixMilli()                        // 结束时间
+	response.UsedTime = response.RequestFinishTime - response.RequestStartTime // 耗时
+	response.CacheKey = cacheKey                                               // 缓存key
+	response.CacheValue = cacheValue                                           // 缓存值
+	return response
+}
+
+// setCacheResult ...
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 16:24 2024/6/3
+func (hc *HttpClient) setCacheResult(response *define.Response) (bool, error) {
+	if nil == response || nil == hc.cacheInstance {
+		return false, nil
+	}
+	// 全局未开启或者当前请求不支持缓存
+	if !hc.cacheInstance.Enable() || !hc.cacheInstance.IsAllow(hc.reqConfig, response) {
+		return false, nil
+	}
+	cacheKey := hc.cacheInstance.GetKey(hc.reqConfig)
+	if err := hc.cacheInstance.SetValue(cacheKey, serialize.JSON.MarshalForString(response)); nil != err {
+		return false, err
+	}
+	return true, nil
+}