概述

在文件上传过程中，实时监控上传进度对于用户体验和系统性能优化至关重要。Golang（Go）作为一种高效的编程语言，提供了丰富的库来处理网络和并发任务。本文将介绍如何使用Golang轻松实现文件上传进度的实时监控。

准备工作

在开始之前，请确保您已安装以下软件和库：

  • Go语言环境
  • net/http 标准库
  • github.com/gin-gonic/gin（可选，用于创建Web服务）

步骤一：创建HTTP服务器

首先，我们需要创建一个简单的HTTP服务器来处理文件上传请求。

package main

import (
    "net/http"
)

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

func uploadFileHandler(w http.ResponseWriter, r *http.Request) {
    // 处理文件上传逻辑
}

步骤二：处理文件上传

接下来，我们需要在uploadFileHandler函数中添加文件上传逻辑。同时，我们将使用io和os标准库来监控文件上传进度。

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

func uploadFileHandler(w http.ResponseWriter, r *http.Request) {
    if r.Method != "POST" {
        http.Error(w, "Only POST method is allowed", http.StatusMethodNotAllowed)
        return
    }

    // 解析文件
    r.ParseMultipartForm(10 << 20) // 10 MB max memory

    file, handler, err := r.FormFile("uploadFile")
    if err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
        return
    }
    defer file.Close()

    // 创建临时文件
    tempFile, err := ioutil.TempFile("", "upload-")
    if err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
        return
    }
    defer tempFile.Close()

    // 创建一个缓冲区
    buffer := make([]byte, 1024)
    bytesRead := 0
    totalSize := r.ContentLength

    // 读取上传文件，并监控进度
    for {
        bytesRead, err = file.Read(buffer)
        if err == io.EOF {
            break
        }
        if err != nil {
            http.Error(w, err.Error(), http.StatusInternalServerError)
            return
        }

        // 写入临时文件
        _, err = tempFile.Write(buffer[:bytesRead])
        if err != nil {
            http.Error(w, err.Error(), http.StatusInternalServerError)
            return
        }

        // 计算并返回上传进度
        uploadedSize := bytesRead + tempFile.Size()
        fmt.Fprintf(w, "Progress: %d%%\n", int(float64(uploadedSize)/totalSize*100))
    }

    // 重命名临时文件
    tempFileName := tempFile.Name()
    finalFileName := handler.Filename
    os.Rename(tempFileName, finalFileName)
    fmt.Fprintf(w, "File uploaded successfully: %s", finalFileName)
}

步骤三：启动HTTP服务器

总结

本文介绍了如何使用Golang轻松实现文件上传进度的实时监控。通过以上步骤，您可以创建一个简单的HTTP服务器来处理文件上传请求，并在上传过程中监控进度。在实际应用中，您可以根据需要调整代码以适应不同的场景。