risingwave_common/test_utils/
rand_bitmap.rs

1// Copyright 2025 RisingWave Labs
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7//     http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15use itertools::Itertools;
16use rand::SeedableRng;
17use rand::seq::SliceRandom;
18
19use crate::bitmap::{Bitmap, BitmapBuilder};
20
21pub fn gen_rand_bitmap(num_bits: usize, count_ones: usize, seed: u64) -> Bitmap {
22    if count_ones == num_bits {
23        return Bitmap::ones(num_bits);
24    }
25    let mut builder = BitmapBuilder::zeroed(num_bits);
26    let mut range = (0..num_bits).collect_vec();
27    range.shuffle(&mut rand::rngs::StdRng::seed_from_u64(seed));
28    let shuffled = range.into_iter().collect_vec();
29    for item in shuffled.iter().take(count_ones) {
30        builder.set(*item, true);
31    }
32    builder.finish()
33}
risingwave_common/test_utils/rand_bitmap.rs

risingwave_common/test_utils/
rand_bitmap.rs
