如何在Go语言中实现文件接收

在Go语言开发中，我们常常需要处理文件上传和下载的需求。本文将介绍如何在Go语言中实现文件接收功能。

文件上传

文件上传是指将本地文件上传到服务器端的操作。在Go语言中，可以使用net/http包来处理文件上传。

首先，我们需要创建一个HTTP请求处理函数：

func uploadFile(w http.ResponseWriter, r *http.Request) {
    file, handler, err := r.FormFile("file")
    defer file.Close()
    
    if err != nil {
        fmt.Println(err)
        return
    }
    // 处理文件逻辑
}

上述代码中，r.FormFile("file")用于获取上传的文件，通过r.FormFile可以获取到文件对象、文件名以及可能的错误。

接下来，我们可以在处理函数中处理文件逻辑。比如将文件保存到服务器的指定路径：

func uploadFile(w http.ResponseWriter, r *http.Request) {
    file, handler, err := r.FormFile("file")
    defer file.Close()
    
    if err != nil {
        fmt.Println(err)
        return
    }
    
    // 创建目标文件
    dst, err := os.Create("./uploads/" + handler.Filename)
    defer dst.Close()
    
    if err != nil {
        fmt.Println(err)
        return
    }
    
    // 将上传的文件内容拷贝到目标文件
    _, err = io.Copy(dst, file)
    
    if err != nil {
        fmt.Println(err)
        return
    }
    
    fmt.Fprintf(w, "File uploaded successfully!")
}

在上述代码中，我们使用os.Create函数创建目标文件，并使用io.Copy函数将上传的文件内容拷贝到目标文件中。

接下来，我们需要实现文件上传的路由：

func main() {
    http.HandleFunc("/upload", uploadFile)
    http.ListenAndServe(":8080", nil)
}

在上述代码中，我们使用http.HandleFunc函数将上传路径和处理函数进行绑定，然后调用http.ListenAndServe函数监听指定地址和端口。

文件下载

文件下载是指将服务器端的文件发送给客户端的操作。在Go语言中，可以使用net/http包来处理文件下载。

我们需要创建一个HTTP请求处理函数：

func downloadFile(w http.ResponseWriter, r *http.Request) {
    file, err := os.Open("./uploads/file.txt")
    defer file.Close()
    
    if err != nil {
        fmt.Println(err)
        return
    }
    
    // 设置响应头
    w.Header().Add("Content-Disposition", "attachment; filename=file.txt")
    w.Header().Add("Content-Type", "application/octet-stream")
    
    // 将文件内容发送给客户端
    _, err = io.Copy(w, file)
    
    if err != nil {
        fmt.Println(err)
        return
    }
}

在上述代码中，我们使用os.Open函数打开文件，并设置响应头的Content-Disposition和Content-Type字段。然后使用io.Copy函数将文件内容发送给客户端。

接下来，我们需要实现文件下载的路由：

func main() {
    http.HandleFunc("/download", downloadFile)
    http.ListenAndServe(":8080", nil)
}

在上述代码中，我们使用http.HandleFunc函数将下载路径和处理函数进行绑定，然后调用http.ListenAndServe函数监听指定地址和端口。

总结

通过使用Go语言提供的net/http包，我们可以方便地实现文件上传和下载功能。文件上传时，我们通过r.FormFile函数获取上传的文件，然后可以对文件进行处理；文件下载时，我们通过os.Open函数打开文件，并将文件内容发送给客户端。希望本文对你理解Go语言中的文件接收有所帮助！