修改后端初始化
This commit is contained in:
163
server/main.go
163
server/main.go
@@ -2,159 +2,30 @@ package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/aliyun/aliyun-oss-go-sdk/oss"
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
// 配置常量
|
||||
const (
|
||||
allowImageTypes = "image/jpeg,image/png,image/gif,image/webp" // 允许的图片类型
|
||||
maxFileSize = 5 << 20 // 最大上传文件大小（5MB）
|
||||
)
|
||||
|
||||
// 阿里云 OSS 配置（请替换为你自己的信息）
|
||||
const (
|
||||
OSS_ENDPOINT = "https://oss-cn-shenzhen.aliyuncs.com" // 你的OSS地域节点
|
||||
OSS_BUCKET_NAME = "hldr-data" // 你的Bucket名称
|
||||
OSS_ACCESS_KEY_ID = "LTAI5tMTSfbaaAtYBS2gFkf9" // 你的AccessKeyId
|
||||
OSS_ACCESS_KEY_SECRET = "dHfxH4XziDO0C8ppN94kbLUREAw1Dc" // 你的AccessKeySecret
|
||||
OSS_OBJECT_PREFIX = "article/" // OSS中的文件前缀（类似文件夹）
|
||||
"github.com/JACKYMYPERSON/hldrCenter/config" // 替换为你的项目模块名
|
||||
"github.com/JACKYMYPERSON/hldrCenter/middleware" // 替换为你的项目模块名
|
||||
"github.com/JACKYMYPERSON/hldrCenter/router" // 替换为你的项目模块名
|
||||
)
|
||||
|
||||
func main() {
|
||||
// 1. 初始化 Gin 引擎
|
||||
r := gin.Default()
|
||||
// 加载配置文件
|
||||
cfg, err := config.LoadConfig("config/config.yaml")
|
||||
if err != nil {
|
||||
fmt.Printf("加载配置失败：%v\n", err)
|
||||
return
|
||||
}
|
||||
|
||||
// 2. 配置跨域
|
||||
r.Use(corsMiddleware())
|
||||
// 设置路由
|
||||
r := router.SetupRouter(cfg)
|
||||
|
||||
// 3. 图片上传接口
|
||||
r.POST("/api/upload/image", uploadImageHandler)
|
||||
r.POST("/api/upload/cover", func(c *gin.Context) {
|
||||
// 直接复用已有的上传逻辑
|
||||
uploadImageHandler(c)
|
||||
})
|
||||
// 应用跨域中间件
|
||||
r.Use(middleware.CorsMiddleware(&cfg.Server))
|
||||
|
||||
// 4. 启动服务
|
||||
fmt.Println("后端服务启动成功，地址：http://localhost:8080")
|
||||
if err := r.Run(":8080"); err != nil {
|
||||
// 启动服务
|
||||
addr := fmt.Sprintf(":%s", cfg.Server.Port)
|
||||
fmt.Printf("后端服务启动成功，地址：http://localhost%s\n", addr)
|
||||
if err := r.Run(addr); err != nil {
|
||||
fmt.Printf("服务启动失败：%v\n", err)
|
||||
}
|
||||
}
|
||||
|
||||
// corsMiddleware 跨域中间件
|
||||
func corsMiddleware() gin.HandlerFunc {
|
||||
return func(c *gin.Context) {
|
||||
c.Writer.Header().Set("Access-Control-Allow-Origin", "http://localhost:5173")
|
||||
c.Writer.Header().Set("Access-Control-Allow-Headers", "Origin, Content-Type, Accept")
|
||||
c.Writer.Header().Set("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, OPTIONS")
|
||||
c.Writer.Header().Set("Access-Control-Allow-Credentials", "true")
|
||||
|
||||
// 处理 OPTIONS 预检请求
|
||||
if c.Request.Method == "OPTIONS" {
|
||||
c.AbortWithStatus(http.StatusOK)
|
||||
return
|
||||
}
|
||||
|
||||
c.Next()
|
||||
}
|
||||
}
|
||||
|
||||
// uploadImageHandler 图片上传到阿里云OSS处理函数
|
||||
func uploadImageHandler(c *gin.Context) {
|
||||
// 获取上传的图片文件
|
||||
fileHeader, err := c.FormFile("image")
|
||||
if err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{
|
||||
"code": 400,
|
||||
"message": "获取图片失败，请重新上传",
|
||||
"error": err.Error(),
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
// 打开文件
|
||||
file, err := fileHeader.Open()
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{
|
||||
"code": 500,
|
||||
"message": "打开图片文件失败",
|
||||
"error": err.Error(),
|
||||
})
|
||||
return
|
||||
}
|
||||
defer file.Close()
|
||||
|
||||
// 检查文件大小
|
||||
if fileHeader.Size > maxFileSize {
|
||||
c.JSON(http.StatusBadRequest, gin.H{
|
||||
"code": 400,
|
||||
"message": fmt.Sprintf("图片大小不能超过 %dMB", maxFileSize>>20),
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
// 检查文件类型
|
||||
fileType := fileHeader.Header.Get("Content-Type")
|
||||
if !strings.Contains(allowImageTypes, fileType) {
|
||||
c.JSON(http.StatusBadRequest, gin.H{
|
||||
"code": 400,
|
||||
"message": fmt.Sprintf("不支持的图片类型，仅允许：%s", allowImageTypes),
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
// 初始化OSS客户端
|
||||
client, err := oss.New(OSS_ENDPOINT, OSS_ACCESS_KEY_ID, OSS_ACCESS_KEY_SECRET)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{
|
||||
"code": 500,
|
||||
"message": "初始化OSS客户端失败",
|
||||
"error": err.Error(),
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
// 获取Bucket
|
||||
bucket, err := client.Bucket(OSS_BUCKET_NAME)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{
|
||||
"code": 500,
|
||||
"message": "获取Bucket失败",
|
||||
"error": err.Error(),
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
// 生成唯一文件名
|
||||
timestamp := time.Now().Format("20060102150405")
|
||||
filename := fmt.Sprintf("%s_%s", timestamp, fileHeader.Filename)
|
||||
objectKey := OSS_OBJECT_PREFIX + filename // OSS中的完整对象键
|
||||
|
||||
// 上传文件到OSS
|
||||
err = bucket.PutObject(objectKey, file)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{
|
||||
"code": 500,
|
||||
"message": "上传图片到OSS失败",
|
||||
"error": err.Error(),
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
// 生成图片访问URL
|
||||
// 注意：需要确保Bucket设置了公共读权限，或者生成带签名的URL
|
||||
host := strings.TrimPrefix(OSS_ENDPOINT, "https://")
|
||||
imageURL := fmt.Sprintf("https://%s.%s/%s", OSS_BUCKET_NAME, host, objectKey)
|
||||
|
||||
// 返回成功响应
|
||||
c.JSON(http.StatusOK, gin.H{
|
||||
"code": 200,
|
||||
"message": "图片上传成功",
|
||||
"data": gin.H{"url": imageURL},
|
||||
})
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user