mirror of
https://github.com/yangjian102621/geekai.git
synced 2025-09-18 01:06:39 +08:00
49 lines
1.0 KiB
Go
49 lines
1.0 KiB
Go
package admin
|
|
|
|
import (
|
|
"chatplus/core"
|
|
"chatplus/handler"
|
|
"chatplus/service/oss"
|
|
"chatplus/store/model"
|
|
"chatplus/utils/resp"
|
|
"github.com/gin-gonic/gin"
|
|
"gorm.io/gorm"
|
|
"time"
|
|
)
|
|
|
|
type UploadHandler struct {
|
|
handler.BaseHandler
|
|
db *gorm.DB
|
|
uploaderManager *oss.UploaderManager
|
|
}
|
|
|
|
func NewUploadHandler(app *core.AppServer, db *gorm.DB, manager *oss.UploaderManager) *UploadHandler {
|
|
adminHandler := &UploadHandler{db: db, uploaderManager: manager}
|
|
adminHandler.App = app
|
|
return adminHandler
|
|
}
|
|
|
|
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)
|
|
}
|