database/doris
2
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

[Feature] Support bitmap_and_not & bitmap_and_not_count (#6910)

Browse Source 
Support bitmap_and_not & bitmap_and_not_count.
This commit is contained in:
Pxl
2021-11-01 10:11:54 +08:00
committed by GitHub
parent f47919136a
commit 28030294f7
10 changed files with 299 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

81
be/test/exprs/bitmap_function_test.cpp
View File

@ -394,6 +394,87 @@ TEST_F(BitmapFunctionsTest, bitmap_not) {
ASSERT_EQ(expected, result);
}
TEST_F(BitmapFunctionsTest, bitmap_and_not) {
{
BitmapValue bitmap1({1, 2, 3});
BitmapValue bitmap2({3, 4, 5});
StringVal bitmap_src = convert_bitmap_to_string(ctx, bitmap1);
StringVal bitmap_dst = convert_bitmap_to_string(ctx, bitmap2);
StringVal bitmap_str = BitmapFunctions::bitmap_and_not(ctx, bitmap_src, bitmap_dst);
BigIntVal result = BitmapFunctions::bitmap_count(ctx, bitmap_str);
BigIntVal expected(2);
ASSERT_EQ(expected, result);
}
{
BitmapValue bitmap1({1, 2, 3});
BitmapValue bitmap2({3, 2, 1});
StringVal bitmap_src = convert_bitmap_to_string(ctx, bitmap1);
StringVal bitmap_dst = convert_bitmap_to_string(ctx, bitmap2);
StringVal bitmap_str = BitmapFunctions::bitmap_and_not(ctx, bitmap_src, bitmap_dst);
BigIntVal result = BitmapFunctions::bitmap_count(ctx, bitmap_str);
BigIntVal expected(0);
ASSERT_EQ(expected, result);
}
{
BitmapValue bitmap1({1, 2, 3});
BitmapValue bitmap2({998, 999, 1000});
StringVal bitmap_src = convert_bitmap_to_string(ctx, bitmap1);
StringVal bitmap_dst = convert_bitmap_to_string(ctx, bitmap2);
StringVal bitmap_str = BitmapFunctions::bitmap_and_not(ctx, bitmap_src, bitmap_dst);
BigIntVal result = BitmapFunctions::bitmap_count(ctx, bitmap_str);
BigIntVal expected(3);
ASSERT_EQ(expected, result);
}
}
TEST_F(BitmapFunctionsTest, bitmap_and_not_count) {
{
BitmapValue bitmap1({1, 2, 3});
BitmapValue bitmap2({3, 4, 5});
StringVal bitmap_src = convert_bitmap_to_string(ctx, bitmap1);
StringVal bitmap_dst = convert_bitmap_to_string(ctx, bitmap2);
BigIntVal result = BitmapFunctions::bitmap_and_not_count(ctx, bitmap_src, bitmap_dst);
BigIntVal expected(2);
ASSERT_EQ(expected, result);
}
{
BitmapValue bitmap1({1, 2, 3});
BitmapValue bitmap2({3, 2, 1});
StringVal bitmap_src = convert_bitmap_to_string(ctx, bitmap1);
StringVal bitmap_dst = convert_bitmap_to_string(ctx, bitmap2);
BigIntVal result = BitmapFunctions::bitmap_and_not_count(ctx, bitmap_src, bitmap_dst);
BigIntVal expected(0);
ASSERT_EQ(expected, result);
}
{
BitmapValue bitmap1({1, 2, 3});
BitmapValue bitmap2({998, 999, 1000});
StringVal bitmap_src = convert_bitmap_to_string(ctx, bitmap1);
StringVal bitmap_dst = convert_bitmap_to_string(ctx, bitmap2);
BigIntVal result = BitmapFunctions::bitmap_and_not_count(ctx, bitmap_src, bitmap_dst);
BigIntVal expected(3);
ASSERT_EQ(expected, result);
}
}
TEST_F(BitmapFunctionsTest, bitmap_contains) {
BitmapValue bitmap({4, 5});
StringVal bitmap_str = convert_bitmap_to_string(ctx, bitmap);