Coverage Report

Created: 2024-11-21 12:22

/root/doris/be/src/olap/tablet_schema.h
Line
Count
Source (jump to first uncovered line)
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
#pragma once
19
20
#include <gen_cpp/Types_types.h>
21
#include <gen_cpp/olap_common.pb.h>
22
#include <gen_cpp/olap_file.pb.h>
23
#include <gen_cpp/segment_v2.pb.h>
24
#include <parallel_hashmap/phmap.h>
25
26
#include <algorithm>
27
#include <map>
28
#include <memory>
29
#include <string>
30
#include <unordered_map>
31
#include <unordered_set>
32
#include <utility>
33
#include <vector>
34
35
#include "common/consts.h"
36
#include "common/status.h"
37
#include "gutil/stringprintf.h"
38
#include "olap/metadata_adder.h"
39
#include "olap/olap_common.h"
40
#include "olap/rowset/segment_v2/options.h"
41
#include "runtime/define_primitive_type.h"
42
#include "runtime/descriptors.h"
43
#include "runtime/memory/lru_cache_policy.h"
44
#include "util/string_util.h"
45
#include "vec/aggregate_functions/aggregate_function.h"
46
#include "vec/common/string_ref.h"
47
#include "vec/common/string_utils/string_utils.h"
48
#include "vec/core/types.h"
49
#include "vec/json/path_in_data.h"
50
51
namespace doris {
52
namespace vectorized {
53
class Block;
54
class PathInData;
55
class IDataType;
56
} // namespace vectorized
57
58
struct OlapTableIndexSchema;
59
class TColumn;
60
class TOlapTableIndex;
61
class TabletColumn;
62
63
using TabletColumnPtr = std::shared_ptr<TabletColumn>;
64
65
class TabletColumn : public MetadataAdder<TabletColumn> {
66
public:
67
    TabletColumn();
68
    TabletColumn(const ColumnPB& column);
69
    TabletColumn(const TColumn& column);
70
    TabletColumn(FieldAggregationMethod agg, FieldType type);
71
    TabletColumn(FieldAggregationMethod agg, FieldType filed_type, bool is_nullable);
72
    TabletColumn(FieldAggregationMethod agg, FieldType filed_type, bool is_nullable,
73
                 int32_t unique_id, size_t length);
74
    void init_from_pb(const ColumnPB& column);
75
    void init_from_thrift(const TColumn& column);
76
    void to_schema_pb(ColumnPB* column) const;
77
78
204k
    int32_t unique_id() const { return _unique_id; }
79
19
    void set_unique_id(int32_t id) { _unique_id = id; }
80
64.5k
    const std::string& name() const { return _col_name; }
81
0
    const std::string& name_lower_case() const { return _col_name_lower_case; }
82
23
    void set_name(std::string col_name) {
83
23
        _col_name = col_name;
84
23
        _col_name_lower_case = to_lower(_col_name);
85
23
    }
86
787k
    FieldType type() const { return _type; }
87
19
    void set_type(FieldType type) { _type = type; }
88
45.1k
    bool is_key() const { return _is_key; }
89
147k
    bool is_nullable() const { return _is_nullable; }
90
0
    bool is_auto_increment() const { return _is_auto_increment; }
91
46.8k
    bool is_variant_type() const { return _type == FieldType::OLAP_FIELD_TYPE_VARIANT; }
92
10.4k
    bool is_bf_column() const { return _is_bf_column; }
93
10.4k
    bool has_bitmap_index() const { return _has_bitmap_index; }
94
0
    bool is_array_type() const { return _type == FieldType::OLAP_FIELD_TYPE_ARRAY; }
95
10.0k
    bool is_agg_state_type() const { return _type == FieldType::OLAP_FIELD_TYPE_AGG_STATE; }
96
0
    bool is_jsonb_type() const { return _type == FieldType::OLAP_FIELD_TYPE_JSONB; }
97
0
    bool is_length_variable_type() const {
98
0
        return _type == FieldType::OLAP_FIELD_TYPE_CHAR ||
99
0
               _type == FieldType::OLAP_FIELD_TYPE_VARCHAR ||
100
0
               _type == FieldType::OLAP_FIELD_TYPE_STRING ||
101
0
               _type == FieldType::OLAP_FIELD_TYPE_HLL ||
102
0
               _type == FieldType::OLAP_FIELD_TYPE_OBJECT ||
103
0
               _type == FieldType::OLAP_FIELD_TYPE_QUANTILE_STATE ||
104
0
               _type == FieldType::OLAP_FIELD_TYPE_AGG_STATE;
105
0
    }
106
    // Such columns are not exist in frontend schema info, so we need to
107
    // add them into tablet_schema for later column indexing.
108
    static TabletColumn create_materialized_variant_column(const std::string& root,
109
                                                           const std::vector<std::string>& paths,
110
                                                           int32_t parent_unique_id);
111
188
    bool has_default_value() const { return _has_default_value; }
112
10.4k
    std::string default_value() const { return _default_value; }
113
31.1k
    size_t length() const { return _length; }
114
0
    void set_length(size_t length) { _length = length; }
115
0
    void set_default_value(const std::string& default_value) {
116
0
        _default_value = default_value;
117
0
        _has_default_value = true;
118
0
    }
119
23.9k
    size_t index_length() const { return _index_length; }
120
1
    void set_index_length(size_t index_length) { _index_length = index_length; }
121
0
    void set_is_key(bool is_key) { _is_key = is_key; }
122
4
    void set_is_nullable(bool is_nullable) { _is_nullable = is_nullable; }
123
0
    void set_is_auto_increment(bool is_auto_increment) { _is_auto_increment = is_auto_increment; }
124
0
    void set_has_default_value(bool has) { _has_default_value = has; }
125
    void set_path_info(const vectorized::PathInData& path);
126
14.2k
    FieldAggregationMethod aggregation() const { return _aggregation; }
127
    vectorized::AggregateFunctionPtr get_aggregate_function_union(
128
            vectorized::DataTypePtr type, int current_be_exec_version) const;
129
    vectorized::AggregateFunctionPtr get_aggregate_function(std::string suffix,
130
                                                            int current_be_exec_version) const;
131
127k
    int precision() const { return _precision; }
132
127k
    int frac() const { return _frac; }
133
0
    inline bool visible() const { return _visible; }
134
135
4
    void set_aggregation_method(FieldAggregationMethod agg) {
136
4
        _aggregation = agg;
137
4
        _aggregation_name = get_string_by_aggregation_type(agg);
138
4
    }
139
140
    /**
141
     * Add a sub column.
142
     */
143
    void add_sub_column(TabletColumn& sub_column);
144
145
10.4k
    uint32_t get_subtype_count() const { return _sub_column_count; }
146
17
    const TabletColumn& get_sub_column(uint32_t i) const { return *_sub_columns[i]; }
147
0
    const std::vector<TabletColumnPtr>& get_sub_columns() const { return _sub_columns; }
148
149
    friend bool operator==(const TabletColumn& a, const TabletColumn& b);
150
    friend bool operator!=(const TabletColumn& a, const TabletColumn& b);
151
152
    static std::string get_string_by_field_type(FieldType type);
153
    static std::string get_string_by_aggregation_type(FieldAggregationMethod aggregation_type);
154
    static FieldType get_field_type_by_string(const std::string& str);
155
    static FieldType get_field_type_by_type(PrimitiveType type);
156
    static FieldAggregationMethod get_aggregation_type_by_string(const std::string& str);
157
    static uint32_t get_field_length_by_type(TPrimitiveType::type type, uint32_t string_length);
158
    bool is_row_store_column() const;
159
10.3k
    std::string get_aggregation_name() const { return _aggregation_name; }
160
10.3k
    bool get_result_is_nullable() const { return _result_is_nullable; }
161
10.4k
    int get_be_exec_version() const { return _be_exec_version; }
162
62.1k
    bool has_path_info() const { return _column_path != nullptr && !_column_path->empty(); }
163
20.6k
    const vectorized::PathInDataPtr& path_info_ptr() const { return _column_path; }
164
    // If it is an extracted column from variant column
165
76.3k
    bool is_extracted_column() const {
166
76.3k
        return _column_path != nullptr && !_column_path->empty() && _parent_col_unique_id > 0;
167
76.3k
    };
168
20.8k
    std::string suffix_path() const {
169
20.8k
        return is_extracted_column() ? _column_path->get_path() : "";
170
20.8k
    }
171
0
    bool is_nested_subcolumn() const {
172
0
        return _column_path != nullptr && _column_path->has_nested_part();
173
0
    }
174
20.6k
    int32_t parent_unique_id() const { return _parent_col_unique_id; }
175
4
    void set_parent_unique_id(int32_t col_unique_id) { _parent_col_unique_id = col_unique_id; }
176
2
    void set_is_bf_column(bool is_bf_column) { _is_bf_column = is_bf_column; }
177
0
    void set_has_bitmap_index(bool has_bitmap_index) { _has_bitmap_index = has_bitmap_index; }
178
    std::shared_ptr<const vectorized::IDataType> get_vec_type() const;
179
180
    void append_sparse_column(TabletColumn column);
181
    const TabletColumn& sparse_column_at(size_t oridinal) const;
182
    const std::vector<TabletColumnPtr>& sparse_columns() const;
183
10.4k
    size_t num_sparse_columns() const { return _num_sparse_columns; }
184
185
0
    Status check_valid() const {
186
0
        if (type() != FieldType::OLAP_FIELD_TYPE_ARRAY &&
187
0
            type() != FieldType::OLAP_FIELD_TYPE_STRUCT &&
188
0
            type() != FieldType::OLAP_FIELD_TYPE_MAP) {
189
0
            return Status::OK();
190
0
        }
191
0
        if (is_bf_column()) {
192
0
            return Status::NotSupported("Do not support bloom filter index, type={}",
193
0
                                        get_string_by_field_type(type()));
194
0
        }
195
0
        if (has_bitmap_index()) {
196
0
            return Status::NotSupported("Do not support bitmap index, type={}",
197
0
                                        get_string_by_field_type(type()));
198
0
        }
199
0
        return Status::OK();
200
0
    }
201
202
private:
203
    int32_t _unique_id = -1;
204
    std::string _col_name;
205
    std::string _col_name_lower_case;
206
    // the field _type will change from TPrimitiveType
207
    // to string by 'EnumToString(TPrimitiveType, tcolumn.column_type.type, data_type);' (reference: TabletMeta::init_column_from_tcolumn)
208
    // to FieldType by 'TabletColumn::get_field_type_by_string' (reference: TabletColumn::init_from_pb).
209
    // And the _type in columnPB is string and it changed from FieldType by 'get_string_by_field_type' (reference: TabletColumn::to_schema_pb).
210
    FieldType _type;
211
    bool _is_key = false;
212
    FieldAggregationMethod _aggregation;
213
    std::string _aggregation_name;
214
    bool _is_nullable = false;
215
    bool _is_auto_increment = false;
216
217
    bool _has_default_value = false;
218
    std::string _default_value;
219
220
    bool _is_decimal = false;
221
    int32_t _precision = -1;
222
    int32_t _frac = -1;
223
224
    int32_t _length = -1;
225
    int32_t _index_length = -1;
226
227
    bool _is_bf_column = false;
228
229
    bool _has_bitmap_index = false;
230
    bool _visible = true;
231
232
    std::vector<TabletColumnPtr> _sub_columns;
233
    uint32_t _sub_column_count = 0;
234
235
    bool _result_is_nullable = false;
236
    int _be_exec_version = -1;
237
238
    // The extracted sub-columns from "variant" contain the following information:
239
    int32_t _parent_col_unique_id = -1;     // "variant" -> col_unique_id
240
    vectorized::PathInDataPtr _column_path; // the path of the sub-columns themselves
241
242
    // Record information about columns merged into a sparse column within a variant
243
    // `{"id": 100, "name" : "jack", "point" : 3.9}`
244
    // If the information mentioned above is inserted into the variant column,
245
    // 'id' and 'name' are correctly extracted, while 'point' is merged into the sparse column due to its sparsity.
246
    // The path_info and type of 'point' will be recorded using the TabletColumn.
247
    // Use shared_ptr for reuse and reducing column memory usage
248
    std::vector<TabletColumnPtr> _sparse_cols;
249
    size_t _num_sparse_columns = 0;
250
};
251
252
bool operator==(const TabletColumn& a, const TabletColumn& b);
253
bool operator!=(const TabletColumn& a, const TabletColumn& b);
254
255
class TabletSchema;
256
257
class TabletIndex : public MetadataAdder<TabletIndex> {
258
public:
259
68
    TabletIndex() = default;
260
    void init_from_thrift(const TOlapTableIndex& index, const TabletSchema& tablet_schema);
261
    void init_from_thrift(const TOlapTableIndex& index, const std::vector<int32_t>& column_uids);
262
    void init_from_pb(const TabletIndexPB& index);
263
    void to_schema_pb(TabletIndexPB* index) const;
264
265
650
    int64_t index_id() const { return _index_id; }
266
0
    const std::string& index_name() const { return _index_name; }
267
2.37k
    IndexType index_type() const { return _index_type; }
268
1.43k
    const vector<int32_t>& col_unique_ids() const { return _col_unique_ids; }
269
723
    const std::map<string, string>& properties() const { return _properties; }
270
0
    int32_t get_gram_size() const {
271
0
        if (_properties.contains("gram_size")) {
272
0
            return std::stoi(_properties.at("gram_size"));
273
0
        }
274
275
0
        return 0;
276
0
    }
277
0
    int32_t get_gram_bf_size() const {
278
0
        if (_properties.contains("bf_size")) {
279
0
            return std::stoi(_properties.at("bf_size"));
280
0
        }
281
282
0
        return 0;
283
0
    }
284
285
1.24k
    const std::string& get_index_suffix() const { return _escaped_index_suffix_path; }
286
287
    void set_escaped_escaped_index_suffix_path(const std::string& name);
288
289
private:
290
    int64_t _index_id = -1;
291
    // Identify the different index with the same _index_id
292
    std::string _escaped_index_suffix_path;
293
    std::string _index_name;
294
    IndexType _index_type;
295
    std::vector<int32_t> _col_unique_ids;
296
    std::map<string, string> _properties;
297
};
298
299
class TabletSchema : public MetadataAdder<TabletSchema> {
300
public:
301
    enum ColumnType { NORMAL = 0, DROPPED = 1, VARIANT = 2 };
302
    // TODO(yingchun): better to make constructor as private to avoid
303
    // manually init members incorrectly, and define a new function like
304
    // void create_from_pb(const TabletSchemaPB& schema, TabletSchema* tablet_schema).
305
    TabletSchema();
306
    virtual ~TabletSchema();
307
308
    // Init from pb
309
    // ignore_extracted_columns: ignore the extracted columns from variant column
310
    // reuse_cached_column: reuse the cached column in the schema if they are the same, to reduce memory usage
311
    void init_from_pb(const TabletSchemaPB& schema, bool ignore_extracted_columns = false,
312
                      bool reuse_cached_column = false);
313
    // Notice: Use deterministic way to serialize protobuf,
314
    // since serialize Map in protobuf may could lead to un-deterministic by default
315
    template <class PbType>
316
1.45k
    static std::string deterministic_string_serialize(const PbType& pb) {
317
1.45k
        std::string output;
318
1.45k
        google::protobuf::io::StringOutputStream string_output_stream(&output);
319
1.45k
        google::protobuf::io::CodedOutputStream output_stream(&string_output_stream);
320
1.45k
        output_stream.SetSerializationDeterministic(true);
321
1.45k
        pb.SerializeToCodedStream(&output_stream);
322
1.45k
        return output;
323
1.45k
    }
_ZN5doris12TabletSchema30deterministic_string_serializeINS_14TabletSchemaPBEEENSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEERKT_
Line
Count
Source
316
1.33k
    static std::string deterministic_string_serialize(const PbType& pb) {
317
1.33k
        std::string output;
318
1.33k
        google::protobuf::io::StringOutputStream string_output_stream(&output);
319
1.33k
        google::protobuf::io::CodedOutputStream output_stream(&string_output_stream);
320
1.33k
        output_stream.SetSerializationDeterministic(true);
321
1.33k
        pb.SerializeToCodedStream(&output_stream);
322
1.33k
        return output;
323
1.33k
    }
_ZN5doris12TabletSchema30deterministic_string_serializeINS_8ColumnPBEEENSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEERKT_
Line
Count
Source
316
124
    static std::string deterministic_string_serialize(const PbType& pb) {
317
124
        std::string output;
318
124
        google::protobuf::io::StringOutputStream string_output_stream(&output);
319
124
        google::protobuf::io::CodedOutputStream output_stream(&string_output_stream);
320
124
        output_stream.SetSerializationDeterministic(true);
321
124
        pb.SerializeToCodedStream(&output_stream);
322
124
        return output;
323
124
    }
324
    void to_schema_pb(TabletSchemaPB* tablet_meta_pb) const;
325
    void append_column(TabletColumn column, ColumnType col_type = ColumnType::NORMAL);
326
    void append_index(TabletIndex&& index);
327
    void update_index(const TabletColumn& column, const IndexType& index_type, TabletIndex&& index);
328
    void remove_index(int64_t index_id);
329
    void clear_index();
330
    // Must make sure the row column is always the last column
331
    void add_row_column();
332
    void copy_from(const TabletSchema& tablet_schema);
333
    void update_index_info_from(const TabletSchema& tablet_schema);
334
    std::string to_key() const;
335
    // Don't use.
336
    // TODO: memory size of TabletSchema cannot be accurately tracked.
337
    // In some places, temporarily use num_columns() as TabletSchema size.
338
139
    int64_t mem_size() const { return _mem_size; }
339
    size_t row_size() const;
340
    int32_t field_index(const std::string& field_name) const;
341
    int32_t field_index(const vectorized::PathInData& path) const;
342
    int32_t field_index(int32_t col_unique_id) const;
343
    const TabletColumn& column(size_t ordinal) const;
344
    Result<const TabletColumn*> column(const std::string& field_name) const;
345
    Status have_column(const std::string& field_name) const;
346
    bool exist_column(const std::string& field_name) const;
347
    const TabletColumn& column_by_uid(int32_t col_unique_id) const;
348
    TabletColumn& mutable_column_by_uid(int32_t col_unique_id);
349
    TabletColumn& mutable_column(size_t ordinal);
350
    void replace_column(size_t pos, TabletColumn new_col);
351
    const std::vector<TabletColumnPtr>& columns() const;
352
622k
    size_t num_columns() const { return _num_columns; }
353
1.07M
    size_t num_key_columns() const { return _num_key_columns; }
354
134k
    const std::vector<uint32_t>& cluster_key_idxes() const { return _cluster_key_idxes; }
355
0
    size_t num_null_columns() const { return _num_null_columns; }
356
4.85k
    size_t num_short_key_columns() const { return _num_short_key_columns; }
357
0
    size_t num_rows_per_row_block() const { return _num_rows_per_row_block; }
358
574
    size_t num_variant_columns() const { return _num_variant_columns; };
359
7.09M
    KeysType keys_type() const { return _keys_type; }
360
4.95k
    SortType sort_type() const { return _sort_type; }
361
0
    size_t sort_col_num() const { return _sort_col_num; }
362
0
    CompressKind compress_kind() const { return _compress_kind; }
363
0
    size_t next_column_unique_id() const { return _next_column_unique_id; }
364
0
    bool has_bf_fpp() const { return _has_bf_fpp; }
365
0
    double bloom_filter_fpp() const { return _bf_fpp; }
366
10.7k
    bool is_in_memory() const { return _is_in_memory; }
367
0
    void set_is_in_memory(bool is_in_memory) { _is_in_memory = is_in_memory; }
368
0
    void set_disable_auto_compaction(bool disable_auto_compaction) {
369
0
        _disable_auto_compaction = disable_auto_compaction;
370
0
    }
371
76
    bool disable_auto_compaction() const { return _disable_auto_compaction; }
372
0
    void set_variant_enable_flatten_nested(bool flatten_nested) {
373
0
        _variant_enable_flatten_nested = flatten_nested;
374
0
    }
375
0
    bool variant_flatten_nested() const { return _variant_enable_flatten_nested; }
376
0
    void set_enable_single_replica_compaction(bool enable_single_replica_compaction) {
377
0
        _enable_single_replica_compaction = enable_single_replica_compaction;
378
0
    }
379
27
    bool enable_single_replica_compaction() const { return _enable_single_replica_compaction; }
380
    // indicate if full row store column(all the columns encodes as row) exists
381
0
    bool has_row_store_for_all_columns() const {
382
0
        return _store_row_column && row_columns_uids().empty();
383
0
    }
384
0
    void set_skip_write_index_on_load(bool skip) { _skip_write_index_on_load = skip; }
385
59
    bool skip_write_index_on_load() const { return _skip_write_index_on_load; }
386
4.15k
    int32_t delete_sign_idx() const { return _delete_sign_idx; }
387
0
    void set_delete_sign_idx(int32_t delete_sign_idx) { _delete_sign_idx = delete_sign_idx; }
388
139k
    bool has_sequence_col() const { return _sequence_col_idx != -1; }
389
64.7k
    int32_t sequence_col_idx() const { return _sequence_col_idx; }
390
0
    void set_version_col_idx(int32_t version_col_idx) { _version_col_idx = version_col_idx; }
391
0
    int32_t version_col_idx() const { return _version_col_idx; }
392
0
    bool has_skip_bitmap_col() const { return _skip_bitmap_col_idx != -1; }
393
0
    int32_t skip_bitmap_col_idx() const { return _skip_bitmap_col_idx; }
394
4.75k
    segment_v2::CompressionTypePB compression_type() const { return _compression_type; }
395
0
    void set_row_store_page_size(long page_size) { _row_store_page_size = page_size; }
396
0
    long row_store_page_size() const { return _row_store_page_size; }
397
0
    void set_storage_page_size(long storage_page_size) { _storage_page_size = storage_page_size; }
398
10.4k
    long storage_page_size() const { return _storage_page_size; }
399
400
15
    const std::vector<const TabletIndex*> inverted_indexes() const {
401
15
        std::vector<const TabletIndex*> inverted_indexes;
402
18
        for (const auto& index : _indexes) {
403
18
            if (index.index_type() == IndexType::INVERTED) {
404
17
                inverted_indexes.emplace_back(&index);
405
17
            }
406
18
        }
407
15
        return inverted_indexes;
408
15
    }
409
9.96k
    bool has_inverted_index() const {
410
9.96k
        for (const auto& index : _indexes) {
411
216
            if (index.index_type() == IndexType::INVERTED) {
412
216
                return true;
413
216
            }
414
216
        }
415
9.74k
        return false;
416
9.96k
    }
417
    bool has_inverted_index_with_index_id(int64_t index_id) const;
418
    // Check whether this column supports inverted index
419
    // Some columns (Float, Double, JSONB ...) from the variant do not support index, but they are listed in TabletIndex.
420
    const TabletIndex* inverted_index(const TabletColumn& col) const;
421
422
    // Regardless of whether this column supports inverted index
423
    // TabletIndex information will be returned as long as it exists.
424
    const TabletIndex* inverted_index(int32_t col_unique_id,
425
                                      const std::string& suffix_path = "") const;
426
    bool has_ngram_bf_index(int32_t col_unique_id) const;
427
    const TabletIndex* get_ngram_bf_index(int32_t col_unique_id) const;
428
    void update_indexes_from_thrift(const std::vector<doris::TOlapTableIndex>& indexes);
429
    // If schema version is not set, it should be -1
430
733
    int32_t schema_version() const { return _schema_version; }
431
    void clear_columns();
432
    vectorized::Block create_block(
433
            const std::vector<uint32_t>& return_columns,
434
            const std::unordered_set<uint32_t>* tablet_columns_need_convert_null = nullptr) const;
435
    vectorized::Block create_block(bool ignore_dropped_col = true) const;
436
0
    void set_schema_version(int32_t version) { _schema_version = version; }
437
0
    void set_auto_increment_column(const std::string& auto_increment_column) {
438
0
        _auto_increment_column = auto_increment_column;
439
0
    }
440
0
    std::string auto_increment_column() const { return _auto_increment_column; }
441
442
25
    void set_table_id(int64_t table_id) { _table_id = table_id; }
443
254
    int64_t table_id() const { return _table_id; }
444
25
    void set_db_id(int64_t db_id) { _db_id = db_id; }
445
0
    int64_t db_id() const { return _db_id; }
446
    void build_current_tablet_schema(int64_t index_id, int32_t version,
447
                                     const OlapTableIndexSchema* index,
448
                                     const TabletSchema& out_tablet_schema);
449
450
    // Merge columns that not exit in current schema, these column is dropped in current schema
451
    // but they are useful in some cases. For example,
452
    // 1. origin schema is  ColA, ColB
453
    // 2. insert values     1, 2
454
    // 3. delete where ColB = 2
455
    // 4. drop ColB
456
    // 5. insert values  3
457
    // 6. add column ColB, although it is name ColB, but it is different with previous ColB, the new ColB we name could call ColB'
458
    // 7. insert value  4, 5
459
    // Then the read schema should be ColA, ColB, ColB' because the delete predicate need ColB to remove related data.
460
    // Because they have same name, so that the dropped column should not be added to the map, only with unique id.
461
    void merge_dropped_columns(const TabletSchema& src_schema);
462
463
    bool is_dropped_column(const TabletColumn& col) const;
464
465
    // copy extracted columns from src_schema
466
    void copy_extracted_columns(const TabletSchema& src_schema);
467
468
    // only reserve extracted columns
469
    void reserve_extracted_columns();
470
471
0
    string get_all_field_names() const {
472
0
        string str = "[";
473
0
        for (auto p : _field_name_to_index) {
474
0
            if (str.size() > 1) {
475
0
                str += ", ";
476
0
            }
477
0
            str += p.first.to_string() + "(" + std::to_string(_cols[p.second]->unique_id()) + ")";
478
0
        }
479
0
        str += "]";
480
0
        return str;
481
0
    }
482
483
    // Dump [(name, type, is_nullable), ...]
484
0
    string dump_structure() const {
485
0
        string str = "[";
486
0
        for (auto p : _cols) {
487
0
            if (str.size() > 1) {
488
0
                str += ", ";
489
0
            }
490
0
            str += "(";
491
0
            str += p->name();
492
0
            str += ", ";
493
0
            str += TabletColumn::get_string_by_field_type(p->type());
494
0
            str += ", ";
495
0
            str += "is_nullable:";
496
0
            str += (p->is_nullable() ? "true" : "false");
497
0
            str += ")";
498
0
        }
499
0
        str += "]";
500
0
        return str;
501
0
    }
502
503
0
    string dump_full_schema() const {
504
0
        string str = "[";
505
0
        for (auto p : _cols) {
506
0
            if (str.size() > 1) {
507
0
                str += ", ";
508
0
            }
509
0
            ColumnPB col_pb;
510
0
            p->to_schema_pb(&col_pb);
511
0
            str += "(";
512
0
            str += col_pb.ShortDebugString();
513
0
            str += ")";
514
0
        }
515
0
        str += "]";
516
0
        return str;
517
0
    }
518
519
    vectorized::Block create_block_by_cids(const std::vector<uint32_t>& cids) const;
520
521
    std::shared_ptr<TabletSchema> copy_without_variant_extracted_columns();
522
5.45k
    InvertedIndexStorageFormatPB get_inverted_index_storage_format() const {
523
5.45k
        return _inverted_index_storage_format;
524
5.45k
    }
525
526
    void update_tablet_columns(const TabletSchema& tablet_schema,
527
                               const std::vector<TColumn>& t_columns);
528
529
0
    const std::vector<int32_t>& row_columns_uids() const { return _row_store_column_unique_ids; }
530
531
    int64_t get_metadata_size() const override;
532
533
private:
534
    friend bool operator==(const TabletSchema& a, const TabletSchema& b);
535
    friend bool operator!=(const TabletSchema& a, const TabletSchema& b);
536
537
    void clear_column_cache_handlers();
538
539
    KeysType _keys_type = DUP_KEYS;
540
    SortType _sort_type = SortType::LEXICAL;
541
    size_t _sort_col_num = 0;
542
    std::vector<TabletColumnPtr> _cols;
543
    std::vector<Cache::Handle*> _column_cache_handlers;
544
545
    std::vector<TabletIndex> _indexes;
546
    std::unordered_map<StringRef, int32_t, StringRefHash> _field_name_to_index;
547
    std::unordered_map<int32_t, int32_t> _field_id_to_index;
548
    std::unordered_map<vectorized::PathInDataRef, int32_t, vectorized::PathInDataRef::Hash>
549
            _field_path_to_index;
550
    size_t _num_columns = 0;
551
    size_t _num_variant_columns = 0;
552
    size_t _num_key_columns = 0;
553
    std::vector<uint32_t> _cluster_key_idxes;
554
    size_t _num_null_columns = 0;
555
    size_t _num_short_key_columns = 0;
556
    size_t _num_rows_per_row_block = 0;
557
    CompressKind _compress_kind = COMPRESS_NONE;
558
    segment_v2::CompressionTypePB _compression_type = segment_v2::CompressionTypePB::LZ4F;
559
    long _row_store_page_size = segment_v2::ROW_STORE_PAGE_SIZE_DEFAULT_VALUE;
560
    long _storage_page_size = segment_v2::STORAGE_PAGE_SIZE_DEFAULT_VALUE;
561
    size_t _next_column_unique_id = 0;
562
    std::string _auto_increment_column;
563
564
    bool _has_bf_fpp = false;
565
    double _bf_fpp = 0;
566
    bool _is_in_memory = false;
567
    int32_t _delete_sign_idx = -1;
568
    int32_t _sequence_col_idx = -1;
569
    int32_t _version_col_idx = -1;
570
    int32_t _skip_bitmap_col_idx = -1;
571
    int32_t _schema_version = -1;
572
    int64_t _table_id = -1;
573
    int64_t _db_id = -1;
574
    bool _disable_auto_compaction = false;
575
    bool _enable_single_replica_compaction = false;
576
    int64_t _mem_size = 0;
577
    bool _store_row_column = false;
578
    bool _skip_write_index_on_load = false;
579
    InvertedIndexStorageFormatPB _inverted_index_storage_format = InvertedIndexStorageFormatPB::V1;
580
581
    // Contains column ids of which columns should be encoded into row store.
582
    // ATTN: For compability reason empty cids means all columns of tablet schema are encoded to row column
583
    std::vector<int32_t> _row_store_column_unique_ids;
584
    bool _variant_enable_flatten_nested = false;
585
    int64_t _vl_field_mem_size {0}; // variable length field
586
};
587
588
bool operator==(const TabletSchema& a, const TabletSchema& b);
589
bool operator!=(const TabletSchema& a, const TabletSchema& b);
590
591
using TabletSchemaSPtr = std::shared_ptr<TabletSchema>;
592
593
} // namespace doris