下载 golang api 中损坏的图像文件

今天golang学习网给大家带来了《下载 golang api 中损坏的图像文件》，其中涉及到的知识点包括等等，无论你是小白还是老手，都适合看一看哦~有好的建议也欢迎大家在评论留言，若是看完有所收获，也希望大家能多多点赞支持呀！一起加油学习~

我正在 golang 中构建一个简单的测试 api，用于上传和下载图像文件（png、jpeg、jpg）：

/pic [post] 用于上传图片并保存到文件夹； /pic [get] 用于将图像下载到客户端。

我已成功构建 /pic [post] 并且图像已成功上传到服务器的文件。我可以打开存储文件夹中的文件。 （在windows localhost服务器和ubuntu服务器中）

但是，当我构建 /pic [get] 用于下载图片时，我可以将文件下载到客户端（我的计算机），但下载的文件在某种程度上已损坏，因为当我尝试使用不同的文件打开它时图像查看器，例如画廊或 photoshop，它显示“看来我们不支持此文件格式”。所以看起来下载没有成功。

邮递员结果：

在图库中打开文件： 关于为什么会发生这种情况以及我应该如何解决它有什么想法吗？

下载图片的golang代码如下（省略错误处理）：

func PicDownload(w http.ResponseWriter, r *http.Request){

   request := make(map[string]string)
   reqBody, _ := ioutil.ReadAll(r.Body)
   err = json.Unmarshal(reqBody, &request)
   // Error handling

   file, err := os.OpenFile("./resources/pic/" + request["filename"], os.O_RDONLY, 0666)
   // Error handling

   buffer := make([]byte, 512)
   _, err = file.Read(buffer)
   // Error handling

   contentType := http.DetectContentType(buffer)

   fileStat, _ := file.Stat()

   // Set header
   w.Header().Set("Content-Disposition", "attachment; filename=" + request["filename"])
   w.Header().Set("Content-Type", contentType)
   w.Header().Set("Content-Length", strconv.FormatInt(fileStat.Size(), 10))

   // Copying the file content to response body
   io.Copy(w, file)

   return
}

解决方案

当您从文件中读取前 512 个字节以确定内容类型时，基础文件流指针向前移动 512 个字节。当您稍后调用 io.Copy 时，将从该位置继续读取。

有两种方法可以纠正此问题。

第一个是在调用 io.Copy() 之前调用 file.Seek(0, io.SeekStart)。这会将指针放回到文件的开头。此解决方案需要的代码量最少，但意味着从文件中读取相同的 512 字节两次，这会导致一些开销。

第二个解决方案是使用 buffer := make([]byte, fileStat.Size() 创建一个包含整个文件的缓冲区，并将该缓冲区用于 http.DetectContentType() 调用并写入输出（使用 w.Write(buffer) 写入，而不是使用 io.Copy()。这种方法可能存在将整个文件一次性加载到内存中的缺点，这对于非常大的文件（io.txt）来说并不理想。 Copy 使用 32KB 块而不是加载整个文件）。

注意：正如 Peter 在评论中提到的，您必须确保用户无法通过发布 ../../ 或其他文件名来遍历您的文件系统。

终于介绍完啦！小伙伴们，这篇关于《下载 golang api 中损坏的图像文件》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！