修改后端结构
This commit is contained in:
@@ -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,
|
||||||
@@ -103,4 +109,5 @@ func UploadImageHandler(c *gin.Context, cfg *config.Config) {
|
|||||||
"message": "图片上传成功",
|
"message": "图片上传成功",
|
||||||
"data": gin.H{"url": imageURL},
|
"data": gin.H{"url": imageURL},
|
||||||
})
|
})
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user