Add buf.MergeMulti

SagerNet · Nov 15, 2024 · d6c9d39 · d6c9d39
1 parent 07f7dca
commit d6c9d39
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/common/buf/multi.go b/common/buf/multi.go
@@ -10,6 +10,22 @@ func LenMulti(buffers []*Buffer) int {
 	return n
 }
 
+func MergeMulti(buffers []*Buffer) *Buffer {
+	switch len(buffers) {
+	case 0:
+		panic("empty buffers")
+	case 1:
+		return buffers[0]
+	default:
+		newBuffer := NewSize(LenMulti(buffers))
+		for _, buffer := range buffers {
+			common.Must1(newBuffer.Write(buffer.Bytes()))
+			buffer.Release()
+		}
+		return newBuffer
+	}
+}
+
 func ToSliceMulti(buffers []*Buffer) [][]byte {
 	return common.Map(buffers, func(it *Buffer) []byte {
 		return it.Bytes()