From 9decd12ee8e403fa525d10d7008ae6697c77d8e6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=99=BD=E8=8C=B6=E6=B8=85=E6=AC=A2?=
 <go_developer@163.com>
Date: Fri, 28 Feb 2025 16:53:02 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E4=BB=A3=E7=A0=81=E7=BB=84?=
 =?UTF-8?q?=E7=BB=87?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 define/response.go | 22 ++++++++++++++++++++++
 router/define.go   | 21 ---------------------
 router/handler.go  |  5 +++--
 3 files changed, 25 insertions(+), 23 deletions(-)

diff --git a/define/response.go b/define/response.go
index 3ff6e63..5829670 100644
--- a/define/response.go
+++ b/define/response.go
@@ -10,6 +10,7 @@ package define
 import (
 	"git.zhangdeman.cn/zhangdeman/consts"
 	"git.zhangdeman.cn/zhangdeman/wrapper"
+	"sync"
 )
 
 var (
@@ -124,3 +125,24 @@ func GetHttpHandleConfig() *HttpHandleConfig {
 		).Value(),
 	}
 }
+
+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)
+}
diff --git a/router/define.go b/router/define.go
index 79723a3..785307b 100644
--- a/router/define.go
+++ b/router/define.go
@@ -66,24 +66,3 @@ 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)
-}
diff --git a/router/handler.go b/router/handler.go
index b3e6a7f..71dd4a7 100644
--- a/router/handler.go
+++ b/router/handler.go
@@ -9,6 +9,7 @@ package router
 
 import (
 	"git.zhangdeman.cn/zhangdeman/exception"
+	"git.zhangdeman.cn/zhangdeman/gin/define"
 	"git.zhangdeman.cn/zhangdeman/gin/request"
 	"git.zhangdeman.cn/zhangdeman/gin/response"
 	"github.com/gin-gonic/gin"
@@ -47,13 +48,13 @@ func RequestHandler(uriCfg UriConfig) gin.HandlerFunc {
 		}
 
 		// 初始化响应之后logic
-		logicAfterResponse := &LogicAfterResponse{
+		logicAfterResponse := &define.LogicAfterResponse{
 			SuccessHookFuncList: make([]func(), 0),
 			FailureHookFuncList: make([]func(), 0),
 			Lock:                &sync.RWMutex{},
 		}
 		// 此处暴露出去,是为了使用方可以获取到对应数据
-		ctx.Set(LogicAfterResponseKey, logicAfterResponse)
+		ctx.Set(define.LogicAfterResponseKey, logicAfterResponse)
 
 		// 执行逻辑
 		inputValue := reflect.ValueOf(formValue)