From d6b1873a1d1d30569620db15812ba37710055d87 Mon Sep 17 00:00:00 2001 From: kevin Date: Thu, 10 Jul 2025 17:15:11 +0800 Subject: [PATCH] =?UTF-8?q?=E6=89=8B=E5=8A=A8=E5=88=A4=E6=96=AD=E6=96=87?= =?UTF-8?q?=E4=BB=B6=E5=90=8D=E6=98=AF=E5=90=A6=E4=B8=BA=E7=A9=BA?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- gin_ops/def_config/error_codes.json | 3 +- gin_ops/routers/routers_flie.go | 174 +++++++++++++++------------- 2 files changed, 95 insertions(+), 82 deletions(-) diff --git a/gin_ops/def_config/error_codes.json b/gin_ops/def_config/error_codes.json index 8f470d9..77b52b7 100644 --- a/gin_ops/def_config/error_codes.json +++ b/gin_ops/def_config/error_codes.json @@ -18,5 +18,6 @@ "warehouses_api_err":15, "warehouses_name_err":16, "warehouses_id_err":17, - "file_id_error":18 + "file_id_error":18, + "file_name_err":19 } \ No newline at end of file diff --git a/gin_ops/routers/routers_flie.go b/gin_ops/routers/routers_flie.go index 5d07ddb..722cbb7 100644 --- a/gin_ops/routers/routers_flie.go +++ b/gin_ops/routers/routers_flie.go @@ -23,19 +23,24 @@ func Router_file(r *gin.RouterGroup) { id := ctx.Param("id") id_int, err := strconv.ParseInt(id, 10, 0) if err == nil { - //fmt.Println(id_int) - file_info := models.File_info{ - ID: uint(id_int), - } - if models.DB.Where(file_info.ID).First(&file_info).Error == nil { - fmt.Println(file_info) + if id_int != 0 { + //fmt.Println(id_int) + file_info := models.File_info{ + ID: uint(id_int), + } + if models.DB.Where(&file_info).First(&file_info).Error == nil { + fmt.Println(file_info) + } else { + fmt.Println("not fund") + } + + Return_json(ctx, "api_ok", map[string]interface{}{ + "data": file_info, + }) } else { - fmt.Println("not fund") + Return_json(ctx, "file_id_error", nil) } - Return_json(ctx, "api_ok", map[string]interface{}{ - "data": file_info, - }) } else { Return_json(ctx, "file_id_error", nil) } @@ -66,104 +71,111 @@ func Router_file(r *gin.RouterGroup) { file, err := ctx.FormFile("file") if err == nil { - //限制文件大小 - if file.Size > 512 { - if file.Size < int64(models.Configs_file.Max_size) { - // 2. 安全获取文件名并处理路径问题 - filename := filepath.Base(file.Filename) // 防御性处理路径分隔符 - //fmt.Println(filename) - // 3. 获取标准后缀名（含点） - //extWithDot := filepath.Ext(filename) + if file.Filename != "" { - //判断文件mime是否合法 - // 打开文件流 - src_mime, _ := file.Open() - defer src_mime.Close() - // 读取前512字节用于MIME检测 - buffer := make([]byte, 512) - io.ReadFull(src_mime, buffer) - // 检测MIME类型 - mimeType := http.DetectContentType(buffer) - file_extname := models.Configs_file.Allow_image_mime[mimeType] - if file_extname != "" { + //限制文件大小 + if file.Size > 512 { + if file.Size < int64(models.Configs_file.Max_size) { + // 2. 安全获取文件名并处理路径问题 + filename := filepath.Base(file.Filename) // 防御性处理路径分隔符 + //fmt.Println(filename) + // 3. 获取标准后缀名（含点） + //extWithDot := filepath.Ext(filename) + + //判断文件mime是否合法 // 打开文件流 - src, _ := file.Open() - defer src.Close() - // 创建SHA256哈希器 - hasher := sha256.New() + src_mime, _ := file.Open() + defer src_mime.Close() + // 读取前512字节用于MIME检测 + buffer := make([]byte, 512) + io.ReadFull(src_mime, buffer) + // 检测MIME类型 + mimeType := http.DetectContentType(buffer) + file_extname := models.Configs_file.Allow_image_mime[mimeType] + if file_extname != "" { + // 打开文件流 + src, _ := file.Open() + defer src.Close() + // 创建SHA256哈希器 + hasher := sha256.New() - // 计算哈希值 - io.Copy(hasher, src) - // 获取哈希结果 - hashBytes := hasher.Sum(nil) - hashString := hex.EncodeToString(hashBytes) + // 计算哈希值 + io.Copy(hasher, src) + // 获取哈希结果 + hashBytes := hasher.Sum(nil) + hashString := hex.EncodeToString(hashBytes) - new_filename := fmt.Sprintf("%s%s", hashString, file_extname) - file.Filename = new_filename + new_filename := fmt.Sprintf("%s%s", hashString, file_extname) + file.Filename = new_filename - //fmt.Println(user_info) + //fmt.Println(user_info) - //这是上传的真实路径 - dst := path.Join(models.Configs_file.Pahts["image"], file.Filename) + //这是上传的真实路径 + dst := path.Join(models.Configs_file.Pahts["image"], file.Filename) - //判断文件是否存在避免重复保存 - if models.File_exists(dst) { - //fmt.Println("文件存在") - - } else { - //fmt.Println("文件no存在") - ferr := ctx.SaveUploadedFile(file, dst) - if ferr == nil { - //文件保存成功 + //判断文件是否存在避免重复保存 + if models.File_exists(dst) { + //fmt.Println("文件存在") } else { + //fmt.Println("文件no存在") + ferr := ctx.SaveUploadedFile(file, dst) + if ferr == nil { + //文件保存成功 - Return_json(ctx, "file_save_err", nil) - ctx.Abort() //end + } else { + + Return_json(ctx, "file_save_err", nil) + ctx.Abort() //end + } } - } - //记录到数据库 + //记录到数据库 - //先检查数据库有没有数据 - fund_file_info := models.File_info{ - Name: filename, - Sha256: hashString, - Mime: mimeType, - Type: "image", - UserID: user_info.UserID, - } - fund_file_info2 := models.File_info{} + //先检查数据库有没有数据 + fund_file_info := models.File_info{ + Name: filename, + Sha256: hashString, + Mime: mimeType, + Type: "image", + UserID: user_info.UserID, + } + fund_file_info2 := models.File_info{} - models.DB.Where(&fund_file_info).Find(&fund_file_info2) + models.DB.Where(&fund_file_info).Find(&fund_file_info2) + + if fund_file_info2.ID != 0 { + fmt.Println(fund_file_info2) + fund_file_info2.Const += 1 + models.DB.Where(&fund_file_info).Updates(&fund_file_info2) + } else { + models.DB.Create(&fund_file_info) // 传入指针 + fund_file_info2 = fund_file_info + } + + red := map[string]interface{}{ + "data": fund_file_info2, + } + + Return_json(ctx, "api_ok", red) - if fund_file_info2.ID != 0 { - fmt.Println(fund_file_info2) - fund_file_info2.Const += 1 - models.DB.Where(&fund_file_info).Updates(&fund_file_info2) } else { - models.DB.Create(&fund_file_info) // 传入指针 - fund_file_info2 = fund_file_info - } - red := map[string]interface{}{ - "data": fund_file_info2, + Return_json(ctx, "file_mime_err", nil) } - Return_json(ctx, "api_ok", red) - } else { - Return_json(ctx, "file_mime_err", nil) + Return_json(ctx, "file_size_err", nil) } - } else { Return_json(ctx, "file_size_err", nil) } - } else { - Return_json(ctx, "file_size_err", nil) + } else { + Return_json(ctx, "file_name_err", nil) } + } else { Return_json(ctx, "file_get_err", nil) }