golang 网络传输文件例子

网络传输文件是开发中常见的操作，有时我们需要将本地文件发送到远程服务器，或者接收远程服务器上的文件到本地。在Golang中，提供了一些简便的方法来实现这个功能。本文将介绍如何在Golang中进行网络传输文件。

1. 文件上传

首先，我们来看一下如何将本地文件上传到远程服务器。Golang中的net/http包中提供了一个方便的方法，即使用multipart/form-data类型的POST请求来上传文件。

首先，我们需要创建一个http包含文件上传功能的服务器端，可以使用http包中的HandleFunc函数来处理上传请求。在请求处理函数中，我们可以通过r.FormFile函数获取上传的文件对象，然后使用io.Copy函数将文件拷贝到服务器指定的目录中。

示例代码如下：

package main

import (
	"io"
	"log"
	"net/http"
	"os"
)

func uploadFile(w http.ResponseWriter, r *http.Request) {
	file, handler, err := r.FormFile("file")
	if err != nil {
		log.Fatal(err)
	}
	defer file.Close()

	f, err := os.OpenFile(handler.Filename, os.O_WRONLY|os.O_CREATE, 0666)
	if err != nil {
		log.Fatal(err)
	}
	defer f.Close()

	io.Copy(f, file)

	w.Write([]byte("File uploaded successfully!"))
}

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

2. 文件下载

接下来，我们将介绍如何实现从远程服务器上下载文件到本地。同样地，Golang中的net/http包提供了下载文件的简便方式，我们可以通过发送GET请求来获取远程服务器上的文件并保存到本地。

我们可以使用http包中的Get函数发送GET请求，并将返回的response.Body对象拷贝到本地文件中。

示例代码如下：

package main

import (
	"io"
	"log"
	"net/http"
	"os"
)

func downloadFile(url string, filepath string) error {
	resp, err := http.Get(url)
	if err != nil {
		return err
	}
	defer resp.Body.Close()

	out, err := os.Create(filepath)
	if err != nil {
		return err
	}
	defer out.Close()

	_, err = io.Copy(out, resp.Body)
	return err
}

func main() {
	err := downloadFile("http://example.com/file.txt", "file.txt")
	if err != nil {
		log.Fatal(err)
	}

	log.Println("File downloaded successfully!")
}

3. 文件传输进度

在文件上传或下载过程中，我们可能需要显示文件传输的进度以及估计剩余时间。Golang中的io包提供了一个ProgressReader类型，它封装了一个io.Reader对象，并可以通过回调函数来获取传输进度。

首先，我们可以实现一个Read、Write方法来封装一个带有进度回调的文件读写器。然后，我们可以将这个文件读写器传递给io.Copy函数来进行文件的拷贝操作，并在回调函数中更新传输进度。

示例代码如下：

package main

import (
	"fmt"
	"io"
	"log"
	"net/http"
	"os"
)

type ProgressReader struct {
	reader     io.Reader
	total      int64
	transferred int64
}

func (pr *ProgressReader) Read(p []byte) (n int, err error) {
	n, err = pr.reader.Read(p)
	pr.transferred += int64(n)
	pr.printProgress()
	return n, err
}

func (pr *ProgressReader) printProgress() {
	progress := float64(pr.transferred) / float64(pr.total) * 100
	fmt.Printf("Progress: %.2f%%\r", progress)
}

func uploadFileWithProgress(url string, filepath string) error {
	file, err := os.Open(filepath)
	if err != nil {
		return err
	}
	defer file.Close()

	fi, err := file.Stat()
	if err != nil {
		return err
	}

	pr := &ProgressReader{
		reader:     file,
		total:      fi.Size(),
		transferred: 0,
	}

	resp, err := http.Post(url, "multipart/form-data", pr)
	if err != nil {
		return err
	}
	defer resp.Body.Close()

	return nil
}

func main() {
	err := uploadFileWithProgress("http://example.com/upload", "file.txt")
	if err != nil {
		log.Fatal(err)
	}

	log.Println("File uploaded successfully!")
}

本文介绍了如何在Golang中进行文件的网络传输，包括文件上传和下载。通过使用net/http包和io包提供的函数和类型，我们可以简单地实现这些功能。此外，本文还演示了如何获取传输进度，并将其显示出来。希望本文对你在Golang中进行网络文件传输有所帮助。