从 Node.js 到 Go：以单个 Zip 方式增强数千个文件的下载量-Golang-PHP中文网

从 Node.js 到 Go：以单个 Zip 方式增强数千个文件的下载量

WBOY

发布： 2024-08-21 12:32:40

原创

826 人浏览过

From Node.js to Go: Supercharging Sownloads of Thousands of Files as a Single Zip

作为开发人员，我们在处理大规模数据处理和交付时经常面临挑战。在 Kamero，我们最近解决了文件传输管道中的一个重大瓶颈。我们的应用程序允许用户将与特定事件相关的数千个文件下载为单个 zip 文件。此功能由基于 Node.js 的 Lambda 函数提供支持，负责从 S3 存储桶中获取和压缩文件，但随着我们用户群的增长，该功能一直在努力解决内存限制和执行时间过长的问题。

这篇文章详细介绍了我们从资源匮乏的 Node.js 实现到高效处理大量 S3 下载的精益且快速的 Go 解决方案的历程。我们将探索如何优化我们的系统，以便在从特定事件请求大量文件时为用户提供无缝体验，所有文件都打包到一个方便的单个 zip 下载中。

挑战

我们最初的 Lambda 函数在处理基于事件的大型文件集时面临几个关键问题：

内存消耗：即使分配了 10GB 内存，在处理较大事件的 20,000+ 个文件时，该功能也会失败。
执行时间：具有大量文件的事件的 Zip 操作花费的时间太长，有时会在完成之前超时。
可扩展性：该功能无法有效地处理不断增加的负载，限制了我们为用户提供流行事件中的大型文件集的能力。
用户体验：缓慢的下载准备时间影响了用户满意度，尤其是对于文件数量较多的事件。

Node.js 实现：快速浏览

我们最初的实现使用 s3-zip 库从 S3 对象创建 zip 文件。这是我们如何处理文件的简化片段：

const s3Zip = require("s3-zip");

// ... other code ...

const body = s3Zip.archive(
  { bucket: bucketName },
  eventId,
  files,
  entryData
);

await uploadZipFile(Upload_Bucket, zipfileKey, body);

登录后复制

虽然这种方法有效，但它会在创建 zip 之前将所有文件加载到内存中，从而导致内存使用率较高，并且大型文件集可能会出现内存不足错误。

输入 Go：改变游戏规则的重写

我们决定用 Go 重写 Lambda 函数，利用其效率和内置并发功能。结果令人震惊：

内存使用量：对于相同的工作负载，从 10GB 降至仅 100MB。
速度：该函数速度提高了大约 10 倍。
可靠性：成功处理 20,000 多个文件，没有出现任何问题。

Go实现中的关键优化

1. 高效的S3操作

我们使用了 AWS SDK for Go v2，与 v1 相比，它提供了更好的性能和更低的内存使用量：

cfg, err := config.LoadDefaultConfig(context.TODO())
s3Client = s3.NewFromConfig(cfg)

登录后复制

2. 并发处理

Go 的 goroutine 允许我们同时处理多个文件：

var wg sync.WaitGroup
sem := make(chan struct{}, 10) // Limit concurrent operations

for _, photo := range photos {
    wg.Add(1)
    go func(photo Photo) {
        defer wg.Done()
        sem <- struct{}{} // Acquire semaphore
        defer func() { <-sem }() // Release semaphore

        // Process photo
    }(photo)
}

wg.Wait()

登录后复制

这种方法允许我们同时处理多个文件，同时控制并发级别以防止系统不堪重负。

3. 流式 Zip 创建

我们不是将所有文件加载到内存中，而是将 zip 内容直接流式传输到 S3：

pipeReader, pipeWriter := io.Pipe()

go func() {
    zipWriter := zip.NewWriter(pipeWriter)
    // Add files to zip
    zipWriter.Close()
    pipeWriter.Close()
}()

// Upload streaming content to S3
uploader.Upload(ctx, &s3.PutObjectInput{
    Bucket: &destBucket,
    Key:    &zipFileKey,
    Body:   pipeReader,
})

登录后复制

这种流式传输方法显着减少了内存使用量，并使我们能够处理更大的文件集。