feat: 添加 MustWithData

opt/opt.go

@@ -0,0 +1,4 @@
+package opt
+
+type Config struct {
+}

resp/resp.go

@@ -3,6 +3,7 @@ package resp
 import (
 	"errors"
 	"github.com/gin-gonic/gin"
+	"github.com/spf13/cast"
 )
 
 type res struct {
@@ -53,6 +54,10 @@ func _r(c *gin.Context, r *res, args ...any) {
 		r.Data = args[1]
 	case 3:
 		r.Err = args[2]
+	case 4:
+		if code, err := cast.ToIntE(args[3]); err == nil {
+			r.Code = code
+		}
 	}
 
 	if r.Msg == "" {
@@ -65,6 +70,7 @@ func _r(c *gin.Context, r *res, args ...any) {
 func R400(c *gin.Context, args ...any) {
 	r := &res{
 		Status: 400,
+		Code:   -1,
 	}
 
 	_r(c, r, args...)
@@ -73,6 +79,7 @@ func R400(c *gin.Context, args ...any) {
 func R401(c *gin.Context, args ...any) {
 	r := &res{
 		Status: 401,
+		Code:   -1,
 	}
 
 	_r(c, r, args...)
@@ -81,6 +88,7 @@ func R401(c *gin.Context, args ...any) {
 func R403(c *gin.Context, args ...any) {
 	r := &res{
 		Status: 403,
+		Code:   -1,
 	}
 
 	_r(c, r, args...)
@@ -89,6 +97,7 @@ func R403(c *gin.Context, args ...any) {
 func R500(c *gin.Context, args ...any) {
 	r := &res{
 		Status: 500,
+		Code:   -1,
 	}
 
 	_r(c, r, args...)
@@ -97,6 +106,7 @@ func R500(c *gin.Context, args ...any) {
 func R501(c *gin.Context, args ...any) {
 	r := &res{
 		Status: 501,
+		Code:   -1,
 	}
 
 	_r(c, r, args...)

tool/gin.go

@@ -0,0 +1,12 @@
+package tool
+
+import "github.com/gin-gonic/gin"
+
+func Local(c *gin.Context, key string) any {
+	data, ok := c.Get(key)
+	if !ok {
+		return nil
+	}
+
+	return data
+}

tool/must.go

@@ -11,3 +11,8 @@ func Must(errs ...error) {
 		}
 	}
 }
+
+func MustWithData[T any](data T, err error) T {
+	Must(err)
+	return data
+}