mirror of
https://github.com/yangjian102621/geekai.git
synced 2026-04-11 21:54:26 +08:00
79 lines
1.9 KiB
Go
79 lines
1.9 KiB
Go
package admin
|
|
|
|
// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
|
// * Copyright 2023 The Geek-AI Authors. All rights reserved.
|
|
// * Use of this source code is governed by a Apache-2.0 license
|
|
// * that can be found in the LICENSE file.
|
|
// * @Author yangjian102621@163.com
|
|
// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
|
|
|
import (
|
|
"geekai/core"
|
|
"geekai/core/middleware"
|
|
"geekai/handler"
|
|
"geekai/service/oss"
|
|
"geekai/store/model"
|
|
"geekai/utils/resp"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type UploadHandler struct {
|
|
handler.BaseHandler
|
|
uploaderManager *oss.UploaderManager
|
|
}
|
|
|
|
func NewUploadHandler(app *core.AppServer, db *gorm.DB, manager *oss.UploaderManager) *UploadHandler {
|
|
return &UploadHandler{BaseHandler: handler.BaseHandler{DB: db, App: app}, uploaderManager: manager}
|
|
}
|
|
|
|
// RegisterRoutes 注册路由
|
|
func (h *UploadHandler) RegisterRoutes() {
|
|
group := h.App.Engine.Group("/api/admin/upload")
|
|
|
|
// 需要管理员授权的接口
|
|
group.Use(middleware.AdminAuthMiddleware(h.App.Config.AdminSession.SecretKey, h.App.Redis))
|
|
{
|
|
group.POST("", h.Upload)
|
|
}
|
|
}
|
|
|
|
func (h *UploadHandler) Upload(c *gin.Context) {
|
|
// 判断文件大小
|
|
f, err := c.FormFile("file")
|
|
if err != nil {
|
|
resp.ERROR(c, err.Error())
|
|
return
|
|
}
|
|
|
|
if h.App.SysConfig.Base.MaxFileSize > 0 && f.Size > int64(h.App.SysConfig.Base.MaxFileSize)*1024*1024 {
|
|
resp.ERROR(c, "文件大小超过限制")
|
|
return
|
|
}
|
|
|
|
file, err := h.uploaderManager.GetUploadHandler().PutFile(c, "file")
|
|
if err != nil {
|
|
resp.ERROR(c, err.Error())
|
|
return
|
|
}
|
|
|
|
userId := 0
|
|
res := h.DB.Create(&model.File{
|
|
UserId: uint(userId),
|
|
Name: file.Name,
|
|
ObjKey: file.ObjKey,
|
|
URL: file.URL,
|
|
Ext: file.Ext,
|
|
Size: file.Size,
|
|
CreatedAt: time.Time{},
|
|
})
|
|
if res.Error != nil || res.RowsAffected == 0 {
|
|
resp.ERROR(c, "error with update database: "+res.Error.Error())
|
|
return
|
|
}
|
|
|
|
resp.SUCCESS(c, file)
|
|
}
|