mirror of
https://github.com/redis/go-redis.git
synced 2025-11-04 02:33:24 +03:00
8d2a022fd5
* Add support for specifying bitcount unit as byte or bit, byte default * Add bitcount test * Test bitcount without unit specified --------- Co-authored-by: wanghongwei5 <wanghongwei5@360.cn> Co-authored-by: ofekshenawa <104765379+ofekshenawa@users.noreply.github.com>
99 lines
2.2 KiB
Go
99 lines
2.2 KiB
Go
package redis_test
|
|
|
|
import (
|
|
. "github.com/bsm/ginkgo/v2"
|
|
. "github.com/bsm/gomega"
|
|
"github.com/redis/go-redis/v9"
|
|
)
|
|
|
|
type bitCountExpected struct {
|
|
Start int64
|
|
End int64
|
|
Expected int64
|
|
}
|
|
|
|
var _ = Describe("BitCountBite", func() {
|
|
var client *redis.Client
|
|
key := "bit_count_test"
|
|
|
|
BeforeEach(func() {
|
|
client = redis.NewClient(redisOptions())
|
|
Expect(client.FlushDB(ctx).Err()).NotTo(HaveOccurred())
|
|
values := []int{0, 1, 0, 0, 1, 0, 1, 0, 1, 1}
|
|
for i, v := range values {
|
|
cmd := client.SetBit(ctx, key, int64(i), v)
|
|
Expect(cmd.Err()).NotTo(HaveOccurred())
|
|
}
|
|
})
|
|
|
|
AfterEach(func() {
|
|
Expect(client.Close()).NotTo(HaveOccurred())
|
|
})
|
|
|
|
It("bit count bite", func() {
|
|
var expected = []bitCountExpected{
|
|
{0, 0, 0},
|
|
{0, 1, 1},
|
|
{0, 2, 1},
|
|
{0, 3, 1},
|
|
{0, 4, 2},
|
|
{0, 5, 2},
|
|
{0, 6, 3},
|
|
{0, 7, 3},
|
|
{0, 8, 4},
|
|
{0, 9, 5},
|
|
}
|
|
|
|
for _, e := range expected {
|
|
cmd := client.BitCount(ctx, key, &redis.BitCount{Start: e.Start, End: e.End, Unit: redis.BitCountIndexBit})
|
|
Expect(cmd.Err()).NotTo(HaveOccurred())
|
|
Expect(cmd.Val()).To(Equal(e.Expected))
|
|
}
|
|
})
|
|
})
|
|
|
|
var _ = Describe("BitCountByte", func() {
|
|
var client *redis.Client
|
|
key := "bit_count_test"
|
|
|
|
BeforeEach(func() {
|
|
client = redis.NewClient(redisOptions())
|
|
Expect(client.FlushDB(ctx).Err()).NotTo(HaveOccurred())
|
|
values := []int{0, 0, 0, 0, 0, 0, 0, 1, 1, 1}
|
|
for i, v := range values {
|
|
cmd := client.SetBit(ctx, key, int64(i), v)
|
|
Expect(cmd.Err()).NotTo(HaveOccurred())
|
|
}
|
|
})
|
|
|
|
AfterEach(func() {
|
|
Expect(client.Close()).NotTo(HaveOccurred())
|
|
})
|
|
|
|
It("bit count byte", func() {
|
|
var expected = []bitCountExpected{
|
|
{0, 0, 1},
|
|
{0, 1, 3},
|
|
}
|
|
|
|
for _, e := range expected {
|
|
cmd := client.BitCount(ctx, key, &redis.BitCount{Start: e.Start, End: e.End, Unit: redis.BitCountIndexByte})
|
|
Expect(cmd.Err()).NotTo(HaveOccurred())
|
|
Expect(cmd.Val()).To(Equal(e.Expected))
|
|
}
|
|
})
|
|
|
|
It("bit count byte with no unit specified", func() {
|
|
var expected = []bitCountExpected{
|
|
{0, 0, 1},
|
|
{0, 1, 3},
|
|
}
|
|
|
|
for _, e := range expected {
|
|
cmd := client.BitCount(ctx, key, &redis.BitCount{Start: e.Start, End: e.End})
|
|
Expect(cmd.Err()).NotTo(HaveOccurred())
|
|
Expect(cmd.Val()).To(Equal(e.Expected))
|
|
}
|
|
})
|
|
})
|