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

Merge branch 'refs/heads/feat-file'

This commit is contained in:
junleea 2024-12-27 20:30:08 +08:00
parent 50fef88276 c9620b64dc
commit 87ae7db41c
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"` 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} file := File{FileStoreName: fileStoreName, FileName: fileName, FileType: fileType, FilePath: filePath, FileSize: fileSize, AuthID: authID, NeedAuth: NeedAuth}
result := DB.Create(&file) result := DB.Create(&file)
if result.Error != nil { if result.Error != nil {
return 0 return 0
} }
return file.ID return file
} }
func DeleteFileByID(id, user int) bool { 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" { } else if authType == "private" {
auth_type_ = true auth_type_ = true
} }
fileID := dao.CreateFile(fileStoreName, fileName, fileType, filePath, fileSize, id1, auth_type_) file_record := dao.CreateFile(fileStoreName, fileName, fileType, filePath, fileSize, id1, auth_type_)
if fileID == 0 { if file_record.ID == 0 {
c.JSON(http.StatusOK, gin.H{"error": "save file info failed", "code": proto.SaveFileInfoFailed, "message": "failed"}) c.JSON(http.StatusOK, gin.H{"error": "save file info failed", "code": proto.SaveFileInfoFailed, "message": "failed"})
return return
} }
ret := map[string]interface{}{ file_record.FilePath = ""
"file_id": fileID, c.JSON(http.StatusOK, gin.H{"code": proto.SuccessCode, "message": "success", "data": file_record})
"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})
} }