From 078b67ccf04018b8affbd1a7ef6e52fef2883cb5 Mon Sep 17 00:00:00 2001
From: xiaoqidun <xiaoqidun@gmail.com>
Date: Fri, 16 Oct 2026 00:23:49 +0800
Subject: [PATCH] =?UTF-8?q?perf(=E5=8E=8B=E7=BC=A9=E6=80=A7=E8=83=BD):=20?=
 =?UTF-8?q?=E6=8F=90=E5=8D=87=E5=BD=92=E6=A1=A3=E5=8E=8B=E7=BC=A9=E6=80=A7?=
 =?UTF-8?q?=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ofdgo_archive.go | 19 ++++++++++++++++---
 1 file changed, 16 insertions(+), 3 deletions(-)

diff --git a/ofdgo_archive.go b/ofdgo_archive.go
index 6436c3c..e17375b 100644
--- a/ofdgo_archive.go
+++ b/ofdgo_archive.go
@@ -55,6 +55,7 @@ func WriteArchive(ctx context.Context, writer io.Writer, files []ArchiveFile, op
 	var count int64
 	archive := zip.NewWriter(convertWriter{context: ctx, writer: writer, count: &count})
 	configureArchive(archive, options.Compression)
+	var buffer []byte
 	for i, file := range files {
 		if err := ctx.Err(); err != nil {
 			return count, err
@@ -77,7 +78,10 @@ func WriteArchive(ctx context.Context, writer io.Writer, files []ArchiveFile, op
 			return count, err
 		}
 		input := convertReader{context: ctx, source: file.Source}
-		written, err := io.Copy(entry, io.NewSectionReader(input, 0, file.Size))
+		if buffer == nil {
+			buffer = make([]byte, 32<<10)
+		}
+		written, err := io.CopyBuffer(entry, io.NewSectionReader(input, 0, file.Size), buffer)
 		if err != nil {
 			return count, err
 		}
@@ -99,10 +103,19 @@ func WriteArchive(ctx context.Context, writer io.Writer, files []ArchiveFile, op
 	return count, ctx.Err()
 }
 
-// configureArchive 统一逐页导出与批量归档的压缩等级
+// configureArchive 统一归档压缩等级，在单个归档内复用压缩器
 // 入参: archive 归档写入器, options 压缩配置
 func configureArchive(archive *zip.Writer, options CompressionOptions) {
 	if options.Mode != CompressionUnchanged {
-		archive.RegisterCompressor(zip.Deflate, func(writer io.Writer) (io.WriteCloser, error) { return flate.NewWriter(writer, flate.BestCompression) })
+		var compressor *flate.Writer
+		archive.RegisterCompressor(zip.Deflate, func(writer io.Writer) (io.WriteCloser, error) {
+			if compressor == nil {
+				var err error
+				compressor, err = flate.NewWriter(writer, flate.BestCompression)
+				return compressor, err
+			}
+			compressor.Reset(writer)
+			return compressor, nil
+		})
 	}
 }