分片上传:客户端按5MB切分文件,携带文件名、序号、总数、唯一ID等元信息通过HTTP POST逐片上传,支持断点续传与大文件传输。
实现一个可靠的 Golang 文件分片上传系统,关键在于分片上传、服务端接收、分片合并与完
整性校验这几个环节。整个流程要支持大文件传输、断点续传和防止数据损坏。下面从核心逻辑出发,说明如何设计和实现。
客户端将大文件按固定大小(如 5MB)切分成多个片段,每个片段独立上传,避免单次请求过大或失败后重传全部内容。
关键步骤:
示例代码片段(客户端):
const chunkSize = 5 << 20 // 5MBfile, _ := os.Open("large-file.zip") defer file.Close()
fi, _ := file.Stat() totalSize := fi.Size() uuid := generateFileID() // 可基于文件路径+时间生成唯一 ID
for i := 0; ; i++ { chunk := make([]byte, chunkSize) n, err := file.Read(chunk) if n <= 0 { break }
// 发送第 i 片 uploadChunk(uuid, i, chunk[:n], totalSize)}
服务端需要提供一个接收分片的接口,把每个上传的片段保存到临时目录,并记录状态,以便后续合并。
处理逻辑:
Go 服务端接收示例(使用 Gin):
func handleUploadChunk(c *gin.Context) {
fileID := c.PostForm("file_id")
indexStr := c.PostForm("chunk_index")
chunkIndex, _ := strconv.Atoi(indexStr)
data, _ := c.FormFile("data")
src, _ := data.Open()
defer src.Close()
dir := filepath.Join("tmp", fileID, "chunks")
os.MkdirAll(dir, 0755)
destPath := filepath.Join(dir, fmt.Sprintf("%d.part", chunkIndex))
dest, _ := os.Create(destPath)
io.Copy(dest, src)
dest.Close()
// 更新已接收分片列表(可用 map 或 Redis)
recordReceivedChunk(fileID, chunkIndex)
c.JSON(200, gin.H{"success": true})}
当服务端检测到某个文件的所有分片都已上传,即可触发合并操作。
合并流程:
合并代码示例:
func mergeChunks(fileID string, total int, outputPath string) error {
outFile, err := os.Create(outputPath)
if err != nil {
return err
}
defer outFile.Close()
for i := 0; i < total; i++ {
chunkPath := filepath.Join("tmp", fileID, "chunks", fmt.Sprintf("%d.part", i))
chunkData, err := os.ReadFile(chunkPath)
if err != nil {
return err
}
outFile.Write(chunkData)
}
// 合并成功后删除临时文件
os.RemoveAll(filepath.Join("tmp", fileID))
return nil}
为防止传输过程中数据出错,必须进行校验。常用方式有:
校验代码示意:
func calculateSHA256(filePath string) (string, error) {
f, err := os.Open(filePath)
if err != nil {
return "", err
}
defer f.Close()
h := sha256.New()
io.Copy(h, f)
return hex.EncodeToString(h.Sum(nil)), nil
}
// 使用方式
originalHash := getOriginalHashFromClient() // 客户端传入
mergedPath := "uploads/" + fileID + ".zip"
actualHash, _ := calculateSHA256(mergedPath)
if originalHash != actualHash {
return errors.New("文件校验失败,数据不一致")
}
基本上就这些。构建完整的系统时,可以加入 Redis 记录上传状态、支持断点续传查询已有分片、前端展示进度条等功能。核心是分片可靠、合并有序、结果可验。整个流程在 Golang 中实现高效且稳定,适合大文件场景。