优化代码组织

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)
+}