From 0d981b32c6d9a4ffb31a1a71c0c2fba3648d1f70 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:57:02 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E8=8E=B7=E5=8F=96logic=20aft?=
 =?UTF-8?q?er=20response=E5=AE=9E=E4=BE=8B=E7=9A=84=E6=96=B9=E6=B3=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 request/wrapper.go | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/request/wrapper.go b/request/wrapper.go
index 85cc7f1..4749e94 100644
--- a/request/wrapper.go
+++ b/request/wrapper.go
@@ -8,10 +8,12 @@
 package request
 
 import (
+	"git.zhangdeman.cn/zhangdeman/gin/define"
 	"git.zhangdeman.cn/zhangdeman/gin/request/parse_body"
 	"git.zhangdeman.cn/zhangdeman/wrapper"
 	"github.com/gin-gonic/gin"
 	"strings"
+	"sync"
 )
 
 var (
@@ -268,3 +270,25 @@ func (wh *wrapperHandle) ParseCookie(ctx *gin.Context) map[string]string {
 	}
 	return cookieData
 }
+
+// GetLogicAfterResponse ...
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 16:54 2025/2/28
+func (wh *wrapperHandle) GetLogicAfterResponse(ctx *gin.Context) *define.LogicAfterResponse {
+	if nil == ctx || nil == ctx.Request {
+		return nil
+	}
+	l, exist := ctx.Get(define.LogicAfterResponseKey)
+	if !exist || nil == l {
+		l = &define.LogicAfterResponse{
+			SuccessHookFuncList: make([]func(), 0),
+			FailureHookFuncList: make([]func(), 0),
+			Lock:                &sync.RWMutex{},
+		}
+		ctx.Set(define.LogicAfterResponseKey, l)
+	}
+	// 就这么写, key值如果被其他人覆盖成非法值, 此处会直接panic
+	return l.(*define.LogicAfterResponse)
+}