在 golang 框架中实现文件分块上传（go 分布式缓存

go mod init myapp

内容来自samhan

go get <a style=color:f60; text-decoration:underline; href="https://www.php.cn/zt/15841.html" target="_blank">git</a>hub.com/gin-gonic/gin 内容来自samhan666

package main

内容来自samhan666

import (

"fmt" copyright zvvq

"github.com/gin-gonic/gin" 内容来自zvvq，别采集哟

"io"

"os" 内容来自zvvq，别采集哟

)

内容来自zvvq，别采集哟

func main() { 内容来自zvvq

// 创建 Gin 路由器 内容来自samhan

r := gin.Default() copyright zvvq

// 分块文件上传路由

内容来自zvvq

r.POST("/upload", func(c gin.Context) {

内容来自samhan666

// 获取上传的文件 zvvq好，好zvvq

file, header, err := c.Request.FormFile("file")

if err != nil { zvvq.cn

c.JSON(500, gin.H{"error": err.Error()}) 内容来自samhan

return

}

defer file.Close()

// 设置块大小

内容来自samhan

blockSize := 5 1024 1024 // 5MB

// 创建临时目录 内容来自samhan

tmpDir, err := os.MkdirTemp("", "upload_")

if err != nil {

copyright zvvq

c.JSON(500, gin.H{"error": err.Error()})

return

内容来自zvvq

} 内容来自samhan

defer os.RemoveAll(tmpDir)

copyright zvvq

// 存储分块文件 zvvq好，好zvvq

partFiles := []string{}

for { zvvq

// 读取块数据

buffer := make([]byte, blockSize) 内容来自samhan666

n, err := file.Read(buffer)

内容来自zvvq

if err == io.EOF { 本文来自zvvq

break

内容来自zvvq

} else if err != nil { 内容来自zvvq，别采集哟

c.JSON(500, gin.H{"error": err.Error()}) zvvq.cn

return 内容来自zvvq，别采集哟

}

// 创建新块文件

内容来自zvvq

partFile, err := os.Create(filepath.Join(tmpDir, header.Filename+"_"+strconv.Itoa(len(partFiles)))) 内容来自zvvq

if err != nil { zvvq.cn

c.JSON(500, gin.H{"error": err.Error()})

内容来自samhan666

return 本文来自zvvq

} copyright zvvq

partFiles = append(partFiles, partFile.Name())

defer partFile.Close()

本文来自zvvq

// 写入块数据 zvvq

if _, err = partFile.Write(buffer[:n]); err != nil {

c.JSON(500, gin.H{"error": err.Error()}) 内容来自zvvq

return 本文来自zvvq

}

zvvq.cn

} zvvq好，好zvvq

// 上传所有分块

// ... 可以在这里实现上传分块文件到存储服务的逻辑 zvvq好，好zvvq

// 成功返回

copyright zvvq

c.JSON(200, gin.H{"message": "文件已上传"}) 本文来自zvvq

}) copyright zvvq

// 启动服务器 本文来自zvvq

if err := r.Run(":8080"); err != nil {

内容来自samhan666

fmt.Println("无法启动服务器：", err) zvvq好，好zvvq

}

} zvvq好，好zvvq

// 上传分块文件 内容来自samhan

func uploadPart(partFile string, bucketName string, s3Key string) (s3manager.UploadPartOutput, error) { 内容来自zvvq，别采集哟

file, err := os.Open(partFile)

内容来自zvvq，别采集哟

if err != nil {

本文来自zvvq

return nil, err

copyright zvvq

} zvvq.cn

defer file.Close() 内容来自samhan

part, err := sess.UploadPart(ctx, &s3.UploadPartInput{

Bucket:        &bucketName, 内容来自zvvq，别采集哟

Key:           &s3Key, copyright zvvq

PartNumber:    int64(partCount),

UploadID:      &uploadID, 内容来自zvvq

Body:          file, 内容来自zvvq

ContentLength: int64(file.Size()),

本文来自zvvq

}) zvvq.cn

if err != nil {

return nil, err zvvq

} copyright zvvq

// 更新已上传的分块列表

parts[partCount-1] = part copyright zvvq

return part, nil

}

copyright zvvq