|
|
@@ -6,25 +6,25 @@ type SuccessReply struct {
|
|
|
Data interface{} `json:"data"`
|
|
|
}
|
|
|
|
|
|
-type ReplyFunc func(req any) any
|
|
|
+type ReplyFunc func(data any) any
|
|
|
|
|
|
-var reply = DefaultFunc
|
|
|
+var wrapReplyFunc ReplyFunc = defaultFunc
|
|
|
|
|
|
-func DefaultFunc(out any) any {
|
|
|
+func defaultFunc(data any) any {
|
|
|
success := &SuccessReply{
|
|
|
Code: 0,
|
|
|
}
|
|
|
|
|
|
- if out != nil {
|
|
|
- success.Data = out
|
|
|
+ if data != nil {
|
|
|
+ success.Data = data
|
|
|
}
|
|
|
return success
|
|
|
}
|
|
|
|
|
|
-func SetReplyFunc(f ReplyFunc) {
|
|
|
- reply = f
|
|
|
+func WrapReply(data any) any {
|
|
|
+ return wrapReplyFunc(data)
|
|
|
}
|
|
|
|
|
|
-func GetReplyFunc() ReplyFunc {
|
|
|
- return reply
|
|
|
+func SetReplyFunc(f ReplyFunc) {
|
|
|
+ wrapReplyFunc = f
|
|
|
}
|