From 497ec3f3e6b910f2dc2715bba611942deb145d7c Mon Sep 17 00:00:00 2001
From: Aliaksandr Valialkin <valyala@victoriametrics.com>
Date: Mon, 19 Jun 2023 22:40:52 -0700
Subject: [PATCH] lib/encoding: add MarshalBool/UnmarshalBool and
 GetUint32s/PutUint32s functions

These functions are going to be used by VictoriaLogs
---
 lib/encoding/int.go | 45 ++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 44 insertions(+), 1 deletion(-)

diff --git a/lib/encoding/int.go b/lib/encoding/int.go
index e60dfe34a..9ba0b9ed3 100644
--- a/lib/encoding/int.go
+++ b/lib/encoding/int.go
@@ -11,7 +11,7 @@ func MarshalUint16(dst []byte, u uint16) []byte {
 	return append(dst, byte(u>>8), byte(u))
 }
 
-// UnmarshalUint16 returns unmarshaled uint32 from src.
+// UnmarshalUint16 returns unmarshaled uint16 from src.
 func UnmarshalUint16(src []byte) uint16 {
 	// This is faster than the manual conversion.
 	return binary.BigEndian.Uint16(src[:2])
@@ -218,6 +218,20 @@ func UnmarshalVarUint64s(dst []uint64, src []byte) ([]byte, error) {
 	return src[idx:], nil
 }
 
+// MarshalBool appends marshaled v to dst and returns the result.
+func MarshalBool(dst []byte, v bool) []byte {
+	x := byte(0)
+	if v {
+		x = 1
+	}
+	return append(dst, x)
+}
+
+// UnmarshalBool unmarshals bool from src.
+func UnmarshalBool(src []byte) bool {
+	return src[0] != 0
+}
+
 // MarshalBytes appends marshaled b to dst and returns the result.
 func MarshalBytes(dst, b []byte) []byte {
 	dst = MarshalVarUint64(dst, uint64(len(b)))
@@ -295,3 +309,32 @@ type Uint64s struct {
 }
 
 var uint64sPool sync.Pool
+
+// GetUint32s returns an uint32 slice with the given size.
+// The slize contents isn't initialized - it may contain garbage.
+func GetUint32s(size int) *Uint32s {
+	v := uint32sPool.Get()
+	if v == nil {
+		return &Uint32s{
+			A: make([]uint32, size),
+		}
+	}
+	is := v.(*Uint32s)
+	if n := size - cap(is.A); n > 0 {
+		is.A = append(is.A[:cap(is.A)], make([]uint32, n)...)
+	}
+	is.A = is.A[:size]
+	return is
+}
+
+// PutUint32s returns is to the pool.
+func PutUint32s(is *Uint32s) {
+	uint32sPool.Put(is)
+}
+
+// Uint32s holds an uint32 slice
+type Uint32s struct {
+	A []uint32
+}
+
+var uint32sPool sync.Pool