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

[feature](function) support bitmap_union/intersect have more columns parameters (#7379)

Browse Source 
support multi bitmap parameter for all bitmap aggregation function
This commit is contained in:
zhangstar333
2021-12-26 11:03:20 +08:00
committed by GitHub
parent fe1d0c1428
commit 0c154733e0
16 changed files with 747 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

156
be/src/exprs/bitmap_function.cpp
View File

@ -473,6 +473,31 @@ StringVal BitmapFunctions::bitmap_or(FunctionContext* ctx, const StringVal& lhs,
return serialize(ctx, &bitmap);
}
StringVal BitmapFunctions::bitmap_or(FunctionContext* ctx, const StringVal& lhs,
int num_args, const StringVal* bitmap_strs) {
DCHECK_GE(num_args, 1);
if (lhs.is_null || bitmap_strs->is_null) {
return StringVal::null();
}
BitmapValue bitmap;
if (lhs.len == 0) {
bitmap |= *reinterpret_cast<BitmapValue*>(lhs.ptr);
} else {
bitmap |= BitmapValue((char*)lhs.ptr);
}
for (int i = 0; i < num_args; ++i) {
if (bitmap_strs[i].is_null) {
return StringVal::null();
}
if (bitmap_strs[i].len == 0) {
bitmap |= *reinterpret_cast<BitmapValue*>(bitmap_strs[i].ptr);
} else {
bitmap |= BitmapValue((char*)bitmap_strs[i].ptr);
}
}
return serialize(ctx, &bitmap);
}
StringVal BitmapFunctions::bitmap_and(FunctionContext* ctx, const StringVal& lhs,
const StringVal& rhs) {
if (lhs.is_null || rhs.is_null) {
@ -492,6 +517,32 @@ StringVal BitmapFunctions::bitmap_and(FunctionContext* ctx, const StringVal& lhs
}
return serialize(ctx, &bitmap);
}
StringVal BitmapFunctions::bitmap_and(FunctionContext* ctx, const StringVal& lhs,
int num_args, const StringVal* bitmap_strs) {
DCHECK_GE(num_args, 1);
if (lhs.is_null || bitmap_strs->is_null) {
return StringVal::null();
}
BitmapValue bitmap;
if (lhs.len == 0) {
bitmap |= *reinterpret_cast<BitmapValue*>(lhs.ptr);
} else {
bitmap |= BitmapValue((char*)lhs.ptr);
}
for (int i = 0; i < num_args; ++i) {
if (bitmap_strs[i].is_null) {
return StringVal::null();
}
if (bitmap_strs[i].len == 0) {
bitmap &= *reinterpret_cast<BitmapValue*>(bitmap_strs[i].ptr);
} else {
bitmap &= BitmapValue((char*)bitmap_strs[i].ptr);
}
}
return serialize(ctx, &bitmap);
}
BigIntVal BitmapFunctions::bitmap_and_count(FunctionContext* ctx, const StringVal& lhs,
const StringVal& rhs) {
if (lhs.is_null || rhs.is_null) {
@ -511,6 +562,32 @@ BigIntVal BitmapFunctions::bitmap_and_count(FunctionContext* ctx, const StringVa
}
}
BigIntVal BitmapFunctions::bitmap_and_count(FunctionContext* ctx, const StringVal& lhs, int num_args,
const StringVal* bitmap_strs) {
DCHECK_GE(num_args, 1);
if (lhs.is_null || bitmap_strs->is_null) {
return BigIntVal::null();
}
BitmapValue bitmap;
if (lhs.len == 0) {
bitmap |= *reinterpret_cast<BitmapValue*>(lhs.ptr);
} else {
bitmap |= BitmapValue((char*)lhs.ptr);
}
for (int i = 0; i < num_args; i++) {
if (bitmap_strs[i].is_null) {
return BigIntVal::null();
}
if (bitmap_strs[i].len == 0) {
bitmap &= *reinterpret_cast<BitmapValue*>(bitmap_strs[i].ptr);
} else {
bitmap &= BitmapValue((char*)bitmap_strs[i].ptr);
}
}
return {static_cast<int64_t>(bitmap.cardinality())};
}
BigIntVal BitmapFunctions::bitmap_or_count(FunctionContext* ctx, const StringVal& lhs,
const StringVal& rhs) {
if (lhs.is_null || rhs.is_null) {
@ -530,6 +607,32 @@ BigIntVal BitmapFunctions::bitmap_or_count(FunctionContext* ctx, const StringVal
}
}
BigIntVal BitmapFunctions::bitmap_or_count(FunctionContext* ctx, const StringVal& lhs, int num_args,
const StringVal* bitmap_strs) {
DCHECK_GE(num_args, 1);
if (lhs.is_null || bitmap_strs->is_null) {
return BigIntVal::null();
}
BitmapValue bitmap;
if (lhs.len == 0) {
bitmap |= *reinterpret_cast<BitmapValue*>(lhs.ptr);
} else {
bitmap |= BitmapValue((char*)lhs.ptr);
}
for (int i = 0; i < num_args; i++) {
if (bitmap_strs[i].is_null) {
return BigIntVal::null();
}
if (bitmap_strs[i].len == 0) {
bitmap |= *reinterpret_cast<BitmapValue*>(bitmap_strs[i].ptr);
} else {
bitmap |= BitmapValue((char*)bitmap_strs[i].ptr);
}
}
return {static_cast<int64_t>(bitmap.cardinality())};
}
StringVal BitmapFunctions::bitmap_xor(FunctionContext* ctx, const StringVal& lhs,
const StringVal& rhs) {
if (lhs.is_null || rhs.is_null) {
@ -550,6 +653,31 @@ StringVal BitmapFunctions::bitmap_xor(FunctionContext* ctx, const StringVal& lhs
return serialize(ctx, &bitmap);
}
StringVal BitmapFunctions::bitmap_xor(FunctionContext* ctx, const StringVal& lhs,
int num_args, const StringVal* bitmap_strs) {
DCHECK_GE(num_args, 1);
if (lhs.is_null || bitmap_strs->is_null) {
return StringVal::null();
}
BitmapValue bitmap;
if (lhs.len == 0) {
bitmap |= *reinterpret_cast<BitmapValue*>(lhs.ptr);
} else {
bitmap |= BitmapValue((char*)lhs.ptr);
}
for (int i = 0; i < num_args; ++i) {
if (bitmap_strs[i].is_null) {
return StringVal::null();
}
if (bitmap_strs[i].len == 0) {
bitmap ^= *reinterpret_cast<BitmapValue*>(bitmap_strs[i].ptr);
} else {
bitmap ^= BitmapValue((char*)bitmap_strs[i].ptr);
}
}
return serialize(ctx, &bitmap);
}
BigIntVal BitmapFunctions::bitmap_xor_count(FunctionContext* ctx, const StringVal& lhs,
const StringVal& rhs) {
if (lhs.is_null || rhs.is_null) {
@ -569,6 +697,32 @@ BigIntVal BitmapFunctions::bitmap_xor_count(FunctionContext* ctx, const StringVa
}
}
BigIntVal BitmapFunctions::bitmap_xor_count(FunctionContext* ctx, const StringVal& lhs, int num_args,
const StringVal* bitmap_strs) {
DCHECK_GE(num_args, 1);
if (lhs.is_null || bitmap_strs->is_null) {
return BigIntVal::null();
}
BitmapValue bitmap;
if (lhs.len == 0) {
bitmap |= *reinterpret_cast<BitmapValue*>(lhs.ptr);
} else {
bitmap |= BitmapValue((char*)lhs.ptr);
}
for (int i = 0; i < num_args; i++) {
if (bitmap_strs[i].is_null) {
return BigIntVal::null();
}
if (bitmap_strs[i].len == 0) {
bitmap ^= *reinterpret_cast<BitmapValue*>(bitmap_strs[i].ptr);
} else {
bitmap ^= BitmapValue((char*)bitmap_strs[i].ptr);
}
}
return {static_cast<int64_t>(bitmap.cardinality())};
}
StringVal BitmapFunctions::bitmap_not(FunctionContext* ctx, const StringVal& lhs,
const StringVal& rhs) {
if (lhs.is_null || rhs.is_null) {
@ -1097,4 +1251,4 @@ template StringVal BitmapFunctions::orthogonal_bitmap_intersect_count_serialize<
FunctionContext* ctx, const StringVal& src);
template StringVal BitmapFunctions::orthogonal_bitmap_intersect_count_serialize<StringValue>(
FunctionContext* ctx, const StringVal& src);
} // namespace doris
} // namespace doris