diff --git a/dao/file.go b/dao/file.go index 2fad62a..06ffc79 100644 --- a/dao/file.go +++ b/dao/file.go @@ -45,7 +45,7 @@ func FindFileByNames(fileName string, auth_id int) File { func FindFileByName(fileName string) File { var file File - DB.Where("file_name = ?", fileName).First(&file) + DB.Where("file_store_name = ?", fileName).First(&file) return file } diff --git a/handler/tool.go b/handler/tool.go index 39d512e..c6f85d6 100644 --- a/handler/tool.go +++ b/handler/tool.go @@ -164,7 +164,14 @@ func UploadFile(c *gin.Context) { c.JSON(http.StatusOK, gin.H{"error": "save file info failed", "code": proto.SaveFileInfoFailed, "message": "failed"}) return } - c.JSON(http.StatusOK, gin.H{"code": proto.SuccessCode, "message": "success", "data": fileID}) + 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}) }