feat: add unit field to bitcount (#538)

* feat: modify contract and commit

* feat: add test

* feat: address comments and convert from ptr to string

* feat: remove unused util

---------

Co-authored-by: Rueian <rueiancsie@gmail.com>
Co-authored-by: Anuragkillswitch <70265851+Anuragkillswitch@users.noreply.github.com>
Signed-off-by: Rueian <rueiancsie@gmail.com>

valkeycompat/adapter.go

@@ -43,7 +43,11 @@ import (
 	"github.com/valkey-io/valkey-go/internal/util"
 )
 
-const KeepTTL = -1
+const (
+	KeepTTL           = -1
+	BitCountIndexByte = "BYTE"
+	BitCountIndexBit  = "BIT"
+)
 
 type Cmdable interface {
 	Cache(ttl time.Duration) CacheCompat
@@ -1096,8 +1100,19 @@ func (c *Compat) BitCount(ctx context.Context, key string, bitCount *BitCount) *
 	var resp valkey.ValkeyResult
 	if bitCount == nil {
 		resp = c.client.Do(ctx, c.client.B().Bitcount().Key(key).Build())
-	} else {
+		return newIntCmd(resp)
+	}
+
+	if bitCount.Unit == "" {
 		resp = c.client.Do(ctx, c.client.B().Bitcount().Key(key).Start(bitCount.Start).End(bitCount.End).Build())
+		return newIntCmd(resp)
+	}
+
+	switch bitCount.Unit {
+	case BitCountIndexByte:
+		resp = c.client.Do(ctx, c.client.B().Bitcount().Key(key).Start(bitCount.Start).End(bitCount.End).Byte().Build())
+	case BitCountIndexBit:
+		resp = c.client.Do(ctx, c.client.B().Bitcount().Key(key).Start(bitCount.Start).End(bitCount.End).Bit().Build())
 	}
 	return newIntCmd(resp)
 }
@@ -4584,8 +4599,19 @@ func (c CacheCompat) BitCount(ctx context.Context, key string, bitCount *BitCoun
 	var resp valkey.ValkeyResult
 	if bitCount == nil {
 		resp = c.client.DoCache(ctx, c.client.B().Bitcount().Key(key).Cache(), c.ttl)
-	} else {
+		return newIntCmd(resp)
+	}
+
+	if bitCount.Unit == "" {
 		resp = c.client.DoCache(ctx, c.client.B().Bitcount().Key(key).Start(bitCount.Start).End(bitCount.End).Cache(), c.ttl)
+		return newIntCmd(resp)
+	}
+
+	switch bitCount.Unit {
+	case BitCountIndexByte:
+		resp = c.client.DoCache(ctx, c.client.B().Bitcount().Key(key).Start(bitCount.Start).End(bitCount.End).Byte().Cache(), c.ttl)
+	case BitCountIndexBit:
+		resp = c.client.DoCache(ctx, c.client.B().Bitcount().Key(key).Start(bitCount.Start).End(bitCount.End).Bit().Cache(), c.ttl)
 	}
 	return newIntCmd(resp)
 }

valkeycompat/adapter_test.go

@@ -1150,6 +1150,24 @@ func testAdapter(resp3 bool) {
 			})
 			Expect(bitCount.Err()).NotTo(HaveOccurred())
 			Expect(bitCount.Val()).To(Equal(int64(6)))
+
+			if resp3 {
+				bitCount = adapter.BitCount(ctx, "key", &BitCount{
+					Start: 1,
+					End:   1,
+					Unit:  "BYTE",
+				})
+				Expect(bitCount.Err()).NotTo(HaveOccurred())
+				Expect(bitCount.Val()).To(Equal(int64(6)))
+
+				bitCount = adapter.BitCount(ctx, "key", &BitCount{
+					Start: 1,
+					End:   1,
+					Unit:  "BIT",
+				})
+				Expect(bitCount.Err()).NotTo(HaveOccurred())
+				Expect(bitCount.Val()).To(Equal(int64(1)))
+			}
 		})
 
 		It("should BitOpAnd", func() {
@@ -7064,6 +7082,24 @@ func testAdapterCache(resp3 bool) {
 			})
 			Expect(bitCount.Err()).NotTo(HaveOccurred())
 			Expect(bitCount.Val()).To(Equal(int64(6)))
+
+			if resp3 {
+				bitCount = adapter.Cache(time.Hour).BitCount(ctx, "key", &BitCount{
+					Start: 1,
+					End:   1,
+					Unit:  "BIT",
+				})
+				Expect(bitCount.Err()).NotTo(HaveOccurred())
+				Expect(bitCount.Val()).To(Equal(int64(1)))
+
+				bitCount = adapter.Cache(time.Hour).BitCount(ctx, "key", &BitCount{
+					Start: 1,
+					End:   1,
+					Unit:  "BYTE",
+				})
+				Expect(bitCount.Err()).NotTo(HaveOccurred())
+				Expect(bitCount.Val()).To(Equal(int64(6)))
+			}
 		})
 
 		It("should BitPos", func() {

valkeycompat/command.go

@@ -1959,6 +1959,7 @@ type SetArgs struct {
 
 type BitCount struct {
 	Start, End int64
+	Unit       string // Stores BIT or BYTE
 }
 
 //type BitPos struct {