Coverage Report

Created: 2026-07-28 22:36

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