包装handler支持响应后触发相关逻辑

router/define.go

@@ -7,7 +7,10 @@
 // Date : 2024-07-20 22:57
 package router
 
-import "reflect"
+import (
+	"reflect"
+	"sync"
+)
 
 const (
 	PrefixFuncName         = "RouterPrefix"     // 路由前缀函数名称
@@ -63,3 +66,24 @@ type UriParam struct {
 const (
 	FieldNameMeta = "Meta" // 元信息字段
 )
+
+const (
+	LogicAfterResponseKey = "__logic_after_response__"
+)
+
+type LogicAfterResponse struct {
+	SuccessHookFuncList []func()      `json:"-"` // 请求最后需要执行的成功hook函数
+	FailureHookFuncList []func()      `json:"-"` // 请求最后需要执行的失败hook函数
+	Lock                *sync.RWMutex `json:"-"` // 逻辑锁
+}
+
+func (logic *LogicAfterResponse) AddSuccessHook(f func()) {
+	logic.Lock.Lock()
+	defer logic.Lock.Unlock()
+	logic.SuccessHookFuncList = append(logic.SuccessHookFuncList, f)
+}
+func (logic *LogicAfterResponse) AddFailureHook(f func()) {
+	logic.Lock.Lock()
+	defer logic.Lock.Unlock()
+	logic.FailureHookFuncList = append(logic.FailureHookFuncList, f)
+}

router/handler.go

@@ -13,6 +13,7 @@ import (
 	"git.zhangdeman.cn/zhangdeman/gin/response"
 	"github.com/gin-gonic/gin"
 	"reflect"
+	"sync"
 )
 
 // RequestHandler 获取请求处理方法
@@ -45,6 +46,15 @@ func RequestHandler(uriCfg UriConfig) gin.HandlerFunc {
 			return
 		}
 
+		// 初始化响应之后logic
+		logicAfterResponse := &LogicAfterResponse{
+			SuccessHookFuncList: make([]func(), 0),
+			FailureHookFuncList: make([]func(), 0),
+			Lock:                &sync.RWMutex{},
+		}
+		// 此处暴露出去,是为了使用方可以获取到对应数据
+		ctx.Set(LogicAfterResponseKey, logicAfterResponse)
+
 		// 执行逻辑
 		inputValue := reflect.ValueOf(formValue)
 		if uriCfg.FormDataType.Kind() != reflect.Ptr {
@@ -54,22 +64,31 @@ func RequestHandler(uriCfg UriConfig) gin.HandlerFunc {
 		if resList[1].IsNil() {
 			// 请求成功
 			response.Success(ctx, resList[0].Interface())
+			// 执行成功之后的逻辑
+			for _, item := range logicAfterResponse.SuccessHookFuncList {
+				if item != nil {
+					item()
+				}
+			}
 			return
 		}
 		// 请求失败
 		if e, ok = resList[1].Interface().(exception.IException); ok {
-			response.SendWithException(ctx, e, nil)
-			return
-		}
-		if err, ok = resList[1].Interface().(error); ok {
+			// 本身就是exception.IException
+		} else if err, ok = resList[1].Interface().(error); ok {
 			e = exception.NewFromError(-1, err)
-			response.SendWithException(ctx, e, nil)
-			return
+		} else {
+			e = exception.NewWithCodeAndData(-1, map[string]any{
+				"err": resList[1].Interface(),
+			})
 		}
-		e = exception.NewWithCodeAndData(-1, map[string]any{
-			"err": resList[1].Interface(),
-		})
 		response.SendWithException(ctx, e, nil)
+		// 执行失败之后的逻辑
+		for _, item := range logicAfterResponse.FailureHookFuncList {
+			if item != nil {
+				item()
+			}
+		}
 		return
 	}
 }