Coverage Report

Created: 2026-07-29 06:28

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