修改后端结构

This commit is contained in:
2025-10-04 21:18:35 +08:00
parent cd9367800a
commit f5c3b26479
3 changed files with 136 additions and 116 deletions

View File

@@ -12,7 +12,9 @@ import (
) )
// UploadImageHandler 处理图片上传到OSS // UploadImageHandler 处理图片上传到OSS
func UploadImageHandler(c *gin.Context, cfg *config.Config) { func UploadImageHandler(cfg *config.Config) gin.HandlerFunc {
// 闭包:内部函数可以访问 cfg 参数
return func(c *gin.Context) {
// 获取上传的图片文件 // 获取上传的图片文件
fileHeader, err := c.FormFile("image") fileHeader, err := c.FormFile("image")
if err != nil { if err != nil {
@@ -36,7 +38,7 @@ func UploadImageHandler(c *gin.Context, cfg *config.Config) {
} }
defer file.Close() defer file.Close()
// 检查文件大小 // 检查文件大小使用配置中的max_file_size
if fileHeader.Size > cfg.Upload.MaxFileSize { if fileHeader.Size > cfg.Upload.MaxFileSize {
c.JSON(http.StatusBadRequest, gin.H{ c.JSON(http.StatusBadRequest, gin.H{
"code": 400, "code": 400,
@@ -45,7 +47,7 @@ func UploadImageHandler(c *gin.Context, cfg *config.Config) {
return return
} }
// 检查文件类型 // 检查文件类型使用配置中的allow_image_types
fileType := fileHeader.Header.Get("Content-Type") fileType := fileHeader.Header.Get("Content-Type")
if !strings.Contains(cfg.Upload.AllowImageTypes, fileType) { if !strings.Contains(cfg.Upload.AllowImageTypes, fileType) {
c.JSON(http.StatusBadRequest, gin.H{ c.JSON(http.StatusBadRequest, gin.H{
@@ -55,8 +57,12 @@ func UploadImageHandler(c *gin.Context, cfg *config.Config) {
return return
} }
// 初始化OSS客户端 // 初始化OSS客户端使用配置中的OSS参数
client, err := oss.New(cfg.OSS.Endpoint, cfg.OSS.AccessKeyID, cfg.OSS.AccessKeySecret) client, err := oss.New(
cfg.OSS.Endpoint,
cfg.OSS.AccessKeyID,
cfg.OSS.AccessKeySecret,
)
if err != nil { if err != nil {
c.JSON(http.StatusInternalServerError, gin.H{ c.JSON(http.StatusInternalServerError, gin.H{
"code": 500, "code": 500,
@@ -104,3 +110,4 @@ func UploadImageHandler(c *gin.Context, cfg *config.Config) {
"data": gin.H{"url": imageURL}, "data": gin.H{"url": imageURL},
}) })
} }
}

View File

@@ -1,40 +1,54 @@
package middleware package middleware
import ( import (
"fmt"
"net/http" "net/http"
"github.com/JACKYMYPERSON/hldrCenter/config" "github.com/JACKYMYPERSON/hldrCenter/config"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
) )
func CorsMiddleware(cfg *config.ServerConfig) gin.HandlerFunc { func CorsMiddleware(serverConfig *config.ServerConfig) gin.HandlerFunc {
return func(c *gin.Context) { return func(c *gin.Context) {
// 处理跨域请求头 // 1. 打印配置的允许源(调试用,确认配置是否正确加载)
fmt.Printf("允许的前端源:%v\n", serverConfig.AllowedOrigins)
// 2. 获取请求的Origin头
origin := c.Request.Header.Get("Origin") origin := c.Request.Header.Get("Origin")
if origin != "" && isAllowedOrigin(origin, cfg.AllowedOrigins) { fmt.Printf("当前请求源:%s\n", origin) // 调试用
c.Writer.Header().Set("Access-Control-Allow-Origin", origin)
// 3. 宽松的跨域匹配逻辑
allowOrigin := ""
if len(serverConfig.AllowedOrigins) > 0 {
for _, allowed := range serverConfig.AllowedOrigins {
// 支持通配符*,或精确匹配
if allowed == "*" || allowed == origin {
allowOrigin = origin
break
}
}
} }
c.Writer.Header().Set("Access-Control-Allow-Headers", "Origin, Content-Type, Accept, Authorization") // 4. 即使没有匹配到,也可以临时设置为*(仅测试用,生产环境需删除)
c.Writer.Header().Set("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, OPTIONS") // if allowOrigin == "" {
c.Writer.Header().Set("Access-Control-Allow-Credentials", "true") // allowOrigin = "*"
// }
// 处理预检请求 // 5. 设置核心跨域头
if allowOrigin != "" {
c.Writer.Header().Set("Access-Control-Allow-Origin", allowOrigin)
}
c.Writer.Header().Set("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, OPTIONS")
c.Writer.Header().Set("Access-Control-Allow-Headers", "Origin, Content-Type, Accept, Authorization")
c.Writer.Header().Set("Access-Control-Allow-Credentials", "true")
c.Writer.Header().Set("Access-Control-Max-Age", "86400") // 24小时缓存预检请求
// 6. 处理OPTIONS预检请求
if c.Request.Method == "OPTIONS" { if c.Request.Method == "OPTIONS" {
c.AbortWithStatus(http.StatusOK) c.AbortWithStatus(http.StatusNoContent) // 使用204更规范
return return
} }
c.Next() c.Next()
} }
} }
// 检查来源是否在允许的列表中
func isAllowedOrigin(origin string, allowedOrigins []string) bool {
for _, allowed := range allowedOrigins {
if allowed == "*" || allowed == origin {
return true
}
}
return false
}

View File

@@ -3,6 +3,7 @@ package router
import ( import (
"github.com/JACKYMYPERSON/hldrCenter/config" "github.com/JACKYMYPERSON/hldrCenter/config"
handler "github.com/JACKYMYPERSON/hldrCenter/handler/uploadimg" handler "github.com/JACKYMYPERSON/hldrCenter/handler/uploadimg"
"github.com/JACKYMYPERSON/hldrCenter/middleware"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
) )
@@ -10,16 +11,14 @@ import (
func SetupRouter(cfg *config.Config) *gin.Engine { func SetupRouter(cfg *config.Config) *gin.Engine {
r := gin.Default() r := gin.Default()
// API路由组 // 关键:跨域中间件必须在所有路由定义之前应用
r.Use(middleware.CorsMiddleware(&cfg.Server))
// 定义路由组(必须在中间件之后)
api := r.Group("/api") api := r.Group("/api")
{ {
// 图片上传路由 api.POST("/upload/image", handler.UploadImageHandler(cfg))
api.POST("/upload/image", func(c *gin.Context) { api.POST("/upload/cover", handler.UploadImageHandler(cfg)) // 复用上传逻辑
handler.UploadImageHandler(c, cfg)
})
api.POST("/upload/cover", func(c *gin.Context) {
handler.UploadImageHandler(c, cfg)
})
} }
return r return r