Merge pull request #5 from coder/testsmallbuffers-batch

fix: use smaller batch sizes under build tag for more efficient tests

Author: dannykopping

conn/batchsize_default.go

@@ -0,0 +1,12 @@
+//go:build !testsmallbatch
+
+/* SPDX-License-Identifier: MIT
+ *
+ * Copyright (C) 2017-2023 WireGuard LLC. All Rights Reserved.
+ */
+
+package conn
+
+const (
+	IdealBatchSize = 128 // maximum number of packets handled per read and write
+)

conn/batchsize_testsmall.go

@@ -0,0 +1,16 @@
+//go:build testsmallbatch
+
+/* SPDX-License-Identifier: MIT
+ *
+ * Copyright (C) 2017-2023 WireGuard LLC. All Rights Reserved.
+ *
+ * Modified by Coder for test environments to reduce memory usage.
+ */
+
+package conn
+
+// IdealBatchSize is reduced for tests to minimize memory usage.
+// In production, this is 128 (see batchsize_default.go), but tests don't need
+// the full batch capacity and this significantly reduces memory overhead
+// from wireguard buffer pools (~8MB per connection down to ~256KB).
+const IdealBatchSize = 4

conn/conn.go

@@ -15,10 +15,6 @@ import (
 	"strings"
 )
 
-const (
-	IdealBatchSize = 128 // maximum number of packets handled per read and write
-)
-
 // A ReceiveFunc receives at least one packet from the network and writes them
 // into packets. On a successful read it returns the number of elements of
 // sizes, packets, and endpoints that should be evaluated. Some elements of