38 lines
988 B
Go
38 lines
988 B
Go
package dao
|
||
|
||
import (
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
type File struct {
|
||
BaseModel
|
||
UUID string `gorm:"unique;type varchar(32);comment:'文件UUID'"`
|
||
UserId int32 `gorm:"comment:'上传者ID'"`
|
||
FileName string `gorm:"type:varchar(255);comment:'文件名称'"`
|
||
FileSize uint32 `gorm:"comment:'文件大小'"`
|
||
FileType string `gorm:"comment:'文件类型，对应html中支持的Content-Type类型'"`
|
||
FileContent []byte `gorm:"type:mediumblob;comment:'文件内容，目前最大为16M'"`
|
||
}
|
||
|
||
func File_Insert(data *File) (uint64, error) {
|
||
if err := s.db.Create(data).Error; err != nil {
|
||
return 0, err
|
||
}
|
||
return uint64(data.ID), nil
|
||
}
|
||
|
||
func File_Query(uuid string) (*File, error) {
|
||
var data File
|
||
if err := s.db.Where("uuid = ?", uuid).First(&data).Error; err != nil {
|
||
if err == gorm.ErrRecordNotFound {
|
||
return nil, nil
|
||
}
|
||
return nil, err
|
||
}
|
||
return &data, nil
|
||
}
|
||
|
||
func File_Delete(uuid string) error {
|
||
return s.db.Where("uuid = ?", uuid).Delete(&File{}).Error
|
||
}
|