mirror of
https://gitee.com/blackfox/geekai.git
synced 2024-12-04 21:27:39 +08:00
53 lines
1.3 KiB
Go
53 lines
1.3 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/handler"
|
|
"geekai/service/oss"
|
|
"geekai/store/model"
|
|
"geekai/utils/resp"
|
|
"github.com/gin-gonic/gin"
|
|
"gorm.io/gorm"
|
|
"time"
|
|
)
|
|
|
|
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}
|
|
}
|
|
|
|
func (h *UploadHandler) Upload(c *gin.Context) {
|
|
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: 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)
|
|
}
|