feat:r2上传文件接口不记录进数据库

yahahaff · Jul 24, 2024 · 0d09439 · 0d09439
1 parent 7cb4757
commit 0d09439
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/internal/middlewares/record_operation.go b/internal/middlewares/record_operation.go
@@ -34,8 +34,14 @@ func RecordOperation() gin.HandlerFunc {
 		c.Writer = w
 		start := time.Now()
 
-		//GET 请求不记录
-		if c.Request.Method != "GET" {
+		notRecordMethods := map[string]bool{
+			"GET": false,
+		}
+
+		specificURL := "/api/cloudflare/r2/file/upload"
+
+		// 判断那些请求方法不被记录
+		if notRecordMethods[c.Request.Method] && c.Request.URL.Path == specificURL {
 			defer func() {
 				requests := string(requestBody)
 				end := time.Now()

diff --git a/internal/routes/router.go b/internal/routes/router.go
@@ -2,13 +2,14 @@ package routes
 
 import (
 	"github.com/gin-gonic/gin"
+	"github.com/yahahaff/rapide/internal/middlewares"
 	"github.com/yahahaff/rapide/internal/routes/http"
 	"github.com/yahahaff/rapide/internal/routes/sys"
 )
 
 // RegisterAPIRoutes 注册分支路由
 func RegisterAPIRoutes(Router *gin.Engine) {
-	//Router.Use(middlewares.RecordOperation())
+	Router.Use(middlewares.RecordOperation())
 	sys.RouterGroup(Router)
 	http.RouterGroup(Router)