脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → Golang →  Go  HTTP 文件上传

 Go 语言实现 HTTP 文件上传和下载

 更新时间：2022年09月06日 10:16:43   作者：小二上酒8  
这篇文章主要介绍了Go语言实现HTTP文件上传和下载，文章围绕主题展开详细的内容戒杀，具有一定的参考价值，需要的小伙伴可以参考一下

前言：

近我使用 Go 语言完成了一个正式的 Web 应用，有一些方面的问题在使用 Go 开发 Web 应用过程中比较重要。过去，我将 Web 开发作为一项职业并且把使用不同的语言和范式开发 Web 应用作为一项爱好，因此对于 Web 开发领域有一些心得体会。

总的来说，我喜欢使用 Go 语言进行 Web 开发，尽管开始一段时间需要去适应它。Go 语言有一些坑，但是正如本篇文章中所要讨论的文件上传与下载，Go 语言的标准库与内置函数，使得开发是种愉快的体验。

在接下来的几篇文章中，我将重点讨论我在 Go 中编写生产级 Web 应用程序时遇到的一些问题，特别是关于身份验证/授权的问题。

这篇文章将展示 HTTP 文件上传和下载的基本示例。我们将一个有 type 文本框和一个 uploadFile 上传框的 HTML 表单作为客户端。

让我们来看下 Go 语言中是如何解决这种在 Web 开发中随处可见的问题的。

代码示例 首先，我们在服务器端设定两个路由， /upload 用于文件上传， /files/* 用于文件下载。

const maxUploadSize = 2 * 1024 * 2014 // 2 MB
const uploadPath = "./tmp"

func main() {
http.HandleFunc("/upload", uploadFileHandler())

fs := http.FileServer(http.Dir(uploadPath))
http.Handle("/files/", http.StripPrefix("/files", fs))

log.Print("Server started on localhost:8080, use /upload for uploading files and /files/{fileName} for downloading files.")
log.Fatal(http.ListenAndServe(":8080", nil))
}

我们还将要上传的目标目录，以及我们接受的最大文件大小定义为常量。注意这里，整个文件服务的概念是如此的简单 —— 我们仅使用标准库中的工具，使用 http.FileServe 创建一个 HTTP 处理程序，它将使用 http.Dir(uploadPath) 提供的目录来上传文件。

现在我们只需要实现 uploadFileHandler 。

这个处理程序将包含以下功能：

  • 验证文件最大值
  • 从请求验证文件和 POST 参数
  • 检查所提供的文件类型（我们只接受图像和 PDF）
  • 创建一个随机文件名
  • 将文件写入硬盘
  • 处理所有错误，如果一切顺利返回成功消息

第一步，我们定义处理程序:

func uploadFileHandler() http.HandlerFunc {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {

然后，我们使用 http.MaxBytesReader 验证文件大小，当文件大小大于设定值时它将返回一个错误。错误将被一个助手程序 renderError 进行处理，它返回错误信息及对应的 HTTP 状态码。

r.Body = http.MaxBytesReader(w, r.Body, maxUploadSize)
if err := r.ParseMultipartForm(maxUploadSize); err != nil {
renderError(w, "FILE_TOO_BIG", http.StatusBadRequest)
return
}

如果文件大小验证通过，我们将检查并解析表单参数类型和上传的文件，并读取文件。在本例中，为了清晰起见，我们不使用花哨的 io.Reader 和 io.Writer 接口，我们只是简单的将文件读取到一个字节数组中，这点我们后面会写到。

fileType := r.PostFormValue("type")
file, _, err := r.FormFile("uploadFile")
if err != nil {
renderError(w, "INVALID_FILE", http.StatusBadRequest)
return
}
defer file.Close()
fileBytes, err := ioutil.ReadAll(file)
if err != nil {
renderError(w, "INVALID_FILE", http.StatusBadRequest)
return
}

现在我们成功的验证了文件的大小，并且读取了文件，接下来我们该检验文件的类型了。一种廉价但是并不安全的方式，只检查文件扩展名，并相信用户没有改变它，但是对于一个正式的项目来讲不应该这么做。

幸运的是，Go 标准库提供给我们一个 http.DetectContentType 函数，这个函数基于 mimesniff 算法，只需要读取文件的前 512 个字节就能够判定文件类型。

filetype := http.DetectContentType(fileBytes)
if filetype != "image/jpeg" && filetype != "image/jpg" &&
filetype != "image/gif" && filetype != "image/png" &&
filetype != "application/pdf" {
renderError(w, "INVALID_FILE_TYPE", http.StatusBadRequest)
return
}

在实际应用程序中，我们可能会使用文件元数据做一些事情，例如将其保存到数据库或将其推送到外部服务——以任何方式，我们将解析和操作元数据。这里我们创建一个随机的新名字（这在实践中可能是一个 UUID）并将新文件名记录下来。

fileName := randToken(12)
fileEndings, err := mime.ExtensionsByType(fileType)
if err != nil {
renderError(w, "CANT_READ_FILE_TYPE", http.StatusInternalServerError)
return
}
newPath := filepath.Join(uploadPath, fileName+fileEndings[0])
fmt.Printf("FileType: %s, File: %s\n", fileType, newPath)

马上就大功告成了，只剩下一个关键步骤-写文件。如上文所提到的，我们只需要复制读取的二进制文件到一个新创建的名为 newFile 的文件处理程序里。

如果所有部分都没问题，我们给用户返回一个 SUCCESS 信息。

newFile, err := os.Create(newPath)
if err != nil {
renderError(w, "CANT_WRITE_FILE", http.StatusInternalServerError)
return
}
defer newFile.Close()
if _, err := newFile.Write(fileBytes); err != nil {
renderError(w, "CANT_WRITE_FILE", http.StatusInternalServerError)
return
}
w.Write([]byte("SUCCESS"))

这样可以了. 你可以对这个简单的例子进行测试，使用虚拟的文件上传 HTML 页面，cURL 或者工具例如 postman [1] 。

这里是完整的代码示例 这里 [2]

结论 这是又一个证明了 Go 如何允许用户为 Web 编写简单而强大的软件，而不必像处理其他语言和生态系统中固有的无数抽象层。

在接下来的篇幅中，我将展示一些在我第一次使用 Go 语言编写正式的 Web 应用中其他细节，敬请期待。；）

// 根据 reddit 用户 lstokeworth 的反馈对部分代码进行了修改。谢谢:)

到此这篇关于 Go 语言完成 HTTP 文件上传和下载的文章就介绍到这了,更多相关 Go HTTP 文件上传内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

相关文章

  • Go语言计算指定年月天数的方法

    Go语言计算指定年月天数的方法

    这篇文章主要介绍了Go语言计算指定年月天数的方法,实例分析了Go语言操作时间的技巧,具有一定参考借鉴价值,需要的朋友可以参考下
    2015-02-02
  • go实现base64编码的四种方式

    go实现base64编码的四种方式

    本文主要介绍了go实现base64编码的四种方式，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2023-03-03
  • golang中Context的使用场景

    golang中Context的使用场景

    context包提供了一种在goroutine之间传递信号的方法,用于管理请求的生命周期和控制并发操作,本文主要介绍了golang中Context的使用场景,下面就来介绍一下,感兴趣的可以了解一下
    2025-05-05
  • Gin与Mysql实现简单Restful风格API实战示例详解

    Gin与Mysql实现简单Restful风格API实战示例详解

    这篇文章主要为大家介绍了Gin与Mysql实现简单Restful风格API示例详解，有需要的朋友可以借鉴参考下希望能够有所帮助，祝大家多多进步
    2021-11-11
  • Golang语言实现gRPC的具体使用

    Golang语言实现gRPC的具体使用

    本文主要介绍了Golang语言实现gRPC的具体使用，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2022-08-08
  • Go语言使用Gob传输数据

    Go语言使用Gob传输数据

    本文主要介绍了Go语言使用Gob传输数据，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2023-04-04
  • Go异步任务解决方案之Asynq库详解

    Go异步任务解决方案之Asynq库详解

    需要在Go应用程序中异步处理任务? Asynq,简单高效的任务队列实现,下面这篇文章主要给大家介绍了关于Go异步任务解决方案之Asynq库的相关资料,需要的朋友可以参考下
    2023-02-02
  • 详解在Go语言单元测试中如何解决文件依赖问题

    详解在Go语言单元测试中如何解决文件依赖问题

    现如今的 Web 应用程序往往采用 RESTful API 接口形式对外提供服务,后端接口直接向前端返回 HTML 文件的情况越来越少,所以在程序中操作文件的场景也变少了,在编写单元测试时,文件就成了被测试代码的外部依赖,本文就来讲解下测试过程中如何解决文件外部依赖问题
    2023-08-08
  • k8s容器互联flannel vxlan通信原理

    k8s容器互联flannel vxlan通信原理

    这篇文章主要为大家介绍了k8s容器互联flannel vxlan通信原理详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
    2023-04-04
  • Goland 2020或2019软件版本去掉a...或fmt...提示的方法

    Goland 2020或2019软件版本去掉a...或fmt...提示的方法

    这篇文章主要介绍了Goland 2020或2019软件版本去掉a...或fmt...提示的方法,本文通过图文并茂的形式给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
    2020-10-10

最新评论