Coverage Report

Created: 2026-07-15 01:43

next uncovered line (L), next uncovered region (R), next uncovered branch (B)
be/src/storage/tablet/tablet_schema.cpp
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
18
#include "storage/tablet/tablet_schema.h"
19
20
#include <gen_cpp/Descriptors_types.h>
21
#include <gen_cpp/olap_file.pb.h>
22
#include <glog/logging.h>
23
#include <google/protobuf/io/coded_stream.h>
24
#include <google/protobuf/io/zero_copy_stream.h>
25
#include <google/protobuf/io/zero_copy_stream_impl_lite.h>
26
27
#include <algorithm>
28
#include <cctype>
29
// IWYU pragma: no_include <bits/std_abs.h>
30
#include <cmath> // IWYU pragma: keep
31
#include <memory>
32
#include <ostream>
33
#include <vector>
34
35
#include "common/compiler_util.h" // IWYU pragma: keep
36
#include "common/consts.h"
37
#include "common/status.h"
38
#include "core/block/block.h"
39
#include "core/column/column_nothing.h"
40
#include "core/data_type/data_type.h"
41
#include "core/data_type/data_type_factory.hpp"
42
#include "core/string_ref.h"
43
#include "exec/common/hex.h"
44
#include "exprs/aggregate/aggregate_function_simple_factory.h"
45
#include "exprs/aggregate/aggregate_function_state_union.h"
46
#include "storage/index/inverted/analyzer/analyzer.h"
47
#include "storage/index/inverted/inverted_index_parser.h"
48
#include "storage/olap_common.h"
49
#include "storage/olap_define.h"
50
#include "storage/tablet/tablet_column_object_pool.h"
51
#include "storage/tablet/tablet_meta.h"
52
#include "storage/tablet_info.h"
53
#include "storage/types.h"
54
#include "storage/utils.h"
55
#include "util/json/path_in_data.h"
56
57
namespace doris {
58
19.3M
FieldType TabletColumn::get_field_type_by_type(PrimitiveType primitiveType) {
59
19.3M
    switch (primitiveType) {
60
0
    case PrimitiveType::INVALID_TYPE:
61
0
        return FieldType::OLAP_FIELD_TYPE_UNKNOWN;
62
0
    case PrimitiveType::TYPE_NULL:
63
0
        return FieldType::OLAP_FIELD_TYPE_NONE;
64
297k
    case PrimitiveType::TYPE_BOOLEAN:
65
297k
        return FieldType::OLAP_FIELD_TYPE_BOOL;
66
3.47M
    case PrimitiveType::TYPE_TINYINT:
67
3.47M
        return FieldType::OLAP_FIELD_TYPE_TINYINT;
68
8.01k
    case PrimitiveType::TYPE_SMALLINT:
69
8.01k
        return FieldType::OLAP_FIELD_TYPE_SMALLINT;
70
269k
    case PrimitiveType::TYPE_INT:
71
269k
        return FieldType::OLAP_FIELD_TYPE_INT;
72
6.83M
    case PrimitiveType::TYPE_BIGINT:
73
6.83M
        return FieldType::OLAP_FIELD_TYPE_BIGINT;
74
73.3k
    case PrimitiveType::TYPE_LARGEINT:
75
73.3k
        return FieldType::OLAP_FIELD_TYPE_LARGEINT;
76
1.39k
    case PrimitiveType::TYPE_FLOAT:
77
1.39k
        return FieldType::OLAP_FIELD_TYPE_FLOAT;
78
2.81M
    case PrimitiveType::TYPE_DOUBLE:
79
2.81M
        return FieldType::OLAP_FIELD_TYPE_DOUBLE;
80
4.28M
    case PrimitiveType::TYPE_VARCHAR:
81
4.28M
        return FieldType::OLAP_FIELD_TYPE_VARCHAR;
82
1.75k
    case PrimitiveType::TYPE_DATE:
83
1.75k
        return FieldType::OLAP_FIELD_TYPE_DATE;
84
1.66k
    case PrimitiveType::TYPE_DATETIME:
85
1.66k
        return FieldType::OLAP_FIELD_TYPE_DATETIME;
86
0
    case PrimitiveType::TYPE_BINARY:
87
0
        return FieldType::OLAP_FIELD_TYPE_UNKNOWN; // Not implemented
88
6.80k
    case PrimitiveType::TYPE_CHAR:
89
6.80k
        return FieldType::OLAP_FIELD_TYPE_CHAR;
90
0
    case PrimitiveType::TYPE_STRUCT:
91
0
        return FieldType::OLAP_FIELD_TYPE_STRUCT;
92
2.22k
    case PrimitiveType::TYPE_ARRAY:
93
2.22k
        return FieldType::OLAP_FIELD_TYPE_ARRAY;
94
1
    case PrimitiveType::TYPE_MAP:
95
1
        return FieldType::OLAP_FIELD_TYPE_MAP;
96
4
    case PrimitiveType::TYPE_HLL:
97
4
        return FieldType::OLAP_FIELD_TYPE_HLL;
98
0
    case PrimitiveType::TYPE_DECIMALV2:
99
0
        return FieldType::OLAP_FIELD_TYPE_UNKNOWN; // Not implemented
100
4
    case PrimitiveType::TYPE_BITMAP:
101
4
        return FieldType::OLAP_FIELD_TYPE_BITMAP;
102
544k
    case PrimitiveType::TYPE_STRING:
103
544k
        return FieldType::OLAP_FIELD_TYPE_STRING;
104
4
    case PrimitiveType::TYPE_QUANTILE_STATE:
105
4
        return FieldType::OLAP_FIELD_TYPE_QUANTILE_STATE;
106
19.1k
    case PrimitiveType::TYPE_DATEV2:
107
19.1k
        return FieldType::OLAP_FIELD_TYPE_DATEV2;
108
483k
    case PrimitiveType::TYPE_DATETIMEV2:
109
483k
        return FieldType::OLAP_FIELD_TYPE_DATETIMEV2;
110
1.94k
    case PrimitiveType::TYPE_TIMESTAMPTZ:
111
1.94k
        return FieldType::OLAP_FIELD_TYPE_TIMESTAMPTZ;
112
0
    case PrimitiveType::TYPE_TIMEV2:
113
0
        return FieldType::OLAP_FIELD_TYPE_TIMEV2;
114
626
    case PrimitiveType::TYPE_DECIMAL32:
115
626
        return FieldType::OLAP_FIELD_TYPE_DECIMAL32;
116
628
    case PrimitiveType::TYPE_DECIMAL64:
117
628
        return FieldType::OLAP_FIELD_TYPE_DECIMAL64;
118
240k
    case PrimitiveType::TYPE_DECIMAL128I:
119
240k
        return FieldType::OLAP_FIELD_TYPE_DECIMAL128I;
120
611
    case PrimitiveType::TYPE_DECIMAL256:
121
611
        return FieldType::OLAP_FIELD_TYPE_DECIMAL256;
122
1.51k
    case PrimitiveType::TYPE_JSONB:
123
1.51k
        return FieldType::OLAP_FIELD_TYPE_JSONB;
124
1
    case PrimitiveType::TYPE_VARIANT:
125
1
        return FieldType::OLAP_FIELD_TYPE_VARIANT;
126
297
    case PrimitiveType::TYPE_IPV4:
127
297
        return FieldType::OLAP_FIELD_TYPE_IPV4;
128
286
    case PrimitiveType::TYPE_IPV6:
129
286
        return FieldType::OLAP_FIELD_TYPE_IPV6;
130
3
    case PrimitiveType::TYPE_AGG_STATE:
131
3
        return FieldType::OLAP_FIELD_TYPE_AGG_STATE;
132
0
    default:
133
0
        return FieldType::OLAP_FIELD_TYPE_UNKNOWN;
134
19.3M
    }
135
19.3M
}
136
137
24.5M
PrimitiveType TabletColumn::get_primitive_type_by_field_type(FieldType type) {
138
24.5M
    static const PrimitiveType mapping[] = {
139
24.5M
            /*  0 */ PrimitiveType::INVALID_TYPE,
140
24.5M
            /*  1 OLAP_FIELD_TYPE_TINYINT           */ PrimitiveType::TYPE_TINYINT,
141
24.5M
            /*  2 OLAP_FIELD_TYPE_UNSIGNED_TINYINT  */ PrimitiveType::INVALID_TYPE,
142
24.5M
            /*  3 OLAP_FIELD_TYPE_SMALLINT          */ PrimitiveType::TYPE_SMALLINT,
143
24.5M
            /*  4 OLAP_FIELD_TYPE_UNSIGNED_SMALLINT */ PrimitiveType::INVALID_TYPE,
144
24.5M
            /*  5 OLAP_FIELD_TYPE_INT               */ PrimitiveType::TYPE_INT,
145
24.5M
            /*  6 OLAP_FIELD_TYPE_UNSIGNED_INT      */ PrimitiveType::INVALID_TYPE,
146
24.5M
            /*  7 OLAP_FIELD_TYPE_BIGINT            */ PrimitiveType::TYPE_BIGINT,
147
24.5M
            /*  8 OLAP_FIELD_TYPE_UNSIGNED_BIGINT   */ PrimitiveType::INVALID_TYPE,
148
24.5M
            /*  9 OLAP_FIELD_TYPE_LARGEINT          */ PrimitiveType::TYPE_LARGEINT,
149
24.5M
            /* 10 OLAP_FIELD_TYPE_FLOAT             */ PrimitiveType::TYPE_FLOAT,
150
24.5M
            /* 11 OLAP_FIELD_TYPE_DOUBLE            */ PrimitiveType::TYPE_DOUBLE,
151
24.5M
            /* 12 OLAP_FIELD_TYPE_DISCRETE_DOUBLE   */ PrimitiveType::INVALID_TYPE,
152
24.5M
            /* 13 OLAP_FIELD_TYPE_CHAR              */ PrimitiveType::TYPE_CHAR,
153
24.5M
            /* 14 OLAP_FIELD_TYPE_DATE              */ PrimitiveType::TYPE_DATE,
154
24.5M
            /* 15 OLAP_FIELD_TYPE_DATETIME          */ PrimitiveType::TYPE_DATETIME,
155
24.5M
            /* 16 OLAP_FIELD_TYPE_DECIMAL           */ PrimitiveType::INVALID_TYPE,
156
24.5M
            /* 17 OLAP_FIELD_TYPE_VARCHAR           */ PrimitiveType::TYPE_VARCHAR,
157
24.5M
            /* 18 OLAP_FIELD_TYPE_STRUCT            */ PrimitiveType::TYPE_STRUCT,
158
24.5M
            /* 19 OLAP_FIELD_TYPE_ARRAY             */ PrimitiveType::TYPE_ARRAY,
159
24.5M
            /* 20 OLAP_FIELD_TYPE_MAP               */ PrimitiveType::TYPE_MAP,
160
24.5M
            /* 21 OLAP_FIELD_TYPE_UNKNOWN           */ PrimitiveType::INVALID_TYPE,
161
24.5M
            /* 22 OLAP_FIELD_TYPE_NONE              */ PrimitiveType::TYPE_NULL,
162
24.5M
            /* 23 OLAP_FIELD_TYPE_HLL               */ PrimitiveType::TYPE_HLL,
163
24.5M
            /* 24 OLAP_FIELD_TYPE_BOOL              */ PrimitiveType::TYPE_BOOLEAN,
164
24.5M
            /* 25 OLAP_FIELD_TYPE_BITMAP            */ PrimitiveType::TYPE_BITMAP,
165
24.5M
            /* 26 OLAP_FIELD_TYPE_STRING            */ PrimitiveType::TYPE_STRING,
166
24.5M
            /* 27 OLAP_FIELD_TYPE_QUANTILE_STATE    */ PrimitiveType::TYPE_QUANTILE_STATE,
167
24.5M
            /* 28 OLAP_FIELD_TYPE_DATEV2            */ PrimitiveType::TYPE_DATEV2,
168
24.5M
            /* 29 OLAP_FIELD_TYPE_DATETIMEV2        */ PrimitiveType::TYPE_DATETIMEV2,
169
24.5M
            /* 30 OLAP_FIELD_TYPE_TIMEV2            */ PrimitiveType::TYPE_TIMEV2,
170
24.5M
            /* 31 OLAP_FIELD_TYPE_DECIMAL32         */ PrimitiveType::TYPE_DECIMAL32,
171
24.5M
            /* 32 OLAP_FIELD_TYPE_DECIMAL64         */ PrimitiveType::TYPE_DECIMAL64,
172
24.5M
            /* 33 OLAP_FIELD_TYPE_DECIMAL128I       */ PrimitiveType::TYPE_DECIMAL128I,
173
24.5M
            /* 34 OLAP_FIELD_TYPE_JSONB             */ PrimitiveType::TYPE_JSONB,
174
24.5M
            /* 35 OLAP_FIELD_TYPE_VARIANT           */ PrimitiveType::TYPE_VARIANT,
175
24.5M
            /* 36 OLAP_FIELD_TYPE_AGG_STATE         */ PrimitiveType::TYPE_AGG_STATE,
176
24.5M
            /* 37 OLAP_FIELD_TYPE_DECIMAL256        */ PrimitiveType::TYPE_DECIMAL256,
177
24.5M
            /* 38 OLAP_FIELD_TYPE_IPV4              */ PrimitiveType::TYPE_IPV4,
178
24.5M
            /* 39 OLAP_FIELD_TYPE_IPV6              */ PrimitiveType::TYPE_IPV6,
179
24.5M
            /* 40 OLAP_FIELD_TYPE_TIMESTAMPTZ       */ PrimitiveType::TYPE_TIMESTAMPTZ,
180
24.5M
    };
181
182
24.5M
    int idx = static_cast<int>(type);
183
24.5M
    return mapping[idx];
184
24.5M
}
185
186
56.8M
FieldType TabletColumn::get_field_type_by_string(const std::string& type_str) {
187
56.8M
    std::string upper_type_str = type_str;
188
56.8M
    std::transform(type_str.begin(), type_str.end(), upper_type_str.begin(),
189
369M
                   [](auto c) { return std::toupper(c); });
190
56.8M
    FieldType type;
191
192
56.8M
    if (0 == upper_type_str.compare("TINYINT")) {
193
3.88M
        type = FieldType::OLAP_FIELD_TYPE_TINYINT;
194
52.9M
    } else if (0 == upper_type_str.compare("SMALLINT")) {
195
729k
        type = FieldType::OLAP_FIELD_TYPE_SMALLINT;
196
52.2M
    } else if (0 == upper_type_str.compare("INT")) {
197
4.52M
        type = FieldType::OLAP_FIELD_TYPE_INT;
198
47.6M
    } else if (0 == upper_type_str.compare("BIGINT")) {
199
10.3M
        type = FieldType::OLAP_FIELD_TYPE_BIGINT;
200
37.3M
    } else if (0 == upper_type_str.compare("LARGEINT")) {
201
749k
        type = FieldType::OLAP_FIELD_TYPE_LARGEINT;
202
36.6M
    } else if (0 == upper_type_str.compare("UNSIGNED_TINYINT")) {
203
0
        type = FieldType::OLAP_FIELD_TYPE_UNSIGNED_TINYINT;
204
36.6M
    } else if (0 == upper_type_str.compare("UNSIGNED_SMALLINT")) {
205
0
        type = FieldType::OLAP_FIELD_TYPE_UNSIGNED_SMALLINT;
206
36.6M
    } else if (0 == upper_type_str.compare("UNSIGNED_INT")) {
207
0
        type = FieldType::OLAP_FIELD_TYPE_UNSIGNED_INT;
208
36.6M
    } else if (0 == upper_type_str.compare("UNSIGNED_BIGINT")) {
209
0
        type = FieldType::OLAP_FIELD_TYPE_UNSIGNED_BIGINT;
210
36.6M
    } else if (0 == upper_type_str.compare("IPV4")) {
211
36.5k
        type = FieldType::OLAP_FIELD_TYPE_IPV4;
212
36.5M
    } else if (0 == upper_type_str.compare("IPV6")) {
213
36.7k
        type = FieldType::OLAP_FIELD_TYPE_IPV6;
214
36.5M
    } else if (0 == upper_type_str.compare("FLOAT")) {
215
622k
        type = FieldType::OLAP_FIELD_TYPE_FLOAT;
216
35.9M
    } else if (0 == upper_type_str.compare("DISCRETE_DOUBLE")) {
217
0
        type = FieldType::OLAP_FIELD_TYPE_DISCRETE_DOUBLE;
218
35.9M
    } else if (0 == upper_type_str.compare("DOUBLE")) {
219
991k
        type = FieldType::OLAP_FIELD_TYPE_DOUBLE;
220
34.9M
    } else if (0 == upper_type_str.compare("CHAR")) {
221
920k
        type = FieldType::OLAP_FIELD_TYPE_CHAR;
222
34.0M
    } else if (0 == upper_type_str.compare("DATE")) {
223
20.5k
        type = FieldType::OLAP_FIELD_TYPE_DATE;
224
33.9M
    } else if (0 == upper_type_str.compare("DATEV2")) {
225
2.18M
        type = FieldType::OLAP_FIELD_TYPE_DATEV2;
226
31.8M
    } else if (0 == upper_type_str.compare("DATETIMEV2")) {
227
3.08M
        type = FieldType::OLAP_FIELD_TYPE_DATETIMEV2;
228
28.7M
    } else if (0 == upper_type_str.compare("DATETIME")) {
229
23.7k
        type = FieldType::OLAP_FIELD_TYPE_DATETIME;
230
28.7M
    } else if (0 == upper_type_str.compare("TIMESTAMPTZ")) {
231
108k
        type = FieldType::OLAP_FIELD_TYPE_TIMESTAMPTZ;
232
28.5M
    } else if (0 == upper_type_str.compare("DECIMAL32")) {
233
498k
        type = FieldType::OLAP_FIELD_TYPE_DECIMAL32;
234
28.0M
    } else if (0 == upper_type_str.compare("DECIMAL64")) {
235
1.47M
        type = FieldType::OLAP_FIELD_TYPE_DECIMAL64;
236
26.6M
    } else if (0 == upper_type_str.compare("DECIMAL128I")) {
237
789k
        type = FieldType::OLAP_FIELD_TYPE_DECIMAL128I;
238
25.8M
    } else if (0 == upper_type_str.compare("DECIMAL256")) {
239
78.2k
        type = FieldType::OLAP_FIELD_TYPE_DECIMAL256;
240
25.7M
    } else if (0 == upper_type_str.compare(0, 7, "DECIMAL")) {
241
46.6k
        type = FieldType::OLAP_FIELD_TYPE_DECIMAL;
242
25.7M
    } else if (0 == upper_type_str.compare(0, 7, "VARCHAR")) {
243
18.7M
        type = FieldType::OLAP_FIELD_TYPE_VARCHAR;
244
18.7M
    } else if (0 == upper_type_str.compare("STRING")) {
245
2.97M
        type = FieldType::OLAP_FIELD_TYPE_STRING;
246
3.98M
    } else if (0 == upper_type_str.compare("JSONB")) {
247
239k
        type = FieldType::OLAP_FIELD_TYPE_JSONB;
248
3.74M
    } else if (0 == upper_type_str.compare("VARIANT")) {
249
71.1k
        type = FieldType::OLAP_FIELD_TYPE_VARIANT;
250
3.67M
    } else if (0 == upper_type_str.compare("BOOLEAN")) {
251
571k
        type = FieldType::OLAP_FIELD_TYPE_BOOL;
252
3.10M
    } else if (0 == upper_type_str.compare(0, 3, "HLL")) {
253
24.7k
        type = FieldType::OLAP_FIELD_TYPE_HLL;
254
3.07M
    } else if (0 == upper_type_str.compare("STRUCT")) {
255
115k
        type = FieldType::OLAP_FIELD_TYPE_STRUCT;
256
2.96M
    } else if (0 == upper_type_str.compare("LIST")) {
257
0
        type = FieldType::OLAP_FIELD_TYPE_ARRAY;
258
2.96M
    } else if (0 == upper_type_str.compare("MAP")) {
259
1.52M
        type = FieldType::OLAP_FIELD_TYPE_MAP;
260
1.52M
    } else if (0 == upper_type_str.compare("OBJECT")) {
261
14.4k
        type = FieldType::OLAP_FIELD_TYPE_BITMAP;
262
1.41M
    } else if (0 == upper_type_str.compare("BITMAP")) {
263
29.5k
        type = FieldType::OLAP_FIELD_TYPE_BITMAP;
264
1.71M
    } else if (0 == upper_type_str.compare("ARRAY")) {
265
1.71M
        type = FieldType::OLAP_FIELD_TYPE_ARRAY;
266
18.4E
    } else if (0 == upper_type_str.compare("QUANTILE_STATE")) {
267
17.8k
        type = FieldType::OLAP_FIELD_TYPE_QUANTILE_STATE;
268
18.4E
    } else if (0 == upper_type_str.compare("AGG_STATE")) {
269
27.5k
        type = FieldType::OLAP_FIELD_TYPE_AGG_STATE;
270
18.4E
    } else {
271
18.4E
        LOG(WARNING) << "invalid type string. [type='" << type_str << "']";
272
18.4E
        type = FieldType::OLAP_FIELD_TYPE_UNKNOWN;
273
18.4E
    }
274
275
56.8M
    return type;
276
56.8M
}
277
278
56.2M
FieldAggregationMethod TabletColumn::get_aggregation_type_by_string(const std::string& str) {
279
56.2M
    std::string upper_str = str;
280
56.2M
    std::transform(str.begin(), str.end(), upper_str.begin(),
281
230M
                   [](auto c) { return std::toupper(c); });
282
56.2M
    FieldAggregationMethod aggregation_type;
283
284
56.2M
    if (0 == upper_str.compare("NONE")) {
285
55.2M
        aggregation_type = FieldAggregationMethod::OLAP_FIELD_AGGREGATION_NONE;
286
55.2M
    } else if (0 == upper_str.compare("SUM")) {
287
337k
        aggregation_type = FieldAggregationMethod::OLAP_FIELD_AGGREGATION_SUM;
288
670k
    } else if (0 == upper_str.compare("MIN")) {
289
18.6k
        aggregation_type = FieldAggregationMethod::OLAP_FIELD_AGGREGATION_MIN;
290
652k
    } else if (0 == upper_str.compare("MAX")) {
291
87.0k
        aggregation_type = FieldAggregationMethod::OLAP_FIELD_AGGREGATION_MAX;
292
622k
    } else if (0 == upper_str.compare("REPLACE")) {
293
622k
        aggregation_type = FieldAggregationMethod::OLAP_FIELD_AGGREGATION_REPLACE;
294
18.4E
    } else if (0 == upper_str.compare("REPLACE_IF_NOT_NULL")) {
295
148k
        aggregation_type = FieldAggregationMethod::OLAP_FIELD_AGGREGATION_REPLACE_IF_NOT_NULL;
296
18.4E
    } else if (0 == upper_str.compare("HLL_UNION")) {
297
22.6k
        aggregation_type = FieldAggregationMethod::OLAP_FIELD_AGGREGATION_HLL_UNION;
298
18.4E
    } else if (0 == upper_str.compare("BITMAP_UNION")) {
299
32.2k
        aggregation_type = FieldAggregationMethod::OLAP_FIELD_AGGREGATION_BITMAP_UNION;
300
18.4E
    } else if (0 == upper_str.compare("QUANTILE_UNION")) {
301
17.3k
        aggregation_type = FieldAggregationMethod::OLAP_FIELD_AGGREGATION_QUANTILE_UNION;
302
18.4E
    } else if (!upper_str.empty()) {
303
13.4k
        aggregation_type = FieldAggregationMethod::OLAP_FIELD_AGGREGATION_GENERIC;
304
18.4E
    } else {
305
18.4E
        aggregation_type = FieldAggregationMethod::OLAP_FIELD_AGGREGATION_UNKNOWN;
306
18.4E
    }
307
308
56.2M
    return aggregation_type;
309
56.2M
}
310
311
52.0M
std::string TabletColumn::get_string_by_field_type(FieldType type) {
312
52.0M
    switch (type) {
313
3.48M
    case FieldType::OLAP_FIELD_TYPE_TINYINT:
314
3.48M
        return "TINYINT";
315
316
0
    case FieldType::OLAP_FIELD_TYPE_UNSIGNED_TINYINT:
317
0
        return "UNSIGNED_TINYINT";
318
319
842k
    case FieldType::OLAP_FIELD_TYPE_SMALLINT:
320
842k
        return "SMALLINT";
321
322
0
    case FieldType::OLAP_FIELD_TYPE_UNSIGNED_SMALLINT:
323
0
        return "UNSIGNED_SMALLINT";
324
325
5.62M
    case FieldType::OLAP_FIELD_TYPE_INT:
326
5.62M
        return "INT";
327
328
0
    case FieldType::OLAP_FIELD_TYPE_UNSIGNED_INT:
329
0
        return "UNSIGNED_INT";
330
331
8.62M
    case FieldType::OLAP_FIELD_TYPE_BIGINT:
332
8.62M
        return "BIGINT";
333
334
963k
    case FieldType::OLAP_FIELD_TYPE_LARGEINT:
335
963k
        return "LARGEINT";
336
337
0
    case FieldType::OLAP_FIELD_TYPE_UNSIGNED_BIGINT:
338
0
        return "UNSIGNED_BIGINT";
339
340
32.2k
    case FieldType::OLAP_FIELD_TYPE_IPV4:
341
32.2k
        return "IPV4";
342
343
32.7k
    case FieldType::OLAP_FIELD_TYPE_IPV6:
344
32.7k
        return "IPV6";
345
346
688k
    case FieldType::OLAP_FIELD_TYPE_FLOAT:
347
688k
        return "FLOAT";
348
349
914k
    case FieldType::OLAP_FIELD_TYPE_DOUBLE:
350
914k
        return "DOUBLE";
351
352
0
    case FieldType::OLAP_FIELD_TYPE_DISCRETE_DOUBLE:
353
0
        return "DISCRETE_DOUBLE";
354
355
891k
    case FieldType::OLAP_FIELD_TYPE_CHAR:
356
891k
        return "CHAR";
357
358
14.3k
    case FieldType::OLAP_FIELD_TYPE_DATE:
359
14.3k
        return "DATE";
360
361
2.34M
    case FieldType::OLAP_FIELD_TYPE_DATEV2:
362
2.34M
        return "DATEV2";
363
364
18.5k
    case FieldType::OLAP_FIELD_TYPE_DATETIME:
365
18.5k
        return "DATETIME";
366
367
2.80M
    case FieldType::OLAP_FIELD_TYPE_DATETIMEV2:
368
2.80M
        return "DATETIMEV2";
369
370
180k
    case FieldType::OLAP_FIELD_TYPE_TIMESTAMPTZ:
371
180k
        return "TIMESTAMPTZ";
372
373
28.0k
    case FieldType::OLAP_FIELD_TYPE_DECIMAL:
374
28.0k
        return "DECIMAL";
375
376
577k
    case FieldType::OLAP_FIELD_TYPE_DECIMAL32:
377
577k
        return "DECIMAL32";
378
379
1.29M
    case FieldType::OLAP_FIELD_TYPE_DECIMAL64:
380
1.29M
        return "DECIMAL64";
381
382
914k
    case FieldType::OLAP_FIELD_TYPE_DECIMAL128I:
383
914k
        return "DECIMAL128I";
384
385
80.5k
    case FieldType::OLAP_FIELD_TYPE_DECIMAL256:
386
80.5k
        return "DECIMAL256";
387
388
14.7M
    case FieldType::OLAP_FIELD_TYPE_VARCHAR:
389
14.7M
        return "VARCHAR";
390
391
363k
    case FieldType::OLAP_FIELD_TYPE_JSONB:
392
363k
        return "JSONB";
393
394
122k
    case FieldType::OLAP_FIELD_TYPE_VARIANT:
395
122k
        return "VARIANT";
396
397
2.97M
    case FieldType::OLAP_FIELD_TYPE_STRING:
398
2.97M
        return "STRING";
399
400
619k
    case FieldType::OLAP_FIELD_TYPE_BOOL:
401
619k
        return "BOOLEAN";
402
403
62.2k
    case FieldType::OLAP_FIELD_TYPE_HLL:
404
62.2k
        return "HLL";
405
406
136k
    case FieldType::OLAP_FIELD_TYPE_STRUCT:
407
136k
        return "STRUCT";
408
409
1.69M
    case FieldType::OLAP_FIELD_TYPE_ARRAY:
410
1.69M
        return "ARRAY";
411
412
1.00M
    case FieldType::OLAP_FIELD_TYPE_MAP:
413
1.00M
        return "MAP";
414
415
83.1k
    case FieldType::OLAP_FIELD_TYPE_BITMAP:
416
83.1k
        return "OBJECT";
417
45.8k
    case FieldType::OLAP_FIELD_TYPE_QUANTILE_STATE:
418
45.8k
        return "QUANTILE_STATE";
419
16.0k
    case FieldType::OLAP_FIELD_TYPE_AGG_STATE:
420
16.0k
        return "AGG_STATE";
421
0
    default:
422
0
        return "UNKNOWN";
423
52.0M
    }
424
52.0M
}
425
426
203k
std::string TabletColumn::get_string_by_aggregation_type(FieldAggregationMethod type) {
427
203k
    switch (type) {
428
134k
    case FieldAggregationMethod::OLAP_FIELD_AGGREGATION_NONE:
429
134k
        return "NONE";
430
431
13.5k
    case FieldAggregationMethod::OLAP_FIELD_AGGREGATION_SUM:
432
13.5k
        return "SUM";
433
434
1.83k
    case FieldAggregationMethod::OLAP_FIELD_AGGREGATION_MIN:
435
1.83k
        return "MIN";
436
437
5.50k
    case FieldAggregationMethod::OLAP_FIELD_AGGREGATION_MAX:
438
5.50k
        return "MAX";
439
440
30.2k
    case FieldAggregationMethod::OLAP_FIELD_AGGREGATION_REPLACE:
441
30.2k
        return "REPLACE";
442
443
14.3k
    case FieldAggregationMethod::OLAP_FIELD_AGGREGATION_REPLACE_IF_NOT_NULL:
444
14.3k
        return "REPLACE_IF_NOT_NULL";
445
446
1.28k
    case FieldAggregationMethod::OLAP_FIELD_AGGREGATION_HLL_UNION:
447
1.28k
        return "HLL_UNION";
448
449
1.89k
    case FieldAggregationMethod::OLAP_FIELD_AGGREGATION_BITMAP_UNION:
450
1.89k
        return "BITMAP_UNION";
451
452
972
    case FieldAggregationMethod::OLAP_FIELD_AGGREGATION_QUANTILE_UNION:
453
972
        return "QUANTILE_UNION";
454
455
2
    default:
456
2
        return "UNKNOWN";
457
203k
    }
458
203k
}
459
460
26.2M
uint32_t TabletColumn::get_field_length_by_type(TPrimitiveType::type type, uint32_t string_length) {
461
26.2M
    switch (type) {
462
1.89M
    case TPrimitiveType::TINYINT:
463
2.13M
    case TPrimitiveType::BOOLEAN:
464
2.13M
        return 1;
465
296k
    case TPrimitiveType::SMALLINT:
466
296k
        return 2;
467
1.68M
    case TPrimitiveType::INT:
468
1.68M
        return 4;
469
4.97M
    case TPrimitiveType::BIGINT:
470
4.97M
        return 8;
471
299k
    case TPrimitiveType::LARGEINT:
472
299k
        return 16;
473
16.5k
    case TPrimitiveType::IPV4:
474
16.5k
        return 4;
475
16.7k
    case TPrimitiveType::IPV6:
476
16.7k
        return 16;
477
9.72k
    case TPrimitiveType::DATE:
478
9.72k
        return 3;
479
833k
    case TPrimitiveType::DATEV2:
480
833k
        return 4;
481
10.1k
    case TPrimitiveType::DATETIME:
482
10.1k
        return 8;
483
1.43M
    case TPrimitiveType::DATETIMEV2:
484
1.47M
    case TPrimitiveType::TIMESTAMPTZ:
485
1.47M
        return 8;
486
264k
    case TPrimitiveType::FLOAT:
487
264k
        return 4;
488
451k
    case TPrimitiveType::DOUBLE:
489
451k
        return 8;
490
4.66k
    case TPrimitiveType::QUANTILE_STATE:
491
18.3k
    case TPrimitiveType::BITMAP:
492
18.3k
        return 16;
493
400k
    case TPrimitiveType::CHAR:
494
400k
        return string_length;
495
9.12M
    case TPrimitiveType::VARCHAR:
496
9.12M
    case TPrimitiveType::HLL:
497
9.13M
    case TPrimitiveType::AGG_STATE:
498
9.13M
        return string_length + sizeof(OLAP_VARCHAR_MAX_LENGTH);
499
1.43M
    case TPrimitiveType::STRING:
500
1.46M
    case TPrimitiveType::VARIANT:
501
1.46M
        return string_length + sizeof(OLAP_STRING_MAX_LENGTH);
502
98.4k
    case TPrimitiveType::JSONB:
503
98.4k
        return string_length + sizeof(OLAP_JSONB_MAX_LENGTH);
504
43.6k
    case TPrimitiveType::STRUCT:
505
        // Note that(xy): this is the length of struct type itself,
506
        // the length of its subtypes are not included.
507
43.6k
        return OLAP_STRUCT_MAX_LENGTH;
508
789k
    case TPrimitiveType::ARRAY:
509
789k
        return OLAP_ARRAY_MAX_LENGTH;
510
746k
    case TPrimitiveType::MAP:
511
746k
        return OLAP_MAP_MAX_LENGTH;
512
170k
    case TPrimitiveType::DECIMAL32:
513
170k
        return 4;
514
615k
    case TPrimitiveType::DECIMAL64:
515
615k
        return 8;
516
314k
    case TPrimitiveType::DECIMAL128I:
517
314k
        return 16;
518
24.6k
    case TPrimitiveType::DECIMAL256:
519
24.6k
        return 32;
520
24.9k
    case TPrimitiveType::DECIMALV2:
521
24.9k
        return 12; // use 12 bytes in olap engine.
522
0
    default:
523
0
        LOG(WARNING) << "unknown field type. [type=" << type << "]";
524
0
        return 0;
525
26.2M
    }
526
26.2M
}
527
528
9
bool TabletColumn::has_char_type() const {
529
9
    switch (_type) {
530
3
    case FieldType::OLAP_FIELD_TYPE_CHAR: {
531
3
        return true;
532
0
    }
533
4
    case FieldType::OLAP_FIELD_TYPE_ARRAY:
534
4
    case FieldType::OLAP_FIELD_TYPE_MAP:
535
4
    case FieldType::OLAP_FIELD_TYPE_STRUCT: {
536
4
        return std::any_of(_sub_columns.begin(), _sub_columns.end(),
537
4
                           [&](const auto& sub) -> bool { return sub->has_char_type(); });
538
4
    }
539
2
    default:
540
2
        return false;
541
9
    }
542
9
}
543
544
34.1M
TabletColumn::TabletColumn() : _aggregation(FieldAggregationMethod::OLAP_FIELD_AGGREGATION_NONE) {}
545
546
21
TabletColumn::TabletColumn(FieldAggregationMethod agg, FieldType type) {
547
21
    _aggregation = agg;
548
21
    _type = type;
549
21
}
550
551
20
TabletColumn::TabletColumn(FieldAggregationMethod agg, FieldType filed_type, bool is_nullable) {
552
20
    _aggregation = agg;
553
20
    _type = filed_type;
554
20
    _length = cast_set<int32_t>(field_type_size(filed_type));
555
20
    _is_nullable = is_nullable;
556
20
}
557
558
TabletColumn::TabletColumn(FieldAggregationMethod agg, FieldType filed_type, bool is_nullable,
559
112k
                           int32_t unique_id, size_t length) {
560
112k
    _aggregation = agg;
561
112k
    _type = filed_type;
562
112k
    _is_nullable = is_nullable;
563
112k
    _unique_id = unique_id;
564
112k
    _length = cast_set<int32_t>(length);
565
112k
}
566
567
5.61k
TabletColumn::TabletColumn(const ColumnPB& column) {
568
5.61k
    init_from_pb(column);
569
5.61k
}
570
571
22.8M
TabletColumn::TabletColumn(const TColumn& column) {
572
22.8M
    init_from_thrift(column);
573
22.8M
}
574
575
23.7M
void TabletColumn::init_from_thrift(const TColumn& tcolumn) {
576
23.7M
    ColumnPB column_pb;
577
23.7M
    TabletMeta::init_column_from_tcolumn(tcolumn.col_unique_id, tcolumn, &column_pb);
578
23.7M
    init_from_pb(column_pb);
579
23.7M
}
580
581
56.1M
void TabletColumn::init_from_pb(const ColumnPB& column) {
582
56.1M
    _unique_id = column.unique_id();
583
56.1M
    _col_name = column.name();
584
56.1M
    _col_name_lower_case = to_lower(_col_name);
585
56.1M
    _type = TabletColumn::get_field_type_by_string(column.type());
586
56.1M
    _is_key = column.is_key();
587
56.1M
    _is_nullable = column.is_nullable();
588
56.1M
    _is_auto_increment = column.is_auto_increment();
589
56.1M
    if (column.has_is_on_update_current_timestamp()) {
590
51.7M
        _is_on_update_current_timestamp = column.is_on_update_current_timestamp();
591
51.7M
    }
592
593
56.1M
    _has_default_value = column.has_default_value();
594
56.1M
    if (_has_default_value) {
595
5.86M
        _default_value = column.default_value();
596
5.86M
    }
597
598
56.4M
    if (column.has_precision()) {
599
56.4M
        _is_decimal = true;
600
56.4M
        _precision = column.precision();
601
18.4E
    } else {
602
18.4E
        _is_decimal = false;
603
18.4E
    }
604
56.4M
    if (column.has_frac()) {
605
56.4M
        _frac = column.frac();
606
56.4M
    }
607
56.1M
    _length = column.length();
608
56.1M
    _index_length = column.index_length();
609
56.1M
    if (column.has_is_bf_column()) {
610
4.78M
        _is_bf_column = column.is_bf_column();
611
51.3M
    } else {
612
51.3M
        _is_bf_column = false;
613
51.3M
    }
614
56.5M
    if (column.has_aggregation()) {
615
56.5M
        _aggregation = get_aggregation_type_by_string(column.aggregation());
616
56.5M
        _aggregation_name = column.aggregation();
617
56.5M
    }
618
619
56.1M
    if (_type == FieldType::OLAP_FIELD_TYPE_AGG_STATE) {
620
14.8k
        _result_is_nullable = column.result_is_nullable();
621
14.8k
        _be_exec_version = column.be_exec_version();
622
14.8k
    }
623
624
56.5M
    if (column.has_visible()) {
625
56.5M
        _visible = column.visible();
626
56.5M
    }
627
56.1M
    if (_type == FieldType::OLAP_FIELD_TYPE_ARRAY) {
628
1.69M
        CHECK(column.children_columns_size() == 1)
629
139
                << "ARRAY type should has 1 children types, but got "
630
139
                << column.children_columns_size();
631
1.69M
    }
632
56.1M
    if (_type == FieldType::OLAP_FIELD_TYPE_MAP) {
633
18.4E
        DCHECK(column.children_columns_size() == 2)
634
18.4E
                << "MAP type should has 2 children types, but got "
635
18.4E
                << column.children_columns_size();
636
1.51M
        if (UNLIKELY(column.children_columns_size() != 2)) {
637
0
            LOG(WARNING) << "MAP type should has 2 children types, but got "
638
0
                         << column.children_columns_size();
639
0
        }
640
1.51M
    }
641
61.5M
    for (int i = 0; i < column.children_columns_size(); i++) {
642
5.44M
        TabletColumn child_column;
643
5.44M
        child_column.init_from_pb(column.children_columns(i));
644
5.44M
        add_sub_column(child_column);
645
5.44M
    }
646
56.1M
    if (column.has_column_path_info()) {
647
38.6k
        _column_path = std::make_shared<PathInData>();
648
38.6k
        _column_path->from_protobuf(column.column_path_info());
649
38.6k
        _parent_col_unique_id = column.column_path_info().parrent_column_unique_id();
650
38.6k
    }
651
56.1M
    if (is_variant_type() && !column.has_column_path_info()) {
652
        // set path info for variant root column, to prevent from missing
653
40.4k
        _column_path = std::make_shared<PathInData>(_col_name_lower_case);
654
        // _parent_col_unique_id = _unique_id;
655
40.4k
    }
656
56.1M
    if (column.has_variant_max_subcolumns_count()) {
657
51.6M
        _variant.max_subcolumns_count = column.variant_max_subcolumns_count();
658
51.6M
    }
659
56.1M
    if (column.has_variant_enable_typed_paths_to_sparse()) {
660
51.6M
        _variant.enable_typed_paths_to_sparse = column.variant_enable_typed_paths_to_sparse();
661
51.6M
    }
662
56.1M
    if (column.has_variant_max_sparse_column_statistics_size()) {
663
51.7M
        _variant.max_sparse_column_statistics_size =
664
51.7M
                column.variant_max_sparse_column_statistics_size();
665
51.7M
    }
666
56.1M
    if (column.has_variant_sparse_hash_shard_count()) {
667
49.3M
        _variant.sparse_hash_shard_count = column.variant_sparse_hash_shard_count();
668
49.3M
    }
669
56.1M
    if (column.has_variant_enable_doc_mode()) {
670
51.8M
        _variant.enable_doc_mode = column.variant_enable_doc_mode();
671
51.8M
    }
672
56.1M
    if (column.has_variant_doc_materialization_min_rows()) {
673
49.3M
        _variant.doc_materialization_min_rows = column.variant_doc_materialization_min_rows();
674
49.3M
    }
675
56.1M
    if (column.has_variant_doc_hash_shard_count()) {
676
49.3M
        _variant.doc_hash_shard_count = column.variant_doc_hash_shard_count();
677
49.3M
    }
678
56.1M
    if (column.has_variant_enable_nested_group()) {
679
49.3M
        _variant.enable_nested_group = column.variant_enable_nested_group();
680
49.3M
    }
681
56.1M
    if (column.has_pattern_type()) {
682
25.5M
        _pattern_type = column.pattern_type();
683
25.5M
    }
684
56.1M
}
685
686
TabletColumn TabletColumn::create_materialized_variant_column(const std::string& root,
687
                                                              const std::vector<std::string>& paths,
688
                                                              int32_t parent_unique_id,
689
                                                              int32_t max_subcolumns_count,
690
7.15k
                                                              bool enable_doc_mode) {
691
7.15k
    TabletColumn subcol;
692
7.15k
    subcol.set_type(FieldType::OLAP_FIELD_TYPE_VARIANT);
693
7.15k
    subcol.set_is_nullable(true);
694
7.15k
    subcol.set_unique_id(-1);
695
7.15k
    subcol.set_parent_unique_id(parent_unique_id);
696
7.15k
    PathInData path(root, paths);
697
7.15k
    subcol.set_path_info(path);
698
7.15k
    subcol.set_name(path.get_path());
699
7.15k
    subcol.set_variant_max_subcolumns_count(max_subcolumns_count);
700
7.15k
    subcol.set_variant_enable_doc_mode(enable_doc_mode);
701
7.15k
    return subcol;
702
7.15k
}
703
704
51.2M
void TabletColumn::to_schema_pb(ColumnPB* column) const {
705
51.2M
    column->set_unique_id(_unique_id);
706
51.2M
    column->set_name(_col_name);
707
51.2M
    column->set_type(get_string_by_field_type(_type));
708
51.2M
    column->set_is_key(_is_key);
709
51.2M
    column->set_is_nullable(_is_nullable);
710
51.2M
    column->set_is_auto_increment(_is_auto_increment);
711
51.2M
    column->set_is_on_update_current_timestamp(_is_on_update_current_timestamp);
712
51.2M
    if (_has_default_value) {
713
7.56M
        column->set_default_value(_default_value);
714
7.56M
    }
715
51.7M
    if (_is_decimal) {
716
51.7M
        column->set_precision(_precision);
717
51.7M
        column->set_frac(_frac);
718
51.7M
    }
719
51.2M
    column->set_length(_length);
720
51.2M
    column->set_index_length(_index_length);
721
51.2M
    if (_is_bf_column) {
722
123k
        column->set_is_bf_column(_is_bf_column);
723
123k
    }
724
51.9M
    if (!_aggregation_name.empty()) {
725
51.9M
        column->set_aggregation(_aggregation_name);
726
51.9M
    }
727
51.2M
    column->set_result_is_nullable(_result_is_nullable);
728
51.2M
    column->set_be_exec_version(_be_exec_version);
729
51.2M
    column->set_visible(_visible);
730
731
51.2M
    if (_type == FieldType::OLAP_FIELD_TYPE_ARRAY) {
732
18.4E
        CHECK(_sub_columns.size() == 1)
733
18.4E
                << "ARRAY type should has 1 children types, but got " << _sub_columns.size();
734
1.68M
    }
735
51.2M
    if (_type == FieldType::OLAP_FIELD_TYPE_MAP) {
736
18.4E
        DCHECK(_sub_columns.size() == 2)
737
18.4E
                << "MAP type should has 2 children types, but got " << _sub_columns.size();
738
994k
        if (UNLIKELY(_sub_columns.size() != 2)) {
739
0
            LOG(WARNING) << "MAP type should has 2 children types, but got " << _sub_columns.size();
740
0
        }
741
994k
    }
742
743
55.8M
    for (size_t i = 0; i < _sub_columns.size(); i++) {
744
4.61M
        ColumnPB* child = column->add_children_columns();
745
4.61M
        _sub_columns[i]->to_schema_pb(child);
746
4.61M
    }
747
748
    // set parts info
749
51.2M
    if (has_path_info()) {
750
        // CHECK_GT(_parent_col_unique_id, 0);
751
137k
        _column_path->to_protobuf(column->mutable_column_path_info(), _parent_col_unique_id);
752
        // Update unstable information for variant columns. Some of the fields in the tablet schema
753
        // are irrelevant for variant sub-columns, but retaining them may lead to an excessive growth
754
        // in the number of tablet schema cache entries.
755
137k
        if (_type == FieldType::OLAP_FIELD_TYPE_STRING) {
756
2.11k
            column->set_length(INT_MAX);
757
2.11k
        }
758
137k
        column->set_index_length(0);
759
137k
    }
760
51.2M
    column->set_variant_max_subcolumns_count(_variant.max_subcolumns_count);
761
51.2M
    column->set_pattern_type(_pattern_type);
762
51.2M
    column->set_variant_enable_typed_paths_to_sparse(_variant.enable_typed_paths_to_sparse);
763
51.2M
    column->set_variant_max_sparse_column_statistics_size(
764
51.2M
            _variant.max_sparse_column_statistics_size);
765
51.2M
    column->set_variant_sparse_hash_shard_count(_variant.sparse_hash_shard_count);
766
51.2M
    column->set_variant_enable_doc_mode(_variant.enable_doc_mode);
767
51.2M
    column->set_variant_doc_materialization_min_rows(_variant.doc_materialization_min_rows);
768
51.2M
    column->set_variant_doc_hash_shard_count(_variant.doc_hash_shard_count);
769
51.2M
    column->set_variant_enable_nested_group(_variant.enable_nested_group);
770
51.2M
}
771
772
5.48M
void TabletColumn::add_sub_column(TabletColumn& sub_column) {
773
5.48M
    _sub_columns.push_back(std::make_shared<TabletColumn>(sub_column));
774
5.48M
    sub_column._parent_col_unique_id = this->_unique_id;
775
5.48M
    _sub_column_count += 1;
776
5.48M
}
777
778
40.2M
bool TabletColumn::is_row_store_column() const {
779
40.2M
    return _col_name == BeConsts::ROW_STORE_COL;
780
40.2M
}
781
782
AggregateFunctionPtr TabletColumn::get_aggregate_function_union(DataTypePtr type,
783
1.39k
                                                                int current_be_exec_version) const {
784
1.39k
    const auto* state_type = assert_cast<const DataTypeAggState*>(type.get());
785
1.39k
    BeExecVersionManager::check_function_compatibility(
786
1.39k
            current_be_exec_version, _be_exec_version,
787
1.39k
            state_type->get_nested_function()->get_name());
788
1.39k
    return AggregateStateUnion::create(state_type->get_nested_function(), {type}, type);
789
1.39k
}
790
791
AggregateFunctionPtr TabletColumn::get_aggregate_function(std::string suffix,
792
69.4k
                                                          int current_be_exec_version) const {
793
69.4k
    AggregateFunctionPtr function = nullptr;
794
795
69.4k
    auto type = DataTypeFactory::instance().create_data_type(*this);
796
69.4k
    if (type && type->get_primitive_type() == PrimitiveType::TYPE_AGG_STATE) {
797
1.39k
        function = get_aggregate_function_union(type, current_be_exec_version);
798
68.0k
    } else {
799
68.0k
        std::string origin_name = TabletColumn::get_string_by_aggregation_type(_aggregation);
800
68.0k
        std::string agg_name = origin_name + suffix;
801
68.0k
        std::transform(agg_name.begin(), agg_name.end(), agg_name.begin(),
802
951k
                       [](unsigned char c) { return std::tolower(c); });
803
68.0k
        function = AggregateFunctionSimpleFactory::instance().get(
804
68.0k
                agg_name, {type}, type, type->is_nullable(),
805
68.0k
                BeExecVersionManager::get_newest_version());
806
68.0k
        if (!function) {
807
0
            LOG(WARNING) << "get column aggregate function failed, aggregation_name=" << origin_name
808
0
                         << ", column_type=" << type->get_name();
809
0
        }
810
68.0k
    }
811
69.4k
    if (function) {
812
69.4k
        function->set_version(_be_exec_version);
813
69.4k
        return function;
814
69.4k
    }
815
18.4E
    return nullptr;
816
69.4k
}
817
818
133k
void TabletColumn::set_path_info(const PathInData& path) {
819
133k
    _column_path = std::make_shared<PathInData>(path);
820
133k
}
821
822
13.2k
DataTypePtr TabletColumn::get_vec_type() const {
823
13.2k
    return DataTypeFactory::instance().create_data_type(*this);
824
13.2k
}
825
826
// escape '.' and '_'
827
56.8M
std::string escape_for_path_name(const std::string& s) {
828
56.8M
    std::string res;
829
56.8M
    const char* pos = s.data();
830
56.8M
    const char* end = pos + s.size();
831
57.3M
    while (pos != end) {
832
553k
        unsigned char c = *pos;
833
553k
        if (c == '.' || c == '_') {
834
79.6k
            res += '%';
835
79.6k
            res += hex_digit_uppercase(c / 16);
836
79.6k
            res += hex_digit_uppercase(c % 16);
837
473k
        } else {
838
473k
            res += c;
839
473k
        }
840
553k
        ++pos;
841
553k
    }
842
56.8M
    return res;
843
56.8M
}
844
845
12.8k
void TabletIndex::set_escaped_escaped_index_suffix_path(const std::string& path_name) {
846
12.8k
    std::string escaped_path = escape_for_path_name(path_name);
847
12.8k
    _escaped_index_suffix_path = escaped_path;
848
12.8k
}
849
850
void TabletIndex::init_from_thrift(const TOlapTableIndex& index,
851
368k
                                   const TabletSchema& tablet_schema) {
852
368k
    _index_id = index.index_id;
853
368k
    _index_name = index.index_name;
854
    // init col_unique_id in index at be side, since col_unique_id may be -1 at fe side
855
    // get column unique id by name
856
368k
    std::vector<int32_t> col_unique_ids(index.columns.size());
857
737k
    for (size_t i = 0; i < index.columns.size(); i++) {
858
369k
        auto column_idx = tablet_schema.field_index(index.columns[i]);
859
369k
        if (column_idx >= 0) {
860
368k
            col_unique_ids[i] = tablet_schema.column(column_idx).unique_id();
861
368k
        } else {
862
            // if column unique id not found by column name, find by column unique id
863
            // column unique id can not found means this column is a new column added by light schema change
864
189
            if (index.__isset.column_unique_ids && !index.column_unique_ids.empty() &&
865
189
                tablet_schema.has_column_unique_id(index.column_unique_ids[i])) {
866
145
                col_unique_ids[i] = index.column_unique_ids[i];
867
145
            } else {
868
44
                col_unique_ids[i] = -1;
869
44
            }
870
189
        }
871
369k
    }
872
368k
    _col_unique_ids = std::move(col_unique_ids);
873
874
368k
    switch (index.index_type) {
875
0
    case TIndexType::BITMAP:
876
0
        _index_type = IndexType::BITMAP;
877
0
        break;
878
321k
    case TIndexType::INVERTED:
879
321k
        _index_type = IndexType::INVERTED;
880
321k
        break;
881
422
    case TIndexType::ANN:
882
422
        _index_type = IndexType::ANN;
883
422
        break;
884
0
    case TIndexType::BLOOMFILTER:
885
0
        _index_type = IndexType::BLOOMFILTER;
886
0
        break;
887
46.6k
    case TIndexType::NGRAM_BF:
888
46.6k
        _index_type = IndexType::NGRAM_BF;
889
46.6k
        break;
890
368k
    }
891
368k
    if (index.__isset.properties) {
892
368k
        for (auto kv : index.properties) {
893
316k
            _properties[kv.first] = kv.second;
894
316k
        }
895
368k
    }
896
368k
}
897
898
void TabletIndex::init_from_thrift(const TOlapTableIndex& index,
899
15.0k
                                   const std::vector<int32_t>& column_uids) {
900
15.0k
    _index_id = index.index_id;
901
15.0k
    _index_name = index.index_name;
902
15.0k
    _col_unique_ids = column_uids;
903
904
15.0k
    switch (index.index_type) {
905
0
    case TIndexType::BITMAP:
906
0
        _index_type = IndexType::BITMAP;
907
0
        break;
908
14.3k
    case TIndexType::INVERTED:
909
14.3k
        _index_type = IndexType::INVERTED;
910
14.3k
        break;
911
132
    case TIndexType::ANN:
912
132
        _index_type = IndexType::ANN;
913
132
        break;
914
0
    case TIndexType::BLOOMFILTER:
915
0
        _index_type = IndexType::BLOOMFILTER;
916
0
        break;
917
580
    case TIndexType::NGRAM_BF:
918
580
        _index_type = IndexType::NGRAM_BF;
919
580
        break;
920
15.0k
    }
921
15.0k
    if (index.__isset.properties) {
922
17.1k
        for (auto kv : index.properties) {
923
17.1k
            _properties[kv.first] = kv.second;
924
17.1k
        }
925
15.0k
    }
926
15.0k
}
927
928
926k
void TabletIndex::init_from_pb(const TabletIndexPB& index) {
929
926k
    _index_id = index.index_id();
930
926k
    _index_name = index.index_name();
931
926k
    _col_unique_ids.clear();
932
927k
    for (auto col_unique_id : index.col_unique_id()) {
933
927k
        _col_unique_ids.push_back(col_unique_id);
934
927k
    }
935
926k
    _index_type = index.index_type();
936
926k
    for (const auto& kv : index.properties()) {
937
622k
        _properties[kv.first] = kv.second;
938
622k
    }
939
926k
    _escaped_index_suffix_path = index.index_suffix_name();
940
926k
}
941
942
1.80M
void TabletIndex::to_schema_pb(TabletIndexPB* index) const {
943
1.80M
    index->set_index_id(_index_id);
944
1.80M
    index->set_index_name(_index_name);
945
1.80M
    index->clear_col_unique_id();
946
1.81M
    for (auto col_unique_id : _col_unique_ids) {
947
1.81M
        index->add_col_unique_id(col_unique_id);
948
1.81M
    }
949
1.80M
    index->set_index_type(_index_type);
950
1.80M
    for (const auto& kv : _properties) {
951
1.44M
        DBUG_EXECUTE_IF("tablet_schema.to_schema_pb", {
952
1.44M
            if (kv.first == INVERTED_INDEX_PARSER_LOWERCASE_KEY) {
953
1.44M
                continue;
954
1.44M
            }
955
1.44M
        })
956
1.44M
        (*index->mutable_properties())[kv.first] = kv.second;
957
1.44M
    }
958
1.80M
    index->set_index_suffix_name(_escaped_index_suffix_path);
959
960
1.80M
    DBUG_EXECUTE_IF("tablet_schema.to_schema_pb", { return; })
961
962
    // Only add lower_case=true default for built-in analyzers/parsers, NOT for custom analyzers
963
    // Custom analyzer: lower_case is determined by analyzer's internal token filter
964
1.80M
    if (!_properties.empty() && !_properties.contains(INVERTED_INDEX_PARSER_LOWERCASE_KEY)) {
965
52.5k
        bool has_parser = _properties.contains(INVERTED_INDEX_PARSER_KEY) ||
966
52.5k
                          _properties.contains(INVERTED_INDEX_PARSER_KEY_ALIAS);
967
52.5k
        std::string analyzer_name = get_analyzer_name_from_properties(_properties);
968
52.5k
        bool is_builtin = analyzer_name.empty() ||
969
52.5k
                          segment_v2::inverted_index::InvertedIndexAnalyzer::is_builtin_analyzer(
970
3.22k
                                  analyzer_name);
971
52.5k
        if (has_parser || is_builtin) {
972
49.2k
            (*index->mutable_properties())[INVERTED_INDEX_PARSER_LOWERCASE_KEY] =
973
49.2k
                    INVERTED_INDEX_PARSER_TRUE;
974
49.2k
        }
975
52.5k
    }
976
1.80M
}
977
978
3.29M
TabletSchema::TabletSchema() = default;
979
980
3.26M
TabletSchema::~TabletSchema() {}
981
982
2.14M
int64_t TabletSchema::get_metadata_size() const {
983
2.14M
    return sizeof(TabletSchema);
984
2.14M
}
985
986
22.9M
void TabletSchema::append_column(TabletColumn column, ColumnType col_type) {
987
22.9M
    if (column.is_key()) {
988
7.36M
        _num_key_columns++;
989
7.36M
    }
990
22.9M
    if (column.is_nullable()) {
991
12.2M
        _num_null_columns++;
992
12.2M
    }
993
22.9M
    if (column.is_variant_type()) {
994
26.2k
        ++_num_variant_columns;
995
26.2k
        if (!column.has_path_info()) {
996
28
            const std::string& col_name = column.name_lower_case();
997
28
            PathInData path(col_name);
998
28
            column.set_path_info(path);
999
28
        }
1000
26.2k
    }
1001
22.9M
    if (UNLIKELY(column.name() == DELETE_SIGN)) {
1002
935k
        _delete_sign_idx = _num_columns;
1003
22.0M
    } else if (UNLIKELY(column.name() == SEQUENCE_COL)) {
1004
8.90k
        _sequence_col_idx = _num_columns;
1005
22.0M
    } else if (UNLIKELY(column.name() == VERSION_COL)) {
1006
935k
        _version_col_idx = _num_columns;
1007
21.1M
    } else if (UNLIKELY(column.name() == SKIP_BITMAP_COL)) {
1008
1.00k
        _skip_bitmap_col_idx = _num_columns;
1009
21.1M
    } else if (UNLIKELY(column.name() == COMMIT_TSO_COL)) {
1010
3
        _commit_tso_col_idx = _num_columns;
1011
21.1M
    } else if (UNLIKELY(column.name() == BINLOG_TSO_COL)) {
1012
0
        _binlog_tso_col_idx = _num_columns;
1013
21.1M
    } else if (UNLIKELY(column.name() == BINLOG_LSN_COL)) {
1014
0
        _binlog_lsn_col_idx = _num_columns;
1015
21.1M
    } else if (UNLIKELY(column.name() == BINLOG_OP_COL)) {
1016
0
        _binlog_op_col_idx = _num_columns;
1017
21.1M
    } else if (UNLIKELY(column.name().starts_with(BeConsts::VIRTUAL_COLUMN_PREFIX))) {
1018
413
        _vir_col_idx_to_unique_id[_num_columns] = column.unique_id();
1019
413
    }
1020
22.9M
    _field_uniqueid_to_index[column.unique_id()] = _num_columns;
1021
22.9M
    _cols.push_back(std::make_shared<TabletColumn>(std::move(column)));
1022
    // The dropped column may have same name with exsiting column, so that
1023
    // not add to name to index map, only for uid to index map
1024
23.0M
    if (col_type == ColumnType::VARIANT || _cols.back()->is_variant_type() ||
1025
23.0M
        _cols.back()->is_extracted_column()) {
1026
30.1k
        _field_name_to_index.emplace(StringRef(_cols.back()->name()), _num_columns);
1027
30.1k
        _field_path_to_index[_cols.back()->path_info_ptr().get()] = _num_columns;
1028
23.0M
    } else if (col_type == ColumnType::NORMAL) {
1029
23.0M
        _field_name_to_index.emplace(StringRef(_cols.back()->name()), _num_columns);
1030
23.0M
    }
1031
22.9M
    _num_columns++;
1032
22.9M
    _num_virtual_columns = _vir_col_idx_to_unique_id.size();
1033
    // generate column index mapping for seq map
1034
22.9M
    if (_seq_col_uid_to_value_cols_uid.contains(column.unique_id())) {
1035
42
        const auto seq_idx = _field_uniqueid_to_index[column.unique_id()];
1036
42
        if (!_seq_col_idx_to_value_cols_idx.contains(seq_idx)) {
1037
38
            _seq_col_idx_to_value_cols_idx[seq_idx] = {};
1038
38
        }
1039
42
    }
1040
22.9M
    if (_value_col_uid_to_seq_col_uid.contains(column.unique_id())) {
1041
102
        const auto seq_uid = _value_col_uid_to_seq_col_uid[column.unique_id()];
1042
102
        if (_field_uniqueid_to_index.contains(seq_uid)) {
1043
48
            bool all_uid_index_found = true;
1044
48
            std::vector<int32_t> value_cols_index;
1045
68
            for (const auto value_col_uid : _seq_col_uid_to_value_cols_uid[seq_uid]) {
1046
68
                if (!_field_uniqueid_to_index.contains(value_col_uid)) {
1047
6
                    all_uid_index_found = false;
1048
6
                    break;
1049
6
                }
1050
62
                value_cols_index.push_back(_field_uniqueid_to_index[value_col_uid]);
1051
62
            }
1052
48
            if (all_uid_index_found) {
1053
42
                const auto seq_idx = _field_uniqueid_to_index[seq_uid];
1054
60
                for (const auto col_idx : value_cols_index) {
1055
60
                    _seq_col_idx_to_value_cols_idx[seq_idx].push_back(col_idx);
1056
60
                    _value_col_idx_to_seq_col_idx[col_idx] = seq_idx;
1057
60
                }
1058
42
                _value_col_idx_to_seq_col_idx[seq_idx] = seq_idx;
1059
42
            }
1060
48
        }
1061
102
    }
1062
22.9M
}
1063
1064
1.50k
void TabletSchema::append_index(TabletIndex&& index) {
1065
1.50k
    size_t index_pos = _indexes.size();
1066
1.50k
    _indexes.push_back(std::make_shared<TabletIndex>(index));
1067
1.51k
    for (int32_t id : _indexes.back()->col_unique_ids()) {
1068
1.51k
        if (auto field_pattern = _indexes.back()->field_pattern(); !field_pattern.empty()) {
1069
9
            auto& pattern_to_index_map = _index_by_unique_id_with_pattern[id];
1070
9
            pattern_to_index_map[field_pattern].emplace_back(_indexes.back());
1071
1.50k
        } else {
1072
1.50k
            IndexKey key = std::make_tuple(_indexes.back()->index_type(), id,
1073
1.50k
                                           _indexes.back()->get_index_suffix());
1074
1.50k
            _col_id_suffix_to_index[key].push_back(index_pos);
1075
1.50k
        }
1076
1.51k
    }
1077
1.50k
}
1078
1079
0
void TabletSchema::replace_column(size_t pos, TabletColumn new_col) {
1080
0
    CHECK_LT(pos, num_columns()) << " outof range";
1081
0
    _cols[pos] = std::make_shared<TabletColumn>(std::move(new_col));
1082
0
}
1083
1084
812
void TabletSchema::clear_index() {
1085
812
    _indexes.clear();
1086
812
    _col_id_suffix_to_index.clear();
1087
812
    _index_by_unique_id_with_pattern.clear();
1088
812
}
1089
1090
9
void TabletSchema::remove_index(int64_t index_id) {
1091
9
    std::vector<TabletIndexPtr> new_indexes;
1092
13
    for (auto& index : _indexes) {
1093
13
        if (index->index_id() != index_id) {
1094
4
            new_indexes.emplace_back(std::move(index));
1095
4
        }
1096
13
    }
1097
9
    _indexes = std::move(new_indexes);
1098
9
    _col_id_suffix_to_index.clear();
1099
9
    _index_by_unique_id_with_pattern.clear();
1100
13
    for (size_t new_pos = 0; new_pos < _indexes.size(); ++new_pos) {
1101
4
        const auto& index = _indexes[new_pos];
1102
4
        for (int32_t col_uid : index->col_unique_ids()) {
1103
4
            if (auto field_pattern = index->field_pattern(); !field_pattern.empty()) {
1104
0
                auto& pattern_to_index_map = _index_by_unique_id_with_pattern[col_uid];
1105
0
                pattern_to_index_map[field_pattern].emplace_back(index);
1106
4
            } else {
1107
4
                IndexKey key = std::make_tuple(_indexes.back()->index_type(), col_uid,
1108
4
                                               _indexes.back()->get_index_suffix());
1109
4
                _col_id_suffix_to_index[key].push_back(new_pos);
1110
4
            }
1111
4
        }
1112
4
    }
1113
9
}
1114
1115
1.47M
void TabletSchema::clear_columns() {
1116
1.47M
    _field_path_to_index.clear();
1117
1.47M
    _field_name_to_index.clear();
1118
1.47M
    _field_uniqueid_to_index.clear();
1119
1.47M
    _num_columns = 0;
1120
1.47M
    _num_variant_columns = 0;
1121
1.47M
    _num_null_columns = 0;
1122
1.47M
    _num_key_columns = 0;
1123
1.47M
    _seq_col_idx_to_value_cols_idx.clear();
1124
1.47M
    _value_col_idx_to_seq_col_idx.clear();
1125
1.47M
    _cols.clear();
1126
1.47M
}
1127
1128
void TabletSchema::init_from_pb(const TabletSchemaPB& schema, bool ignore_extracted_columns,
1129
2.03M
                                bool reuse_cache_column) {
1130
2.03M
    _keys_type = schema.keys_type();
1131
2.03M
    _num_columns = 0;
1132
2.03M
    _num_variant_columns = 0;
1133
2.03M
    _num_key_columns = 0;
1134
2.03M
    _num_null_columns = 0;
1135
2.03M
    _cols.clear();
1136
2.03M
    _indexes.clear();
1137
2.03M
    _index_by_unique_id_with_pattern.clear();
1138
2.03M
    _col_id_suffix_to_index.clear();
1139
2.03M
    _field_name_to_index.clear();
1140
2.03M
    _field_uniqueid_to_index.clear();
1141
2.03M
    _cluster_key_uids.clear();
1142
2.03M
    for (const auto& i : schema.cluster_key_uids()) {
1143
56.2k
        _cluster_key_uids.push_back(i);
1144
56.2k
    }
1145
27.3M
    for (auto& column_pb : schema.column()) {
1146
27.3M
        TabletColumnPtr column;
1147
27.3M
        if (reuse_cache_column) {
1148
515k
            auto pair = TabletColumnObjectPool::instance()->insert(
1149
515k
                    deterministic_string_serialize(column_pb));
1150
515k
            column = pair.second;
1151
            // Release the handle quickly, because we use shared ptr to manage column.
1152
            // It often core during tablet schema copy to another schema because handle's
1153
            // reference count should be managed mannually.
1154
515k
            TabletColumnObjectPool::instance()->release(pair.first);
1155
26.8M
        } else {
1156
26.8M
            column = std::make_shared<TabletColumn>();
1157
26.8M
            column->init_from_pb(column_pb);
1158
26.8M
        }
1159
27.3M
        if (ignore_extracted_columns && column->is_extracted_column()) {
1160
0
            continue;
1161
0
        }
1162
27.3M
        if (column->is_key()) {
1163
8.09M
            _num_key_columns++;
1164
8.09M
        }
1165
27.3M
        if (column->is_nullable()) {
1166
14.8M
            _num_null_columns++;
1167
14.8M
        }
1168
27.3M
        if (column->is_variant_type()) {
1169
44.1k
            ++_num_variant_columns;
1170
44.1k
        }
1171
1172
27.3M
        _cols.emplace_back(std::move(column));
1173
27.3M
        if (!_cols.back()->is_extracted_column()) {
1174
27.3M
            _field_name_to_index.emplace(StringRef(_cols.back()->name()), _num_columns);
1175
27.3M
            _field_uniqueid_to_index[_cols.back()->unique_id()] = _num_columns;
1176
27.3M
        }
1177
27.3M
        _num_columns++;
1178
27.3M
    }
1179
2.03M
    for (const auto& index_pb : schema.index()) {
1180
952k
        TabletIndexPtr index;
1181
952k
        if (reuse_cache_column) {
1182
42.0k
            auto pair = TabletColumnObjectPool::instance()->insert_index(
1183
42.0k
                    deterministic_string_serialize(index_pb));
1184
42.0k
            index = pair.second;
1185
            //  Only need the value to be cached by the pool, release it quickly because the handle need
1186
            // record reference count mannually, or it will core during tablet schema copy method.
1187
42.0k
            TabletColumnObjectPool::instance()->release(pair.first);
1188
910k
        } else {
1189
910k
            index = std::make_shared<TabletIndex>();
1190
910k
            index->init_from_pb(index_pb);
1191
910k
        }
1192
952k
        size_t index_pos = _indexes.size();
1193
952k
        _indexes.emplace_back(std::move(index));
1194
952k
        for (int32_t col_uid : _indexes.back()->col_unique_ids()) {
1195
951k
            if (auto field_pattern = _indexes.back()->field_pattern(); !field_pattern.empty()) {
1196
32.1k
                auto& pattern_to_index_map = _index_by_unique_id_with_pattern[col_uid];
1197
32.1k
                pattern_to_index_map[field_pattern].emplace_back(_indexes.back());
1198
919k
            } else {
1199
919k
                IndexKey key = std::make_tuple(_indexes.back()->index_type(), col_uid,
1200
919k
                                               _indexes.back()->get_index_suffix());
1201
919k
                _col_id_suffix_to_index[key].push_back(index_pos);
1202
919k
            }
1203
951k
        }
1204
952k
    }
1205
2.03M
    _num_short_key_columns = schema.num_short_key_columns();
1206
2.03M
    _num_rows_per_row_block = schema.num_rows_per_row_block();
1207
2.03M
    _compress_kind = schema.compress_kind();
1208
2.03M
    _next_column_unique_id = schema.next_column_unique_id();
1209
2.03M
    if (schema.has_bf_fpp()) {
1210
1.32M
        _has_bf_fpp = true;
1211
1.32M
        _bf_fpp = schema.bf_fpp();
1212
1.32M
    } else {
1213
716k
        _has_bf_fpp = false;
1214
716k
        _bf_fpp = BLOOM_FILTER_DEFAULT_FPP;
1215
716k
    }
1216
2.03M
    _is_in_memory = schema.is_in_memory();
1217
2.03M
    _disable_auto_compaction = schema.disable_auto_compaction();
1218
2.03M
    _store_row_column = schema.store_row_column();
1219
2.03M
    _skip_write_index_on_load = schema.skip_write_index_on_load();
1220
2.03M
    _delete_sign_idx = schema.delete_sign_idx();
1221
2.03M
    _sequence_col_idx = schema.sequence_col_idx();
1222
2.03M
    _version_col_idx = schema.version_col_idx();
1223
2.03M
    _skip_bitmap_col_idx = schema.skip_bitmap_col_idx();
1224
2.03M
    _commit_tso_col_idx = schema.commit_tso_col_idx();
1225
2.03M
    _binlog_tso_col_idx = schema.binlog_tso_col_idx();
1226
2.03M
    _binlog_lsn_col_idx = schema.binlog_lsn_col_idx();
1227
2.03M
    _binlog_op_col_idx = schema.binlog_op_col_idx();
1228
2.03M
    _sort_type = schema.sort_type();
1229
2.03M
    _sort_col_num = schema.sort_col_num();
1230
2.03M
    _compression_type = schema.compression_type();
1231
2.03M
    _row_store_page_size = schema.row_store_page_size();
1232
2.03M
    _storage_page_size = schema.storage_page_size();
1233
2.03M
    _storage_dict_page_size = schema.storage_dict_page_size();
1234
2.03M
    _schema_version = schema.schema_version();
1235
2.03M
    if (schema.has_seq_map()) {
1236
2.03M
        auto column_groups_pb = schema.seq_map();
1237
2.03M
        _seq_col_uid_to_value_cols_uid.clear();
1238
2.03M
        _value_col_uid_to_seq_col_uid.clear();
1239
2.03M
        _seq_col_idx_to_value_cols_idx.clear();
1240
2.03M
        _value_col_idx_to_seq_col_idx.clear();
1241
        /*
1242
         * ColumnGroupsPB is a list of cg_pb, and
1243
         * ColumnGroupsPB do not have begin() or end() method.
1244
         * we must use for(i=0;i<xx;i++) loop
1245
         */
1246
2.03M
        for (int i = 0; i < column_groups_pb.cg_size(); i++) {
1247
374
            ColumnGroupPB cg_pb = column_groups_pb.cg(i);
1248
374
            uint32_t key_uid = cg_pb.sequence_column();
1249
374
            auto found = _field_uniqueid_to_index.find(key_uid);
1250
374
            DCHECK(found != _field_uniqueid_to_index.end())
1251
0
                    << "could not find sequence col with unique id = " << key_uid
1252
0
                    << " table_id=" << _table_id;
1253
374
            int32_t seq_index = found->second;
1254
374
            _seq_col_uid_to_value_cols_uid[key_uid] = {};
1255
374
            _seq_col_idx_to_value_cols_idx[seq_index] = {};
1256
557
            for (auto val_uid : cg_pb.columns_in_group()) {
1257
557
                _seq_col_uid_to_value_cols_uid[key_uid].push_back(val_uid);
1258
557
                found = _field_uniqueid_to_index.find(val_uid);
1259
557
                DCHECK(found != _field_uniqueid_to_index.end())
1260
0
                        << "could not find value col with unique id = " << key_uid
1261
0
                        << " table_id=" << _table_id;
1262
557
                int32_t val_index = found->second;
1263
557
                _seq_col_idx_to_value_cols_idx[seq_index].push_back(val_index);
1264
557
            }
1265
374
        }
1266
1267
2.03M
        if (!_seq_col_uid_to_value_cols_uid.empty()) {
1268
            /*
1269
                |** KEY **|        ** VALUE **     |
1270
                ------------------------------------
1271
                |** KEY **|  CDE is value| sequence|
1272
                |----|----|----|----|----|----|----|
1273
                A    B    C    D    E   S1   S2
1274
                0    1    2    3    4    5    6
1275
                for example: _seq_map is {5:{2,3}, 6:{4}}
1276
                then, _value_to_seq = {2:5,3:5,5:5,4:6,6:6}
1277
            */
1278
374
            for (auto& [seq_uid, cols_uid] : _seq_col_uid_to_value_cols_uid) {
1279
557
                for (auto col_uid : cols_uid) {
1280
557
                    _value_col_uid_to_seq_col_uid[col_uid] = seq_uid;
1281
557
                }
1282
374
                _value_col_uid_to_seq_col_uid[seq_uid] = seq_uid;
1283
374
            }
1284
1285
374
            for (auto& [seq_idx, value_cols_idx] : _seq_col_idx_to_value_cols_idx) {
1286
557
                for (auto col_idx : value_cols_idx) {
1287
557
                    _value_col_idx_to_seq_col_idx[col_idx] = seq_idx;
1288
557
                }
1289
374
                _value_col_idx_to_seq_col_idx[seq_idx] = seq_idx;
1290
374
            }
1291
261
        }
1292
2.03M
    }
1293
    // Default to V1 inverted index storage format for backward compatibility if not specified in schema.
1294
2.03M
    if (!schema.has_inverted_index_storage_format()) {
1295
531
        _inverted_index_storage_format = InvertedIndexStorageFormatPB::V1;
1296
2.03M
    } else {
1297
2.03M
        _inverted_index_storage_format = schema.inverted_index_storage_format();
1298
2.03M
    }
1299
1300
2.03M
    _row_store_column_unique_ids.assign(schema.row_store_column_unique_ids().begin(),
1301
2.03M
                                        schema.row_store_column_unique_ids().end());
1302
2.03M
    _deprecated_enable_variant_flatten_nested = schema.enable_variant_flatten_nested();
1303
2.03M
    if (schema.has_storage_format()) {
1304
2.02M
        _storage_format = schema.storage_format();
1305
2.02M
    } else if (schema.is_external_segment_column_meta_used() ||
1306
9.29k
               schema.integer_type_default_use_plain_encoding() ||
1307
9.29k
               schema.binary_plain_encoding_default_impl() ==
1308
9.19k
                       BinaryPlainEncodingTypePB::BINARY_PLAIN_ENCODING_V2) {
1309
        // Old PB without storage_format: any of the three legacy V3-flavor flags implies V3.
1310
1
        _storage_format = TabletStorageFormatPB::TABLET_STORAGE_FORMAT_V3;
1311
9.29k
    } else {
1312
9.29k
        _storage_format = TabletStorageFormatPB::TABLET_STORAGE_FORMAT_V2;
1313
9.29k
    }
1314
2.03M
    update_metadata_size();
1315
2.03M
}
1316
1317
1.28M
void TabletSchema::copy_from(const TabletSchema& tablet_schema) {
1318
1.28M
    TabletSchemaPB tablet_schema_pb;
1319
1.28M
    tablet_schema.to_schema_pb(&tablet_schema_pb);
1320
1.28M
    init_from_pb(tablet_schema_pb);
1321
1.28M
    _table_id = tablet_schema.table_id();
1322
1.28M
    _path_set_info_map = tablet_schema._path_set_info_map;
1323
1.28M
}
1324
1325
83.0k
void TabletSchema::shawdow_copy_without_columns(const TabletSchema& tablet_schema) {
1326
83.0k
    *this = tablet_schema;
1327
83.0k
    _field_path_to_index.clear();
1328
83.0k
    _field_name_to_index.clear();
1329
83.0k
    _field_uniqueid_to_index.clear();
1330
83.0k
    _num_columns = 0;
1331
83.0k
    _num_variant_columns = 0;
1332
83.0k
    _num_null_columns = 0;
1333
83.0k
    _num_key_columns = 0;
1334
83.0k
    _cols.clear();
1335
83.0k
    _delete_sign_idx = -1;
1336
83.0k
    _sequence_col_idx = -1;
1337
83.0k
    _version_col_idx = -1;
1338
83.0k
    _skip_bitmap_col_idx = -1;
1339
83.0k
    _commit_tso_col_idx = -1;
1340
83.0k
    _binlog_tso_col_idx = -1;
1341
83.0k
    _binlog_lsn_col_idx = -1;
1342
83.0k
    _binlog_op_col_idx = -1;
1343
83.0k
}
1344
1345
20
void TabletSchema::update_index_info_from(const TabletSchema& tablet_schema) {
1346
80
    for (auto& col : _cols) {
1347
80
        if (col->unique_id() < 0) {
1348
0
            continue;
1349
0
        }
1350
80
        const auto iter = tablet_schema._field_uniqueid_to_index.find(col->unique_id());
1351
80
        if (iter == tablet_schema._field_uniqueid_to_index.end()) {
1352
0
            continue;
1353
0
        }
1354
80
        auto col_idx = iter->second;
1355
80
        if (col_idx < 0 || col_idx >= tablet_schema._cols.size()) {
1356
0
            continue;
1357
0
        }
1358
80
        col->set_is_bf_column(tablet_schema._cols[col_idx]->is_bf_column());
1359
80
    }
1360
20
}
1361
1362
1.20M
std::string TabletSchema::to_key() const {
1363
1.20M
    TabletSchemaPB pb;
1364
1.20M
    to_schema_pb(&pb);
1365
1.20M
    return TabletSchema::deterministic_string_serialize(pb);
1366
1.20M
}
1367
1368
void TabletSchema::build_current_tablet_schema(int64_t index_id, int32_t version,
1369
                                               const OlapTableIndexSchema* index,
1370
60.8k
                                               const TabletSchema& ori_tablet_schema) {
1371
    // copy from ori_tablet_schema
1372
60.8k
    _keys_type = ori_tablet_schema.keys_type();
1373
60.8k
    _num_short_key_columns = ori_tablet_schema.num_short_key_columns();
1374
60.8k
    _num_rows_per_row_block = ori_tablet_schema.num_rows_per_row_block();
1375
60.8k
    _compress_kind = ori_tablet_schema.compress_kind();
1376
1377
    // todo(yixiu): unique_id
1378
60.8k
    _next_column_unique_id = ori_tablet_schema.next_column_unique_id();
1379
60.8k
    _is_in_memory = ori_tablet_schema.is_in_memory();
1380
60.8k
    _disable_auto_compaction = ori_tablet_schema.disable_auto_compaction();
1381
60.8k
    _skip_write_index_on_load = ori_tablet_schema.skip_write_index_on_load();
1382
60.8k
    _sort_type = ori_tablet_schema.sort_type();
1383
60.8k
    _sort_col_num = ori_tablet_schema.sort_col_num();
1384
60.8k
    _row_store_page_size = ori_tablet_schema.row_store_page_size();
1385
60.8k
    _storage_page_size = ori_tablet_schema.storage_page_size();
1386
60.8k
    _storage_dict_page_size = ori_tablet_schema.storage_dict_page_size();
1387
60.8k
    _deprecated_enable_variant_flatten_nested =
1388
60.8k
            ori_tablet_schema.deprecated_variant_flatten_nested();
1389
1390
    // copy from table_schema_param
1391
60.8k
    _schema_version = version;
1392
60.8k
    _num_columns = 0;
1393
60.8k
    _num_variant_columns = 0;
1394
60.8k
    _num_key_columns = 0;
1395
60.8k
    _num_null_columns = 0;
1396
60.8k
    bool has_bf_columns = false;
1397
60.8k
    _cols.clear();
1398
60.8k
    _indexes.clear();
1399
60.8k
    _col_id_suffix_to_index.clear();
1400
60.8k
    _index_by_unique_id_with_pattern.clear();
1401
60.8k
    _field_name_to_index.clear();
1402
60.8k
    _field_uniqueid_to_index.clear();
1403
60.8k
    _delete_sign_idx = -1;
1404
60.8k
    _sequence_col_idx = -1;
1405
60.8k
    _version_col_idx = -1;
1406
60.8k
    _skip_bitmap_col_idx = -1;
1407
60.8k
    _commit_tso_col_idx = -1;
1408
60.8k
    _binlog_tso_col_idx = -1;
1409
60.8k
    _binlog_lsn_col_idx = -1;
1410
60.8k
    _binlog_op_col_idx = -1;
1411
60.8k
    _cluster_key_uids.clear();
1412
60.8k
    for (const auto& i : ori_tablet_schema._cluster_key_uids) {
1413
1.36k
        _cluster_key_uids.push_back(i);
1414
1.36k
    }
1415
713k
    for (auto& column : index->columns) {
1416
713k
        if (column->is_key()) {
1417
112k
            _num_key_columns++;
1418
112k
        }
1419
713k
        if (column->is_nullable()) {
1420
403k
            _num_null_columns++;
1421
403k
        }
1422
713k
        if (column->is_bf_column()) {
1423
6.20k
            has_bf_columns = true;
1424
6.20k
        }
1425
713k
        if (column->is_variant_type()) {
1426
1.92k
            ++_num_variant_columns;
1427
1.92k
        }
1428
713k
        if (UNLIKELY(column->name() == DELETE_SIGN)) {
1429
11.7k
            _delete_sign_idx = _num_columns;
1430
702k
        } else if (UNLIKELY(column->name() == SEQUENCE_COL)) {
1431
728
            _sequence_col_idx = _num_columns;
1432
701k
        } else if (UNLIKELY(column->name() == VERSION_COL)) {
1433
11.5k
            _version_col_idx = _num_columns;
1434
689k
        } else if (UNLIKELY(column->name() == SKIP_BITMAP_COL)) {
1435
117
            _skip_bitmap_col_idx = _num_columns;
1436
689k
        } else if (UNLIKELY(column->name() == COMMIT_TSO_COL)) {
1437
1
            _commit_tso_col_idx = _num_columns;
1438
689k
        } else if (UNLIKELY(column->name() == BINLOG_TSO_COL)) {
1439
0
            _binlog_tso_col_idx = _num_columns;
1440
689k
        } else if (UNLIKELY(column->name() == BINLOG_LSN_COL)) {
1441
0
            _binlog_lsn_col_idx = _num_columns;
1442
689k
        } else if (UNLIKELY(column->name() == BINLOG_OP_COL)) {
1443
0
            _binlog_op_col_idx = _num_columns;
1444
0
        }
1445
        // Reuse TabletColumn object from pool to reduce memory consumption
1446
713k
        TabletColumnPtr new_column;
1447
713k
        ColumnPB column_pb;
1448
713k
        column->to_schema_pb(&column_pb);
1449
713k
        auto pair = TabletColumnObjectPool::instance()->insert(
1450
713k
                deterministic_string_serialize(column_pb));
1451
713k
        new_column = pair.second;
1452
        // Release the handle quickly, because we use shared ptr to manage column
1453
713k
        TabletColumnObjectPool::instance()->release(pair.first);
1454
713k
        _cols.emplace_back(std::move(new_column));
1455
713k
        _field_name_to_index.emplace(StringRef(_cols.back()->name()), _num_columns);
1456
713k
        _field_uniqueid_to_index[_cols.back()->unique_id()] = _num_columns;
1457
713k
        _num_columns++;
1458
713k
    }
1459
1460
66.9k
    for (const auto& i : index->indexes) {
1461
66.9k
        size_t index_pos = _indexes.size();
1462
        // Reuse TabletIndex object from pool to reduce memory consumption
1463
66.9k
        TabletIndexPtr new_index;
1464
66.9k
        TabletIndexPB index_pb;
1465
66.9k
        i->to_schema_pb(&index_pb);
1466
66.9k
        auto pair = TabletColumnObjectPool::instance()->insert_index(
1467
66.9k
                deterministic_string_serialize(index_pb));
1468
66.9k
        new_index = pair.second;
1469
        // Release the handle quickly, because we use shared ptr to manage index
1470
66.9k
        TabletColumnObjectPool::instance()->release(pair.first);
1471
66.9k
        _indexes.emplace_back(std::move(new_index));
1472
66.9k
        for (int32_t col_uid : _indexes.back()->col_unique_ids()) {
1473
66.9k
            if (auto field_pattern = _indexes.back()->field_pattern(); !field_pattern.empty()) {
1474
1.28k
                auto& pattern_to_index_map = _index_by_unique_id_with_pattern[col_uid];
1475
1.28k
                pattern_to_index_map[field_pattern].emplace_back(_indexes.back());
1476
65.7k
            } else {
1477
65.7k
                IndexKey key = std::make_tuple(_indexes.back()->index_type(), col_uid,
1478
65.7k
                                               _indexes.back()->get_index_suffix());
1479
65.7k
                _col_id_suffix_to_index[key].push_back(index_pos);
1480
65.7k
            }
1481
66.9k
        }
1482
66.9k
    }
1483
1484
60.8k
    if (has_bf_columns) {
1485
5.90k
        _has_bf_fpp = true;
1486
5.90k
        _bf_fpp = ori_tablet_schema.bloom_filter_fpp();
1487
54.9k
    } else {
1488
54.9k
        _has_bf_fpp = false;
1489
54.9k
        _bf_fpp = BLOOM_FILTER_DEFAULT_FPP;
1490
54.9k
    }
1491
60.8k
}
1492
1493
7.16k
void TabletSchema::merge_dropped_columns(const TabletSchema& src_schema) {
1494
    // If they are the same tablet schema object, then just return
1495
7.16k
    if (this == &src_schema) {
1496
0
        return;
1497
0
    }
1498
53.0k
    for (const auto& src_col : src_schema.columns()) {
1499
53.0k
        if (_field_uniqueid_to_index.find(src_col->unique_id()) == _field_uniqueid_to_index.end()) {
1500
52
            CHECK(!src_col->is_key())
1501
0
                    << src_col->name() << " is key column, should not be dropped.";
1502
52
            ColumnPB src_col_pb;
1503
            // There are some pointer in tablet column, not sure the reference relation, so
1504
            // that deep copy it.
1505
52
            src_col->to_schema_pb(&src_col_pb);
1506
52
            TabletColumn new_col(src_col_pb);
1507
52
            append_column(new_col, TabletSchema::ColumnType::DROPPED);
1508
52
        }
1509
53.0k
    }
1510
7.16k
}
1511
1512
8.45k
TabletSchemaSPtr TabletSchema::copy_without_variant_extracted_columns() {
1513
8.45k
    TabletSchemaSPtr copy = std::make_shared<TabletSchema>();
1514
8.45k
    copy->shawdow_copy_without_columns(*this);
1515
59.1k
    for (auto& col : this->columns()) {
1516
59.1k
        if (col->is_extracted_column()) {
1517
2.75k
            continue;
1518
2.75k
        }
1519
56.3k
        copy->append_column(*col);
1520
56.3k
    }
1521
8.45k
    return copy;
1522
8.45k
}
1523
1524
// Dropped column is in _field_uniqueid_to_index but not in _field_name_to_index
1525
// Could refer to append_column method
1526
587k
bool TabletSchema::is_dropped_column(const TabletColumn& col) const {
1527
18.4E
    CHECK(_field_uniqueid_to_index.find(col.unique_id()) != _field_uniqueid_to_index.end())
1528
18.4E
            << "could not find col with unique id = " << col.unique_id()
1529
18.4E
            << " and name = " << col.name() << " table_id=" << _table_id;
1530
587k
    auto it = _field_name_to_index.find(StringRef {col.name()});
1531
590k
    return it == _field_name_to_index.end() || _cols[it->second]->unique_id() != col.unique_id();
1532
587k
}
1533
1534
103
void TabletSchema::copy_extracted_columns(const TabletSchema& src_schema) {
1535
103
    std::unordered_set<int32_t> variant_columns;
1536
354
    for (const auto& col : columns()) {
1537
354
        if (col->is_variant_type()) {
1538
220
            variant_columns.insert(col->unique_id());
1539
220
        }
1540
354
    }
1541
296
    for (const TabletColumnPtr& col : src_schema.columns()) {
1542
296
        if (col->is_extracted_column() && variant_columns.contains(col->parent_unique_id())) {
1543
0
            ColumnPB col_pb;
1544
0
            col->to_schema_pb(&col_pb);
1545
0
            TabletColumn new_col(col_pb);
1546
0
            append_column(new_col, ColumnType::VARIANT);
1547
0
        }
1548
296
    }
1549
103
}
1550
1551
106
void TabletSchema::reserve_extracted_columns() {
1552
623
    for (auto it = _cols.begin(); it != _cols.end();) {
1553
517
        if (!(*it)->is_extracted_column()) {
1554
217
            it = _cols.erase(it);
1555
300
        } else {
1556
300
            ++it;
1557
300
        }
1558
517
    }
1559
106
}
1560
1561
3.80M
void TabletSchema::to_schema_pb(TabletSchemaPB* tablet_schema_pb) const {
1562
3.80M
    for (const auto& i : _cluster_key_uids) {
1563
92.2k
        tablet_schema_pb->add_cluster_key_uids(i);
1564
92.2k
    }
1565
3.80M
    tablet_schema_pb->set_keys_type(_keys_type);
1566
43.8M
    for (const auto& col : _cols) {
1567
43.8M
        ColumnPB* column = tablet_schema_pb->add_column();
1568
43.8M
        col->to_schema_pb(column);
1569
43.8M
    }
1570
3.80M
    for (const auto& index : _indexes) {
1571
1.61M
        auto* index_pb = tablet_schema_pb->add_index();
1572
1.61M
        index->to_schema_pb(index_pb);
1573
1.61M
    }
1574
3.80M
    tablet_schema_pb->set_num_short_key_columns(cast_set<int32_t>(_num_short_key_columns));
1575
3.80M
    tablet_schema_pb->set_num_rows_per_row_block(cast_set<int32_t>(_num_rows_per_row_block));
1576
3.80M
    tablet_schema_pb->set_compress_kind(_compress_kind);
1577
3.80M
    if (_has_bf_fpp) {
1578
1.55M
        tablet_schema_pb->set_bf_fpp(_bf_fpp);
1579
1.55M
    }
1580
3.80M
    tablet_schema_pb->set_next_column_unique_id(cast_set<uint32_t>(_next_column_unique_id));
1581
3.80M
    tablet_schema_pb->set_is_in_memory(_is_in_memory);
1582
3.80M
    tablet_schema_pb->set_disable_auto_compaction(_disable_auto_compaction);
1583
3.80M
    tablet_schema_pb->set_store_row_column(_store_row_column);
1584
3.80M
    tablet_schema_pb->set_skip_write_index_on_load(_skip_write_index_on_load);
1585
3.80M
    tablet_schema_pb->set_delete_sign_idx(_delete_sign_idx);
1586
3.80M
    tablet_schema_pb->set_sequence_col_idx(_sequence_col_idx);
1587
3.80M
    tablet_schema_pb->set_sort_type(_sort_type);
1588
3.80M
    tablet_schema_pb->set_sort_col_num(cast_set<int32_t>(_sort_col_num));
1589
3.80M
    tablet_schema_pb->set_schema_version(_schema_version);
1590
3.80M
    tablet_schema_pb->set_compression_type(_compression_type);
1591
3.80M
    tablet_schema_pb->set_row_store_page_size(_row_store_page_size);
1592
3.80M
    tablet_schema_pb->set_storage_page_size(_storage_page_size);
1593
3.80M
    tablet_schema_pb->set_storage_dict_page_size(_storage_dict_page_size);
1594
3.80M
    tablet_schema_pb->set_version_col_idx(_version_col_idx);
1595
3.80M
    tablet_schema_pb->set_skip_bitmap_col_idx(_skip_bitmap_col_idx);
1596
3.80M
    tablet_schema_pb->set_commit_tso_col_idx(_commit_tso_col_idx);
1597
3.80M
    tablet_schema_pb->set_binlog_tso_col_idx(_binlog_tso_col_idx);
1598
3.80M
    tablet_schema_pb->set_binlog_lsn_col_idx(_binlog_lsn_col_idx);
1599
3.80M
    tablet_schema_pb->set_binlog_op_col_idx(_binlog_op_col_idx);
1600
3.80M
    tablet_schema_pb->set_inverted_index_storage_format(_inverted_index_storage_format);
1601
3.80M
    tablet_schema_pb->mutable_row_store_column_unique_ids()->Assign(
1602
3.80M
            _row_store_column_unique_ids.begin(), _row_store_column_unique_ids.end());
1603
3.80M
    tablet_schema_pb->set_enable_variant_flatten_nested(_deprecated_enable_variant_flatten_nested);
1604
3.80M
    tablet_schema_pb->set_storage_format(_storage_format);
1605
    // Backward downgrade safety: if a new BE rewrites tablet_meta.json carrying only
1606
    // storage_format and the deployment is then rolled back to an old BE, the old BE
1607
    // does not know the new field and would default-derive V2 for a V3 tablet, causing
1608
    // it to write V2-encoded segments into a V3 tablet. Redundantly emit the three
1609
    // legacy V3-flavor flags so old BEs can recover the format via the prior "any of
1610
    // these implies V3" rule. ~3 bytes per schema PB; only paid for V3 tablets.
1611
3.80M
    if (_storage_format == TabletStorageFormatPB::TABLET_STORAGE_FORMAT_V3) {
1612
1.41M
        tablet_schema_pb->set_is_external_segment_column_meta_used(true);
1613
1.41M
        tablet_schema_pb->set_integer_type_default_use_plain_encoding(true);
1614
1.41M
        tablet_schema_pb->set_binary_plain_encoding_default_impl(
1615
1.41M
                BinaryPlainEncodingTypePB::BINARY_PLAIN_ENCODING_V2);
1616
1.41M
    }
1617
3.80M
    auto column_groups_pb = tablet_schema_pb->mutable_seq_map();
1618
3.80M
    for (const auto& it : _seq_col_uid_to_value_cols_uid) {
1619
640
        uint32_t key = it.first;
1620
640
        ColumnGroupPB* cg_pb = column_groups_pb->add_cg(); // ColumnGroupPB {key: {v1, v2, v3}}
1621
640
        cg_pb->set_sequence_column(key);
1622
993
        for (auto v : it.second) {
1623
993
            cg_pb->add_columns_in_group(v);
1624
993
        }
1625
640
    }
1626
3.80M
}
1627
1628
243k
size_t TabletSchema::row_size() const {
1629
243k
    size_t size = 0;
1630
2.20M
    for (const auto& column : _cols) {
1631
2.20M
        size += column->length();
1632
2.20M
    }
1633
243k
    size += (_num_columns + 7) / 8;
1634
1635
243k
    return size;
1636
243k
}
1637
1638
11.5M
int32_t TabletSchema::field_index(const std::string& field_name) const {
1639
11.5M
    const auto& found = _field_name_to_index.find(StringRef(field_name));
1640
11.5M
    return (found == _field_name_to_index.end()) ? -1 : found->second;
1641
11.5M
}
1642
1643
13.6k
int32_t TabletSchema::field_index(const PathInData& path) const {
1644
13.6k
    const auto& found = _field_path_to_index.find(PathInDataRef(&path));
1645
13.6k
    return (found == _field_path_to_index.end()) ? -1 : found->second;
1646
13.6k
}
1647
1648
22.6M
int32_t TabletSchema::field_index(int32_t col_unique_id) const {
1649
22.6M
    const auto& found = _field_uniqueid_to_index.find(col_unique_id);
1650
22.6M
    return (found == _field_uniqueid_to_index.end()) ? -1 : found->second;
1651
22.6M
}
1652
1653
48.6M
const std::vector<TabletColumnPtr>& TabletSchema::columns() const {
1654
48.6M
    return _cols;
1655
48.6M
}
1656
1657
158M
const TabletColumn& TabletSchema::column(size_t ordinal) const {
1658
18.4E
    DCHECK(ordinal < _num_columns) << "ordinal:" << ordinal << ", _num_columns:" << _num_columns;
1659
158M
    return *_cols[ordinal];
1660
158M
}
1661
1662
2.05M
const TabletColumn& TabletSchema::column_by_uid(int32_t col_unique_id) const {
1663
2.05M
    return *_cols.at(_field_uniqueid_to_index.at(col_unique_id));
1664
2.05M
}
1665
1666
41
TabletColumn& TabletSchema::mutable_column_by_uid(int32_t col_unique_id) {
1667
41
    return *_cols.at(_field_uniqueid_to_index.at(col_unique_id));
1668
41
}
1669
1670
72.6k
TabletColumn& TabletSchema::mutable_column(size_t ordinal) {
1671
72.6k
    return *_cols.at(ordinal);
1672
72.6k
}
1673
1674
1.46M
void TabletSchema::update_indexes_from_thrift(const std::vector<doris::TOlapTableIndex>& tindexes) {
1675
1.46M
    std::vector<TabletIndexPtr> indexes;
1676
1.46M
    for (const auto& tindex : tindexes) {
1677
368k
        TabletIndex index;
1678
368k
        index.init_from_thrift(tindex, *this);
1679
368k
        indexes.emplace_back(std::make_shared<TabletIndex>(std::move(index)));
1680
368k
    }
1681
1.46M
    _indexes = std::move(indexes);
1682
1.46M
    _col_id_suffix_to_index.clear();
1683
1.46M
    _index_by_unique_id_with_pattern.clear();
1684
1.46M
    size_t index_pos = 0;
1685
1.46M
    for (auto& index : _indexes) {
1686
368k
        for (int32_t col_uid : index->col_unique_ids()) {
1687
367k
            if (auto field_pattern = index->field_pattern(); !field_pattern.empty()) {
1688
4.28k
                auto& pattern_to_index_map = _index_by_unique_id_with_pattern[col_uid];
1689
4.28k
                pattern_to_index_map[field_pattern].emplace_back(index);
1690
363k
            } else {
1691
363k
                IndexKey key =
1692
363k
                        std::make_tuple(index->index_type(), col_uid, index->get_index_suffix());
1693
363k
                _col_id_suffix_to_index[key].push_back(index_pos);
1694
363k
            }
1695
367k
        }
1696
368k
        index_pos++;
1697
368k
    }
1698
1.46M
}
1699
1700
4.75k
bool TabletSchema::exist_column(const std::string& field_name) const {
1701
4.75k
    return _field_name_to_index.contains(StringRef {field_name});
1702
4.75k
}
1703
1704
28.5M
bool TabletSchema::has_column_unique_id(int32_t col_unique_id) const {
1705
28.5M
    return _field_uniqueid_to_index.contains(col_unique_id);
1706
28.5M
}
1707
1708
4.17k
Status TabletSchema::have_column(const std::string& field_name) const {
1709
4.17k
    if (!_field_name_to_index.contains(StringRef(field_name))) {
1710
4.04k
        return Status::Error<ErrorCode::INTERNAL_ERROR>(
1711
4.04k
                "Not found field_name, field_name:{}, schema:{}", field_name,
1712
4.04k
                get_all_field_names());
1713
4.04k
    }
1714
131
    return Status::OK();
1715
4.17k
}
1716
1717
5.90k
Result<const TabletColumn*> TabletSchema::column(const std::string& field_name) const {
1718
5.90k
    auto it = _field_name_to_index.find(StringRef {field_name});
1719
5.90k
    if (it == _field_name_to_index.end()) {
1720
0
        DCHECK(false) << "field_name=" << field_name << ", table_id=" << _table_id
1721
0
                      << ", field_name_to_index=" << get_all_field_names();
1722
0
        return ResultError(
1723
0
                Status::InternalError("column not found, name={}, table_id={}, schema_version={}",
1724
0
                                      field_name, _table_id, _schema_version));
1725
0
    }
1726
5.90k
    return _cols[it->second].get();
1727
5.90k
}
1728
1729
void TabletSchema::update_tablet_columns(const TabletSchema& tablet_schema,
1730
13.4k
                                         const std::vector<TColumn>& t_columns) {
1731
13.4k
    copy_from(tablet_schema);
1732
13.4k
    if (!t_columns.empty() && t_columns[0].col_unique_id >= 0) {
1733
13.3k
        clear_columns();
1734
134k
        for (const auto& column : t_columns) {
1735
134k
            append_column(TabletColumn(column));
1736
134k
        }
1737
13.3k
    }
1738
13.4k
}
1739
1740
88
bool TabletSchema::has_inverted_index_with_index_id(int64_t index_id) const {
1741
107
    for (size_t i = 0; i < _indexes.size(); i++) {
1742
60
        if ((_indexes[i]->index_type() == IndexType::INVERTED ||
1743
60
             _indexes[i]->index_type() == IndexType::ANN) &&
1744
60
            _indexes[i]->index_id() == index_id) {
1745
41
            return true;
1746
41
        }
1747
60
    }
1748
47
    return false;
1749
88
}
1750
1751
std::vector<const TabletIndex*> TabletSchema::inverted_indexs(
1752
28.5M
        int32_t col_unique_id, const std::string& suffix_path) const {
1753
28.5M
    std::vector<const TabletIndex*> result;
1754
28.5M
    const std::string escaped_suffix = escape_for_path_name(suffix_path);
1755
28.5M
    auto it = _col_id_suffix_to_index.find(
1756
28.5M
            std::make_tuple(IndexType::INVERTED, col_unique_id, escaped_suffix));
1757
28.5M
    if (it != _col_id_suffix_to_index.end()) {
1758
160k
        for (size_t pos : it->second) {
1759
160k
            if (pos < _indexes.size()) {
1760
160k
                result.push_back(_indexes[pos].get());
1761
160k
            }
1762
160k
        }
1763
159k
    }
1764
28.5M
    return result;
1765
28.5M
}
1766
1767
std::vector<TabletIndexPtr> TabletSchema::inverted_index_by_field_pattern(
1768
12.7k
        int32_t col_unique_id, const std::string& field_pattern) const {
1769
12.7k
    auto id_to_pattern_map = _index_by_unique_id_with_pattern.find(col_unique_id);
1770
12.7k
    if (id_to_pattern_map == _index_by_unique_id_with_pattern.end()) {
1771
6.82k
        return {};
1772
6.82k
    }
1773
5.90k
    auto pattern_to_index_map = id_to_pattern_map->second.find(field_pattern);
1774
5.90k
    if (pattern_to_index_map == id_to_pattern_map->second.end()) {
1775
1.20k
        return {};
1776
1.20k
    }
1777
4.69k
    return pattern_to_index_map->second;
1778
5.90k
}
1779
1780
28.3M
std::vector<const TabletIndex*> TabletSchema::inverted_indexs(const TabletColumn& col) const {
1781
    // Some columns(Float, Double, JSONB ...) from the variant do not support inverted index
1782
28.3M
    if (!segment_v2::IndexColumnWriter::check_support_inverted_index(col)) {
1783
59.5k
        return {};
1784
59.5k
    }
1785
1786
    // TODO use more efficient impl
1787
    // Use parent id if unique not assigned, this could happend when accessing subcolumns of variants
1788
28.2M
    int32_t col_unique_id = col.is_extracted_column() ? col.parent_unique_id() : col.unique_id();
1789
28.2M
    std::vector<const TabletIndex*> result;
1790
28.2M
    if (result = inverted_indexs(col_unique_id, escape_for_path_name(col.suffix_path()));
1791
28.2M
        !result.empty()) {
1792
119k
        return result;
1793
119k
    }
1794
    // variant's typed column has it's own index
1795
28.1M
    else if (col.is_extracted_column() && col.path_info_ptr()->get_is_typed()) {
1796
492
        std::string relative_path = col.path_info_ptr()->copy_pop_front().get_path();
1797
492
        if (_path_set_info_map.find(col_unique_id) == _path_set_info_map.end()) {
1798
0
            return result;
1799
0
        }
1800
492
        const auto& path_set_info = _path_set_info_map.at(col_unique_id);
1801
492
        if (path_set_info.typed_path_set.find(relative_path) ==
1802
492
            path_set_info.typed_path_set.end()) {
1803
0
            return result;
1804
0
        }
1805
492
        for (const auto& index : path_set_info.typed_path_set.at(relative_path).indexes) {
1806
58
            result.push_back(index.get());
1807
58
        }
1808
492
        return result;
1809
492
    }
1810
    // variant's subcolumns has it's own index
1811
28.1M
    else if (col.is_extracted_column()) {
1812
1.53k
        std::string relative_path = col.path_info_ptr()->copy_pop_front().get_path();
1813
1.53k
        if (_path_set_info_map.find(col_unique_id) == _path_set_info_map.end()) {
1814
5
            return result;
1815
5
        }
1816
1.53k
        const auto& path_set_info = _path_set_info_map.at(col_unique_id);
1817
1.53k
        if (path_set_info.subcolumn_indexes.find(relative_path) ==
1818
1.53k
            path_set_info.subcolumn_indexes.end()) {
1819
806
            return result;
1820
806
        }
1821
728
        for (const auto& index : path_set_info.subcolumn_indexes.at(relative_path)) {
1822
40
            result.push_back(index.get());
1823
40
        }
1824
728
    }
1825
28.1M
    return result;
1826
28.2M
}
1827
1828
const TabletIndex* TabletSchema::ann_index(int32_t col_unique_id,
1829
97.7k
                                           const std::string& suffix_path) const {
1830
171k
    for (size_t i = 0; i < _indexes.size(); i++) {
1831
73.6k
        if (_indexes[i]->index_type() == IndexType::ANN) {
1832
219
            for (int32_t id : _indexes[i]->col_unique_ids()) {
1833
219
                if (id == col_unique_id &&
1834
219
                    _indexes[i]->get_index_suffix() == escape_for_path_name(suffix_path)) {
1835
219
                    return _indexes[i].get();
1836
219
                }
1837
219
            }
1838
218
        }
1839
73.6k
    }
1840
97.4k
    return nullptr;
1841
97.7k
}
1842
1843
27.3M
const TabletIndex* TabletSchema::ann_index(const TabletColumn& col) const {
1844
27.3M
    if (!segment_v2::IndexColumnWriter::check_support_ann_index(col)) {
1845
27.2M
        return nullptr;
1846
27.2M
    }
1847
    // TODO use more efficient impl
1848
    // Use parent id if unique not assigned, this could happend when accessing subcolumns of variants
1849
94.6k
    int32_t col_unique_id = col.is_extracted_column() ? col.parent_unique_id() : col.unique_id();
1850
94.6k
    return ann_index(col_unique_id, escape_for_path_name(col.suffix_path()));
1851
27.3M
}
1852
1853
0
bool TabletSchema::has_ngram_bf_index(int32_t col_unique_id) const {
1854
0
    IndexKey index_key(IndexType::NGRAM_BF, col_unique_id, "");
1855
0
    auto it = _col_id_suffix_to_index.find(index_key);
1856
0
    return it != _col_id_suffix_to_index.end();
1857
0
}
1858
1859
828k
const TabletIndex* TabletSchema::get_ngram_bf_index(int32_t col_unique_id) const {
1860
    // Get the ngram bf index for the given column unique id
1861
828k
    IndexKey index_key(IndexType::NGRAM_BF, col_unique_id, "");
1862
828k
    auto it = _col_id_suffix_to_index.find(index_key);
1863
828k
    if (it != _col_id_suffix_to_index.end()) {
1864
3.35k
        if (!it->second.empty() && it->second[0] < _indexes.size()) {
1865
3.35k
            return _indexes[it->second[0]].get();
1866
3.35k
        }
1867
3.35k
    }
1868
824k
    return nullptr;
1869
828k
}
1870
1871
const TabletIndex* TabletSchema::get_index(int32_t col_unique_id, IndexType index_type,
1872
14
                                           const std::string& suffix_path) const {
1873
14
    IndexKey index_key(index_type, col_unique_id, suffix_path);
1874
14
    auto it = _col_id_suffix_to_index.find(index_key);
1875
14
    if (it != _col_id_suffix_to_index.end()) {
1876
12
        if (!it->second.empty() && it->second[0] < _indexes.size()) {
1877
12
            return _indexes[it->second[0]].get();
1878
12
        }
1879
12
    }
1880
2
    return nullptr;
1881
14
}
1882
1883
Block TabletSchema::create_block(
1884
        const std::vector<uint32_t>& return_columns,
1885
3.57M
        const std::unordered_set<uint32_t>* tablet_columns_need_convert_null) const {
1886
3.57M
    Block block;
1887
46.4M
    for (int i = 0; i < return_columns.size(); ++i) {
1888
42.8M
        const ColumnId cid = return_columns[i];
1889
42.8M
        const auto& col = *_cols[cid];
1890
42.8M
        bool is_nullable = (tablet_columns_need_convert_null != nullptr &&
1891
42.8M
                            tablet_columns_need_convert_null->find(cid) !=
1892
42.7M
                                    tablet_columns_need_convert_null->end());
1893
42.8M
        auto data_type = DataTypeFactory::instance().create_data_type(col, is_nullable);
1894
42.8M
        if (col.type() == FieldType::OLAP_FIELD_TYPE_STRUCT ||
1895
42.8M
            col.type() == FieldType::OLAP_FIELD_TYPE_MAP ||
1896
42.8M
            col.type() == FieldType::OLAP_FIELD_TYPE_ARRAY) {
1897
144k
            if (_pruned_columns_data_type.contains(col.unique_id())) {
1898
141k
                data_type = _pruned_columns_data_type.at(col.unique_id());
1899
141k
            }
1900
144k
        }
1901
1902
42.8M
        if (_vir_col_idx_to_unique_id.contains(cid)) {
1903
346
            block.insert({ColumnNothing::create(0), data_type, col.name()});
1904
346
            VLOG_DEBUG << fmt::format(
1905
1
                    "Create block from tablet schema, column cid {} is virtual column, col_name: "
1906
1
                    "{}, col_unique_id: {}, type {}",
1907
1
                    cid, col.name(), col.unique_id(), data_type->get_name());
1908
42.8M
        } else {
1909
42.8M
            block.insert({data_type->create_column(), data_type, col.name()});
1910
42.8M
        }
1911
42.8M
    }
1912
3.57M
    return block;
1913
3.57M
}
1914
1915
56.1k
Block TabletSchema::create_block() const {
1916
56.1k
    Block block;
1917
591k
    for (const auto& col : _cols) {
1918
591k
        if (is_dropped_column(*col)) {
1919
13
            continue;
1920
13
        }
1921
1922
591k
        auto data_type = DataTypeFactory::instance().create_data_type(*col);
1923
591k
        if (col->type() == FieldType::OLAP_FIELD_TYPE_STRUCT) {
1924
1.05k
            if (_pruned_columns_data_type.contains(col->unique_id())) {
1925
0
                data_type = _pruned_columns_data_type.at(col->unique_id());
1926
0
            }
1927
1.05k
        }
1928
591k
        block.insert({data_type->create_column(), data_type, col->name()});
1929
591k
    }
1930
56.1k
    return block;
1931
56.1k
}
1932
1933
2.92k
Block TabletSchema::create_block_by_cids(const std::vector<uint32_t>& cids) const {
1934
2.92k
    Block block;
1935
18.1k
    for (const auto& cid : cids) {
1936
18.1k
        const auto& col = *_cols[cid];
1937
18.1k
        auto data_type = DataTypeFactory::instance().create_data_type(col);
1938
18.1k
        if (col.type() == FieldType::OLAP_FIELD_TYPE_STRUCT) {
1939
17
            if (_pruned_columns_data_type.contains(col.unique_id())) {
1940
0
                data_type = _pruned_columns_data_type.at(col.unique_id());
1941
0
            }
1942
17
        }
1943
18.1k
        block.insert({data_type->create_column(), data_type, col.name()});
1944
18.1k
    }
1945
2.92k
    return block;
1946
2.92k
}
1947
1948
827
bool operator==(const TabletColumn& a, const TabletColumn& b) {
1949
827
    if (a._unique_id != b._unique_id) return false;
1950
827
    if (a._col_name != b._col_name) return false;
1951
827
    if (a._type != b._type) return false;
1952
827
    if (a._is_key != b._is_key) return false;
1953
827
    if (a._aggregation != b._aggregation) return false;
1954
827
    if (a._is_nullable != b._is_nullable) return false;
1955
827
    if (a._has_default_value != b._has_default_value) return false;
1956
827
    if (a._has_default_value) {
1957
415
        if (a._default_value != b._default_value) return false;
1958
415
    }
1959
827
    if (a._is_decimal != b._is_decimal) return false;
1960
834
    if (a._is_decimal) {
1961
834
        if (a._precision != b._precision) return false;
1962
834
        if (a._frac != b._frac) return false;
1963
834
    }
1964
827
    if (a._length != b._length) return false;
1965
827
    if (a._index_length != b._index_length) return false;
1966
827
    if (a._is_bf_column != b._is_bf_column) return false;
1967
827
    if (a._column_path == nullptr && a._column_path != nullptr) return false;
1968
827
    if (b._column_path == nullptr && a._column_path != nullptr) return false;
1969
827
    if (b._column_path != nullptr && a._column_path != nullptr &&
1970
827
        *a._column_path != *b._column_path)
1971
0
        return false;
1972
827
    return true;
1973
827
}
1974
1975
834
bool operator!=(const TabletColumn& a, const TabletColumn& b) {
1976
834
    return !(a == b);
1977
834
}
1978
1979
109
bool operator==(const TabletSchema& a, const TabletSchema& b) {
1980
109
    if (a._keys_type != b._keys_type) return false;
1981
109
    if (a._cols.size() != b._cols.size()) return false;
1982
940
    for (int i = 0; i < a._cols.size(); ++i) {
1983
831
        if (*a._cols[i] != *b._cols[i]) return false;
1984
831
    }
1985
109
    if (a._num_columns != b._num_columns) return false;
1986
109
    if (a._num_key_columns != b._num_key_columns) return false;
1987
109
    if (a._num_null_columns != b._num_null_columns) return false;
1988
109
    if (a._num_short_key_columns != b._num_short_key_columns) return false;
1989
109
    if (a._num_rows_per_row_block != b._num_rows_per_row_block) return false;
1990
109
    if (a._compress_kind != b._compress_kind) return false;
1991
109
    if (a._next_column_unique_id != b._next_column_unique_id) return false;
1992
109
    if (a._has_bf_fpp != b._has_bf_fpp) return false;
1993
109
    if (a._has_bf_fpp) {
1994
9
        if (std::abs(a._bf_fpp - b._bf_fpp) > 1e-6) return false;
1995
9
    }
1996
109
    if (a._is_in_memory != b._is_in_memory) return false;
1997
109
    if (a._delete_sign_idx != b._delete_sign_idx) return false;
1998
109
    if (a._sequence_col_idx != b._sequence_col_idx) return false;
1999
109
    if (a._version_col_idx != b._version_col_idx) return false;
2000
109
    if (a._skip_bitmap_col_idx != b._skip_bitmap_col_idx) return false;
2001
109
    if (a._commit_tso_col_idx != b._commit_tso_col_idx) return false;
2002
109
    if (a._binlog_tso_col_idx != b._binlog_tso_col_idx) return false;
2003
109
    if (a._binlog_lsn_col_idx != b._binlog_lsn_col_idx) return false;
2004
109
    if (a._binlog_op_col_idx != b._binlog_op_col_idx) return false;
2005
109
    if (a._disable_auto_compaction != b._disable_auto_compaction) return false;
2006
109
    if (a._store_row_column != b._store_row_column) return false;
2007
109
    if (a._row_store_page_size != b._row_store_page_size) return false;
2008
109
    if (a._storage_page_size != b._storage_page_size) return false;
2009
109
    if (a._storage_dict_page_size != b._storage_dict_page_size) return false;
2010
109
    if (a._skip_write_index_on_load != b._skip_write_index_on_load) return false;
2011
109
    if (a._deprecated_enable_variant_flatten_nested !=
2012
109
        b._deprecated_enable_variant_flatten_nested) {
2013
0
        return false;
2014
0
    }
2015
109
    if (a._storage_format != b._storage_format) return false;
2016
109
    return true;
2017
109
}
2018
2019
108
bool operator!=(const TabletSchema& a, const TabletSchema& b) {
2020
108
    return !(a == b);
2021
108
}
2022
} // namespace doris