feat: convert go-zero middleware to gin

zeromicro · May 23, 2024 · 1bdc1f7 · 1bdc1f7
1 parent 1deb9fc
commit 1bdc1f7
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/router/gin/middleware.go b/router/gin/middleware.go
@@ -0,0 +1,25 @@
+package gin
+
+import (
+	"net/http"
+
+	"github.com/gin-gonic/gin"
+)
+
+type (
+	Middleware func(next http.Handler) http.Handler
+)
+
+func HandlerFunc(ctx *gin.Context) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		ctx.Request = r
+		ctx.Next()
+	}
+}
+
+func ZeroMiddleware(middleware Middleware) func(ctx *gin.Context) {
+	return func(ctx *gin.Context) {
+		next := HandlerFunc(ctx)
+		middleware(next).ServeHTTP(ctx.Writer, ctx.Request)
+	}
+}