be/src/exec/common/field_visitors.h
Line | Count | Source |
1 | | // Licensed to the Apache Software Foundation (ASF) under one |
2 | | // or more contributor license agreements. See the NOTICE file |
3 | | // distributed with this work for additional information |
4 | | // regarding copyright ownership. The ASF licenses this file |
5 | | // to you under the Apache License, Version 2.0 (the |
6 | | // "License"); you may not use this file except in compliance |
7 | | // with the License. You may obtain a copy of the License at |
8 | | // |
9 | | // http://www.apache.org/licenses/LICENSE-2.0 |
10 | | // |
11 | | // Unless required by applicable law or agreed to in writing, |
12 | | // software distributed under the License is distributed on an |
13 | | // "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
14 | | // KIND, either express or implied. See the License for the |
15 | | // specific language governing permissions and limitations |
16 | | // under the License. |
17 | | // This file is copied from |
18 | | // https://github.com/ClickHouse/ClickHouse/blob/master/src/Common/FieldVisitors.h |
19 | | // and modified by Doris |
20 | | |
21 | | #pragma once |
22 | | |
23 | | #include "common/demangle.h" |
24 | | #include "common/exception.h" |
25 | | #include "common/status.h" |
26 | | #include "core/accurate_comparison.h" |
27 | | #include "core/data_type/primitive_type.h" |
28 | | #include "core/field.h" |
29 | | |
30 | | namespace doris { |
31 | | |
32 | | /** StaticVisitor (and its descendants) - class with overloaded operator() for all types of fields. |
33 | | * You could call visitor for field using function 'apply_visitor'. |
34 | | * Also "binary visitor" is supported - its operator() takes two arguments. |
35 | | */ |
36 | | template <typename R = void> |
37 | | struct StaticVisitor { |
38 | | using ResultType = R; |
39 | | }; |
40 | | |
41 | | /// F is template parameter, to allow universal reference for field, that is useful for const and non-const values. |
42 | | template <typename Visitor, typename F> |
43 | 0 | typename std::decay_t<Visitor>::ResultType apply_visitor(Visitor&& visitor, F&& field) { |
44 | 0 | switch (field.get_type()) { |
45 | 0 | case PrimitiveType::TYPE_NULL: |
46 | 0 | return visitor.template apply<PrimitiveType::TYPE_NULL>(field.template get<TYPE_NULL>()); |
47 | 0 | case PrimitiveType::TYPE_DATETIMEV2: |
48 | 0 | return visitor.template apply<PrimitiveType::TYPE_DATETIMEV2>( |
49 | 0 | field.template get<TYPE_DATETIMEV2>()); |
50 | 0 | case PrimitiveType::TYPE_TIMESTAMP_NS: |
51 | 0 | return visitor.template apply<PrimitiveType::TYPE_TIMESTAMP_NS>( |
52 | 0 | field.template get<TYPE_TIMESTAMP_NS>()); |
53 | 0 | case PrimitiveType::TYPE_TIMESTAMPTZ: |
54 | 0 | return visitor.template apply<PrimitiveType::TYPE_TIMESTAMPTZ>( |
55 | 0 | field.template get<TYPE_TIMESTAMPTZ>()); |
56 | 0 | case PrimitiveType::TYPE_LARGEINT: |
57 | 0 | return visitor.template apply<PrimitiveType::TYPE_LARGEINT>( |
58 | 0 | field.template get<TYPE_LARGEINT>()); |
59 | 0 | case PrimitiveType::TYPE_DATETIME: |
60 | 0 | return visitor.template apply<PrimitiveType::TYPE_DATETIME>( |
61 | 0 | field.template get<TYPE_DATETIME>()); |
62 | 0 | case PrimitiveType::TYPE_DATE: |
63 | 0 | return visitor.template apply<PrimitiveType::TYPE_DATE>(field.template get<TYPE_DATE>()); |
64 | 0 | case PrimitiveType::TYPE_DATEV2: |
65 | 0 | return visitor.template apply<PrimitiveType::TYPE_DATEV2>( |
66 | 0 | field.template get<TYPE_DATEV2>()); |
67 | 0 | case PrimitiveType::TYPE_BOOLEAN: |
68 | 0 | return visitor.template apply<PrimitiveType::TYPE_BOOLEAN>( |
69 | 0 | field.template get<TYPE_BOOLEAN>()); |
70 | 0 | case PrimitiveType::TYPE_TINYINT: |
71 | 0 | return visitor.template apply<PrimitiveType::TYPE_TINYINT>( |
72 | 0 | field.template get<TYPE_TINYINT>()); |
73 | 0 | case PrimitiveType::TYPE_SMALLINT: |
74 | 0 | return visitor.template apply<PrimitiveType::TYPE_SMALLINT>( |
75 | 0 | field.template get<TYPE_SMALLINT>()); |
76 | 0 | case PrimitiveType::TYPE_INT: |
77 | 0 | return visitor.template apply<PrimitiveType::TYPE_INT>(field.template get<TYPE_INT>()); |
78 | 0 | case PrimitiveType::TYPE_BIGINT: |
79 | 0 | return visitor.template apply<PrimitiveType::TYPE_BIGINT>( |
80 | 0 | field.template get<TYPE_BIGINT>()); |
81 | 0 | case PrimitiveType::TYPE_UINT32: |
82 | 0 | return visitor.template apply<PrimitiveType::TYPE_UINT32>( |
83 | 0 | field.template get<TYPE_UINT32>()); |
84 | 0 | case PrimitiveType::TYPE_UINT64: |
85 | 0 | return visitor.template apply<PrimitiveType::TYPE_UINT64>( |
86 | 0 | field.template get<TYPE_UINT64>()); |
87 | 0 | case PrimitiveType::TYPE_FLOAT: |
88 | 0 | return visitor.template apply<PrimitiveType::TYPE_FLOAT>(field.template get<TYPE_FLOAT>()); |
89 | 0 | case PrimitiveType::TYPE_TIMEV2: |
90 | 0 | return visitor.template apply<PrimitiveType::TYPE_TIMEV2>( |
91 | 0 | field.template get<TYPE_TIMEV2>()); |
92 | 0 | case PrimitiveType::TYPE_DOUBLE: |
93 | 0 | return visitor.template apply<PrimitiveType::TYPE_DOUBLE>( |
94 | 0 | field.template get<TYPE_DOUBLE>()); |
95 | 0 | case PrimitiveType::TYPE_IPV4: |
96 | 0 | return visitor.template apply<PrimitiveType::TYPE_IPV4>(field.template get<TYPE_IPV4>()); |
97 | 0 | case PrimitiveType::TYPE_IPV6: |
98 | 0 | return visitor.template apply<PrimitiveType::TYPE_IPV6>(field.template get<TYPE_IPV6>()); |
99 | 0 | case PrimitiveType::TYPE_STRING: |
100 | 0 | return visitor.template apply<PrimitiveType::TYPE_STRING>( |
101 | 0 | field.template get<TYPE_STRING>()); |
102 | 0 | case PrimitiveType::TYPE_CHAR: |
103 | 0 | return visitor.template apply<PrimitiveType::TYPE_CHAR>(field.template get<TYPE_CHAR>()); |
104 | 0 | case PrimitiveType::TYPE_VARCHAR: |
105 | 0 | return visitor.template apply<PrimitiveType::TYPE_VARCHAR>( |
106 | 0 | field.template get<TYPE_VARCHAR>()); |
107 | 0 | case PrimitiveType::TYPE_VARBINARY: |
108 | 0 | return visitor.template apply<PrimitiveType::TYPE_VARBINARY>( |
109 | 0 | field.template get<TYPE_VARBINARY>()); |
110 | 0 | case PrimitiveType::TYPE_ARRAY: |
111 | 0 | return visitor.template apply<PrimitiveType::TYPE_ARRAY>(field.template get<TYPE_ARRAY>()); |
112 | 0 | case PrimitiveType::TYPE_STRUCT: |
113 | 0 | return visitor.template apply<PrimitiveType::TYPE_STRUCT>( |
114 | 0 | field.template get<TYPE_STRUCT>()); |
115 | 0 | case PrimitiveType::TYPE_MAP: |
116 | 0 | return visitor.template apply<PrimitiveType::TYPE_MAP>(field.template get<TYPE_MAP>()); |
117 | 0 | case PrimitiveType::TYPE_VARIANT: |
118 | 0 | return visitor.template apply<PrimitiveType::TYPE_VARIANT>( |
119 | 0 | field.template get<TYPE_VARIANT>()); |
120 | 0 | case PrimitiveType::TYPE_DECIMAL32: |
121 | 0 | return visitor.template apply<PrimitiveType::TYPE_DECIMAL32>( |
122 | 0 | field.template get<TYPE_DECIMAL32>()); |
123 | 0 | case PrimitiveType::TYPE_DECIMAL64: |
124 | 0 | return visitor.template apply<PrimitiveType::TYPE_DECIMAL64>( |
125 | 0 | field.template get<TYPE_DECIMAL64>()); |
126 | 0 | case PrimitiveType::TYPE_DECIMALV2: |
127 | 0 | return visitor.template apply<PrimitiveType::TYPE_DECIMALV2>( |
128 | 0 | field.template get<TYPE_DECIMALV2>()); |
129 | 0 | case PrimitiveType::TYPE_DECIMAL128I: |
130 | 0 | return visitor.template apply<PrimitiveType::TYPE_DECIMAL128I>( |
131 | 0 | field.template get<TYPE_DECIMAL128I>()); |
132 | 0 | case PrimitiveType::TYPE_DECIMAL256: |
133 | 0 | return visitor.template apply<PrimitiveType::TYPE_DECIMAL256>( |
134 | 0 | field.template get<TYPE_DECIMAL256>()); |
135 | 0 | case PrimitiveType::TYPE_BITMAP: |
136 | 0 | return visitor.template apply<PrimitiveType::TYPE_BITMAP>( |
137 | 0 | field.template get<TYPE_BITMAP>()); |
138 | 0 | case PrimitiveType::TYPE_HLL: |
139 | 0 | return visitor.template apply<PrimitiveType::TYPE_HLL>(field.template get<TYPE_HLL>()); |
140 | 0 | case PrimitiveType::TYPE_QUANTILE_STATE: |
141 | 0 | return visitor.template apply<PrimitiveType::TYPE_QUANTILE_STATE>( |
142 | 0 | field.template get<TYPE_QUANTILE_STATE>()); |
143 | 0 | case PrimitiveType::TYPE_JSONB: |
144 | 0 | return visitor.template apply<PrimitiveType::TYPE_JSONB>(field.template get<TYPE_JSONB>()); |
145 | 0 | default: |
146 | 0 | throw doris::Exception(ErrorCode::INTERNAL_ERROR, "Bad type of Field {}", |
147 | 0 | static_cast<int>(field.get_type())); |
148 | 0 | return {}; |
149 | 0 | } |
150 | 0 | } Unexecuted instantiation: _ZN5doris13apply_visitorIRNS_26FieldVisitorToStringSimpleERKNS_5FieldEEENSt5decayIT_E4type10ResultTypeEOS7_OT0_ Unexecuted instantiation: _ZN5doris13apply_visitorIRNS_12variant_util24FieldVisitorToScalarTypeERKNS_5FieldEEENSt5decayIT_E4type10ResultTypeEOS8_OT0_ Unexecuted instantiation: _ZN5doris13apply_visitorINS_12variant_util32FieldVisitorToNumberOfDimensionsERKNS_5FieldEEENSt5decayIT_E4type10ResultTypeEOS7_OT0_ Unexecuted instantiation: _ZN5doris13apply_visitorIRNS_12variant_util32FieldVisitorToNumberOfDimensionsERKNS_5FieldEEENSt5decayIT_E4type10ResultTypeEOS8_OT0_ Unexecuted instantiation: _ZN5doris13apply_visitorIRNS_12variant_util30SimpleFieldVisitorToScalarTypeERKNS_5FieldEEENSt5decayIT_E4type10ResultTypeEOS8_OT0_ |
151 | | |
152 | | } // namespace doris |