ljydlv
/
videoplayer
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

文件上传返回值修改

This commit is contained in:
junleea 2024-12-27 20:30:05 +08:00
parent 7a650394b0
commit c9620b64dc
2 changed files with 6 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
dao/file.go
View File

@ -18,13 +18,13 @@ type File struct {
Md5 string `gorm:"column:md5;type:varchar(255);uniqueIndex:idx_file_name"`
}
func CreateFile(fileStoreName, fileName, fileType, filePath string, fileSize, authID int, NeedAuth bool) uint {
func CreateFile(fileStoreName, fileName, fileType, filePath string, fileSize, authID int, NeedAuth bool) File {
file := File{FileStoreName: fileStoreName, FileName: fileName, FileType: fileType, FilePath: filePath, FileSize: fileSize, AuthID: authID, NeedAuth: NeedAuth}
result := DB.Create(&file)
if result.Error != nil {
return 0
}
return file.ID
return file
}
func DeleteFileByID(id, user int) bool {

14
handler/tool.go
View File

@ -173,19 +173,13 @@ func UploadFile(c *gin.Context) {
} else if authType == "private" {
auth_type_ = true
}
fileID := dao.CreateFile(fileStoreName, fileName, fileType, filePath, fileSize, id1, auth_type_)
if fileID == 0 {
file_record := dao.CreateFile(fileStoreName, fileName, fileType, filePath, fileSize, id1, auth_type_)
if file_record.ID == 0 {
c.JSON(http.StatusOK, gin.H{"error": "save file info failed", "code": proto.SaveFileInfoFailed, "message": "failed"})
return
}
ret := map[string]interface{}{
"file_id": fileID,
"file_store_name": fileStoreName,
"file_name": fileName,
"file_size": fileSize,
"file_type": fileType,
}
c.JSON(http.StatusOK, gin.H{"code": proto.SuccessCode, "message": "success", "data": ret})
file_record.FilePath = ""
c.JSON(http.StatusOK, gin.H{"code": proto.SuccessCode, "message": "success", "data": file_record})
}