From 2f398c73b35c487d3f0b0b775aa82d523e7ed769 Mon Sep 17 00:00:00 2001
From: yanweidong <david.yan@qq.com>
Date: Tue, 2 Sep 2025 10:01:09 +0800
Subject: [PATCH] add middleware for CORS and mode configuration

---
 middleware/cors.go | 18 ++++++++++++++++++
 middleware/mode.go | 16 ++++++++++++++++
 2 files changed, 34 insertions(+)
 create mode 100644 middleware/cors.go
 create mode 100644 middleware/mode.go

diff --git a/middleware/cors.go b/middleware/cors.go
new file mode 100644
index 0000000..8244dba
--- /dev/null
+++ b/middleware/cors.go
@@ -0,0 +1,18 @@
+package middleware
+
+import (
+	"github.com/gin-contrib/cors"
+	"github.com/gin-gonic/gin"
+)
+
+func Cors() gin.HandlerFunc {
+	return cors.New(cors.Config{
+		AllowAllOrigins: true,
+		AllowHeaders: []string{
+			"Origin", "Content-Length", "Content-Type", "Workspace", "Request-Id", "Authorization", "Token",
+		},
+		AllowMethods: []string{
+			"GET", "POST", "PUT", "DELETE", "HEAD", "OPTIONS",
+		},
+	})
+}
diff --git a/middleware/mode.go b/middleware/mode.go
new file mode 100644
index 0000000..c52abc8
--- /dev/null
+++ b/middleware/mode.go
@@ -0,0 +1,16 @@
+package middleware
+
+import (
+	"git.apinb.com/bsm-sdk/core/env"
+	"git.apinb.com/bsm-sdk/core/vars"
+	"github.com/gin-gonic/gin"
+)
+
+func Mode(app *gin.Engine) {
+	// 设置gin模式
+	if env.Runtime.Mode == vars.RUN_MODE_PROD {
+		gin.SetMode(gin.ReleaseMode)
+	} else {
+		gin.SetMode(gin.DebugMode)
+	}
+}