Dima/orion_greptime_proto
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: pql custom node (#305)

Browse Source 
* feat: histogram fold&union distinct on

Signed-off-by: discord9 <discord9@163.com>

* make all

Signed-off-by: discord9 <discord9@163.com>

* c

Signed-off-by: discord9 <discord9@163.com>

* make all

Signed-off-by: discord9 <discord9@163.com>

---------

Signed-off-by: discord9 <discord9@163.com>
This commit is contained in:
discord9
2026-01-21 20:10:53 +08:00
committed by GitHub
parent 7755821c27
commit 4a14b140f2
5 changed files with 2856 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
c++/substrait_extension/promql_plan.pb.cc
+562 -6
View File
@@ -169,8 +169,39 @@ struct LabelPairDefaultTypeInternal {
};
};
PROTOBUF_ATTRIBUTE_NO_DESTROY PROTOBUF_CONSTINIT PROTOBUF_ATTRIBUTE_INIT_PRIORITY1 LabelPairDefaultTypeInternal _LabelPair_default_instance_;
PROTOBUF_CONSTEXPR HistogramFold::HistogramFold(
::_pbi::ConstantInitialized): _impl_{
/*decltype(_impl_.le_column_idx_)*/uint64_t{0u}
, /*decltype(_impl_.ts_column_idx_)*/uint64_t{0u}
, /*decltype(_impl_.field_column_idx_)*/uint64_t{0u}
, /*decltype(_impl_.quantile_)*/0
, /*decltype(_impl_._cached_size_)*/{}} {}
struct HistogramFoldDefaultTypeInternal {
PROTOBUF_CONSTEXPR HistogramFoldDefaultTypeInternal()
: _instance(::_pbi::ConstantInitialized{}) {}
~HistogramFoldDefaultTypeInternal() {}
union {
HistogramFold _instance;
};
};
PROTOBUF_ATTRIBUTE_NO_DESTROY PROTOBUF_CONSTINIT PROTOBUF_ATTRIBUTE_INIT_PRIORITY1 HistogramFoldDefaultTypeInternal _HistogramFold_default_instance_;
PROTOBUF_CONSTEXPR UnionDistinctOn::UnionDistinctOn(
::_pbi::ConstantInitialized): _impl_{
/*decltype(_impl_.compare_key_indices_)*/{}
, /*decltype(_impl_._compare_key_indices_cached_byte_size_)*/{0}
, /*decltype(_impl_.ts_col_idx_)*/uint64_t{0u}
, /*decltype(_impl_._cached_size_)*/{}} {}
struct UnionDistinctOnDefaultTypeInternal {
PROTOBUF_CONSTEXPR UnionDistinctOnDefaultTypeInternal()
: _instance(::_pbi::ConstantInitialized{}) {}
~UnionDistinctOnDefaultTypeInternal() {}
union {
UnionDistinctOn _instance;
};
};
PROTOBUF_ATTRIBUTE_NO_DESTROY PROTOBUF_CONSTINIT PROTOBUF_ATTRIBUTE_INIT_PRIORITY1 UnionDistinctOnDefaultTypeInternal _UnionDistinctOn_default_instance_;
} // namespace substrait_extension
static ::_pb::Metadata file_level_metadata_substrait_5fextension_2fpromql_5fplan_2eproto[8];
static ::_pb::Metadata file_level_metadata_substrait_5fextension_2fpromql_5fplan_2eproto[10];
static constexpr ::_pb::EnumDescriptor const** file_level_enum_descriptors_substrait_5fextension_2fpromql_5fplan_2eproto = nullptr;
static constexpr ::_pb::ServiceDescriptor const** file_level_service_descriptors_substrait_5fextension_2fpromql_5fplan_2eproto = nullptr;
@@ -271,6 +302,24 @@ const uint32_t TableStruct_substrait_5fextension_2fpromql_5fplan_2eproto::offset
~0u, // no _inlined_string_donated_
PROTOBUF_FIELD_OFFSET(::substrait_extension::LabelPair, _impl_.key_),
PROTOBUF_FIELD_OFFSET(::substrait_extension::LabelPair, _impl_.value_),
~0u, // no _has_bits_
PROTOBUF_FIELD_OFFSET(::substrait_extension::HistogramFold, _internal_metadata_),
~0u, // no _extensions_
~0u, // no _oneof_case_
~0u, // no _weak_field_map_
~0u, // no _inlined_string_donated_
PROTOBUF_FIELD_OFFSET(::substrait_extension::HistogramFold, _impl_.le_column_idx_),
PROTOBUF_FIELD_OFFSET(::substrait_extension::HistogramFold, _impl_.ts_column_idx_),
PROTOBUF_FIELD_OFFSET(::substrait_extension::HistogramFold, _impl_.field_column_idx_),
PROTOBUF_FIELD_OFFSET(::substrait_extension::HistogramFold, _impl_.quantile_),
~0u, // no _has_bits_
PROTOBUF_FIELD_OFFSET(::substrait_extension::UnionDistinctOn, _internal_metadata_),
~0u, // no _extensions_
~0u, // no _oneof_case_
~0u, // no _weak_field_map_
~0u, // no _inlined_string_donated_
PROTOBUF_FIELD_OFFSET(::substrait_extension::UnionDistinctOn, _impl_.compare_key_indices_),
PROTOBUF_FIELD_OFFSET(::substrait_extension::UnionDistinctOn, _impl_.ts_col_idx_),
};
static const ::_pbi::MigrationSchema schemas[] PROTOBUF_SECTION_VARIABLE(protodesc_cold) = {
{ 0, -1, -1, sizeof(::substrait_extension::EmptyMetric)},
@@ -281,6 +330,8 @@ static const ::_pbi::MigrationSchema schemas[] PROTOBUF_SECTION_VARIABLE(protode
{ 59, -1, -1, sizeof(::substrait_extension::ScalarCalculate)},
{ 74, -1, -1, sizeof(::substrait_extension::Absent)},
{ 88, -1, -1, sizeof(::substrait_extension::LabelPair)},
{ 96, -1, -1, sizeof(::substrait_extension::HistogramFold)},
{ 106, -1, -1, sizeof(::substrait_extension::UnionDistinctOn)},
};
static const ::_pb::Message* const file_default_instances[] = {
@@ -292,6 +343,8 @@ static const ::_pb::Message* const file_default_instances[] = {
&::substrait_extension::_ScalarCalculate_default_instance_._instance,
&::substrait_extension::_Absent_default_instance_._instance,
&::substrait_extension::_LabelPair_default_instance_._instance,
&::substrait_extension::_HistogramFold_default_instance_._instance,
&::substrait_extension::_UnionDistinctOn_default_instance_._instance,
};
const char descriptor_table_protodef_substrait_5fextension_2fpromql_5fplan_2eproto[] PROTOBUF_SECTION_VARIABLE(protodesc_cold) =
@@ -326,15 +379,19 @@ const char descriptor_table_protodef_substrait_5fextension_2fpromql_5fplan_2epro
"labels\030\006 \003(\0132\036.substrait_extension.Label"
"Pair\022\035\n\025time_index_column_idx\030\007 \001(\004\022\030\n\020v"
"alue_column_idx\030\010 \001(\004\"\'\n\tLabelPair\022\013\n\003ke"
"y\030\001 \001(\t\022\r\n\005value\030\002 \001(\tB\?Z=github.com/Gre"
"ptimeTeam/greptime-proto/go/substrait_ex"
"tensionb\006proto3"
"y\030\001 \001(\t\022\r\n\005value\030\002 \001(\t\"i\n\rHistogramFold\022"
"\025\n\rle_column_idx\030\001 \001(\004\022\025\n\rts_column_idx\030"
"\002 \001(\004\022\030\n\020field_column_idx\030\003 \001(\004\022\020\n\010quant"
"ile\030\004 \001(\001\"B\n\017UnionDistinctOn\022\033\n\023compare_"
"key_indices\030\001 \003(\004\022\022\n\nts_col_idx\030\002 \001(\004B\?Z"
"=github.com/GreptimeTeam/greptime-proto/"
"go/substrait_extensionb\006proto3"
;
static ::_pbi::once_flag descriptor_table_substrait_5fextension_2fpromql_5fplan_2eproto_once;
const ::_pbi::DescriptorTable descriptor_table_substrait_5fextension_2fpromql_5fplan_2eproto = {
false, false, 1335, descriptor_table_protodef_substrait_5fextension_2fpromql_5fplan_2eproto,
false, false, 1510, descriptor_table_protodef_substrait_5fextension_2fpromql_5fplan_2eproto,
"substrait_extension/promql_plan.proto",
&descriptor_table_substrait_5fextension_2fpromql_5fplan_2eproto_once, nullptr, 0, 8,
&descriptor_table_substrait_5fextension_2fpromql_5fplan_2eproto_once, nullptr, 0, 10,
schemas, file_default_instances, TableStruct_substrait_5fextension_2fpromql_5fplan_2eproto::offsets,
file_level_metadata_substrait_5fextension_2fpromql_5fplan_2eproto, file_level_enum_descriptors_substrait_5fextension_2fpromql_5fplan_2eproto,
file_level_service_descriptors_substrait_5fextension_2fpromql_5fplan_2eproto,
@@ -3224,6 +3281,497 @@ void LabelPair::InternalSwap(LabelPair* other) {
file_level_metadata_substrait_5fextension_2fpromql_5fplan_2eproto[7]);
}
// ===================================================================
class HistogramFold::_Internal {
public:
};
HistogramFold::HistogramFold(::PROTOBUF_NAMESPACE_ID::Arena* arena,
bool is_message_owned)
: ::PROTOBUF_NAMESPACE_ID::Message(arena, is_message_owned) {
SharedCtor(arena, is_message_owned);
// @@protoc_insertion_point(arena_constructor:substrait_extension.HistogramFold)
}
HistogramFold::HistogramFold(const HistogramFold& from)
: ::PROTOBUF_NAMESPACE_ID::Message() {
HistogramFold* const _this = this; (void)_this;
new (&_impl_) Impl_{
decltype(_impl_.le_column_idx_){}
, decltype(_impl_.ts_column_idx_){}
, decltype(_impl_.field_column_idx_){}
, decltype(_impl_.quantile_){}
, /*decltype(_impl_._cached_size_)*/{}};
_internal_metadata_.MergeFrom<::PROTOBUF_NAMESPACE_ID::UnknownFieldSet>(from._internal_metadata_);
::memcpy(&_impl_.le_column_idx_, &from._impl_.le_column_idx_,
static_cast<size_t>(reinterpret_cast<char*>(&_impl_.quantile_) -
reinterpret_cast<char*>(&_impl_.le_column_idx_)) + sizeof(_impl_.quantile_));
// @@protoc_insertion_point(copy_constructor:substrait_extension.HistogramFold)
}
inline void HistogramFold::SharedCtor(
::_pb::Arena* arena, bool is_message_owned) {
(void)arena;
(void)is_message_owned;
new (&_impl_) Impl_{
decltype(_impl_.le_column_idx_){uint64_t{0u}}
, decltype(_impl_.ts_column_idx_){uint64_t{0u}}
, decltype(_impl_.field_column_idx_){uint64_t{0u}}
, decltype(_impl_.quantile_){0}
, /*decltype(_impl_._cached_size_)*/{}
};
}
HistogramFold::~HistogramFold() {
// @@protoc_insertion_point(destructor:substrait_extension.HistogramFold)
if (auto *arena = _internal_metadata_.DeleteReturnArena<::PROTOBUF_NAMESPACE_ID::UnknownFieldSet>()) {
(void)arena;
return;
}
SharedDtor();
}
inline void HistogramFold::SharedDtor() {
GOOGLE_DCHECK(GetArenaForAllocation() == nullptr);
}
void HistogramFold::SetCachedSize(int size) const {
_impl_._cached_size_.Set(size);
}
void HistogramFold::Clear() {
// @@protoc_insertion_point(message_clear_start:substrait_extension.HistogramFold)
uint32_t cached_has_bits = 0;
// Prevent compiler warnings about cached_has_bits being unused
(void) cached_has_bits;
::memset(&_impl_.le_column_idx_, 0, static_cast<size_t>(
reinterpret_cast<char*>(&_impl_.quantile_) -
reinterpret_cast<char*>(&_impl_.le_column_idx_)) + sizeof(_impl_.quantile_));
_internal_metadata_.Clear<::PROTOBUF_NAMESPACE_ID::UnknownFieldSet>();
}
const char* HistogramFold::_InternalParse(const char* ptr, ::_pbi::ParseContext* ctx) {
#define CHK_(x) if (PROTOBUF_PREDICT_FALSE(!(x))) goto failure
while (!ctx->Done(&ptr)) {
uint32_t tag;
ptr = ::_pbi::ReadTag(ptr, &tag);
switch (tag >> 3) {
// uint64 le_column_idx = 1;
case 1:
if (PROTOBUF_PREDICT_TRUE(static_cast<uint8_t>(tag) == 8)) {
_impl_.le_column_idx_ = ::PROTOBUF_NAMESPACE_ID::internal::ReadVarint64(&ptr);
CHK_(ptr);
} else
goto handle_unusual;
continue;
// uint64 ts_column_idx = 2;
case 2:
if (PROTOBUF_PREDICT_TRUE(static_cast<uint8_t>(tag) == 16)) {
_impl_.ts_column_idx_ = ::PROTOBUF_NAMESPACE_ID::internal::ReadVarint64(&ptr);
CHK_(ptr);
} else
goto handle_unusual;
continue;
// uint64 field_column_idx = 3;
case 3:
if (PROTOBUF_PREDICT_TRUE(static_cast<uint8_t>(tag) == 24)) {
_impl_.field_column_idx_ = ::PROTOBUF_NAMESPACE_ID::internal::ReadVarint64(&ptr);
CHK_(ptr);
} else
goto handle_unusual;
continue;
// double quantile = 4;
case 4:
if (PROTOBUF_PREDICT_TRUE(static_cast<uint8_t>(tag) == 33)) {
_impl_.quantile_ = ::PROTOBUF_NAMESPACE_ID::internal::UnalignedLoad<double>(ptr);
ptr += sizeof(double);
} else
goto handle_unusual;
continue;
default:
goto handle_unusual;
} // switch
handle_unusual:
if ((tag == 0) || ((tag & 7) == 4)) {
CHK_(ptr);
ctx->SetLastTag(tag);
goto message_done;
}
ptr = UnknownFieldParse(
tag,
_internal_metadata_.mutable_unknown_fields<::PROTOBUF_NAMESPACE_ID::UnknownFieldSet>(),
ptr, ctx);
CHK_(ptr != nullptr);
} // while
message_done:
return ptr;
failure:
ptr = nullptr;
goto message_done;
#undef CHK_
}
uint8_t* HistogramFold::_InternalSerialize(
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const {
// @@protoc_insertion_point(serialize_to_array_start:substrait_extension.HistogramFold)
uint32_t cached_has_bits = 0;
(void) cached_has_bits;
// uint64 le_column_idx = 1;
if (this->_internal_le_column_idx() != 0) {
target = stream->EnsureSpace(target);
target = ::_pbi::WireFormatLite::WriteUInt64ToArray(1, this->_internal_le_column_idx(), target);
}
// uint64 ts_column_idx = 2;
if (this->_internal_ts_column_idx() != 0) {
target = stream->EnsureSpace(target);
target = ::_pbi::WireFormatLite::WriteUInt64ToArray(2, this->_internal_ts_column_idx(), target);
}
// uint64 field_column_idx = 3;
if (this->_internal_field_column_idx() != 0) {
target = stream->EnsureSpace(target);
target = ::_pbi::WireFormatLite::WriteUInt64ToArray(3, this->_internal_field_column_idx(), target);
}
// double quantile = 4;
static_assert(sizeof(uint64_t) == sizeof(double), "Code assumes uint64_t and double are the same size.");
double tmp_quantile = this->_internal_quantile();
uint64_t raw_quantile;
memcpy(&raw_quantile, &tmp_quantile, sizeof(tmp_quantile));
if (raw_quantile != 0) {
target = stream->EnsureSpace(target);
target = ::_pbi::WireFormatLite::WriteDoubleToArray(4, this->_internal_quantile(), target);
}
if (PROTOBUF_PREDICT_FALSE(_internal_metadata_.have_unknown_fields())) {
target = ::_pbi::WireFormat::InternalSerializeUnknownFieldsToArray(
_internal_metadata_.unknown_fields<::PROTOBUF_NAMESPACE_ID::UnknownFieldSet>(::PROTOBUF_NAMESPACE_ID::UnknownFieldSet::default_instance), target, stream);
}
// @@protoc_insertion_point(serialize_to_array_end:substrait_extension.HistogramFold)
return target;
}
size_t HistogramFold::ByteSizeLong() const {
// @@protoc_insertion_point(message_byte_size_start:substrait_extension.HistogramFold)
size_t total_size = 0;
uint32_t cached_has_bits = 0;
// Prevent compiler warnings about cached_has_bits being unused
(void) cached_has_bits;
// uint64 le_column_idx = 1;
if (this->_internal_le_column_idx() != 0) {
total_size += ::_pbi::WireFormatLite::UInt64SizePlusOne(this->_internal_le_column_idx());
}
// uint64 ts_column_idx = 2;
if (this->_internal_ts_column_idx() != 0) {
total_size += ::_pbi::WireFormatLite::UInt64SizePlusOne(this->_internal_ts_column_idx());
}
// uint64 field_column_idx = 3;
if (this->_internal_field_column_idx() != 0) {
total_size += ::_pbi::WireFormatLite::UInt64SizePlusOne(this->_internal_field_column_idx());
}
// double quantile = 4;
static_assert(sizeof(uint64_t) == sizeof(double), "Code assumes uint64_t and double are the same size.");
double tmp_quantile = this->_internal_quantile();
uint64_t raw_quantile;
memcpy(&raw_quantile, &tmp_quantile, sizeof(tmp_quantile));
if (raw_quantile != 0) {
total_size += 1 + 8;
}
return MaybeComputeUnknownFieldsSize(total_size, &_impl_._cached_size_);
}
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData HistogramFold::_class_data_ = {
::PROTOBUF_NAMESPACE_ID::Message::CopyWithSourceCheck,
HistogramFold::MergeImpl
};
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*HistogramFold::GetClassData() const { return &_class_data_; }
void HistogramFold::MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg) {
auto* const _this = static_cast<HistogramFold*>(&to_msg);
auto& from = static_cast<const HistogramFold&>(from_msg);
// @@protoc_insertion_point(class_specific_merge_from_start:substrait_extension.HistogramFold)
GOOGLE_DCHECK_NE(&from, _this);
uint32_t cached_has_bits = 0;
(void) cached_has_bits;
if (from._internal_le_column_idx() != 0) {
_this->_internal_set_le_column_idx(from._internal_le_column_idx());
}
if (from._internal_ts_column_idx() != 0) {
_this->_internal_set_ts_column_idx(from._internal_ts_column_idx());
}
if (from._internal_field_column_idx() != 0) {
_this->_internal_set_field_column_idx(from._internal_field_column_idx());
}
static_assert(sizeof(uint64_t) == sizeof(double), "Code assumes uint64_t and double are the same size.");
double tmp_quantile = from._internal_quantile();
uint64_t raw_quantile;
memcpy(&raw_quantile, &tmp_quantile, sizeof(tmp_quantile));
if (raw_quantile != 0) {
_this->_internal_set_quantile(from._internal_quantile());
}
_this->_internal_metadata_.MergeFrom<::PROTOBUF_NAMESPACE_ID::UnknownFieldSet>(from._internal_metadata_);
}
void HistogramFold::CopyFrom(const HistogramFold& from) {
// @@protoc_insertion_point(class_specific_copy_from_start:substrait_extension.HistogramFold)
if (&from == this) return;
Clear();
MergeFrom(from);
}
bool HistogramFold::IsInitialized() const {
return true;
}
void HistogramFold::InternalSwap(HistogramFold* other) {
using std::swap;
_internal_metadata_.InternalSwap(&other->_internal_metadata_);
::PROTOBUF_NAMESPACE_ID::internal::memswap<
PROTOBUF_FIELD_OFFSET(HistogramFold, _impl_.quantile_)
+ sizeof(HistogramFold::_impl_.quantile_)
- PROTOBUF_FIELD_OFFSET(HistogramFold, _impl_.le_column_idx_)>(
reinterpret_cast<char*>(&_impl_.le_column_idx_),
reinterpret_cast<char*>(&other->_impl_.le_column_idx_));
}
::PROTOBUF_NAMESPACE_ID::Metadata HistogramFold::GetMetadata() const {
return ::_pbi::AssignDescriptors(
&descriptor_table_substrait_5fextension_2fpromql_5fplan_2eproto_getter, &descriptor_table_substrait_5fextension_2fpromql_5fplan_2eproto_once,
file_level_metadata_substrait_5fextension_2fpromql_5fplan_2eproto[8]);
}
// ===================================================================
class UnionDistinctOn::_Internal {
public:
};
UnionDistinctOn::UnionDistinctOn(::PROTOBUF_NAMESPACE_ID::Arena* arena,
bool is_message_owned)
: ::PROTOBUF_NAMESPACE_ID::Message(arena, is_message_owned) {
SharedCtor(arena, is_message_owned);
// @@protoc_insertion_point(arena_constructor:substrait_extension.UnionDistinctOn)
}
UnionDistinctOn::UnionDistinctOn(const UnionDistinctOn& from)
: ::PROTOBUF_NAMESPACE_ID::Message() {
UnionDistinctOn* const _this = this; (void)_this;
new (&_impl_) Impl_{
decltype(_impl_.compare_key_indices_){from._impl_.compare_key_indices_}
, /*decltype(_impl_._compare_key_indices_cached_byte_size_)*/{0}
, decltype(_impl_.ts_col_idx_){}
, /*decltype(_impl_._cached_size_)*/{}};
_internal_metadata_.MergeFrom<::PROTOBUF_NAMESPACE_ID::UnknownFieldSet>(from._internal_metadata_);
_this->_impl_.ts_col_idx_ = from._impl_.ts_col_idx_;
// @@protoc_insertion_point(copy_constructor:substrait_extension.UnionDistinctOn)
}
inline void UnionDistinctOn::SharedCtor(
::_pb::Arena* arena, bool is_message_owned) {
(void)arena;
(void)is_message_owned;
new (&_impl_) Impl_{
decltype(_impl_.compare_key_indices_){arena}
, /*decltype(_impl_._compare_key_indices_cached_byte_size_)*/{0}
, decltype(_impl_.ts_col_idx_){uint64_t{0u}}
, /*decltype(_impl_._cached_size_)*/{}
};
}
UnionDistinctOn::~UnionDistinctOn() {
// @@protoc_insertion_point(destructor:substrait_extension.UnionDistinctOn)
if (auto *arena = _internal_metadata_.DeleteReturnArena<::PROTOBUF_NAMESPACE_ID::UnknownFieldSet>()) {
(void)arena;
return;
}
SharedDtor();
}
inline void UnionDistinctOn::SharedDtor() {
GOOGLE_DCHECK(GetArenaForAllocation() == nullptr);
_impl_.compare_key_indices_.~RepeatedField();
}
void UnionDistinctOn::SetCachedSize(int size) const {
_impl_._cached_size_.Set(size);
}
void UnionDistinctOn::Clear() {
// @@protoc_insertion_point(message_clear_start:substrait_extension.UnionDistinctOn)
uint32_t cached_has_bits = 0;
// Prevent compiler warnings about cached_has_bits being unused
(void) cached_has_bits;
_impl_.compare_key_indices_.Clear();
_impl_.ts_col_idx_ = uint64_t{0u};
_internal_metadata_.Clear<::PROTOBUF_NAMESPACE_ID::UnknownFieldSet>();
}
const char* UnionDistinctOn::_InternalParse(const char* ptr, ::_pbi::ParseContext* ctx) {
#define CHK_(x) if (PROTOBUF_PREDICT_FALSE(!(x))) goto failure
while (!ctx->Done(&ptr)) {
uint32_t tag;
ptr = ::_pbi::ReadTag(ptr, &tag);
switch (tag >> 3) {
// repeated uint64 compare_key_indices = 1;
case 1:
if (PROTOBUF_PREDICT_TRUE(static_cast<uint8_t>(tag) == 10)) {
ptr = ::PROTOBUF_NAMESPACE_ID::internal::PackedUInt64Parser(_internal_mutable_compare_key_indices(), ptr, ctx);
CHK_(ptr);
} else if (static_cast<uint8_t>(tag) == 8) {
_internal_add_compare_key_indices(::PROTOBUF_NAMESPACE_ID::internal::ReadVarint64(&ptr));
CHK_(ptr);
} else
goto handle_unusual;
continue;
// uint64 ts_col_idx = 2;
case 2:
if (PROTOBUF_PREDICT_TRUE(static_cast<uint8_t>(tag) == 16)) {
_impl_.ts_col_idx_ = ::PROTOBUF_NAMESPACE_ID::internal::ReadVarint64(&ptr);
CHK_(ptr);
} else
goto handle_unusual;
continue;
default:
goto handle_unusual;
} // switch
handle_unusual:
if ((tag == 0) || ((tag & 7) == 4)) {
CHK_(ptr);
ctx->SetLastTag(tag);
goto message_done;
}
ptr = UnknownFieldParse(
tag,
_internal_metadata_.mutable_unknown_fields<::PROTOBUF_NAMESPACE_ID::UnknownFieldSet>(),
ptr, ctx);
CHK_(ptr != nullptr);
} // while
message_done:
return ptr;
failure:
ptr = nullptr;
goto message_done;
#undef CHK_
}
uint8_t* UnionDistinctOn::_InternalSerialize(
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const {
// @@protoc_insertion_point(serialize_to_array_start:substrait_extension.UnionDistinctOn)
uint32_t cached_has_bits = 0;
(void) cached_has_bits;
// repeated uint64 compare_key_indices = 1;
{
int byte_size = _impl_._compare_key_indices_cached_byte_size_.load(std::memory_order_relaxed);
if (byte_size > 0) {
target = stream->WriteUInt64Packed(
1, _internal_compare_key_indices(), byte_size, target);
}
}
// uint64 ts_col_idx = 2;
if (this->_internal_ts_col_idx() != 0) {
target = stream->EnsureSpace(target);
target = ::_pbi::WireFormatLite::WriteUInt64ToArray(2, this->_internal_ts_col_idx(), target);
}
if (PROTOBUF_PREDICT_FALSE(_internal_metadata_.have_unknown_fields())) {
target = ::_pbi::WireFormat::InternalSerializeUnknownFieldsToArray(
_internal_metadata_.unknown_fields<::PROTOBUF_NAMESPACE_ID::UnknownFieldSet>(::PROTOBUF_NAMESPACE_ID::UnknownFieldSet::default_instance), target, stream);
}
// @@protoc_insertion_point(serialize_to_array_end:substrait_extension.UnionDistinctOn)
return target;
}
size_t UnionDistinctOn::ByteSizeLong() const {
// @@protoc_insertion_point(message_byte_size_start:substrait_extension.UnionDistinctOn)
size_t total_size = 0;
uint32_t cached_has_bits = 0;
// Prevent compiler warnings about cached_has_bits being unused
(void) cached_has_bits;
// repeated uint64 compare_key_indices = 1;
{
size_t data_size = ::_pbi::WireFormatLite::
UInt64Size(this->_impl_.compare_key_indices_);
if (data_size > 0) {
total_size += 1 +
::_pbi::WireFormatLite::Int32Size(static_cast<int32_t>(data_size));
}
int cached_size = ::_pbi::ToCachedSize(data_size);
_impl_._compare_key_indices_cached_byte_size_.store(cached_size,
std::memory_order_relaxed);
total_size += data_size;
}
// uint64 ts_col_idx = 2;
if (this->_internal_ts_col_idx() != 0) {
total_size += ::_pbi::WireFormatLite::UInt64SizePlusOne(this->_internal_ts_col_idx());
}
return MaybeComputeUnknownFieldsSize(total_size, &_impl_._cached_size_);
}
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData UnionDistinctOn::_class_data_ = {
::PROTOBUF_NAMESPACE_ID::Message::CopyWithSourceCheck,
UnionDistinctOn::MergeImpl
};
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*UnionDistinctOn::GetClassData() const { return &_class_data_; }
void UnionDistinctOn::MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg) {
auto* const _this = static_cast<UnionDistinctOn*>(&to_msg);
auto& from = static_cast<const UnionDistinctOn&>(from_msg);
// @@protoc_insertion_point(class_specific_merge_from_start:substrait_extension.UnionDistinctOn)
GOOGLE_DCHECK_NE(&from, _this);
uint32_t cached_has_bits = 0;
(void) cached_has_bits;
_this->_impl_.compare_key_indices_.MergeFrom(from._impl_.compare_key_indices_);
if (from._internal_ts_col_idx() != 0) {
_this->_internal_set_ts_col_idx(from._internal_ts_col_idx());
}
_this->_internal_metadata_.MergeFrom<::PROTOBUF_NAMESPACE_ID::UnknownFieldSet>(from._internal_metadata_);
}
void UnionDistinctOn::CopyFrom(const UnionDistinctOn& from) {
// @@protoc_insertion_point(class_specific_copy_from_start:substrait_extension.UnionDistinctOn)
if (&from == this) return;
Clear();
MergeFrom(from);
}
bool UnionDistinctOn::IsInitialized() const {
return true;
}
void UnionDistinctOn::InternalSwap(UnionDistinctOn* other) {
using std::swap;
_internal_metadata_.InternalSwap(&other->_internal_metadata_);
_impl_.compare_key_indices_.InternalSwap(&other->_impl_.compare_key_indices_);
swap(_impl_.ts_col_idx_, other->_impl_.ts_col_idx_);
}
::PROTOBUF_NAMESPACE_ID::Metadata UnionDistinctOn::GetMetadata() const {
return ::_pbi::AssignDescriptors(
&descriptor_table_substrait_5fextension_2fpromql_5fplan_2eproto_getter, &descriptor_table_substrait_5fextension_2fpromql_5fplan_2eproto_once,
file_level_metadata_substrait_5fextension_2fpromql_5fplan_2eproto[9]);
}
// @@protoc_insertion_point(namespace_scope)
} // namespace substrait_extension
PROTOBUF_NAMESPACE_OPEN
@@ -3259,6 +3807,14 @@ template<> PROTOBUF_NOINLINE ::substrait_extension::LabelPair*
Arena::CreateMaybeMessage< ::substrait_extension::LabelPair >(Arena* arena) {
return Arena::CreateMessageInternal< ::substrait_extension::LabelPair >(arena);
}
template<> PROTOBUF_NOINLINE ::substrait_extension::HistogramFold*
Arena::CreateMaybeMessage< ::substrait_extension::HistogramFold >(Arena* arena) {
return Arena::CreateMessageInternal< ::substrait_extension::HistogramFold >(arena);
}
template<> PROTOBUF_NOINLINE ::substrait_extension::UnionDistinctOn*
Arena::CreateMaybeMessage< ::substrait_extension::UnionDistinctOn >(Arena* arena) {
return Arena::CreateMessageInternal< ::substrait_extension::UnionDistinctOn >(arena);
}
PROTOBUF_NAMESPACE_CLOSE
// @@protoc_insertion_point(global_scope)
c++/substrait_extension/promql_plan.pb.h
+521
View File
@@ -51,6 +51,9 @@ extern AbsentDefaultTypeInternal _Absent_default_instance_;
class EmptyMetric;
struct EmptyMetricDefaultTypeInternal;
extern EmptyMetricDefaultTypeInternal _EmptyMetric_default_instance_;
class HistogramFold;
struct HistogramFoldDefaultTypeInternal;
extern HistogramFoldDefaultTypeInternal _HistogramFold_default_instance_;
class InstantManipulate;
struct InstantManipulateDefaultTypeInternal;
extern InstantManipulateDefaultTypeInternal _InstantManipulate_default_instance_;
@@ -69,16 +72,21 @@ extern SeriesDivideDefaultTypeInternal _SeriesDivide_default_instance_;
class SeriesNormalize;
struct SeriesNormalizeDefaultTypeInternal;
extern SeriesNormalizeDefaultTypeInternal _SeriesNormalize_default_instance_;
class UnionDistinctOn;
struct UnionDistinctOnDefaultTypeInternal;
extern UnionDistinctOnDefaultTypeInternal _UnionDistinctOn_default_instance_;
} // namespace substrait_extension
PROTOBUF_NAMESPACE_OPEN
template<> ::substrait_extension::Absent* Arena::CreateMaybeMessage<::substrait_extension::Absent>(Arena*);
template<> ::substrait_extension::EmptyMetric* Arena::CreateMaybeMessage<::substrait_extension::EmptyMetric>(Arena*);
template<> ::substrait_extension::HistogramFold* Arena::CreateMaybeMessage<::substrait_extension::HistogramFold>(Arena*);
template<> ::substrait_extension::InstantManipulate* Arena::CreateMaybeMessage<::substrait_extension::InstantManipulate>(Arena*);
template<> ::substrait_extension::LabelPair* Arena::CreateMaybeMessage<::substrait_extension::LabelPair>(Arena*);
template<> ::substrait_extension::RangeManipulate* Arena::CreateMaybeMessage<::substrait_extension::RangeManipulate>(Arena*);
template<> ::substrait_extension::ScalarCalculate* Arena::CreateMaybeMessage<::substrait_extension::ScalarCalculate>(Arena*);
template<> ::substrait_extension::SeriesDivide* Arena::CreateMaybeMessage<::substrait_extension::SeriesDivide>(Arena*);
template<> ::substrait_extension::SeriesNormalize* Arena::CreateMaybeMessage<::substrait_extension::SeriesNormalize>(Arena*);
template<> ::substrait_extension::UnionDistinctOn* Arena::CreateMaybeMessage<::substrait_extension::UnionDistinctOn>(Arena*);
PROTOBUF_NAMESPACE_CLOSE
namespace substrait_extension {
@@ -1886,6 +1894,360 @@ class LabelPair final :
union { Impl_ _impl_; };
friend struct ::TableStruct_substrait_5fextension_2fpromql_5fplan_2eproto;
};
// -------------------------------------------------------------------
class HistogramFold final :
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:substrait_extension.HistogramFold) */ {
public:
inline HistogramFold() : HistogramFold(nullptr) {}
~HistogramFold() override;
explicit PROTOBUF_CONSTEXPR HistogramFold(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
HistogramFold(const HistogramFold& from);
HistogramFold(HistogramFold&& from) noexcept
: HistogramFold() {
*this = ::std::move(from);
}
inline HistogramFold& operator=(const HistogramFold& from) {
CopyFrom(from);
return *this;
}
inline HistogramFold& operator=(HistogramFold&& from) noexcept {
if (this == &from) return *this;
if (GetOwningArena() == from.GetOwningArena()
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
&& GetOwningArena() != nullptr
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
) {
InternalSwap(&from);
} else {
CopyFrom(from);
}
return *this;
}
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
return GetDescriptor();
}
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
return default_instance().GetMetadata().descriptor;
}
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
return default_instance().GetMetadata().reflection;
}
static const HistogramFold& default_instance() {
return *internal_default_instance();
}
static inline const HistogramFold* internal_default_instance() {
return reinterpret_cast<const HistogramFold*>(
&_HistogramFold_default_instance_);
}
static constexpr int kIndexInFileMessages =
8;
friend void swap(HistogramFold& a, HistogramFold& b) {
a.Swap(&b);
}
inline void Swap(HistogramFold* other) {
if (other == this) return;
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
if (GetOwningArena() != nullptr &&
GetOwningArena() == other->GetOwningArena()) {
#else // PROTOBUF_FORCE_COPY_IN_SWAP
if (GetOwningArena() == other->GetOwningArena()) {
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
InternalSwap(other);
} else {
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
}
}
void UnsafeArenaSwap(HistogramFold* other) {
if (other == this) return;
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
InternalSwap(other);
}
// implements Message ----------------------------------------------
HistogramFold* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
return CreateMaybeMessage<HistogramFold>(arena);
}
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
void CopyFrom(const HistogramFold& from);
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
void MergeFrom( const HistogramFold& from) {
HistogramFold::MergeImpl(*this, from);
}
private:
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
public:
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
bool IsInitialized() const final;
size_t ByteSizeLong() const final;
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
uint8_t* _InternalSerialize(
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
private:
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
void SharedDtor();
void SetCachedSize(int size) const final;
void InternalSwap(HistogramFold* other);
private:
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
return "substrait_extension.HistogramFold";
}
protected:
explicit HistogramFold(::PROTOBUF_NAMESPACE_ID::Arena* arena,
bool is_message_owned = false);
public:
static const ClassData _class_data_;
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
// nested types ----------------------------------------------------
// accessors -------------------------------------------------------
enum : int {
kLeColumnIdxFieldNumber = 1,
kTsColumnIdxFieldNumber = 2,
kFieldColumnIdxFieldNumber = 3,
kQuantileFieldNumber = 4,
};
// uint64 le_column_idx = 1;
void clear_le_column_idx();
uint64_t le_column_idx() const;
void set_le_column_idx(uint64_t value);
private:
uint64_t _internal_le_column_idx() const;
void _internal_set_le_column_idx(uint64_t value);
public:
// uint64 ts_column_idx = 2;
void clear_ts_column_idx();
uint64_t ts_column_idx() const;
void set_ts_column_idx(uint64_t value);
private:
uint64_t _internal_ts_column_idx() const;
void _internal_set_ts_column_idx(uint64_t value);
public:
// uint64 field_column_idx = 3;
void clear_field_column_idx();
uint64_t field_column_idx() const;
void set_field_column_idx(uint64_t value);
private:
uint64_t _internal_field_column_idx() const;
void _internal_set_field_column_idx(uint64_t value);
public:
// double quantile = 4;
void clear_quantile();
double quantile() const;
void set_quantile(double value);
private:
double _internal_quantile() const;
void _internal_set_quantile(double value);
public:
// @@protoc_insertion_point(class_scope:substrait_extension.HistogramFold)
private:
class _Internal;
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
typedef void InternalArenaConstructable_;
typedef void DestructorSkippable_;
struct Impl_ {
uint64_t le_column_idx_;
uint64_t ts_column_idx_;
uint64_t field_column_idx_;
double quantile_;
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
};
union { Impl_ _impl_; };
friend struct ::TableStruct_substrait_5fextension_2fpromql_5fplan_2eproto;
};
// -------------------------------------------------------------------
class UnionDistinctOn final :
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:substrait_extension.UnionDistinctOn) */ {
public:
inline UnionDistinctOn() : UnionDistinctOn(nullptr) {}
~UnionDistinctOn() override;
explicit PROTOBUF_CONSTEXPR UnionDistinctOn(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
UnionDistinctOn(const UnionDistinctOn& from);
UnionDistinctOn(UnionDistinctOn&& from) noexcept
: UnionDistinctOn() {
*this = ::std::move(from);
}
inline UnionDistinctOn& operator=(const UnionDistinctOn& from) {
CopyFrom(from);
return *this;
}
inline UnionDistinctOn& operator=(UnionDistinctOn&& from) noexcept {
if (this == &from) return *this;
if (GetOwningArena() == from.GetOwningArena()
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
&& GetOwningArena() != nullptr
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
) {
InternalSwap(&from);
} else {
CopyFrom(from);
}
return *this;
}
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
return GetDescriptor();
}
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
return default_instance().GetMetadata().descriptor;
}
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
return default_instance().GetMetadata().reflection;
}
static const UnionDistinctOn& default_instance() {
return *internal_default_instance();
}
static inline const UnionDistinctOn* internal_default_instance() {
return reinterpret_cast<const UnionDistinctOn*>(
&_UnionDistinctOn_default_instance_);
}
static constexpr int kIndexInFileMessages =
9;
friend void swap(UnionDistinctOn& a, UnionDistinctOn& b) {
a.Swap(&b);
}
inline void Swap(UnionDistinctOn* other) {
if (other == this) return;
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
if (GetOwningArena() != nullptr &&
GetOwningArena() == other->GetOwningArena()) {
#else // PROTOBUF_FORCE_COPY_IN_SWAP
if (GetOwningArena() == other->GetOwningArena()) {
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
InternalSwap(other);
} else {
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
}
}
void UnsafeArenaSwap(UnionDistinctOn* other) {
if (other == this) return;
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
InternalSwap(other);
}
// implements Message ----------------------------------------------
UnionDistinctOn* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
return CreateMaybeMessage<UnionDistinctOn>(arena);
}
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
void CopyFrom(const UnionDistinctOn& from);
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
void MergeFrom( const UnionDistinctOn& from) {
UnionDistinctOn::MergeImpl(*this, from);
}
private:
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
public:
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
bool IsInitialized() const final;
size_t ByteSizeLong() const final;
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
uint8_t* _InternalSerialize(
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
private:
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
void SharedDtor();
void SetCachedSize(int size) const final;
void InternalSwap(UnionDistinctOn* other);
private:
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
return "substrait_extension.UnionDistinctOn";
}
protected:
explicit UnionDistinctOn(::PROTOBUF_NAMESPACE_ID::Arena* arena,
bool is_message_owned = false);
public:
static const ClassData _class_data_;
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
// nested types ----------------------------------------------------
// accessors -------------------------------------------------------
enum : int {
kCompareKeyIndicesFieldNumber = 1,
kTsColIdxFieldNumber = 2,
};
// repeated uint64 compare_key_indices = 1;
int compare_key_indices_size() const;
private:
int _internal_compare_key_indices_size() const;
public:
void clear_compare_key_indices();
private:
uint64_t _internal_compare_key_indices(int index) const;
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< uint64_t >&
_internal_compare_key_indices() const;
void _internal_add_compare_key_indices(uint64_t value);
::PROTOBUF_NAMESPACE_ID::RepeatedField< uint64_t >*
_internal_mutable_compare_key_indices();
public:
uint64_t compare_key_indices(int index) const;
void set_compare_key_indices(int index, uint64_t value);
void add_compare_key_indices(uint64_t value);
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< uint64_t >&
compare_key_indices() const;
::PROTOBUF_NAMESPACE_ID::RepeatedField< uint64_t >*
mutable_compare_key_indices();
// uint64 ts_col_idx = 2;
void clear_ts_col_idx();
uint64_t ts_col_idx() const;
void set_ts_col_idx(uint64_t value);
private:
uint64_t _internal_ts_col_idx() const;
void _internal_set_ts_col_idx(uint64_t value);
public:
// @@protoc_insertion_point(class_scope:substrait_extension.UnionDistinctOn)
private:
class _Internal;
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
typedef void InternalArenaConstructable_;
typedef void DestructorSkippable_;
struct Impl_ {
::PROTOBUF_NAMESPACE_ID::RepeatedField< uint64_t > compare_key_indices_;
mutable std::atomic<int> _compare_key_indices_cached_byte_size_;
uint64_t ts_col_idx_;
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
};
union { Impl_ _impl_; };
friend struct ::TableStruct_substrait_5fextension_2fpromql_5fplan_2eproto;
};
// ===================================================================
@@ -3563,6 +3925,161 @@ inline void LabelPair::set_allocated_value(std::string* value) {
// @@protoc_insertion_point(field_set_allocated:substrait_extension.LabelPair.value)
}
// -------------------------------------------------------------------
// HistogramFold
// uint64 le_column_idx = 1;
inline void HistogramFold::clear_le_column_idx() {
_impl_.le_column_idx_ = uint64_t{0u};
}
inline uint64_t HistogramFold::_internal_le_column_idx() const {
return _impl_.le_column_idx_;
}
inline uint64_t HistogramFold::le_column_idx() const {
// @@protoc_insertion_point(field_get:substrait_extension.HistogramFold.le_column_idx)
return _internal_le_column_idx();
}
inline void HistogramFold::_internal_set_le_column_idx(uint64_t value) {
_impl_.le_column_idx_ = value;
}
inline void HistogramFold::set_le_column_idx(uint64_t value) {
_internal_set_le_column_idx(value);
// @@protoc_insertion_point(field_set:substrait_extension.HistogramFold.le_column_idx)
}
// uint64 ts_column_idx = 2;
inline void HistogramFold::clear_ts_column_idx() {
_impl_.ts_column_idx_ = uint64_t{0u};
}
inline uint64_t HistogramFold::_internal_ts_column_idx() const {
return _impl_.ts_column_idx_;
}
inline uint64_t HistogramFold::ts_column_idx() const {
// @@protoc_insertion_point(field_get:substrait_extension.HistogramFold.ts_column_idx)
return _internal_ts_column_idx();
}
inline void HistogramFold::_internal_set_ts_column_idx(uint64_t value) {
_impl_.ts_column_idx_ = value;
}
inline void HistogramFold::set_ts_column_idx(uint64_t value) {
_internal_set_ts_column_idx(value);
// @@protoc_insertion_point(field_set:substrait_extension.HistogramFold.ts_column_idx)
}
// uint64 field_column_idx = 3;
inline void HistogramFold::clear_field_column_idx() {
_impl_.field_column_idx_ = uint64_t{0u};
}
inline uint64_t HistogramFold::_internal_field_column_idx() const {
return _impl_.field_column_idx_;
}
inline uint64_t HistogramFold::field_column_idx() const {
// @@protoc_insertion_point(field_get:substrait_extension.HistogramFold.field_column_idx)
return _internal_field_column_idx();
}
inline void HistogramFold::_internal_set_field_column_idx(uint64_t value) {
_impl_.field_column_idx_ = value;
}
inline void HistogramFold::set_field_column_idx(uint64_t value) {
_internal_set_field_column_idx(value);
// @@protoc_insertion_point(field_set:substrait_extension.HistogramFold.field_column_idx)
}
// double quantile = 4;
inline void HistogramFold::clear_quantile() {
_impl_.quantile_ = 0;
}
inline double HistogramFold::_internal_quantile() const {
return _impl_.quantile_;
}
inline double HistogramFold::quantile() const {
// @@protoc_insertion_point(field_get:substrait_extension.HistogramFold.quantile)
return _internal_quantile();
}
inline void HistogramFold::_internal_set_quantile(double value) {
_impl_.quantile_ = value;
}
inline void HistogramFold::set_quantile(double value) {
_internal_set_quantile(value);
// @@protoc_insertion_point(field_set:substrait_extension.HistogramFold.quantile)
}
// -------------------------------------------------------------------
// UnionDistinctOn
// repeated uint64 compare_key_indices = 1;
inline int UnionDistinctOn::_internal_compare_key_indices_size() const {
return _impl_.compare_key_indices_.size();
}
inline int UnionDistinctOn::compare_key_indices_size() const {
return _internal_compare_key_indices_size();
}
inline void UnionDistinctOn::clear_compare_key_indices() {
_impl_.compare_key_indices_.Clear();
}
inline uint64_t UnionDistinctOn::_internal_compare_key_indices(int index) const {
return _impl_.compare_key_indices_.Get(index);
}
inline uint64_t UnionDistinctOn::compare_key_indices(int index) const {
// @@protoc_insertion_point(field_get:substrait_extension.UnionDistinctOn.compare_key_indices)
return _internal_compare_key_indices(index);
}
inline void UnionDistinctOn::set_compare_key_indices(int index, uint64_t value) {
_impl_.compare_key_indices_.Set(index, value);
// @@protoc_insertion_point(field_set:substrait_extension.UnionDistinctOn.compare_key_indices)
}
inline void UnionDistinctOn::_internal_add_compare_key_indices(uint64_t value) {
_impl_.compare_key_indices_.Add(value);
}
inline void UnionDistinctOn::add_compare_key_indices(uint64_t value) {
_internal_add_compare_key_indices(value);
// @@protoc_insertion_point(field_add:substrait_extension.UnionDistinctOn.compare_key_indices)
}
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< uint64_t >&
UnionDistinctOn::_internal_compare_key_indices() const {
return _impl_.compare_key_indices_;
}
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< uint64_t >&
UnionDistinctOn::compare_key_indices() const {
// @@protoc_insertion_point(field_list:substrait_extension.UnionDistinctOn.compare_key_indices)
return _internal_compare_key_indices();
}
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< uint64_t >*
UnionDistinctOn::_internal_mutable_compare_key_indices() {
return &_impl_.compare_key_indices_;
}
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< uint64_t >*
UnionDistinctOn::mutable_compare_key_indices() {
// @@protoc_insertion_point(field_mutable_list:substrait_extension.UnionDistinctOn.compare_key_indices)
return _internal_mutable_compare_key_indices();
}
// uint64 ts_col_idx = 2;
inline void UnionDistinctOn::clear_ts_col_idx() {
_impl_.ts_col_idx_ = uint64_t{0u};
}
inline uint64_t UnionDistinctOn::_internal_ts_col_idx() const {
return _impl_.ts_col_idx_;
}
inline uint64_t UnionDistinctOn::ts_col_idx() const {
// @@protoc_insertion_point(field_get:substrait_extension.UnionDistinctOn.ts_col_idx)
return _internal_ts_col_idx();
}
inline void UnionDistinctOn::_internal_set_ts_col_idx(uint64_t value) {
_impl_.ts_col_idx_ = value;
}
inline void UnionDistinctOn::set_ts_col_idx(uint64_t value) {
_internal_set_ts_col_idx(value);
// @@protoc_insertion_point(field_set:substrait_extension.UnionDistinctOn.ts_col_idx)
}
#ifdef __GNUC__
#pragma GCC diagnostic pop
#endif // __GNUC__
@@ -3580,6 +4097,10 @@ inline void LabelPair::set_allocated_value(std::string* value) {
// -------------------------------------------------------------------
// -------------------------------------------------------------------
// -------------------------------------------------------------------
// @@protoc_insertion_point(namespace_scope)
go/substrait_extension/promql_plan.pb.go
+181 -7
View File
@@ -828,6 +828,138 @@ func (x *LabelPair) GetValue() string {
return ""
}
type HistogramFold struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
// Column index of the `le` column (histogram bucket bounds)
LeColumnIdx uint64 `protobuf:"varint,1,opt,name=le_column_idx,json=leColumnIdx,proto3" json:"le_column_idx,omitempty"`
// Column index of the timestamp column
TsColumnIdx uint64 `protobuf:"varint,2,opt,name=ts_column_idx,json=tsColumnIdx,proto3" json:"ts_column_idx,omitempty"`
// Column index of the field column (histogram values)
FieldColumnIdx uint64 `protobuf:"varint,3,opt,name=field_column_idx,json=fieldColumnIdx,proto3" json:"field_column_idx,omitempty"`
// Quantile value
Quantile float64 `protobuf:"fixed64,4,opt,name=quantile,proto3" json:"quantile,omitempty"`
}
func (x *HistogramFold) Reset() {
*x = HistogramFold{}
if protoimpl.UnsafeEnabled {
mi := &file_substrait_extension_promql_plan_proto_msgTypes[8]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *HistogramFold) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*HistogramFold) ProtoMessage() {}
func (x *HistogramFold) ProtoReflect() protoreflect.Message {
mi := &file_substrait_extension_promql_plan_proto_msgTypes[8]
if protoimpl.UnsafeEnabled && x != nil {
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
if ms.LoadMessageInfo() == nil {
ms.StoreMessageInfo(mi)
}
return ms
}
return mi.MessageOf(x)
}
// Deprecated: Use HistogramFold.ProtoReflect.Descriptor instead.
func (*HistogramFold) Descriptor() ([]byte, []int) {
return file_substrait_extension_promql_plan_proto_rawDescGZIP(), []int{8}
}
func (x *HistogramFold) GetLeColumnIdx() uint64 {
if x != nil {
return x.LeColumnIdx
}
return 0
}
func (x *HistogramFold) GetTsColumnIdx() uint64 {
if x != nil {
return x.TsColumnIdx
}
return 0
}
func (x *HistogramFold) GetFieldColumnIdx() uint64 {
if x != nil {
return x.FieldColumnIdx
}
return 0
}
func (x *HistogramFold) GetQuantile() float64 {
if x != nil {
return x.Quantile
}
return 0
}
type UnionDistinctOn struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
// Indices of columns to compare for equality
CompareKeyIndices []uint64 `protobuf:"varint,1,rep,packed,name=compare_key_indices,json=compareKeyIndices,proto3" json:"compare_key_indices,omitempty"`
// Column index of the timestamp column
TsColIdx uint64 `protobuf:"varint,2,opt,name=ts_col_idx,json=tsColIdx,proto3" json:"ts_col_idx,omitempty"`
}
func (x *UnionDistinctOn) Reset() {
*x = UnionDistinctOn{}
if protoimpl.UnsafeEnabled {
mi := &file_substrait_extension_promql_plan_proto_msgTypes[9]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *UnionDistinctOn) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*UnionDistinctOn) ProtoMessage() {}
func (x *UnionDistinctOn) ProtoReflect() protoreflect.Message {
mi := &file_substrait_extension_promql_plan_proto_msgTypes[9]
if protoimpl.UnsafeEnabled && x != nil {
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
if ms.LoadMessageInfo() == nil {
ms.StoreMessageInfo(mi)
}
return ms
}
return mi.MessageOf(x)
}
// Deprecated: Use UnionDistinctOn.ProtoReflect.Descriptor instead.
func (*UnionDistinctOn) Descriptor() ([]byte, []int) {
return file_substrait_extension_promql_plan_proto_rawDescGZIP(), []int{9}
}
func (x *UnionDistinctOn) GetCompareKeyIndices() []uint64 {
if x != nil {
return x.CompareKeyIndices
}
return nil
}
func (x *UnionDistinctOn) GetTsColIdx() uint64 {
if x != nil {
return x.TsColIdx
}
return 0
}
var File_substrait_extension_promql_plan_proto protoreflect.FileDescriptor
var file_substrait_extension_promql_plan_proto_rawDesc = []byte{
@@ -944,11 +1076,27 @@ var file_substrait_extension_promql_plan_proto_rawDesc = []byte{
0x33, 0x0a, 0x09, 0x4c, 0x61, 0x62, 0x65, 0x6c, 0x50, 0x61, 0x69, 0x72, 0x12, 0x10, 0x0a, 0x03,
0x6b, 0x65, 0x79, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x03, 0x6b, 0x65, 0x79, 0x12, 0x14,
0x0a, 0x05, 0x76, 0x61, 0x6c, 0x75, 0x65, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x05, 0x76,
0x61, 0x6c, 0x75, 0x65, 0x42, 0x3f, 0x5a, 0x3d, 0x67, 0x69, 0x74, 0x68, 0x75, 0x62, 0x2e, 0x63,
0x6f, 0x6d, 0x2f, 0x47, 0x72, 0x65, 0x70, 0x74, 0x69, 0x6d, 0x65, 0x54, 0x65, 0x61, 0x6d, 0x2f,
0x67, 0x72, 0x65, 0x70, 0x74, 0x69, 0x6d, 0x65, 0x2d, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x2f, 0x67,
0x6f, 0x2f, 0x73, 0x75, 0x62, 0x73, 0x74, 0x72, 0x61, 0x69, 0x74, 0x5f, 0x65, 0x78, 0x74, 0x65,
0x6e, 0x73, 0x69, 0x6f, 0x6e, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33,
0x61, 0x6c, 0x75, 0x65, 0x22, 0x9d, 0x01, 0x0a, 0x0d, 0x48, 0x69, 0x73, 0x74, 0x6f, 0x67, 0x72,
0x61, 0x6d, 0x46, 0x6f, 0x6c, 0x64, 0x12, 0x22, 0x0a, 0x0d, 0x6c, 0x65, 0x5f, 0x63, 0x6f, 0x6c,
0x75, 0x6d, 0x6e, 0x5f, 0x69, 0x64, 0x78, 0x18, 0x01, 0x20, 0x01, 0x28, 0x04, 0x52, 0x0b, 0x6c,
0x65, 0x43, 0x6f, 0x6c, 0x75, 0x6d, 0x6e, 0x49, 0x64, 0x78, 0x12, 0x22, 0x0a, 0x0d, 0x74, 0x73,
0x5f, 0x63, 0x6f, 0x6c, 0x75, 0x6d, 0x6e, 0x5f, 0x69, 0x64, 0x78, 0x18, 0x02, 0x20, 0x01, 0x28,
0x04, 0x52, 0x0b, 0x74, 0x73, 0x43, 0x6f, 0x6c, 0x75, 0x6d, 0x6e, 0x49, 0x64, 0x78, 0x12, 0x28,
0x0a, 0x10, 0x66, 0x69, 0x65, 0x6c, 0x64, 0x5f, 0x63, 0x6f, 0x6c, 0x75, 0x6d, 0x6e, 0x5f, 0x69,
0x64, 0x78, 0x18, 0x03, 0x20, 0x01, 0x28, 0x04, 0x52, 0x0e, 0x66, 0x69, 0x65, 0x6c, 0x64, 0x43,
0x6f, 0x6c, 0x75, 0x6d, 0x6e, 0x49, 0x64, 0x78, 0x12, 0x1a, 0x0a, 0x08, 0x71, 0x75, 0x61, 0x6e,
0x74, 0x69, 0x6c, 0x65, 0x18, 0x04, 0x20, 0x01, 0x28, 0x01, 0x52, 0x08, 0x71, 0x75, 0x61, 0x6e,
0x74, 0x69, 0x6c, 0x65, 0x22, 0x5f, 0x0a, 0x0f, 0x55, 0x6e, 0x69, 0x6f, 0x6e, 0x44, 0x69, 0x73,
0x74, 0x69, 0x6e, 0x63, 0x74, 0x4f, 0x6e, 0x12, 0x2e, 0x0a, 0x13, 0x63, 0x6f, 0x6d, 0x70, 0x61,
0x72, 0x65, 0x5f, 0x6b, 0x65, 0x79, 0x5f, 0x69, 0x6e, 0x64, 0x69, 0x63, 0x65, 0x73, 0x18, 0x01,
0x20, 0x03, 0x28, 0x04, 0x52, 0x11, 0x63, 0x6f, 0x6d, 0x70, 0x61, 0x72, 0x65, 0x4b, 0x65, 0x79,
0x49, 0x6e, 0x64, 0x69, 0x63, 0x65, 0x73, 0x12, 0x1c, 0x0a, 0x0a, 0x74, 0x73, 0x5f, 0x63, 0x6f,
0x6c, 0x5f, 0x69, 0x64, 0x78, 0x18, 0x02, 0x20, 0x01, 0x28, 0x04, 0x52, 0x08, 0x74, 0x73, 0x43,
0x6f, 0x6c, 0x49, 0x64, 0x78, 0x42, 0x3f, 0x5a, 0x3d, 0x67, 0x69, 0x74, 0x68, 0x75, 0x62, 0x2e,
0x63, 0x6f, 0x6d, 0x2f, 0x47, 0x72, 0x65, 0x70, 0x74, 0x69, 0x6d, 0x65, 0x54, 0x65, 0x61, 0x6d,
0x2f, 0x67, 0x72, 0x65, 0x70, 0x74, 0x69, 0x6d, 0x65, 0x2d, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x2f,
0x67, 0x6f, 0x2f, 0x73, 0x75, 0x62, 0x73, 0x74, 0x72, 0x61, 0x69, 0x74, 0x5f, 0x65, 0x78, 0x74,
0x65, 0x6e, 0x73, 0x69, 0x6f, 0x6e, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33,
}
var (
@@ -963,7 +1111,7 @@ func file_substrait_extension_promql_plan_proto_rawDescGZIP() []byte {
return file_substrait_extension_promql_plan_proto_rawDescData
}
var file_substrait_extension_promql_plan_proto_msgTypes = make([]protoimpl.MessageInfo, 8)
var file_substrait_extension_promql_plan_proto_msgTypes = make([]protoimpl.MessageInfo, 10)
var file_substrait_extension_promql_plan_proto_goTypes = []interface{}{
(*EmptyMetric)(nil), // 0: substrait_extension.EmptyMetric
(*InstantManipulate)(nil), // 1: substrait_extension.InstantManipulate
@@ -973,6 +1121,8 @@ var file_substrait_extension_promql_plan_proto_goTypes = []interface{}{
(*ScalarCalculate)(nil), // 5: substrait_extension.ScalarCalculate
(*Absent)(nil), // 6: substrait_extension.Absent
(*LabelPair)(nil), // 7: substrait_extension.LabelPair
(*HistogramFold)(nil), // 8: substrait_extension.HistogramFold
(*UnionDistinctOn)(nil), // 9: substrait_extension.UnionDistinctOn
}
var file_substrait_extension_promql_plan_proto_depIdxs = []int32{
7, // 0: substrait_extension.Absent.fake_labels:type_name -> substrait_extension.LabelPair
@@ -1085,6 +1235,30 @@ func file_substrait_extension_promql_plan_proto_init() {
return nil
}
}
file_substrait_extension_promql_plan_proto_msgTypes[8].Exporter = func(v interface{}, i int) interface{} {
switch v := v.(*HistogramFold); i {
case 0:
return &v.state
case 1:
return &v.sizeCache
case 2:
return &v.unknownFields
default:
return nil
}
}
file_substrait_extension_promql_plan_proto_msgTypes[9].Exporter = func(v interface{}, i int) interface{} {
switch v := v.(*UnionDistinctOn); i {
case 0:
return &v.state
case 1:
return &v.sizeCache
case 2:
return &v.unknownFields
default:
return nil
}
}
}
type x struct{}
out := protoimpl.TypeBuilder{
@@ -1092,7 +1266,7 @@ func file_substrait_extension_promql_plan_proto_init() {
GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
RawDescriptor: file_substrait_extension_promql_plan_proto_rawDesc,
NumEnums: 0,
NumMessages: 8,
NumMessages: 10,
NumExtensions: 0,
NumServices: 0,
},
java/src/main/java/substrait_extension/PromqlPlan.java
+1574 -3
View File
File diff suppressed because it is too large Load Diff
proto/substrait_extension/promql_plan.proto
+18
View File
@@ -148,3 +148,21 @@ message LabelPair {
string key = 1;
string value = 2;
}
message HistogramFold {
// Column index of the `le` column (histogram bucket bounds)
uint64 le_column_idx = 1;
// Column index of the timestamp column
uint64 ts_column_idx = 2;
// Column index of the field column (histogram values)
uint64 field_column_idx = 3;
// Quantile value
double quantile = 4;
}
message UnionDistinctOn {
// Indices of columns to compare for equality
repeated uint64 compare_key_indices = 1;
// Column index of the timestamp column
uint64 ts_col_idx = 2;
}