sync

compressConn.go

@@ -7,6 +7,7 @@ import (
 	"strings"
 	"time"
 
+	"github.com/klauspost/compress/zstd"
 	"github.com/mholt/archives"
 )
 
@@ -38,11 +39,29 @@ func NewCompressionConn(decode string, conn net.Conn) (net.Conn, error) {
 }
 
 func newZstdConn(conn net.Conn) (io.ReadCloser, io.WriteCloser, error) {
-	r, err := archives.Zstd{}.OpenReader(conn)
+	r, err := archives.Zstd{
+		EncoderOptions: []zstd.EOption{
+			zstd.WithEncoderLevel(zstd.SpeedFastest),
+			zstd.WithWindowSize(zstd.MinWindowSize), // 1MB 窗口，减少内存
+			zstd.WithEncoderConcurrency(1),          // 单线程，减少内存
+		},
+		DecoderOptions: []zstd.DOption{
+			zstd.WithDecodeBuffersBelow(zstd.MinWindowSize),
+		},
+	}.OpenReader(conn)
 	if err != nil {
 		return nil, nil, err
 	}
-	w, err := archives.Zstd{}.OpenWriter(conn)
+	w, err := archives.Zstd{
+		EncoderOptions: []zstd.EOption{
+			zstd.WithEncoderLevel(zstd.SpeedFastest),
+			zstd.WithWindowSize(zstd.MinWindowSize), // 1MB 窗口，减少内存
+			zstd.WithEncoderConcurrency(1),          // 单线程，减少内存
+		},
+		DecoderOptions: []zstd.DOption{
+			zstd.WithDecodeBuffersBelow(zstd.MinWindowSize),
+		},
+	}.OpenWriter(conn)
 	if err != nil {
 		return nil, nil, err
 	}

go.mod

@@ -13,6 +13,7 @@ require (
 	github.com/gospider007/re v0.0.0-20250217075352-bcb79f285d6c
 	github.com/gospider007/tools v0.0.0-20250529114105-be4d4dbf36a2
 	github.com/gospider007/websocket v0.0.0-20250429035144-b1cf6819063a
+	github.com/klauspost/compress v1.18.0
 	github.com/mholt/archives v0.1.2
 	github.com/quic-go/quic-go v0.52.0
 	github.com/refraction-networking/uquic v0.0.6
@@ -47,7 +48,6 @@ require (
 	github.com/gospider007/kinds v0.0.0-20250217075226-10f199f7215d // indirect
 	github.com/hashicorp/golang-lru/v2 v2.0.7 // indirect
 	github.com/json-iterator/go v1.1.12 // indirect
-	github.com/klauspost/compress v1.18.0 // indirect
 	github.com/klauspost/cpuid/v2 v2.2.10 // indirect
 	github.com/klauspost/pgzip v1.2.6 // indirect
 	github.com/kr/pretty v0.3.1 // indirect