golang 网络传输文件例子

admin 2025-03-18 01:22:06 编程 来源:ZONE.CI 全球网 0 阅读模式

网络传输文件是开发中常见的操作,有时我们需要将本地文件发送到远程服务器,或者接收远程服务器上的文件到本地。在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中进行网络文件传输有所帮助。

以太坊cppgolang区别 编程

以太坊cppgolang区别

以太坊是一种去中心化的开源平台,它采用智能合约技术,旨在构建和运行不受干扰的分布式应用程序。作为目前最受欢迎的区块链平台之一,以太坊提供了多种编程语言的支持,其
progolang 编程

progolang

Go语言(Golang)是由Google开发的一门静态类型编程语言。作为一名专业的Golang开发者,我深知这门语言的优势和特点。在本文中,我将介绍Golang
golangn个发送者 编程

golangn个发送者

Golang是一种开源的编程语言,由Google团队开发,旨在提高程序的并发性和简化软件开发过程。在Go语言中,有时需要向多个接收者发送信息。本文将介绍如何在G
golang技能图谱 编程

golang技能图谱

从互联网行业的快速发展到人工智能技术的日益成熟,各种编程语言也应运而生。而在这众多的编程语言中,Golang(即Go)作为一门强大且高效的开发语言备受关注。Go
评论:0   参与:  8