Coverage Report

Created: 2026-03-20 13:00

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