reply.go 543 B

123456789101112131415161718192021222324252627282930313233
  1. package reply
  2. type SuccessReply struct {
  3. Code int32 `json:"code"`
  4. Message string `json:"message"`
  5. Data interface{} `json:"data"`
  6. }
  7. type ReplyFunc func(data any) any
  8. var wrapReplyFunc ReplyFunc = defaultFunc
  9. func defaultFunc(data any) any {
  10. success := &SuccessReply{
  11. Code: 0,
  12. }
  13. if data != nil {
  14. success.Data = data
  15. }
  16. return success
  17. }
  18. func WrapReply(data any) any {
  19. return wrapReplyFunc(data)
  20. }
  21. func SetReplyFunc(f ReplyFunc) {
  22. if f == nil {
  23. panic("reply: SetReplyFunc: f is nil")
  24. }
  25. wrapReplyFunc = f
  26. }