Go 语言中获取上传文件大小的两种实现方法

核心原理与目标

在处理 HTTP 上传请求时，Go 的标准库 net/http 提供了 Request.FormFile 方法来解析 multipart/form-data 表单中的数据。该方法返回一个 io.Reader 和文件名。为了获取文件大小，我们需要确保返回的对象支持 Stat() 方法（通常由 multipart.File 接口提供），或者检查是否实现了自定义的 Size() 接口。

准备条件

• Go 环境已安装（推荐版本 1.12+）。
• 具备基础的 HTTP 服务器知识。
• 理解 multipart/form-data 上传机制。

真实操作顺序

第1步：定义文件信息接口

虽然 Go 标准库中的 multipart.File 已经包含了 Stat() 方法，但在某些封装或特定场景下，显式定义接口有助于类型断言。我们可以定义两个基础接口用于后续判断：

package main
import (
	"fmt"
	"io"
	"net/http"
	"log"
	"os"
)
// 获取文件大小的接口
type Size interface {
	Size() int64
}
// 获取文件信息的接口
type Stat interface {
	Stat() (os.FileInfo, error)
}

第2步：构建 HTTP 服务端逻辑

编写处理上传的核心函数 HelloServer。该函数首先判断请求方法是否为 POST，然后调用 r.FormFile 获取上传的文件句柄。接着，通过类型断言检查文件对象是否支持 Stat 或 Size 接口，从而提取文件大小。

// hello world, the web server
func HelloServer(w http.ResponseWriter, r *http.Request) {
	if "POST" == r.Method {
		file, _, err := r.FormFile("userfile")
		if err != nil {
			http.Error(w, err.Error(), 500)
			return
		}
		// 关闭文件释放资源
		defer file.Close()

		// 方法一：通过 Stat 接口获取 FileInfo
		if statInterface, ok := file.(Stat); ok {
			fileInfo, _ := statInterface.Stat()
			fmt.Fprintf(w, "通过Stat接口获取-上传文件的大小为: %d\n", fileInfo.Size())
		}
		
		// 方法二：直接调用 Size 接口（如果存在）
		// 注意：标准的 multipart.File 主要依靠 Stat().Size() 
		// 这里演示对可能存在的 Size 接口的兼容判断
		if sizeInterface, ok := file.(Size); ok {
			fmt.Fprintf(w, "通过Size接口获取-上传文件的大小为: %d\n", sizeInterface.Size())
		}
		return
	}
	
	// 非 POST 请求时返回上传页面
	w.Header().Add("Content-Type", "text/html")
	w.WriteHeader(200)
	html := `
<form enctype="multipart/form-data" action="/hello" method="POST">
	Send this file: <input name="userfile" type="file" />
	<input type="submit" value="Send File" />
</form>
`
	io.WriteString(w, html)
}

关键点说明：
1. file.Close()：务必记得关闭文件句柄，防止资源泄漏。
2. statInterface.Stat()：这是获取文件大小最可靠的方式，因为所有实现了 io.ReadSeeker 且来自 multipart 的文件通常都具备此方法。

第3步：启动服务器

在 main 函数中注册路由并监听端口。以下代码将服务绑定到端口 :12345。

func main() {
	http.HandleFunc("/hello", HelloServer)
	err := http.ListenAndServe(":12345", nil)
	if err != nil {
		log.Fatal("ListenAndServe: ", err)
	}
}

完成判断

1. 运行程序后，访问 http://localhost:12345/hello。
2. 页面应显示一个文件选择框和一个提交按钮。
3. 选择一个文件并提交，服务器响应中将包含该文件的具体大小（字节数）。
4. 检查控制台日志，确认没有报错输出。

简要总结

通过本例可以看出，在 Go 中获取上传文件大小并不复杂。关键在于使用 r.FormFile 正确接收数据，并利用 Stat() 方法从文件信息中提取尺寸。这种方式既符合 Go 的语言习惯，又确保了资源的正确管理。以上就是 Go 语言简单获取上传文件大小的详细内容，更多关于 Go 网络编程的资料请关注本站其它相关文章！