使用goroutine和channel实现Go语言多文件并发上传,通过限制并发数、设置超时与重试机制提升稳定性。

Go语言实现多文件并发上传,核心是结合HTTP客户端与并发控制机制。直接使用 goroutine 发起多个上传请求能提升效率,但需注意资源占用和连接管理。下面分步骤说明如何安全高效地实现。
1. 基础上传逻辑封装
每个文件上传可封装成独立函数,接收文件路径和目标URL。使用 multipart/form-data 格式编码数据:
func uploadFile(filePath, url, fieldName string) error {
file, err := os.Open(filePath)
if err != nil {
return err
}
defer file.Close()
body := &bytes.Buffer{}
writer := multipart.NewWriter(body)
part, err := writer.CreateFormFile(fieldName, filepath.Base(filePath))
if err != nil {
return err
}
_, err = io.Copy(part, file)
if err != nil {
return err
}
err = writer.Close()
if err != nil {
return err
}
req, err := http.NewRequest("POST", url, body)
if err != nil {
return err
}
req.Header.Set("Content-Type", writer.FormDataContentType())
client := &http.Client{}
resp, err := client.Do(req)
if err != nil {
return err
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
return fmt.Errorf("upload failed: %s", resp.Status)
}
return nil}
2. 使用 Goroutine 并发执行
将多个文件的上传任务用 goroutine 启动,通过 channel 控制完成状态或错误反馈:
立即学习“go语言免费学习笔记(深入)”;
files := []string{"file1.jpg", "file2.pdf", "file3.txt"}
url := "https://example.com/upload"
sem := make(chan struct{}, 5) // 控制最大并发数为5
errCh := make(chan error, len(files))
for _, f := range files {
go func(filePath string) {
sem <- struct{}{} // 获取信号量
defer func() { <-sem }()
err := uploadFile(filePath, url, "file")
if err != nil {
errCh <- fmt.Errorf("%s: %w", filePath, err)
return
}
errCh <- nil
}(f)}
// 等待所有任务完成
for i := 0; i
3. 优化建议
-
限制并发数量:避免创建过多 goroutine 导致系统负载过高,使用带缓冲的 channel 或第三方库如
golang.org/x/sync/semaphore控制并发度。 - 设置超时:给 HTTP 客户端设置合理的超时时间,防止卡住:
client := &http.Client{
Timeout: 30 * time.Second,
}-
进度反馈:若需显示上传进度,可在
io.Copy时包装io.Reader,定期更新已传输字节数。 - 重试机制:对网络失败的请求添加有限重试,提升稳定性。
基本上就这些。Go 的轻量级协程加上良好的控制手段,能让多文件上传既快又稳。










