Go语言原生HTTP文件上传实现

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>上传文件</title>
</head>
<body>
    <form action="/file/upload" method="post" enctype="multipart/form-data">
        <p><input type="file" name="file" value=""></p>
        <p><input type="submit" value="submit"></p>
    </form>
</body>
</html>

该页面保存至 static/view/index.html 路径下，当用户访问该页面时，可通过浏览器发起文件选择与提交操作。

第2步：编写文件上传Handler

Handler是处理HTTP请求的核心逻辑单元。我们需要一个函数同时处理GET请求（渲染页面）和POST请求（执行保存），并利用 r.Method 进行区分。

具体动作

1. 初始化依赖： 引入 net/http 用于处理请求响应，io 用于高效复制数据流，os 用于创建本地文件。
2. 判断请求方式： 如果是 GET 请求，读取静态HTML文件内容并通过 http.ResponseWriter 返回，触发表单显示。
3. 提取文件数据： 如果是 POST 请求，调用 r.FormFile("file") 获取文件句柄及元数据。file 需对应前端表单中 type="file" 元素的 name 属性。
4. 创建目标文件： 使用 os.Create() 在服务器指定目录（如 /tmp/）创建新文件，文件名取自原始文件的 head.Filename。
5. 流式拷贝： 使用 io.Copy(newFile, file) 将上传的文件流直接拷贝到新建的本地文件中，这种方式内存占用低且效率高。
6. 资源清理： 务必使用 defer 语句关闭上传文件的句柄以及新创建的本地文件，防止文件描述符泄露。

package handler

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

// 注意：Go中首字母大写的函数名才可被其他包引用
func UploadHandler(w http.ResponseWriter, r *http.Request) {
	fmt.Printf("Handler entry\n")
	
	if r.Method == "GET" {
		data, err := ioutil.ReadFile("static/view/index.html")
		if err != nil {
			_, _ = io.WriteString(w, "something wrong!")
			return
		}
		_, _ = io.WriteString(w, string(data))
	} else if r.Method == "POST" {
		// 获取上传的文件
		file, head, err := r.FormFile("file")
		if err != nil {
			fmt.Printf("Failed to get file data: %s\n", err.Error())
			return
		}
		defer file.Close()

		// 在本地创建新文件
		newFile, err := os.Create("/tmp/" + head.Filename)
		if err != nil {
			fmt.Printf("Failed to create new file: %s\n", err.Error())
			return
		}
		defer newFile.Close()

		// 将上传的数据拷贝到本地文件
		_, err = io.Copy(newFile, file)
		if err != nil {
			fmt.Printf("Failed to save file: %s\n", err.Error())
			return
		}

		// 上传成功后重定向
		http.Redirect(w, r, "/file/upload/suc", http.StatusFound)
	}
}

// 上传成功后的简单响应处理
func UploadSucHandler(w http.ResponseWriter, r *http.Request) {
	_, _ = io.WriteString(w, "Upload Succeed!")
}

第3步：配置路由并启动服务

在Go语言的 main 包中，通过 http.HandleFunc 将URL路径映射到对应的处理函数。最后启动监听器即可。

操作说明

• 入口： 程序主函数。
• 参数映射：
• /file/upload -> handler.UploadHandler (处理表单展示与文件接收)
• /file/upload/suc -> handler.UploadSucHandler (处理成功反馈)
• 端口配置： 示例中使用端口 :8081，可根据实际需求修改。

package main

import (
	"log"
	"net/http"
	"zone/src/handler" // 假设handler包在此路径下
)

func main() {
	// 注册路由规则
	http.HandleFunc("/file/upload", handler.UploadHandler)
	http.HandleFunc("/file/upload/suc", handler.UploadSucHandler)
	
	// 开启监听，日志模式包裹以便捕获错误
	log.Fatal(http.ListenAndServe(":8081", nil))
}

运行测试与验证

完成上述代码后，确保项目结构正确（即HTML文件存在于 static/view/ 目录下）。运行 go run main.go 启动服务，然后在浏览器访问 http://localhost:8081/file/upload。

此时应看到上传表单页面，选择文件并提交后，页面将跳转至成功提示页，同时检查服务器端的 /tmp/ 目录，可发现已存在刚上传的同名文件。

控制台输出也将按顺序反映请求进入、表单渲染及文件保存的过程。

以上就是Go语言原生实现HTTP文件上传的详细内容。掌握了标准库中的 FormFile 与 io.Copy 配合用法，即可在此基础上扩展出支持文件大小限制、类型校验及安全存储的高级文件处理模块，更多关于Go网络编程的资料请关注本站其它相关文章！