当前位置：首页 > 文章列表 > Golang > Go教程 > 教程实战：使用Golang进行HTTP文件上传下载

教程实战：使用Golang进行HTTP文件上传下载

2023-06-21 10:29:23 0浏览收藏

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

随着互联网技术的不断发展，文件上传和下载已成为现代应用程序设计不可或缺的一部分。本文将介绍如何使用Golang编写一个简单的HTTP文件上传和下载服务器。

在开始之前，你需要安装一个最新版本的Golang。安装方法请参考官方文档。

第一步：创建服务器

我们从创建一个简单的HTTP服务器开始。在你喜欢的编辑器中打开一个新文件，在文件头部添加以下代码：

package main

import (
      "fmt"
      "net/http"
)

func main() {
      http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
            fmt.Fprintf(w, "Welcome to the HTTP File Upload/Download server.")
      })

      err := http.ListenAndServe(":8080", nil)
    
    if err != nil {
          fmt.Println(err)
      }
}

代码中的http.HandleFunc()函数为HTTP请求提供路由规则。如果HTTP请求的路径和路由规则匹配，对应的函数会被执行。在本例中，我们为根路径"/"提供一个欢迎信息。然后，我们使用http.ListenAndServe()函数为服务器设置监听端口，应当与你的应用程序通信的端口。

将代码保存为main.go文件。执行以下命令运行HTTP服务器：

go run main.go

运行完该命令后，你的HTTP服务器就可以响应基本的HTTP请求。

第二步：添加文件上传功能

接下来，让我们为HTTP服务器添加文件上传功能。在http.HandleFunc()中，我们可以为HTTP POST请求路径"/upload"添加一个路由规则：

http.HandleFunc("/upload", func(w http.ResponseWriter, r *http.Request) {
      if r.Method == "POST" {
            file, handler, err := r.FormFile("file")
          if err != nil {
                 fmt.Println("Error Retrieving the File")
                 fmt.Println(err)
                 return
          }
      defer file.Close()
      fmt.Printf("Uploaded File: %+v
", handler.Filename)
      fmt.Printf("File Size: %+v
", handler.Size)
      fmt.Printf("MIME Type: %+v
", handler.Header.Get("Content-Type"))
      // 处理上传文件
      }
})

这段代码的功能是解析HTTP POST请求中的文件上传，先判断请求类型是否为POST，不是的话则不处理。然后，通过读取r.FormFile()函数中的file参数访问文件上传内容。如果出现错误，则会返回错误消息。如果没有错误，文件将关闭。接下来，我们打印文件名称，大小和MIME类型。

现在，我们需要将上传的文件保存在服务器上。在上传文件的部分，我们将使用Go的os包和io包。我们将读取文件，并使用os.Create()函数创建新文件。将内容写入文件后，文件将关闭。

// ...

file, handler, err := r.FormFile("file")
if err != nil {
   fmt.Println("Error Retrieving the File")
   fmt.Println(err)
   return
}
defer file.Close()
fmt.Printf("Uploaded File: %+v
", handler.Filename)
fmt.Printf("File Size: %+v
", handler.Size)
fmt.Printf("MIME Type: %+v
", handler.Header.Get("Content-Type"))

// 创建一个新文件
f, err := os.OpenFile("./uploads/"+handler.Filename, os.O_WRONLY|os.O_CREATE, 0666)
if err != nil {
   fmt.Println(err)
   return
}
defer f.Close()

// 将内容写入文件
_, err = io.Copy(f, file)
if err != nil {
   fmt.Println(err)
   return
}

// ...

为了将文件保存到服务器上，我们使用os.OpenFile()函数创建一个新文件，然后将其保存在指定的目录中。使用io.Copy()函数将上传文件的内容复制到新文件中。我们将使用handler.Filename变量来命名新文件。你可以自定义命名方式。

第三步：添加文件下载功能

现在，我们需要为HTTP服务器添加文件下载功能。这意味着我们需要创建另一个HTTP路由来处理文件下载。我们将HTTP GET请求路径"/download"分配给下载文件的路由：

http.HandleFunc("/download", func(w http.ResponseWriter, r *http.Request) {
      fileName := "./uploads/sample.txt"
      w.Header().Set("Content-Disposition", "attachment; filename="+fileName)
      w.Header().Set("Content-Type", "application/octet-stream")
      http.ServeFile(w, r, fileName)
})

上面代码中的http.ServeFile()函数负责文件下载的功能。该函数将服务器上的文件发送到客户端，从而实现文件的下载功能。

第四步：测试HTTP服务器

现在，我们已经完成了文件上传和下载的HTTP服务器。输入命令运行HTTP服务器：