Method SetBits

structure/bitmap.go:90–129

SetBits function sets multiple bits at the specified offsets in the bitmap for the provided key

(k string, off ...uint)

// SetBits function sets multiple bits at the specified offsets in the bitmap for the provided key
func (b *BitmapStructure) SetBits(k string, off ...uint) error {
	// Convert the key to bytes
	key := stringToBytesWithKey(k)
	// Create a new bitset
	bit := bitset.New(1)
	// Get the current bits for the key
	value, err := b.getBits(key)
	if err != nil && !errors.Is(err, _const.ErrKeyNotFound) {
		return err
	}
	// If the key is not found, initialize it with the new bitset
	if err == _const.ErrKeyNotFound {
		buf, _ := bitsetToByteArray(bit)
		err = b.db.Put(key, buf)
		if err != nil {
			return err
		}
		value, err = b.getBits(key)
		if err != nil {
			return err
		}
	}

	// Unmarshal the current bits into the bitset
	err = bit.UnmarshalBinary(value)
	if err != nil {
		return err
	}
	// Set the bits at the specified offsets
	for _, u := range off {
		bit.Set(u)
	}
	// Convert the bitset to a byte array
	buf, err := bitsetToByteArray(bit)
	if err != nil {
		return err
	}
	// Store the updated bitset in the database
	return b.db.Put(key, buf)
}

// GetBits function retrieves the bits for the provided key
func (b *BitmapStructure) GetBits(k string) (*BitSet, error) {