Coverage Report

Created: 2026-09-29 06:08

next uncovered line (L), next uncovered region (R), next uncovered branch (B)
be/src/format_v2/table_reader.h
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
#pragma once
19
20
#include <bvar/status.h>
21
22
#include <algorithm>
23
#include <exception>
24
#include <map>
25
#include <memory>
26
#include <optional>
27
#include <string>
28
#include <string_view>
29
#include <utility>
30
#include <vector>
31
32
#include "common/cast_set.h"
33
#include "common/exception.h"
34
#include "common/logging.h"
35
#include "common/status.h"
36
#include "core/assert_cast.h"
37
#include "core/block/block.h"
38
#include "core/column/column_array.h"
39
#include "core/column/column_const.h"
40
#include "core/column/column_map.h"
41
#include "core/column/column_nullable.h"
42
#include "core/column/column_struct.h"
43
#include "core/column/column_vector.h"
44
#include "core/data_type/data_type.h"
45
#include "core/data_type/data_type_array.h"
46
#include "core/data_type/data_type_map.h"
47
#include "core/data_type/data_type_nullable.h"
48
#include "core/data_type/data_type_number.h"
49
#include "core/data_type/data_type_string.h"
50
#include "core/data_type/data_type_struct.h"
51
#include "core/field.h"
52
#include "exec/common/stringop_substring.h"
53
#include "exprs/vexpr.h"
54
#include "exprs/vexpr_context.h"
55
#include "exprs/vexpr_fwd.h"
56
#include "exprs/vslot_ref.h"
57
#include "format/table/deletion_vector.h"
58
#include "format_v2/column_data.h"
59
#include "format_v2/column_mapper.h"
60
#include "format_v2/expr/cast.h"
61
#include "format_v2/expr/delete_predicate.h"
62
#include "format_v2/file_reader.h"
63
#include "format_v2/parquet/reader/column_reader.h"
64
#include "format_v2/schema_projection.h"
65
#include "gen_cpp/PlanNodes_types.h"
66
#include "io/io_common.h"
67
#include "runtime/descriptors.h"
68
#include "storage/segment/condition_cache.h"
69
70
namespace doris {
71
class Block;
72
struct DeleteFileDesc;
73
class RuntimeState;
74
} // namespace doris
75
76
namespace doris::format {
77
78
using DeleteRows = std::vector<int64_t>;
79
80
// Row-level predicates on table/global schema. They are rewritten to file-local expressions when
81
// possible, and remain the source of row-level filtering after localization.
82
struct TableFilter {
83
    VExprContextSPtr conjunct;
84
    std::vector<GlobalIndex> global_indices;
85
    bool metadata_pruning_safe = true;
86
};
87
88
struct ScanTask {
89
1.02k
    virtual ~ScanTask() = default;
90
91
    std::unique_ptr<io::FileDescription> data_file;
92
};
93
94
struct ProjectedColumnBuildContext {
95
    const TFileScanRangeParams* scan_params = nullptr;
96
    const TFileRangeDesc* range = nullptr;
97
    RuntimeState* runtime_state = nullptr;
98
    const SlotDescriptor* slot_desc = nullptr;
99
    std::optional<ColumnDefinition> schema_column = std::nullopt;
100
    size_t next_file_column_idx = 0;
101
};
102
103
struct ReadProfile {
104
    RuntimeProfile::Counter* total_timer = nullptr;
105
    RuntimeProfile::Counter* init_timer = nullptr;
106
    RuntimeProfile::Counter* num_delete_files = nullptr;
107
    RuntimeProfile::Counter* num_delete_rows = nullptr;
108
    RuntimeProfile::Counter* parse_delete_file_time = nullptr;
109
    RuntimeProfile::Counter* equality_delete_index_cache_hit_count = nullptr;
110
    RuntimeProfile::Counter* equality_delete_index_cache_miss_count = nullptr;
111
    RuntimeProfile::Counter* equality_delete_hash_index_memory = nullptr;
112
    RuntimeProfile::Counter* decoded_dv_cache_hit_count = nullptr;
113
    RuntimeProfile::Counter* decoded_dv_cache_miss_count = nullptr;
114
    RuntimeProfile::Counter* dv_file_cache_hit_count = nullptr;
115
    RuntimeProfile::Counter* dv_file_cache_miss_count = nullptr;
116
    RuntimeProfile::Counter* dv_file_cache_peer_read_count = nullptr;
117
    RuntimeProfile::Counter* exec_timer = nullptr;
118
    RuntimeProfile::Counter* prepare_split_timer = nullptr;
119
    RuntimeProfile::Counter* finalize_timer = nullptr;
120
    RuntimeProfile::Counter* create_reader_timer = nullptr;
121
    RuntimeProfile::Counter* pushdown_agg_timer = nullptr;
122
    RuntimeProfile::Counter* open_reader_timer = nullptr;
123
    RuntimeProfile::Counter* refresh_conjuncts_timer = nullptr;
124
    RuntimeProfile::Counter* runtime_filter_partition_prune_timer = nullptr;
125
    RuntimeProfile::Counter* runtime_filter_partition_pruned_range_counter = nullptr;
126
    RuntimeProfile::Counter* close_timer = nullptr;
127
    RuntimeProfile::Counter* file_reader_total_timer = nullptr;
128
    RuntimeProfile::Counter* file_reader_init_timer = nullptr;
129
    RuntimeProfile::Counter* file_reader_schema_timer = nullptr;
130
    RuntimeProfile::Counter* file_reader_mapper_timer = nullptr;
131
    RuntimeProfile::Counter* file_reader_open_timer = nullptr;
132
    RuntimeProfile::Counter* file_reader_refresh_timer = nullptr;
133
    RuntimeProfile::Counter* file_reader_get_block_timer = nullptr;
134
    RuntimeProfile::Counter* file_reader_aggregate_timer = nullptr;
135
    RuntimeProfile::Counter* file_reader_close_timer = nullptr;
136
};
137
138
struct TableReadOptions {
139
    // Columns need to be read from file and output by table reader. They are all in table/global
140
    // schema semantics.
141
    const std::vector<ColumnDefinition> projected_columns;
142
    // All complex conjuncts from scan operator
143
    const VExprContextSPtrs conjuncts;
144
    // File format of the underlying data files, needed for reader initialization and reader-level
145
    // filter pushdown.
146
    const FileFormat format;
147
    TFileScanRangeParams* scan_params;
148
    std::shared_ptr<io::IOContext> io_ctx;
149
    RuntimeState* runtime_state;
150
    RuntimeProfile* scanner_profile;
151
    // File formats without complete self-describing metadata, such as CSV, Text, and JSON, need
152
    // the FE-planned physical file slots to build their file-local schema and deserialize values.
153
    const std::vector<SlotDescriptor*>* file_slot_descs = nullptr;
154
    // Push-down aggregate type.
155
    const TPushAggOp::type push_down_agg_type = TPushAggOp::type::NONE;
156
    // Table/global indices of explicit COUNT arguments. nullopt means an old FE did not send the
157
    // semantic argument field, while an explicit empty vector means COUNT(*)/COUNT(1). Keeping
158
    // those states separate prevents a rolling-upgrade plan from being reinterpreted by a new BE.
159
    const std::optional<std::vector<GlobalIndex>> push_down_count_columns = std::nullopt;
160
    // Initial digest of predicates available during scanner open. Scanner-driven splits override it
161
    // with SplitReadOptions::condition_cache_digest after collecting late-arrival runtime filters.
162
    // A zero digest disables condition cache.
163
    uint64_t condition_cache_digest = 0;
164
};
165
166
struct SplitReadOptions {
167
    // Split-level information for reader initialization, which may include file path, partition values, delete file info, etc. The content is table format specific and opaque to table reader base class; it's the responsibility of the concrete table reader implementation to parse necessary information for reader initialization and filter pushdown.
168
    std::map<std::string, Field> partition_values;
169
    // Latest scanner conjuncts rewritten to table/global column indices. Runtime filters may
170
    // arrive after TableReader::init(), so scanner-driven splits replace the initial snapshot.
171
    // nullopt preserves the initial snapshot for standalone TableReader callers.
172
    std::optional<VExprContextSPtrs> conjuncts = std::nullopt;
173
    // Independent clones used for partition pruning because evaluation prepares and opens them
174
    // against a synthetic partition block before the file reader opens its row-level conjuncts.
175
    VExprContextSPtrs partition_prune_conjuncts;
176
    // Table-level COUNT may emit one metadata-derived batch and resume on a later scheduler turn.
177
    // It is safe only after every runtime filter assigned to the scanner has arrived; otherwise a
178
    // filter could arrive after synthetic rows have already been returned and those rows cannot be
179
    // retracted. Standalone TableReader callers have no scanner runtime-filter lifecycle.
180
    bool all_runtime_filters_applied = true;
181
    // Digest for the exact scanner conjunct snapshot attached to this split. FileScannerV2 rebuilds
182
    // it after collecting late-arrival RFs, so different RF payloads cannot share a cache entry. A
183
    // zero value explicitly disables condition cache for this split.
184
    std::optional<uint64_t> condition_cache_digest;
185
    ShardedKVCache* cache = nullptr;
186
    TFileRangeDesc current_range;
187
    FileFormat current_split_format = FileFormat::PARQUET;
188
    std::optional<GlobalRowIdContext> global_rowid_context;
189
    // Optional absolute file-row selection used by TopN two-phase materialization. TableReader
190
    // carries it unchanged into the format-neutral FileScanRequest.
191
    std::optional<std::vector<int64_t>> row_ids = std::nullopt;
192
};
193
194
// Base class for table-level readers.
195
// This layer owns common table-level orchestration, such as split iteration, dynamic partition
196
// pruning, delete handling and conversion from file-local blocks to table-schema blocks. Concrete
197
// table-format readers only need to provide format-specific hooks for opening readers and parsing
198
// split metadata.
199
class TableReader {
200
public:
201
1.30k
    virtual ~TableReader() = default;
202
203
    // Initialize common runtime options for the table reader. Subclasses may call this from their
204
    // own init(options); table-format schema and split metadata are provided later per split.
205
    virtual Status init(TableReadOptions&& options);
206
207
    // FileScannerV2 adjusts this before each get_block() using an adaptive bytes-per-row estimate.
208
    // Store it here as well as forwarding to the current reader so newly opened split readers start
209
    // with the latest predicted batch size.
210
1.92k
    virtual void set_batch_size(size_t batch_size) {
211
1.92k
        _batch_size = std::max<size_t>(1, batch_size);
212
1.92k
        if (_data_reader.reader != nullptr) {
213
754
            _data_reader.reader->set_batch_size(_batch_size);
214
754
        }
215
1.92k
    }
216
217
#ifdef BE_TEST
218
    size_t TEST_batch_size() const { return _batch_size; }
219
    void TEST_set_condition_cache_hit_count(int64_t hits) { _condition_cache_hit_count = hits; }
220
    bool TEST_current_data_file_is_immutable() const {
221
        DORIS_CHECK(_current_task != nullptr);
222
        DORIS_CHECK(_current_task->data_file != nullptr);
223
        DORIS_CHECK(_current_file_description.has_value());
224
        DORIS_CHECK(_current_task->data_file->is_immutable ==
225
                    _current_file_description->is_immutable);
226
        return _current_task->data_file->is_immutable;
227
    }
228
#endif
229
230
    // Prepare for reading a new split/task.
231
    // 1. Pass a new split/task to reader, which will be used in subsequent open_reader() to initialize the underlying file reader.
232
    // 2. Parse delete predicates from split/task information, which will be used for later dynamic filtering and delete handling.
233
    virtual Status prepare_split(const SplitReadOptions& options);
234
235
    // Refresh row-level predicates for an already prepared split. Physical readers that support
236
    // this operation decide the safe boundary at which the new immutable request becomes active.
237
    virtual Status refresh_conjuncts(VExprContextSPtrs conjuncts);
238
239
797
    virtual bool current_split_pruned() const { return _current_split_pruned; }
240
2.08k
    virtual bool current_split_uses_metadata_count() const {
241
2.08k
        return _current_split_uses_metadata_count;
242
2.08k
    }
243
244
    // Discard the active split after the caller decides an error is ignorable, for example a
245
    // stale external-table file listing that returns NOT_FOUND. The next prepare_split() must start
246
    // with no concrete reader or split-local state left from the failed split.
247
2
    virtual Status abort_split() {
248
        // Ignored open failures still spend time closing partially initialized readers. Include
249
        // that recovery path in the common lifecycle profile so NOT_FOUND cannot become invisible.
250
2
        SCOPED_TIMER(_profile.total_timer);
251
2
        SCOPED_TIMER(_profile.close_timer);
252
2
        if (_data_reader.reader != nullptr) {
253
2
            RETURN_IF_ERROR(close_current_reader());
254
2
        } else {
255
0
            _current_task.reset();
256
0
            _current_file_description.reset();
257
0
        }
258
2
        _delete_rows = nullptr;
259
2
        _remaining_table_level_count = -1;
260
2
        _remaining_file_level_count = -1;
261
2
        _current_split_uses_metadata_count = false;
262
2
        _current_split_pruned = false;
263
2
        return Status::OK();
264
2
    }
265
266
    // Public entry point for reading a table-schema block. The base class opens the current reader,
267
    // advances across EOF, and closes exhausted readers. Subclasses provide protected hooks for
268
    // table-format-specific behavior.
269
1.89k
    virtual Status get_block(Block* block, bool* eos) {
270
1.89k
        SCOPED_TIMER(_profile.total_timer);
271
1.89k
        SCOPED_TIMER(_profile.exec_timer);
272
1.89k
        DORIS_CHECK(block->columns() == _projected_columns.size());
273
1.89k
        block->clear_column_data(_projected_columns.size());
274
275
2.62k
        while (true) {
276
2.62k
            if (*eos) {
277
0
                return Status::OK();
278
0
            }
279
2.62k
            if (_io_ctx != nullptr && _io_ctx->should_stop) {
280
0
                *eos = true;
281
0
                return Status::OK();
282
0
            }
283
2.62k
            if (!_data_reader.reader) {
284
1.69k
                if (_is_table_level_count_active()) {
285
12
                    RETURN_IF_ERROR(_read_table_level_count(block, eos));
286
12
                    return Status::OK();
287
12
                }
288
1.68k
                if (_is_file_level_count_active()) {
289
8
                    RETURN_IF_ERROR(_read_file_level_count(block, eos));
290
8
                    return Status::OK();
291
8
                }
292
1.67k
                RETURN_IF_ERROR(create_next_reader(eos));
293
1.67k
                if (!_data_reader.reader) {
294
722
                    DCHECK(*eos);
295
722
                    return Status::OK();
296
722
                }
297
1.67k
            }
298
299
            // Materialize a reduced row set for upper aggregate operators when aggregate
300
            // pushdown can be applied. This is not the final aggregate result: COUNT emits
301
            // `count` default rows for the upper COUNT(*), and MIN/MAX emits two rows containing
302
            // file-level min/max values for the upper MIN/MAX.
303
1.87k
            if (!_aggregate_pushdown_tried) {
304
949
                SCOPED_TIMER(_profile.pushdown_agg_timer);
305
949
                bool pushed_down = false;
306
949
                const auto status = _try_materialize_aggregate_pushdown_rows(block, &pushed_down);
307
949
                if (!status.ok()) {
308
2
                    if (_io_ctx != nullptr && _io_ctx->should_stop &&
309
2
                        status.is<ErrorCode::END_OF_FILE>()) {
310
2
                        *eos = true;
311
2
                        return Status::OK();
312
2
                    }
313
0
                    return status;
314
2
                }
315
947
                if (pushed_down) {
316
22
                    return Status::OK();
317
22
                }
318
947
            }
319
320
1.84k
            bool current_eof = false;
321
1.84k
            _data_reader.block_template.clear_column_data(
322
1.84k
                    cast_set<int64_t>(_data_reader.file_block_layout.size()));
323
1.84k
            size_t current_rows = 0;
324
1.84k
            {
325
1.84k
                SCOPED_TIMER(_profile.file_reader_total_timer);
326
1.84k
                SCOPED_TIMER(_profile.file_reader_get_block_timer);
327
1.84k
                RETURN_IF_ERROR(_data_reader.reader->get_block(&_data_reader.block_template,
328
1.84k
                                                               &current_rows, &current_eof));
329
1.84k
            }
330
1.84k
            const bool stopped_during_read = _io_ctx != nullptr && _io_ctx->should_stop;
331
1.84k
            if (current_rows == 0) {
332
722
                if (current_eof) {
333
718
                    _current_reader_reached_eof = !stopped_during_read;
334
718
                    RETURN_IF_ERROR(close_current_reader());
335
718
                }
336
722
                continue;
337
722
            }
338
1.84k
            DCHECK_EQ(_data_reader.block_template.columns(), _data_reader.file_block_layout.size())
339
0
                    << _data_reader.block_template.dump_structure();
340
1.12k
#ifndef NDEBUG
341
1.12k
            RETURN_IF_ERROR(_check_file_block_columns("after file reader get_block", current_rows));
342
1.12k
#endif
343
1.12k
            DORIS_CHECK(block->columns() == _data_reader.column_mapper->mappings().size());
344
1.12k
            RETURN_IF_ERROR(finalize_chunk(block, current_rows));
345
1.12k
#ifndef NDEBUG
346
1.12k
            RETURN_IF_ERROR(
347
1.12k
                    _check_table_block_columns("after finalize_chunk", block, current_rows));
348
1.12k
#endif
349
1.12k
            if (current_eof) {
350
62
                _current_reader_reached_eof = !stopped_during_read;
351
62
                RETURN_IF_ERROR(close_current_reader());
352
62
            }
353
1.12k
            return Status::OK();
354
1.12k
        }
355
1.89k
    }
356
357
    // Close the table reader and the currently active file reader. Subclasses that hold additional
358
    // table-format resources should override this and call TableReader::close() first.
359
966
    virtual Status close() {
360
966
        SCOPED_TIMER(_profile.total_timer);
361
966
        SCOPED_TIMER(_profile.close_timer);
362
966
        if (_data_reader.reader) {
363
150
            RETURN_IF_ERROR(close_current_reader());
364
150
        }
365
966
        _current_task.reset();
366
966
        _current_file_description.reset();
367
966
        _remaining_table_level_count = -1;
368
966
        _remaining_file_level_count = -1;
369
966
        _current_split_uses_metadata_count = false;
370
966
        return Status::OK();
371
966
    }
372
373
1.14k
    virtual int64_t condition_cache_hit_count() const { return _condition_cache_hit_count; }
374
375
    virtual std::string debug_string() const;
376
377
    virtual Status annotate_projected_column(const TFileScanSlotInfo& slot_info,
378
                                             ProjectedColumnBuildContext* context,
379
                                             ColumnDefinition* column) const;
380
381
576
    virtual Status validate_projected_columns(const ProjectedColumnBuildContext& context) const {
382
576
        (void)context;
383
576
        return Status::OK();
384
576
    }
385
386
protected:
387
    // TableReader keeps the active file description both in the scan task and separately for
388
    // creating the physical reader. Table-format readers must update both copies when their
389
    // snapshot protocol guarantees that a file path is never overwritten with different bytes.
390
    // This guarantee lets readers safely build cache keys without mtime; it must not be used for
391
    // ordinary Hive/TVF files whose paths may be overwritten in place.
392
218
    void mark_current_data_file_immutable() {
393
218
        DORIS_CHECK(_current_task != nullptr);
394
218
        DORIS_CHECK(_current_task->data_file != nullptr);
395
218
        DORIS_CHECK(_current_file_description.has_value());
396
218
        _current_task->data_file->is_immutable = true;
397
218
        _current_file_description->is_immutable = true;
398
218
    }
399
400
    std::optional<ColumnDefinition> _find_table_column_by_field_id(
401
            int32_t field_id, DataTypePtr type, bool include_historical_schemas) const;
402
    std::optional<std::vector<ColumnDefinition>> _find_table_column_path_by_field_id(
403
            int32_t field_id, DataTypePtr leaf_type, bool include_historical_schemas) const;
404
    std::optional<std::vector<ColumnDefinition>> _find_table_column_identity_path_by_field_id(
405
            int32_t field_id, bool include_historical_schemas) const;
406
407
    // Parse deletion vector information from table format specific file description.
408
    virtual Status _parse_deletion_vector_file(const TTableFormatFileDesc& t_desc,
409
814
                                               DeleteFileDesc* desc, bool* has_delete_file) {
410
814
        *has_delete_file = false;
411
814
        return Status::OK();
412
814
    }
413
414
    // Advance to the next reader. This closes the current reader first and then opens the next
415
    // concrete reader. Subclasses should not duplicate this loop.
416
    Status create_next_reader(bool* eos);
417
    virtual Status create_file_reader(std::unique_ptr<FileReader>* reader);
418
771
    virtual TableColumnMappingMode mapping_mode() const { return TableColumnMappingMode::BY_NAME; }
419
774
    virtual void configure_mapper_options(TableColumnMapperOptions*) const {}
420
940
    virtual Status annotate_file_schema(std::vector<ColumnDefinition>* file_schema) {
421
940
        DORIS_CHECK(file_schema != nullptr);
422
940
        return Status::OK();
423
940
    }
424
791
    virtual Status validate_file_mapping(const TableColumnMapper&) const { return Status::OK(); }
425
426
    // Open the concrete reader for the current split/task and build the file-local scan request.
427
955
    virtual Status open_reader() {
428
955
        SCOPED_TIMER(_profile.open_reader_timer);
429
        // 1. Get file schema and create column mapping.
430
955
        std::vector<ColumnDefinition> file_schema;
431
955
        {
432
955
            SCOPED_TIMER(_profile.file_reader_total_timer);
433
955
            SCOPED_TIMER(_profile.file_reader_schema_timer);
434
955
            RETURN_IF_ERROR(_data_reader.reader->get_schema(&file_schema));
435
955
        }
436
        // For Paimon/Hudi, FE can provide field ids through `history_schema_info`. Annotate the
437
        // file schema before column mapping when the table format maps columns by field id.
438
955
        RETURN_IF_ERROR(annotate_file_schema(&file_schema));
439
955
        _data_reader.file_schema = file_schema;
440
955
        _mapper_options.mode = mapping_mode();
441
955
        configure_mapper_options(&_mapper_options);
442
443
955
        {
444
955
            SCOPED_TIMER(_profile.file_reader_total_timer);
445
955
            SCOPED_TIMER(_profile.file_reader_mapper_timer);
446
955
            _data_reader.column_mapper = _data_reader.reader->create_column_mapper(_mapper_options);
447
955
        }
448
955
        DORIS_CHECK(_data_reader.column_mapper != nullptr);
449
955
        RETURN_IF_ERROR(_data_reader.column_mapper->create_mapping(_projected_columns,
450
955
                                                                   _partition_values, file_schema));
451
955
        DORIS_CHECK(_data_reader.column_mapper->mappings().size() == _projected_columns.size());
452
453
        // 2. Build table filters based on conjuncts and column predicates.
454
955
        RETURN_IF_ERROR(_build_table_filters_from_conjuncts());
455
456
        // 3. Create file scan request based on column mapping and table filters, then open file
457
        // reader with the request. File scan request carries row-level expression filters and
458
        // file-level pruning hints. Only expression filters decide returned rows.
459
955
        auto file_request = std::make_shared<FileScanRequest>();
460
955
        RETURN_IF_ERROR(_data_reader.column_mapper->create_scan_request(
461
955
                _table_filters, _projected_columns, file_request.get(), _runtime_state));
462
955
        file_request->row_ids = _row_ids;
463
955
        _constant_pruning_safe_filter_count =
464
955
                std::min(_constant_pruning_safe_filter_count,
465
955
                         file_request->constant_pruning_safe_table_filter_count);
466
955
        bool constant_filter_pruned_split = false;
467
955
        RETURN_IF_ERROR(_evaluate_constant_filters(&constant_filter_pruned_split));
468
955
        if (constant_filter_pruned_split) {
469
2
            RETURN_IF_ERROR(close_current_reader());
470
2
            return Status::OK();
471
2
        }
472
        // COUNT(*) has no semantic column argument, but Nereids retains a minimum-width scan slot
473
        // so the scan node still has an output tuple. Record only the current non-predicate file
474
        // columns before table-format hooks add row-position or equality-delete dependencies. This
475
        // marker is independent of aggregate eligibility: with position deletes, for example,
476
        // metadata COUNT must fall back to reading rows, but an arbitrary unsupported TIME_MILLIS
477
        // placeholder still must not be validated or decoded merely to carry the surviving count.
478
        // Pending runtime filters may later target this retained slot, so placeholder values are
479
        // safe only after every filter for the split has arrived.
480
953
        if (_push_down_agg_type == TPushAggOp::type::COUNT &&
481
953
            _push_down_count_columns.has_value() && _push_down_count_columns->empty() &&
482
953
            _all_runtime_filters_applied_for_split) {
483
24
            file_request->count_star_placeholder_columns.reserve(
484
24
                    file_request->non_predicate_columns.size());
485
24
            for (const auto& column : file_request->non_predicate_columns) {
486
18
                file_request->count_star_placeholder_columns.push_back(column.column_id());
487
18
            }
488
24
        }
489
953
        RETURN_IF_ERROR(customize_file_scan_request(file_request.get()));
490
951
        RETURN_IF_ERROR(_data_reader.column_mapper->reconcile_scan_request_after_customization(
491
951
                file_request.get()));
492
951
        RETURN_IF_ERROR(validate_file_mapping(*_data_reader.column_mapper));
493
949
        RETURN_IF_ERROR(_open_local_filter_exprs(*file_request));
494
949
        _data_reader.file_block_layout.clear();
495
949
        _data_reader.block_template.clear();
496
949
        _file_scan_request.reset();
497
949
        _data_reader.file_block_layout.resize(file_request->block_column_count());
498
499
        // 4. Build file block layout from file schema and column mapping. The layout describes
500
        // the block returned by file reader before table-column materialization.
501
949
        auto add_file_block_column = [&](const LocalColumnIndex& projection,
502
4.22k
                                         LocalIndex block_position) -> Status {
503
4.22k
            DORIS_CHECK(block_position.value() < _data_reader.file_block_layout.size());
504
4.22k
            const auto file_column_id = projection.column_id();
505
4.22k
            const auto* field = _find_column_definition(_data_reader.file_schema, file_column_id);
506
4.22k
            DORIS_CHECK(field != nullptr);
507
508
4.22k
            ColumnDefinition projected_field;
509
4.22k
            RETURN_IF_ERROR(project_column_definition(*field, projection, &projected_field));
510
4.22k
            _data_reader.file_block_layout[block_position.value()] = {
511
4.22k
                    .file_column_id = file_column_id,
512
4.22k
                    .name = projected_field.name,
513
4.22k
                    .type = projected_field.type,
514
4.22k
            };
515
4.22k
            DORIS_CHECK(_data_reader.file_block_layout[block_position.value()].type != nullptr);
516
4.22k
            return Status::OK();
517
4.22k
        };
518
949
        for (const auto& projection : file_request->predicate_columns) {
519
221
            RETURN_IF_ERROR(add_file_block_column(
520
221
                    projection, file_request->local_positions.at(projection.column_id())));
521
221
        }
522
3.99k
        for (const auto& projection : file_request->non_predicate_columns) {
523
3.99k
            RETURN_IF_ERROR(add_file_block_column(
524
3.99k
                    projection, file_request->non_predicate_position(projection.column_id())));
525
3.99k
        }
526
527
        // 5. Prepare block template from file block layout. The block template stores the block
528
        // returned by file reader before table-column materialization.
529
949
        _data_reader.block_template.reserve(_data_reader.file_block_layout.size());
530
4.22k
        for (const auto& column : _data_reader.file_block_layout) {
531
4.22k
            _data_reader.block_template.insert(
532
4.22k
                    {column.type->create_column(), column.type, column.name});
533
4.22k
        }
534
949
        if (VLOG_DEBUG_IS_ON) {
535
0
            VLOG_DEBUG << "TableReader debug: " << debug_string();
536
0
        }
537
949
        RETURN_IF_ERROR(_open_mapping_exprs());
538
949
        {
539
949
            SCOPED_TIMER(_profile.file_reader_total_timer);
540
949
            SCOPED_TIMER(_profile.file_reader_open_timer);
541
949
            RETURN_IF_ERROR(_data_reader.reader->open(file_request));
542
949
        }
543
949
        _file_scan_request = std::move(file_request);
544
949
        RETURN_IF_ERROR(_init_reader_condition_cache(*_file_scan_request));
545
949
        return Status::OK();
546
949
    }
547
548
    Status _build_table_filters_from_conjuncts();
549
    Status _evaluate_partition_prune_conjuncts(const VExprContextSPtrs& conjuncts,
550
                                               bool* can_filter_all);
551
    static bool _is_safe_to_pre_execute(const VExprContextSPtr& conjunct);
552
    Status _build_partition_prune_block(Block* block) const;
553
    Status _open_local_filter_exprs(const FileScanRequest& file_request);
554
    Status _init_reader_condition_cache(const FileScanRequest& file_request);
555
    void _finalize_reader_condition_cache();
556
    bool _should_enable_condition_cache(const FileScanRequest& file_request) const;
557
558
956
    Status _evaluate_constant_filters(bool* can_filter_all) {
559
956
        DORIS_CHECK(can_filter_all != nullptr);
560
956
        DORIS_CHECK_LE(_constant_pruning_safe_filter_count, _table_filters.size());
561
956
        *can_filter_all = false;
562
        // The bound was derived from the original `_conjuncts` order, which includes slotless
563
        // expressions omitted from `_table_filters`. Iterating only this prefix therefore cannot
564
        // skip an unsafe row-level predicate and pre-execute a later constant predicate.
565
1.04k
        for (size_t i = 0; i < _constant_pruning_safe_filter_count; ++i) {
566
93
            const auto& table_filter = _table_filters[i];
567
93
            if (table_filter.conjunct == nullptr) {
568
0
                continue;
569
0
            }
570
93
            DORIS_CHECK(_is_safe_to_pre_execute(table_filter.conjunct));
571
            // RuntimeFilterExpr does not implement execute_column_impl(); it is evaluated by the
572
            // row-level filter path through execute_filter(). Constant split pruning uses
573
            // VExprContext::execute() on a one-row synthetic block, so runtime filters must not be
574
            // pre-executed here even when their referenced slot maps to a constant value.
575
93
            if (table_filter.conjunct->root()->is_rf_wrapper() ||
576
93
                !_table_filter_has_only_constant_entries(table_filter)) {
577
89
                continue;
578
89
            }
579
4
            Block eval_block;
580
4
            RETURN_IF_ERROR(_build_constant_filter_block(table_filter, &eval_block));
581
4
            RowDescriptor row_desc;
582
4
            RETURN_IF_ERROR(table_filter.conjunct->prepare(_runtime_state, row_desc));
583
4
            RETURN_IF_ERROR(table_filter.conjunct->open(_runtime_state));
584
4
            int result_column_id = -1;
585
4
            RETURN_IF_ERROR(table_filter.conjunct->execute(&eval_block, &result_column_id));
586
4
            DORIS_CHECK(result_column_id >= 0);
587
4
            if (_filter_result_filters_all(eval_block.get_by_position(result_column_id).column)) {
588
2
                *can_filter_all = true;
589
2
                return Status::OK();
590
2
            }
591
4
        }
592
954
        return Status::OK();
593
956
    }
594
595
83
    bool _table_filter_has_only_constant_entries(const TableFilter& table_filter) const {
596
83
        const auto& filter_entries = _data_reader.column_mapper->filter_entries();
597
83
        for (const auto global_index : table_filter.global_indices) {
598
83
            const auto entry_it = filter_entries.find(global_index);
599
83
            if (entry_it == filter_entries.end() || !entry_it->second.is_constant()) {
600
79
                return false;
601
79
            }
602
83
        }
603
4
        return !table_filter.global_indices.empty();
604
83
    }
605
606
4
    Status _build_constant_filter_block(const TableFilter& table_filter, Block* eval_block) {
607
4
        DORIS_CHECK(eval_block != nullptr);
608
4
        eval_block->clear();
609
4
        const auto& mappings = _data_reader.column_mapper->mappings();
610
4
        const auto& filter_entries = _data_reader.column_mapper->filter_entries();
611
4
        DORIS_CHECK(mappings.size() == _projected_columns.size());
612
8
        for (size_t column_idx = 0; column_idx < mappings.size(); ++column_idx) {
613
4
            const auto global_index = GlobalIndex(column_idx);
614
4
            const auto& mapping = mappings[column_idx];
615
4
            const auto entry_it = filter_entries.find(global_index);
616
4
            const bool referenced_by_filter =
617
4
                    std::find(table_filter.global_indices.begin(),
618
4
                              table_filter.global_indices.end(),
619
4
                              global_index) != table_filter.global_indices.end();
620
4
            if (referenced_by_filter && entry_it != filter_entries.end() &&
621
4
                entry_it->second.is_constant()) {
622
4
                ColumnPtr constant_column;
623
4
                RETURN_IF_ERROR(_materialize_constant_filter_column(
624
4
                        entry_it->second.constant_index(), &constant_column));
625
4
                eval_block->insert({std::move(constant_column), mapping.table_type,
626
4
                                    mapping.table_column_name});
627
4
            } else {
628
0
                eval_block->insert({mapping.table_type->create_column_const_with_default_value(1),
629
0
                                    mapping.table_type, mapping.table_column_name});
630
0
            }
631
4
        }
632
4
        return Status::OK();
633
4
    }
634
635
4
    Status _materialize_constant_filter_column(ConstantIndex constant_index, ColumnPtr* column) {
636
4
        DORIS_CHECK(column != nullptr);
637
4
        const auto& constant_entry = _data_reader.column_mapper->constant_map().get(constant_index);
638
4
        DORIS_CHECK(constant_entry.expr != nullptr);
639
4
        DORIS_CHECK(constant_entry.type != nullptr);
640
4
        RowDescriptor row_desc;
641
4
        RETURN_IF_ERROR(constant_entry.expr->prepare(_runtime_state, row_desc));
642
4
        RETURN_IF_ERROR(constant_entry.expr->open(_runtime_state));
643
4
        Block eval_block;
644
4
        eval_block.insert({constant_entry.type->create_column_const_with_default_value(1),
645
4
                           constant_entry.type, "__table_reader_constant_filter"});
646
4
        int result_column_id = -1;
647
4
        RETURN_IF_ERROR(constant_entry.expr->execute(&eval_block, &result_column_id));
648
4
        DORIS_CHECK(result_column_id >= 0);
649
4
        *column = eval_block.get_by_position(result_column_id).column;
650
4
        DORIS_CHECK((*column)->size() == 1);
651
4
        return Status::OK();
652
4
    }
653
654
4
    static bool _filter_result_filters_all(const ColumnPtr& filter_column) {
655
4
        DORIS_CHECK(filter_column.get() != nullptr);
656
4
        DORIS_CHECK(filter_column->size() == 1);
657
4
        return !filter_column->get_bool(0);
658
4
    }
659
660
969
    virtual Status customize_file_scan_request(FileScanRequest* file_request) {
661
969
        return _append_delete_predicate(file_request);
662
969
    }
663
664
2.92k
    bool _is_table_level_count_active() const { return _remaining_table_level_count >= 0; }
665
666
1.68k
    bool _is_file_level_count_active() const { return _remaining_file_level_count >= 0; }
667
668
28
    Status _materialize_count_rows(size_t rows, Block* block) const {
669
28
        DORIS_CHECK(block != nullptr);
670
28
        DORIS_CHECK(block->columns() > 0 || rows == 0);
671
56
        for (size_t column_idx = 0; column_idx < block->columns(); ++column_idx) {
672
28
            auto column = block->get_by_position(column_idx).type->create_column();
673
28
            if (auto* nullable = check_and_get_column<ColumnNullable>(*column)) {
674
                // Metadata COUNT emits synthetic input rows for the unchanged upper aggregate.
675
                // They must be non-NULL for COUNT(nullable_col), and constructing them explicitly
676
                // also keeps every nullable null map boolean-valid in debug/ASAN block checks.
677
28
                nullable->get_nested_column().insert_many_defaults(rows);
678
28
                nullable->get_null_map_data().resize_fill(rows, 0);
679
28
            } else {
680
0
                column->insert_many_defaults(rows);
681
0
            }
682
28
            block->replace_by_position(column_idx, std::move(column));
683
28
        }
684
28
        return Status::OK();
685
28
    }
686
687
28
    Status _materialize_next_count_batch(int64_t* remaining_rows, Block* block) const {
688
28
        DORIS_CHECK(remaining_rows != nullptr);
689
28
        DORIS_CHECK(*remaining_rows > 0);
690
28
        const int64_t batch_size = _runtime_state == nullptr
691
28
                                           ? *remaining_rows
692
28
                                           : static_cast<int64_t>(_runtime_state->batch_size());
693
28
        const auto rows = std::min(*remaining_rows, batch_size);
694
28
        RETURN_IF_ERROR(_materialize_count_rows(cast_set<size_t>(rows), block));
695
28
        *remaining_rows -= rows;
696
28
        return Status::OK();
697
28
    }
698
699
20
    Status _read_count_batch(int64_t* remaining_rows, Block* block, bool* eos) {
700
20
        DORIS_CHECK(block != nullptr);
701
20
        DORIS_CHECK(eos != nullptr);
702
20
        DORIS_CHECK(_push_down_agg_type == TPushAggOp::type::COUNT);
703
20
        DORIS_CHECK(remaining_rows != nullptr);
704
20
        DORIS_CHECK(*remaining_rows >= 0);
705
20
        if (*remaining_rows == 0) {
706
8
            *remaining_rows = -1;
707
8
            _current_task.reset();
708
8
            *eos = true;
709
8
            return Status::OK();
710
8
        }
711
12
        RETURN_IF_ERROR(_materialize_next_count_batch(remaining_rows, block));
712
12
        *eos = false;
713
12
        return Status::OK();
714
12
    }
715
716
12
    Status _read_table_level_count(Block* block, bool* eos) {
717
12
        return _read_count_batch(&_remaining_table_level_count, block, eos);
718
12
    }
719
720
8
    Status _read_file_level_count(Block* block, bool* eos) {
721
8
        return _read_count_batch(&_remaining_file_level_count, block, eos);
722
8
    }
723
724
    void _append_file_scan_column(FileScanRequest* request, LocalColumnId column_id,
725
158
                                  std::vector<LocalColumnIndex>* scan_columns) {
726
158
        DORIS_CHECK(request != nullptr);
727
158
        DORIS_CHECK(scan_columns != nullptr);
728
158
        FileScanRequestBuilder builder(request);
729
158
        Status status;
730
158
        if (scan_columns == &request->predicate_columns) {
731
128
            status = builder.add_predicate_column(column_id);
732
128
        } else {
733
30
            DORIS_CHECK(scan_columns == &request->non_predicate_columns);
734
30
            status = builder.add_non_predicate_column(column_id);
735
30
        }
736
158
        DORIS_CHECK(status.ok()) << status.to_string();
737
158
        if (column_id == LocalColumnId(ROW_POSITION_COLUMN_ID) &&
738
158
            _find_column_definition(_data_reader.file_schema, column_id) == nullptr) {
739
88
            _data_reader.file_schema.push_back(row_position_column_definition());
740
88
        }
741
158
    }
742
743
    // Append DeletePredicate to file scan request if there are deletes. The predicate will be evaluated in file reader level and filter out deleted rows before returning data to table reader.
744
969
    Status _append_delete_predicate(FileScanRequest* request) {
745
969
        DORIS_CHECK(request != nullptr);
746
969
        if ((_delete_rows == nullptr || _delete_rows->empty()) &&
747
969
            (_deletion_vector == nullptr || _deletion_vector->isEmpty())) {
748
935
            return Status::OK();
749
935
        }
750
34
        const auto row_position_column_id = LocalColumnId(ROW_POSITION_COLUMN_ID);
751
34
        _append_file_scan_column(request, row_position_column_id, &request->predicate_columns);
752
753
34
        const auto block_position = request->local_positions.at(row_position_column_id);
754
34
        auto append_predicate = [&](auto& deleted_rows) {
755
34
            auto delete_predicate = std::make_shared<DeletePredicate>(deleted_rows);
756
34
            delete_predicate->add_child(VSlotRef::create_shared(
757
34
                    cast_set<int>(block_position.value()), cast_set<int>(block_position.value()),
758
34
                    -1, std::make_shared<DataTypeInt64>(), ROW_POSITION_COLUMN_NAME));
759
34
            request->delete_conjuncts.push_back(
760
34
                    VExprContext::create_shared(std::move(delete_predicate)));
761
34
        };
_ZZN5doris6format11TableReader24_append_delete_predicateEPNS0_15FileScanRequestEENKUlRT_E_clISt6vectorIlSaIlEEEEDaS5_
Line
Count
Source
754
26
        auto append_predicate = [&](auto& deleted_rows) {
755
26
            auto delete_predicate = std::make_shared<DeletePredicate>(deleted_rows);
756
26
            delete_predicate->add_child(VSlotRef::create_shared(
757
26
                    cast_set<int>(block_position.value()), cast_set<int>(block_position.value()),
758
26
                    -1, std::make_shared<DataTypeInt64>(), ROW_POSITION_COLUMN_NAME));
759
26
            request->delete_conjuncts.push_back(
760
26
                    VExprContext::create_shared(std::move(delete_predicate)));
761
26
        };
_ZZN5doris6format11TableReader24_append_delete_predicateEPNS0_15FileScanRequestEENKUlRT_E_clIN7roaring12Roaring64MapEEEDaS5_
Line
Count
Source
754
8
        auto append_predicate = [&](auto& deleted_rows) {
755
8
            auto delete_predicate = std::make_shared<DeletePredicate>(deleted_rows);
756
8
            delete_predicate->add_child(VSlotRef::create_shared(
757
8
                    cast_set<int>(block_position.value()), cast_set<int>(block_position.value()),
758
8
                    -1, std::make_shared<DataTypeInt64>(), ROW_POSITION_COLUMN_NAME));
759
8
            request->delete_conjuncts.push_back(
760
8
                    VExprContext::create_shared(std::move(delete_predicate)));
761
8
        };
762
34
        if (_delete_rows != nullptr && !_delete_rows->empty()) {
763
26
            append_predicate(*_delete_rows);
764
26
        }
765
34
        if (_deletion_vector != nullptr && !_deletion_vector->isEmpty()) {
766
8
            append_predicate(*_deletion_vector);
767
8
        }
768
34
        return Status::OK();
769
969
    }
770
771
    // Close the current concrete reader. This hook is called by both create_next_reader() and
772
    // close(), so it should remain idempotent.
773
956
    virtual Status close_current_reader() {
774
956
        _finalize_reader_condition_cache();
775
956
        {
776
956
            SCOPED_TIMER(_profile.file_reader_total_timer);
777
956
            SCOPED_TIMER(_profile.file_reader_close_timer);
778
956
            RETURN_IF_ERROR(_data_reader.reader->close());
779
956
        }
780
956
        _data_reader.reader.reset();
781
956
        if (_data_reader.column_mapper != nullptr) {
782
954
            _data_reader.column_mapper->clear();
783
954
            _data_reader.column_mapper.reset();
784
954
        }
785
956
        _table_filters.clear();
786
956
        _constant_pruning_safe_filter_count = 0;
787
956
        _data_reader.file_schema.clear();
788
956
        _data_reader.file_block_layout.clear();
789
956
        _data_reader.block_template.clear();
790
956
        _file_scan_request.reset();
791
956
        _current_task.reset();
792
956
        _current_file_description.reset();
793
956
        _current_reader_reached_eof = false;
794
956
        return Status::OK();
795
956
    }
796
797
6
    void _record_scan_rows(size_t rows) {
798
6
        if (_io_ctx != nullptr && _io_ctx->file_reader_stats != nullptr) {
799
6
            _io_ctx->file_reader_stats->read_rows += rows;
800
6
        }
801
6
    }
802
803
    // Finalize file-local block to table/global schema block.
804
1.12k
    Status finalize_chunk(Block* block, const size_t rows) {
805
1.12k
        SCOPED_TIMER(_profile.finalize_timer);
806
1.12k
        size_t idx = 0;
807
1.12k
        const auto& mappings = _data_reader.column_mapper->mappings();
808
7.54k
        for (const auto& mapping : mappings) {
809
7.54k
            ColumnPtr column;
810
7.54k
            RETURN_IF_ERROR(_materialize_mapping_column(mapping, &_data_reader.block_template, rows,
811
7.54k
                                                        &column, idx + 1 == mappings.size()));
812
7.54k
            block->replace_by_position(idx, IColumn::mutate(std::move(column)));
813
7.54k
            idx++;
814
7.54k
        }
815
1.12k
        RETURN_IF_ERROR(materialize_virtual_columns(block));
816
        // Enforce CHAR/VARCHAR length declared by the table schema after all file-to-table
817
        // materialization has finished.
818
1.12k
        RETURN_IF_ERROR(_truncate_char_or_varchar_columns(block));
819
1.12k
        return Status::OK();
820
1.12k
    }
821
822
    // Materialize virtual columns in the table block, such as Iceberg _row_id and
823
    // _last_updated_sequence_number. This runs after normal column materialization so finalize
824
    // expressions can reference those virtual columns.
825
969
    virtual Status materialize_virtual_columns(Block* table_block) { return Status::OK(); }
826
827
#ifndef NDEBUG
828
1.12k
    Status _check_file_block_columns(std::string_view stage, size_t rows) {
829
1.12k
        DORIS_CHECK(_data_reader.block_template.columns() == _data_reader.file_block_layout.size());
830
8.70k
        for (size_t idx = 0; idx < _data_reader.block_template.columns(); ++idx) {
831
7.58k
            const auto& file_block_column = _data_reader.file_block_layout[idx];
832
7.58k
            const auto& column_with_type = _data_reader.block_template.get_by_position(idx);
833
7.58k
            const auto* column = column_with_type.column.get();
834
7.58k
            try {
835
7.58k
                if (column == nullptr) {
836
0
                    auto st = Status::InternalError(
837
0
                            "Invalid file block column {} at {}: file_column_id={}, name='{}', "
838
0
                            "type={}, column=null, expected_rows={}, reader={}",
839
0
                            idx, stage, file_block_column.file_column_id.value(),
840
0
                            file_block_column.name,
841
0
                            file_block_column.type == nullptr ? "null"
842
0
                                                              : file_block_column.type->get_name(),
843
0
                            rows, debug_string());
844
0
                    LOG(WARNING) << st;
845
0
                    return st;
846
0
                }
847
7.58k
                column->sanity_check();
848
7.58k
                auto st = column_with_type.check_type_and_column_match();
849
7.58k
                if (!st.ok()) {
850
0
                    auto contextual_status = Status::InternalError(
851
0
                            "Invalid file block column {} at {}: file_column_id={}, name='{}', "
852
0
                            "type={}, column={}, column_size={}, expected_rows={}, error={}, "
853
0
                            "reader={}",
854
0
                            idx, stage, file_block_column.file_column_id.value(),
855
0
                            file_block_column.name,
856
0
                            file_block_column.type == nullptr ? "null"
857
0
                                                              : file_block_column.type->get_name(),
858
0
                            column->get_name(), column->size(), rows, st.to_string(),
859
0
                            debug_string());
860
0
                    LOG(WARNING) << contextual_status;
861
0
                    return contextual_status;
862
0
                }
863
7.58k
            } catch (const Exception& e) {
864
0
                auto st = Status::InternalError(
865
0
                        "Invalid file block column {} at {}: file_column_id={}, name='{}', "
866
0
                        "type={}, column={}, column_size={}, expected_rows={}, error={}, "
867
0
                        "reader={}",
868
0
                        idx, stage, file_block_column.file_column_id.value(),
869
0
                        file_block_column.name,
870
0
                        file_block_column.type == nullptr ? "null"
871
0
                                                          : file_block_column.type->get_name(),
872
0
                        column == nullptr ? "null" : column->get_name(),
873
0
                        column == nullptr ? 0 : column->size(), rows, e.to_string(),
874
0
                        debug_string());
875
0
                LOG(WARNING) << st;
876
0
                return st;
877
0
            } catch (const std::exception& e) {
878
0
                auto st = Status::InternalError(
879
0
                        "Invalid file block column {} at {}: file_column_id={}, name='{}', "
880
0
                        "type={}, column={}, column_size={}, expected_rows={}, error={}, "
881
0
                        "reader={}",
882
0
                        idx, stage, file_block_column.file_column_id.value(),
883
0
                        file_block_column.name,
884
0
                        file_block_column.type == nullptr ? "null"
885
0
                                                          : file_block_column.type->get_name(),
886
0
                        column == nullptr ? "null" : column->get_name(),
887
0
                        column == nullptr ? 0 : column->size(), rows, e.what(), debug_string());
888
0
                LOG(WARNING) << st;
889
0
                return st;
890
0
            }
891
7.58k
        }
892
1.12k
        return Status::OK();
893
1.12k
    }
894
895
1.12k
    Status _check_table_block_columns(std::string_view stage, const Block* block, size_t rows) {
896
1.12k
        DORIS_CHECK(block != nullptr);
897
1.12k
        DORIS_CHECK(block->columns() == _data_reader.column_mapper->mappings().size());
898
8.66k
        for (size_t idx = 0; idx < block->columns(); ++idx) {
899
7.54k
            const auto& mapping = _data_reader.column_mapper->mappings()[idx];
900
7.54k
            const auto& column_with_type = block->get_by_position(idx);
901
7.54k
            const auto* column = column_with_type.column.get();
902
7.54k
            try {
903
7.54k
                if (column == nullptr) {
904
0
                    auto st = Status::InternalError(
905
0
                            "Invalid table block column {} at {}: table_column='{}', "
906
0
                            "global_index={}, type={}, column=null, expected_rows={}, mapping={}",
907
0
                            idx, stage, mapping.table_column_name, mapping.global_index.value(),
908
0
                            mapping.table_type == nullptr ? "null" : mapping.table_type->get_name(),
909
0
                            rows, mapping.debug_string());
910
0
                    LOG(WARNING) << st;
911
0
                    return st;
912
0
                }
913
7.54k
                column->sanity_check();
914
7.54k
                auto st = column_with_type.check_type_and_column_match();
915
7.54k
                if (!st.ok()) {
916
0
                    auto contextual_status = Status::InternalError(
917
0
                            "Invalid table block column {} at {}: table_column='{}', "
918
0
                            "global_index={}, type={}, column={}, column_size={}, "
919
0
                            "expected_rows={}, error={}, mapping={}",
920
0
                            idx, stage, mapping.table_column_name, mapping.global_index.value(),
921
0
                            mapping.table_type == nullptr ? "null" : mapping.table_type->get_name(),
922
0
                            column->get_name(), column->size(), rows, st.to_string(),
923
0
                            mapping.debug_string());
924
0
                    LOG(WARNING) << contextual_status;
925
0
                    return contextual_status;
926
0
                }
927
7.54k
            } catch (const Exception& e) {
928
0
                auto st = Status::InternalError(
929
0
                        "Invalid table block column {} at {}: table_column='{}', global_index={}, "
930
0
                        "type={}, column={}, column_size={}, expected_rows={}, error={}, "
931
0
                        "mapping={}",
932
0
                        idx, stage, mapping.table_column_name, mapping.global_index.value(),
933
0
                        mapping.table_type == nullptr ? "null" : mapping.table_type->get_name(),
934
0
                        column == nullptr ? "null" : column->get_name(),
935
0
                        column == nullptr ? 0 : column->size(), rows, e.to_string(),
936
0
                        mapping.debug_string());
937
0
                LOG(WARNING) << st;
938
0
                return st;
939
0
            } catch (const std::exception& e) {
940
0
                auto st = Status::InternalError(
941
0
                        "Invalid table block column {} at {}: table_column='{}', global_index={}, "
942
0
                        "type={}, column={}, column_size={}, expected_rows={}, error={}, "
943
0
                        "mapping={}",
944
0
                        idx, stage, mapping.table_column_name, mapping.global_index.value(),
945
0
                        mapping.table_type == nullptr ? "null" : mapping.table_type->get_name(),
946
0
                        column == nullptr ? "null" : column->get_name(),
947
0
                        column == nullptr ? 0 : column->size(), rows, e.what(),
948
0
                        mapping.debug_string());
949
0
                LOG(WARNING) << st;
950
0
                return st;
951
0
            }
952
7.54k
        }
953
1.12k
        return Status::OK();
954
1.12k
    }
955
#endif
956
957
1.12k
    Status _truncate_char_or_varchar_columns(Block* block) {
958
1.12k
        DORIS_CHECK(block != nullptr);
959
1.12k
        if (_runtime_state == nullptr ||
960
1.12k
            !_runtime_state->query_options().truncate_char_or_varchar_columns) {
961
1.12k
            return Status::OK();
962
1.12k
        }
963
0
        DORIS_CHECK(block->columns() == _data_reader.column_mapper->mappings().size());
964
0
        for (size_t idx = 0; idx < _data_reader.column_mapper->mappings().size(); ++idx) {
965
0
            const auto& mapping = _data_reader.column_mapper->mappings()[idx];
966
0
            if (!_should_truncate_char_or_varchar_column(mapping)) {
967
0
                continue;
968
0
            }
969
0
            const auto target_len =
970
0
                    assert_cast<const DataTypeString*>(remove_nullable(mapping.table_type).get())
971
0
                            ->len();
972
0
            _truncate_char_or_varchar_column(block, idx, target_len);
973
0
        }
974
0
        return Status::OK();
975
1.12k
    }
976
977
    // Return true when the table schema has a bounded CHAR/VARCHAR length that is stricter than
978
    // the file-side type. Examples:
979
    // - table VARCHAR(10), file VARCHAR(20): truncate to 10;
980
    // - table VARCHAR(10), file STRING: truncate to 10 because STRING has no declared bound;
981
    // - table STRING, any file type: no truncation because the target has no bound.
982
10
    static bool _should_truncate_char_or_varchar_column(const ColumnMapping& mapping) {
983
10
        if (mapping.table_type == nullptr) {
984
0
            return false;
985
0
        }
986
10
        const auto table_type = remove_nullable(mapping.table_type);
987
10
        const auto primitive_type = table_type->get_primitive_type();
988
10
        if (primitive_type != TYPE_VARCHAR && primitive_type != TYPE_CHAR) {
989
2
            return false;
990
2
        }
991
8
        const auto target_len = assert_cast<const DataTypeString*>(table_type.get())->len();
992
8
        if (target_len <= 0) {
993
0
            return false;
994
0
        }
995
8
        if (mapping.file_type == nullptr) {
996
0
            return true;
997
0
        }
998
8
        const auto file_type = remove_nullable(mapping.file_type);
999
8
        DORIS_CHECK(file_type != nullptr);
1000
8
        int file_len = -1;
1001
8
        if (file_type->get_primitive_type() == TYPE_VARCHAR ||
1002
8
            file_type->get_primitive_type() == TYPE_CHAR ||
1003
8
            file_type->get_primitive_type() == TYPE_STRING) {
1004
6
            file_len = assert_cast<const DataTypeString*>(file_type.get())->len();
1005
6
        }
1006
1007
8
        return file_len < 0 || target_len < file_len;
1008
8
    }
1009
1010
    // Truncate a materialized CHAR/VARCHAR column in place by reusing the vectorized substring
1011
    // implementation: substring(column, 1, len). Nullable columns are unwrapped before substring
1012
    // execution and wrapped back with the original null map afterward, because substring operates
1013
    // on the nested string payload only.
1014
2
    static void _truncate_char_or_varchar_column(Block* block, size_t idx, int len) {
1015
2
        DORIS_CHECK(block != nullptr);
1016
2
        auto int_type = std::make_shared<DataTypeInt32>();
1017
2
        const auto num_columns_without_result = cast_set<uint32_t>(block->columns());
1018
2
        auto& target = block->get_by_position(idx);
1019
2
        const bool is_nullable = target.type->is_nullable();
1020
2
        ColumnPtr input_column = target.column;
1021
2
        ColumnPtr null_map_column;
1022
2
        if (is_nullable) {
1023
2
            const auto* nullable_column = assert_cast<const ColumnNullable*>(target.column.get());
1024
2
            input_column = nullable_column->get_nested_column_ptr();
1025
2
            null_map_column = nullable_column->get_null_map_column_ptr();
1026
2
        }
1027
2
        block->replace_by_position(idx, std::move(input_column));
1028
2
        block->insert({int_type->create_column_const(block->rows(), to_field<TYPE_INT>(1)),
1029
2
                       int_type, "const 1"});
1030
2
        block->insert({int_type->create_column_const(block->rows(), to_field<TYPE_INT>(len)),
1031
2
                       int_type, "const len"});
1032
2
        block->insert({nullptr, std::make_shared<DataTypeString>(), "result"});
1033
1034
2
        ColumnNumbers temp_arguments(3);
1035
2
        temp_arguments[0] = cast_set<uint32_t>(idx);
1036
2
        temp_arguments[1] = num_columns_without_result;
1037
2
        temp_arguments[2] = num_columns_without_result + 1;
1038
2
        const uint32_t result_column_id = num_columns_without_result + 2;
1039
2
        SubstringUtil::substring_execute(*block, temp_arguments, result_column_id, block->rows());
1040
1041
2
        ColumnPtr result_column = block->get_by_position(result_column_id).column;
1042
2
        if (is_nullable) {
1043
2
            result_column = ColumnNullable::create(std::move(result_column), null_map_column);
1044
2
        }
1045
2
        block->replace_by_position(idx, std::move(result_column));
1046
2
        block->erase_tail(num_columns_without_result);
1047
2
    }
1048
1049
949
    Status _try_materialize_aggregate_pushdown_rows(Block* block, bool* pushed_down) {
1050
949
        DORIS_CHECK(block != nullptr);
1051
949
        DORIS_CHECK(pushed_down != nullptr);
1052
949
        *pushed_down = false;
1053
949
        block->clear_column_data(_projected_columns.size());
1054
949
        _aggregate_pushdown_tried = true;
1055
949
        if (_row_ids.has_value() || !_supports_aggregate_pushdown(_push_down_agg_type)) {
1056
924
            return Status::OK();
1057
924
        }
1058
1059
25
        FileAggregateRequest file_request;
1060
25
        RETURN_IF_ERROR(_build_file_aggregate_request(_push_down_agg_type, &file_request));
1061
25
        FileAggregateResult file_result;
1062
25
        Status status;
1063
25
        {
1064
25
            SCOPED_TIMER(_profile.file_reader_total_timer);
1065
25
            SCOPED_TIMER(_profile.file_reader_aggregate_timer);
1066
25
            status = _data_reader.reader->get_aggregate_result(file_request, &file_result);
1067
25
        }
1068
25
        if (status.is<ErrorCode::NOT_IMPLEMENTED_ERROR>()) {
1069
2
            return Status::OK();
1070
2
        }
1071
23
        RETURN_IF_ERROR(status);
1072
21
        if (_push_down_agg_type == TPushAggOp::type::COUNT) {
1073
16
            DORIS_CHECK(file_result.count >= 0);
1074
            // The upper aggregate consumes synthetic input rows, but emitting the whole metadata
1075
            // count in one block bypasses the runtime batch contract and can allocate by file size.
1076
            // Keep the remaining cardinality as split state and expose at most one batch per call.
1077
16
            _remaining_file_level_count = file_result.count;
1078
16
            _current_split_uses_metadata_count = true;
1079
16
            if (_remaining_file_level_count > 0) {
1080
16
                RETURN_IF_ERROR(_materialize_next_count_batch(&_remaining_file_level_count, block));
1081
16
            }
1082
16
        } else {
1083
5
            RETURN_IF_ERROR(
1084
5
                    _materialize_aggregate_pushdown_rows(_push_down_agg_type, file_result, block));
1085
5
        }
1086
21
        *pushed_down = true;
1087
21
        RETURN_IF_ERROR(close_current_reader());
1088
21
        return Status::OK();
1089
21
    }
1090
1091
964
    virtual bool _supports_aggregate_pushdown(TPushAggOp::type agg_type) const {
1092
        // Only COUNT and MIN/MAX can be push down.
1093
964
        if (agg_type != TPushAggOp::type::COUNT && agg_type != TPushAggOp::type::MINMAX) {
1094
868
            return false;
1095
868
        }
1096
        // Aggregate pushdown returns reduced synthetic rows and may close the physical reader
1097
        // before the next scheduler turn. If a runtime filter is still pending, those rows could
1098
        // escape before the filter arrives and cannot later be reconstructed from real file rows.
1099
        // This is the same irreversibility constraint as table-level metadata COUNT, and applies
1100
        // to COUNT and MIN/MAX for Parquet/ORC as well as COUNT for text readers.
1101
96
        if (!_all_runtime_filters_applied_for_split) {
1102
6
            return false;
1103
6
        }
1104
        // Scanner owns the original conjunct list and evaluates it after TableReader finalizes
1105
        // rows. Even a slotless conjunct that cannot become a TableFilter must see every source
1106
        // row before an aggregate reduces the stream to synthetic COUNT/MINMAX rows.
1107
90
        if (!_conjuncts.empty()) {
1108
10
            return false;
1109
10
        }
1110
        // Only support aggregate pushdown when there is no delete or filter, so
1111
        // the reduced rows consumed by the upper aggregate remain semantically equivalent to a
1112
        // normal scan.
1113
80
        if ((_delete_rows != nullptr && !_delete_rows->empty()) ||
1114
80
            (_deletion_vector != nullptr && !_deletion_vector->isEmpty())) {
1115
10
            return false;
1116
10
        }
1117
70
        if (!_table_filters.empty()) {
1118
0
            return false;
1119
0
        }
1120
70
        if (agg_type == TPushAggOp::type::COUNT) {
1121
            // Old FEs do not serialize push_down_count_slot_ids. During the supported BE-first
1122
            // rolling upgrade, nullopt therefore means "COUNT semantics are unknown", not
1123
            // COUNT(*). Fall back to reading rows until the FE explicitly sends either an empty
1124
            // list for COUNT(*) or one slot for COUNT(col).
1125
48
            if (!_push_down_count_columns.has_value()) {
1126
6
                return false;
1127
6
            }
1128
            // COUNT(*) needs no column metadata. COUNT(col) currently supports one direct file
1129
            // column; multiple COUNT arguments fall back to the normal scan so every upper
1130
            // aggregate receives the original rows.
1131
42
            if (_push_down_count_columns->empty()) {
1132
20
                return true;
1133
20
            }
1134
22
            if (_push_down_count_columns->size() != 1) {
1135
2
                return false;
1136
2
            }
1137
20
            const auto& mapping = _push_down_count_mapping();
1138
            // Metadata COUNT skips TableReader's normal materialization path. Only a trivial
1139
            // mapping is safe: for example, a nullable Parquet INT mapped to a NOT NULL table
1140
            // BIGINT normally needs both an INT->BIGINT cast and nullability validation. Counting
1141
            // footer values directly would bypass both operations and could hide invalid data.
1142
20
            return mapping.file_local_id.has_value() && mapping.file_type != nullptr &&
1143
20
                   mapping.table_type != nullptr && mapping.is_trivial &&
1144
20
                   mapping.virtual_column_type == TableVirtualColumnType::INVALID &&
1145
20
                   mapping.default_expr == nullptr;
1146
22
        }
1147
        // For MIN/MAX, only support direct file-to-table column mappings. The two emitted rows
1148
        // must be enough for the upper MIN/MAX aggregate without evaluating default expressions or
1149
        // virtual columns.
1150
26
        for (const auto& mapping : _data_reader.column_mapper->mappings()) {
1151
26
            if (!mapping.file_local_id.has_value() ||
1152
26
                mapping.virtual_column_type != TableVirtualColumnType::INVALID ||
1153
26
                mapping.default_expr != nullptr || mapping.file_type == nullptr ||
1154
26
                mapping.table_type == nullptr) {
1155
2
                return false;
1156
2
            }
1157
24
            if (!_can_push_down_minmax_for_mapping(mapping)) {
1158
4
                return false;
1159
4
            }
1160
24
        }
1161
16
        return true;
1162
22
    }
1163
1164
7.58k
    static ColumnPtr _detach_column(ColumnPtr column) {
1165
7.58k
        DORIS_CHECK(column.get() != nullptr);
1166
7.58k
        return IColumn::mutate(std::move(column));
1167
7.58k
    }
1168
1169
1.08k
    static ColumnPtr _take_and_detach_block_column(Block* block, int position) {
1170
1.08k
        DORIS_CHECK(block != nullptr);
1171
1.08k
        DORIS_CHECK(position >= 0 && position < static_cast<int>(block->columns()));
1172
1.08k
        auto& source = block->get_by_position(position);
1173
1.08k
        ColumnPtr column = source.column;
1174
        // The final mapping no longer needs the file block. Release its COW owner before mutate(),
1175
        // otherwise nested MAP/STRING columns are deep-copied and a multi-GB payload can OOM.
1176
1.08k
        block->replace_by_position(position, source.type->create_column());
1177
1.08k
        return _detach_column(std::move(column));
1178
1.08k
    }
1179
1180
    static Status _align_column_nullability(ColumnPtr* column, const DataTypePtr& table_type,
1181
536
                                            const NullMap* nullable_parent_null_map = nullptr) {
1182
536
        DORIS_CHECK(column != nullptr);
1183
536
        DORIS_CHECK(column->get() != nullptr);
1184
536
        DORIS_CHECK(table_type != nullptr);
1185
        // Must return non-const column
1186
536
        *column = (*column)->convert_to_full_column_if_const();
1187
536
        if (table_type->is_nullable()) {
1188
213
            const auto& nested_type =
1189
213
                    assert_cast<const DataTypeNullable&>(*table_type).get_nested_type();
1190
213
            if (!(*column)->is_nullable()) {
1191
4
                RETURN_IF_ERROR(
1192
4
                        _align_column_nullability(column, nested_type, nullable_parent_null_map));
1193
4
                *column = make_nullable(*column);
1194
4
                return Status::OK();
1195
4
            }
1196
209
            const auto& nullable_column = assert_cast<const ColumnNullable&>(**column);
1197
209
            ColumnPtr nested_column = nullable_column.get_nested_column_ptr();
1198
209
            NullMap combined_null_map;
1199
209
            const NullMap* nested_parent_null_map = &nullable_column.get_null_map_data();
1200
209
            if (nullable_parent_null_map != nullptr) {
1201
114
                const auto& own_null_map = nullable_column.get_null_map_data();
1202
114
                DORIS_CHECK(nullable_parent_null_map->size() == own_null_map.size());
1203
                // Required descendants are hidden when either this nullable container or any
1204
                // inherited nullable ancestor masks the row, so preserve the union recursively.
1205
114
                combined_null_map.resize(own_null_map.size());
1206
388
                for (size_t i = 0; i < own_null_map.size(); ++i) {
1207
274
                    combined_null_map[i] = own_null_map[i] || (*nullable_parent_null_map)[i];
1208
274
                }
1209
114
                nested_parent_null_map = &combined_null_map;
1210
114
            }
1211
209
            RETURN_IF_ERROR(
1212
209
                    _align_column_nullability(&nested_column, nested_type, nested_parent_null_map));
1213
209
            *column = ColumnNullable::create(nested_column,
1214
209
                                             nullable_column.get_null_map_column_ptr());
1215
209
            return Status::OK();
1216
209
        }
1217
323
        if ((*column)->is_nullable()) {
1218
41
            const auto& nullable_column = assert_cast<const ColumnNullable&>(**column);
1219
41
            if (nullable_column.has_null()) {
1220
22
                const auto& null_map = nullable_column.get_null_map_data();
1221
22
                if (nullable_parent_null_map == nullptr ||
1222
22
                    nullable_parent_null_map->size() != null_map.size()) {
1223
2
                    return Status::InternalError(
1224
2
                            "Default expression produced NULL for non-nullable table column");
1225
2
                }
1226
46
                for (size_t i = 0; i < null_map.size(); ++i) {
1227
                    // A required child may contain a physical NULL placeholder only when its
1228
                    // nullable parent masks that row from the logical value.
1229
34
                    if (null_map[i] && !(*nullable_parent_null_map)[i]) {
1230
8
                        return Status::InternalError(
1231
8
                                "Default expression produced NULL for non-nullable table column");
1232
8
                    }
1233
34
                }
1234
20
            }
1235
31
            ColumnPtr nested_column = nullable_column.get_nested_column_ptr();
1236
31
            RETURN_IF_ERROR(_align_column_nullability(&nested_column, table_type,
1237
31
                                                      nullable_parent_null_map));
1238
31
            *column = nested_column;
1239
31
            return Status::OK();
1240
31
        }
1241
282
        if (const auto* array_type = typeid_cast<const DataTypeArray*>(table_type.get())) {
1242
4
            const auto& array_column = assert_cast<const ColumnArray&>(**column);
1243
4
            ColumnPtr nested_column = array_column.get_data_ptr();
1244
4
            NullMap descendant_parent_null_map;
1245
            // Collection entries use offset coordinates, so inherited row masks must be projected
1246
            // only when a required descendant can consume them. This avoids scratch proportional
1247
            // to all array entries for the common all-required schema.
1248
4
            const NullMap* descendant_parent_null_map_ptr = nullptr;
1249
4
            if (_requires_collection_parent_null_map(
1250
4
                        nullable_parent_null_map, nested_column, array_type->get_nested_type(),
1251
4
                        array_column.size(), array_column.get_offsets())) {
1252
2
                descendant_parent_null_map_ptr =
1253
2
                        _project_collection_parent_null_map_for_hidden_entries(
1254
2
                                nullptr, nullable_parent_null_map, array_column.size(),
1255
2
                                array_column.get_offsets(), nested_column->size(),
1256
2
                                &descendant_parent_null_map);
1257
2
            }
1258
4
            RETURN_IF_ERROR(_align_column_nullability(&nested_column, array_type->get_nested_type(),
1259
4
                                                      descendant_parent_null_map_ptr));
1260
4
            *column = ColumnArray::create(nested_column, array_column.get_offsets_ptr());
1261
4
            return Status::OK();
1262
4
        }
1263
278
        if (const auto* map_type = typeid_cast<const DataTypeMap*>(table_type.get())) {
1264
2
            const auto& map_column = assert_cast<const ColumnMap&>(**column);
1265
2
            ColumnPtr key_column = map_column.get_keys_ptr();
1266
2
            ColumnPtr value_column = map_column.get_values_ptr();
1267
2
            NullMap descendant_parent_null_map;
1268
2
            const NullMap* descendant_parent_null_map_ptr = nullptr;
1269
2
            if (_requires_collection_parent_null_map(nullable_parent_null_map, key_column,
1270
2
                                                     map_type->get_key_type(), map_column.size(),
1271
2
                                                     map_column.get_offsets()) ||
1272
2
                _requires_collection_parent_null_map(nullable_parent_null_map, value_column,
1273
2
                                                     map_type->get_value_type(), map_column.size(),
1274
2
                                                     map_column.get_offsets())) {
1275
                // Keys and values share offsets, so one projected mask safely covers both streams.
1276
0
                descendant_parent_null_map_ptr =
1277
0
                        _project_collection_parent_null_map_for_hidden_entries(
1278
0
                                nullptr, nullable_parent_null_map, map_column.size(),
1279
0
                                map_column.get_offsets(), key_column->size(),
1280
0
                                &descendant_parent_null_map);
1281
0
            }
1282
2
            RETURN_IF_ERROR(_align_column_nullability(&key_column, map_type->get_key_type(),
1283
2
                                                      descendant_parent_null_map_ptr));
1284
2
            RETURN_IF_ERROR(_align_column_nullability(&value_column, map_type->get_value_type(),
1285
2
                                                      descendant_parent_null_map_ptr));
1286
2
            *column = ColumnMap::create(key_column, value_column, map_column.get_offsets_ptr());
1287
2
            return Status::OK();
1288
2
        }
1289
276
        if (const auto* struct_type = typeid_cast<const DataTypeStruct*>(table_type.get())) {
1290
30
            const auto& struct_column = assert_cast<const ColumnStruct&>(**column);
1291
30
            Columns columns = struct_column.get_columns_copy();
1292
30
            DORIS_CHECK(columns.size() == struct_type->get_elements().size());
1293
84
            for (size_t i = 0; i < columns.size(); ++i) {
1294
54
                RETURN_IF_ERROR(_align_column_nullability(&columns[i], struct_type->get_element(i),
1295
54
                                                          nullable_parent_null_map));
1296
54
            }
1297
30
            *column = ColumnStruct::create(columns);
1298
30
            return Status::OK();
1299
30
        }
1300
246
        return Status::OK();
1301
276
    }
1302
1303
    static Status _execute_default_expr_without_root_type_check(
1304
            const VExprContextSPtr& default_expr, const Block* block,
1305
32
            ColumnWithTypeAndName* result_data) {
1306
32
        DORIS_CHECK(default_expr != nullptr);
1307
32
        DORIS_CHECK(block != nullptr);
1308
32
        DORIS_CHECK(result_data != nullptr);
1309
32
        ColumnPtr result_column;
1310
32
        Status st;
1311
32
        RETURN_IF_CATCH_EXCEPTION({
1312
32
            st = default_expr->root()->execute_column_impl(default_expr.get(), block, nullptr,
1313
32
                                                           block->rows(), result_column);
1314
32
        });
1315
32
        RETURN_IF_ERROR(st);
1316
32
        DORIS_CHECK(result_column.get() != nullptr);
1317
32
        if (result_column->size() != block->rows()) {
1318
0
            return Status::InternalError(
1319
0
                    "Default expr {} return column size {} not equal to expected size {}",
1320
0
                    default_expr->expr_name(), result_column->size(), block->rows());
1321
0
        }
1322
32
        result_data->column = result_column;
1323
32
        result_data->type = default_expr->execute_type(block);
1324
32
        result_data->name = default_expr->expr_name();
1325
32
        return Status::OK();
1326
32
    }
1327
1328
    Status _cast_column_to_type(ColumnPtr* column, const DataTypePtr& file_type,
1329
                                const DataTypePtr& table_type,
1330
48
                                const std::string& column_name) const {
1331
48
        DORIS_CHECK(column != nullptr);
1332
48
        DORIS_CHECK(column->get() != nullptr);
1333
48
        DORIS_CHECK(file_type != nullptr);
1334
48
        DORIS_CHECK(table_type != nullptr);
1335
48
        if (file_type->equals(*table_type) ||
1336
48
            remove_nullable(file_type)->equals(*remove_nullable(table_type))) {
1337
23
            return Status::OK();
1338
23
        }
1339
1340
25
        DataTypePtr input_type = file_type;
1341
        // Cast wrappers unwrap nullable inputs according to the declared input type, so keep the
1342
        // root nullability of the declared input aligned with the actual column shape. When the
1343
        // runtime column is nullable, also keep the cast target nullable; the caller applies the
1344
        // table's final nullability after value conversion. Casting a nullable runtime column
1345
        // directly to a non-nullable target would pass ColumnNullable to CastToImpl.
1346
25
        if ((*column)->is_nullable() && !input_type->is_nullable()) {
1347
0
            input_type = make_nullable(input_type);
1348
25
        } else if (!(*column)->is_nullable() && input_type->is_nullable()) {
1349
2
            input_type = remove_nullable(input_type);
1350
2
        }
1351
25
        DataTypePtr cast_type = table_type;
1352
25
        if ((*column)->is_nullable() && !cast_type->is_nullable()) {
1353
16
            cast_type = make_nullable(cast_type);
1354
16
        }
1355
25
        Block cast_block;
1356
25
        cast_block.insert({*column, input_type, column_name});
1357
25
        auto slot_ref = VSlotRef::create_shared(0, 0, -1, input_type, column_name);
1358
25
        auto cast_expr = Cast::create_shared(cast_type);
1359
25
        cast_expr->add_child(std::move(slot_ref));
1360
25
        auto cast_ctx = VExprContext::create_shared(std::move(cast_expr));
1361
25
        RowDescriptor row_desc;
1362
25
        RETURN_IF_ERROR(cast_ctx->prepare(_runtime_state, row_desc));
1363
25
        RETURN_IF_ERROR(cast_ctx->open(_runtime_state));
1364
25
        ColumnPtr cast_column;
1365
25
        RETURN_IF_ERROR(cast_ctx->execute(&cast_block, cast_column));
1366
25
        *column = std::move(cast_column);
1367
25
        return Status::OK();
1368
25
    }
1369
1370
    Status _try_materialize_scalar_cast_with_runtime_nullability(const ColumnMapping& mapping,
1371
                                                                 const Block* current_block,
1372
                                                                 ColumnPtr* column,
1373
7.52k
                                                                 bool* handled) const {
1374
7.52k
        DORIS_CHECK(column != nullptr);
1375
7.52k
        DORIS_CHECK(handled != nullptr);
1376
7.52k
        *handled = false;
1377
7.52k
        if (mapping.projection == nullptr || !mapping.file_local_id.has_value() ||
1378
7.52k
            !mapping.child_mappings.empty()) {
1379
1.48k
            return Status::OK();
1380
1.48k
        }
1381
1382
6.04k
        const auto& root = mapping.projection->root();
1383
6.04k
        if (root == nullptr || root->node_type() != TExprNodeType::CAST_EXPR) {
1384
6.00k
            return Status::OK();
1385
6.00k
        }
1386
31
        DORIS_CHECK(root->get_num_children() == 1);
1387
31
        const auto* slot = dynamic_cast<const VSlotRef*>(root->get_child(0).get());
1388
31
        DORIS_CHECK(slot != nullptr);
1389
31
        DORIS_CHECK(current_block != nullptr);
1390
31
        DORIS_CHECK(slot->column_id() >= 0);
1391
31
        DORIS_CHECK(cast_set<size_t>(slot->column_id()) < current_block->columns());
1392
31
        const auto& source = current_block->get_by_position(slot->column_id());
1393
31
        DORIS_CHECK(source.column.get() != nullptr);
1394
31
        DORIS_CHECK(slot->data_type() != nullptr);
1395
31
        DORIS_CHECK(mapping.table_type != nullptr);
1396
31
        const bool runtime_input_mismatch =
1397
31
                source.column->is_nullable() != slot->data_type()->is_nullable();
1398
31
        const bool nullable_input_to_required_table =
1399
31
                source.column->is_nullable() && !mapping.table_type->is_nullable();
1400
31
        if (!runtime_input_mismatch && !nullable_input_to_required_table) {
1401
14
            return Status::OK();
1402
14
        }
1403
1404
        // File readers can return a nullable runtime column even when the physical schema marks the
1405
        // leaf required. A pre-built Cast binds to the declared file type and can therefore pass a
1406
        // ColumnNullable to a non-nullable CastToImpl. Rebuild only when that runtime shape differs
1407
        // from the declared input, or when a declared nullable file field maps to a required table
1408
        // field. Keep the cast target nullable while converting values, then let
1409
        // _align_column_nullability() reject an actual NULL before removing the wrapper.
1410
17
        ColumnPtr result_column = source.column;
1411
17
        RETURN_IF_ERROR(_cast_column_to_type(&result_column, slot->data_type(), mapping.table_type,
1412
17
                                             mapping.file_column_name));
1413
17
        RETURN_IF_ERROR(_align_column_nullability(&result_column, mapping.table_type));
1414
15
        *column = _detach_column(std::move(result_column));
1415
15
        *handled = true;
1416
15
        return Status::OK();
1417
17
    }
1418
1419
    Status _materialize_present_child_mapping_column(
1420
            const ColumnMapping& mapping, const ColumnPtr& file_column, const size_t rows,
1421
119
            ColumnPtr* column, const NullMap* nullable_parent_null_map = nullptr) {
1422
119
        DORIS_CHECK(column != nullptr);
1423
119
        DORIS_CHECK(mapping.file_type != nullptr);
1424
119
        DORIS_CHECK(mapping.table_type != nullptr);
1425
119
        *column = file_column;
1426
119
        if (!mapping.is_trivial) {
1427
59
            if (!mapping.child_mappings.empty()) {
1428
32
                RETURN_IF_ERROR(_materialize_complex_mapping_column(mapping, *column, rows, column,
1429
32
                                                                    nullable_parent_null_map));
1430
32
            } else {
1431
27
                RETURN_IF_ERROR(_cast_column_to_type(column, mapping.file_type, mapping.table_type,
1432
27
                                                     mapping.file_column_name));
1433
27
            }
1434
59
        }
1435
113
        RETURN_IF_ERROR(
1436
113
                _align_column_nullability(column, mapping.table_type, nullable_parent_null_map));
1437
105
        return Status::OK();
1438
113
    }
1439
1440
    Status _materialize_default_or_missing_column(
1441
            const ColumnMapping& mapping, const Block* current_block, const size_t rows,
1442
94
            ColumnPtr* column, const NullMap* nullable_parent_null_map = nullptr) {
1443
94
        DORIS_CHECK(mapping.table_type != nullptr);
1444
94
        DORIS_CHECK(column != nullptr);
1445
94
        if (mapping.default_expr != nullptr) {
1446
32
            Block synthetic_block;
1447
32
            const Block* eval_block = current_block;
1448
32
            if (eval_block == nullptr || eval_block->rows() != rows) {
1449
                // Nested ARRAY/MAP children use element/entry cardinality rather than the root
1450
                // block's row count. Iceberg initial defaults are typed literals, so a synthetic
1451
                // block with the desired row count is sufficient and avoids a top-level
1452
                // ConstantMap dependency for nested mappings.
1453
28
                synthetic_block.insert(
1454
28
                        {mapping.table_type->create_column_const_with_default_value(rows),
1455
28
                         mapping.table_type, "__table_reader_nested_default_rows"});
1456
28
                eval_block = &synthetic_block;
1457
28
            }
1458
32
            ColumnWithTypeAndName result;
1459
32
            RETURN_IF_ERROR(_execute_default_expr_without_root_type_check(mapping.default_expr,
1460
32
                                                                          eval_block, &result));
1461
32
            ColumnPtr result_column = result.column;
1462
32
            RETURN_IF_ERROR(_align_column_nullability(&result_column, mapping.table_type,
1463
32
                                                      nullable_parent_null_map));
1464
32
            *column = _detach_column(std::move(result_column));
1465
32
            return Status::OK();
1466
32
        }
1467
62
        ColumnPtr result_column = mapping.table_type->create_column_const_with_default_value(rows);
1468
62
        RETURN_IF_ERROR(_align_column_nullability(&result_column, mapping.table_type,
1469
62
                                                  nullable_parent_null_map));
1470
62
        *column = _detach_column(std::move(result_column));
1471
62
        return Status::OK();
1472
62
    }
1473
1474
    Status _materialize_mapping_column(const ColumnMapping& mapping, Block* current_block,
1475
                                       const size_t rows, ColumnPtr* column,
1476
7.56k
                                       bool take_projection_result = false) {
1477
7.56k
        if (!mapping.is_trivial && mapping.file_local_id.has_value() &&
1478
7.56k
            !mapping.child_mappings.empty()) {
1479
43
            DCHECK(mapping.projection != nullptr);
1480
43
            int res_id;
1481
43
            auto st = mapping.projection->execute(current_block, &res_id);
1482
43
            if (!st.ok()) {
1483
0
                return Status::InternalError(
1484
0
                        "Failed to execute complex mapping projection for table column '{}' "
1485
0
                        "(global_index={}, file_local_id={}, rows={}): {}, mapping={}",
1486
0
                        mapping.table_column_name, mapping.global_index.value(),
1487
0
                        *mapping.file_local_id, rows, st.to_string(), mapping.debug_string());
1488
0
            }
1489
43
            ColumnPtr result_column = take_projection_result
1490
43
                                              ? _take_and_detach_block_column(current_block, res_id)
1491
43
                                              : current_block->get_by_position(res_id).column;
1492
43
            RETURN_IF_ERROR(
1493
43
                    _materialize_complex_mapping_column(mapping, result_column, rows, column));
1494
37
            return Status::OK();
1495
43
        }
1496
7.52k
        bool runtime_nullability_cast_handled = false;
1497
7.52k
        RETURN_IF_ERROR(_try_materialize_scalar_cast_with_runtime_nullability(
1498
7.52k
                mapping, current_block, column, &runtime_nullability_cast_handled));
1499
7.52k
        if (runtime_nullability_cast_handled) {
1500
15
            return Status::OK();
1501
15
        }
1502
7.50k
        if (mapping.projection != nullptr) {
1503
7.43k
            int res_id;
1504
7.43k
            auto st = mapping.projection->execute(current_block, &res_id);
1505
7.43k
            if (!st.ok()) {
1506
0
                std::string file_local_id = "null";
1507
0
                if (mapping.file_local_id.has_value()) {
1508
0
                    file_local_id = std::to_string(*mapping.file_local_id);
1509
0
                }
1510
0
                return Status::InternalError(
1511
0
                        "Failed to execute mapping projection for table column '{}' "
1512
0
                        "(global_index={}, file_local_id={}, rows={}): {}, mapping={}",
1513
0
                        mapping.table_column_name, mapping.global_index.value(), file_local_id,
1514
0
                        rows, st.to_string(), mapping.debug_string());
1515
0
            }
1516
7.43k
            if (take_projection_result) {
1517
1.04k
                *column = _take_and_detach_block_column(current_block, res_id);
1518
6.39k
            } else {
1519
6.39k
                ColumnPtr result_column = current_block->get_by_position(res_id).column;
1520
6.39k
                *column = _detach_column(std::move(result_column));
1521
6.39k
            }
1522
7.43k
            return Status::OK();
1523
7.43k
        }
1524
68
        return _materialize_default_or_missing_column(mapping, current_block, rows, column);
1525
7.50k
    }
1526
1527
    Status _materialize_complex_mapping_column(const ColumnMapping& mapping,
1528
                                               const ColumnPtr& file_column, const size_t rows,
1529
                                               ColumnPtr* column,
1530
75
                                               const NullMap* nullable_parent_null_map = nullptr) {
1531
75
        DORIS_CHECK(mapping.table_type != nullptr);
1532
75
        DORIS_CHECK(file_column.get() != nullptr);
1533
75
        const auto table_type = remove_nullable(mapping.table_type);
1534
75
        switch (table_type->get_primitive_type()) {
1535
52
        case TYPE_STRUCT:
1536
52
            RETURN_IF_ERROR(_materialize_struct_mapping_column(mapping, file_column, rows, column,
1537
52
                                                               nullable_parent_null_map));
1538
46
            break;
1539
46
        case TYPE_ARRAY:
1540
15
            RETURN_IF_ERROR(_materialize_array_mapping_column(mapping, file_column, rows, column,
1541
15
                                                              nullable_parent_null_map));
1542
9
            break;
1543
9
        case TYPE_MAP:
1544
8
            RETURN_IF_ERROR(_materialize_map_mapping_column(mapping, file_column, rows, column,
1545
8
                                                            nullable_parent_null_map));
1546
8
            break;
1547
8
        default:
1548
0
            *column = _detach_column(file_column);
1549
0
            break;
1550
75
        }
1551
63
        return Status::OK();
1552
75
    }
1553
1554
    static std::vector<const ColumnMapping*> _present_child_mappings_in_file_order(
1555
62
            const std::vector<ColumnMapping>& child_mappings) {
1556
62
        std::vector<const ColumnMapping*> result;
1557
62
        result.reserve(child_mappings.size());
1558
100
        for (const auto& child_mapping : child_mappings) {
1559
100
            if (child_mapping.file_local_id.has_value()) {
1560
74
                result.push_back(&child_mapping);
1561
74
            }
1562
100
        }
1563
62
        std::ranges::sort(result, [](const ColumnMapping* lhs, const ColumnMapping* rhs) {
1564
28
            DORIS_CHECK(lhs->file_local_id.has_value());
1565
28
            DORIS_CHECK(rhs->file_local_id.has_value());
1566
28
            return *lhs->file_local_id < *rhs->file_local_id;
1567
28
        });
1568
62
        return result;
1569
62
    }
1570
1571
    static size_t _file_child_ordinal_for_mapping(
1572
            const ColumnMapping& mapping, const ColumnMapping& child_mapping,
1573
72
            const std::vector<const ColumnMapping*>& file_ordered_children) {
1574
72
        DORIS_CHECK(child_mapping.file_local_id.has_value());
1575
72
        if (!mapping.projected_file_children.empty()) {
1576
50
            const auto child_it = std::ranges::find_if(
1577
68
                    mapping.projected_file_children, [&](const ColumnDefinition& file_child) {
1578
68
                        return file_child.file_local_id() == *child_mapping.file_local_id;
1579
68
                    });
1580
50
            DORIS_CHECK(child_it != mapping.projected_file_children.end());
1581
50
            return static_cast<size_t>(
1582
50
                    std::distance(mapping.projected_file_children.begin(), child_it));
1583
50
        }
1584
22
        const auto child_it = std::ranges::find(file_ordered_children, &child_mapping);
1585
22
        DORIS_CHECK(child_it != file_ordered_children.end());
1586
22
        return static_cast<size_t>(std::distance(file_ordered_children.begin(), child_it));
1587
72
    }
1588
1589
    static std::vector<const ColumnMapping*> _child_mappings_in_table_type_order(
1590
62
            const ColumnMapping& mapping, const DataTypeStruct& table_type) {
1591
62
        std::vector<const ColumnMapping*> result;
1592
62
        result.reserve(mapping.child_mappings.size());
1593
162
        for (size_t child_idx = 0; child_idx < table_type.get_elements().size(); ++child_idx) {
1594
100
            const auto& child_name = table_type.get_element_name(child_idx);
1595
100
            const auto child_it = std::ranges::find_if(
1596
144
                    mapping.child_mappings, [&](const ColumnMapping& child_mapping) {
1597
144
                        return child_mapping.table_column_name == child_name;
1598
144
                    });
1599
100
            DORIS_CHECK(child_it != mapping.child_mappings.end())
1600
0
                    << mapping.debug_string() << ", table_child_name=" << child_name;
1601
100
            result.push_back(&*child_it);
1602
100
        }
1603
62
        return result;
1604
62
    }
1605
1606
    static const IColumn* _nested_column_if_nullable(const ColumnPtr& column,
1607
101
                                                     const NullMap** null_map) {
1608
101
        DORIS_CHECK(column.get() != nullptr);
1609
101
        if (const auto* nullable_column = check_and_get_column<ColumnNullable>(*column)) {
1610
93
            if (null_map != nullptr) {
1611
93
                *null_map = &nullable_column->get_null_map_data();
1612
93
            }
1613
93
            return &nullable_column->get_nested_column();
1614
93
        }
1615
8
        return column.get();
1616
101
    }
1617
1618
    static bool _requires_parent_null_map_for_alignment(const ColumnPtr& column,
1619
0
                                                        const DataTypePtr& table_type) {
1620
0
        DORIS_CHECK(column.get() != nullptr);
1621
0
        DORIS_CHECK(table_type != nullptr);
1622
0
        if (table_type->is_nullable()) {
1623
0
            const auto& nested_type =
1624
0
                    assert_cast<const DataTypeNullable&>(*table_type).get_nested_type();
1625
0
            if (const auto* nullable_column = check_and_get_column<ColumnNullable>(*column)) {
1626
0
                return _requires_parent_null_map_for_alignment(
1627
0
                        nullable_column->get_nested_column_ptr(), nested_type);
1628
0
            }
1629
0
            return _requires_parent_null_map_for_alignment(column, nested_type);
1630
0
        }
1631
0
        if (const auto* nullable_column = check_and_get_column<ColumnNullable>(*column)) {
1632
0
            if (nullable_column->has_null()) {
1633
0
                return true;
1634
0
            }
1635
0
            return _requires_parent_null_map_for_alignment(nullable_column->get_nested_column_ptr(),
1636
0
                                                           table_type);
1637
0
        }
1638
0
        if (const auto* array_type = typeid_cast<const DataTypeArray*>(table_type.get())) {
1639
0
            const auto& array_column = assert_cast<const ColumnArray&>(*column);
1640
0
            return _requires_parent_null_map_for_alignment(array_column.get_data_ptr(),
1641
0
                                                           array_type->get_nested_type());
1642
0
        }
1643
0
        if (const auto* map_type = typeid_cast<const DataTypeMap*>(table_type.get())) {
1644
0
            const auto& map_column = assert_cast<const ColumnMap&>(*column);
1645
0
            return _requires_parent_null_map_for_alignment(map_column.get_keys_ptr(),
1646
0
                                                           map_type->get_key_type()) ||
1647
0
                   _requires_parent_null_map_for_alignment(map_column.get_values_ptr(),
1648
0
                                                           map_type->get_value_type());
1649
0
        }
1650
0
        if (const auto* struct_type = typeid_cast<const DataTypeStruct*>(table_type.get())) {
1651
0
            const auto& struct_column = assert_cast<const ColumnStruct&>(*column);
1652
0
            DORIS_CHECK(struct_column.tuple_size() == struct_type->get_elements().size());
1653
0
            for (size_t i = 0; i < struct_column.tuple_size(); ++i) {
1654
0
                if (_requires_parent_null_map_for_alignment(struct_column.get_column_ptr(i),
1655
0
                                                            struct_type->get_element(i))) {
1656
0
                    return true;
1657
0
                }
1658
0
            }
1659
0
        }
1660
0
        return false;
1661
0
    }
1662
1663
    static bool _requires_parent_null_map_for_alignment_at(const ColumnPtr& column,
1664
                                                           const DataTypePtr& table_type,
1665
34
                                                           const size_t row) {
1666
34
        DORIS_CHECK(column.get() != nullptr);
1667
34
        DORIS_CHECK(table_type != nullptr);
1668
34
        DORIS_CHECK(row < column->size());
1669
34
        if (table_type->is_nullable()) {
1670
10
            const auto& nested_type =
1671
10
                    assert_cast<const DataTypeNullable&>(*table_type).get_nested_type();
1672
10
            if (const auto* nullable_column = check_and_get_column<ColumnNullable>(*column)) {
1673
                // A nearer nullable wrapper already protects its descendants at this entry, so an
1674
                // inherited collection mask cannot be needed there.
1675
10
                if (nullable_column->is_null_at(row)) {
1676
0
                    return false;
1677
0
                }
1678
10
                return _requires_parent_null_map_for_alignment_at(
1679
10
                        nullable_column->get_nested_column_ptr(), nested_type, row);
1680
10
            }
1681
0
            return _requires_parent_null_map_for_alignment_at(column, nested_type, row);
1682
10
        }
1683
24
        if (const auto* nullable_column = check_and_get_column<ColumnNullable>(*column)) {
1684
8
            if (nullable_column->is_null_at(row)) {
1685
4
                return true;
1686
4
            }
1687
4
            return _requires_parent_null_map_for_alignment_at(
1688
4
                    nullable_column->get_nested_column_ptr(), table_type, row);
1689
8
        }
1690
16
        if (const auto* array_type = typeid_cast<const DataTypeArray*>(table_type.get())) {
1691
0
            const auto& array_column = assert_cast<const ColumnArray&>(*column);
1692
0
            const auto& offsets = array_column.get_offsets();
1693
0
            const size_t begin = row == 0 ? 0 : offsets[row - 1];
1694
0
            const size_t end = offsets[row];
1695
0
            for (size_t child_row = begin; child_row < end; ++child_row) {
1696
0
                if (_requires_parent_null_map_for_alignment_at(array_column.get_data_ptr(),
1697
0
                                                               array_type->get_nested_type(),
1698
0
                                                               child_row)) {
1699
0
                    return true;
1700
0
                }
1701
0
            }
1702
0
            return false;
1703
0
        }
1704
16
        if (const auto* map_type = typeid_cast<const DataTypeMap*>(table_type.get())) {
1705
0
            const auto& map_column = assert_cast<const ColumnMap&>(*column);
1706
0
            const auto& offsets = map_column.get_offsets();
1707
0
            const size_t begin = row == 0 ? 0 : offsets[row - 1];
1708
0
            const size_t end = offsets[row];
1709
0
            for (size_t child_row = begin; child_row < end; ++child_row) {
1710
0
                if (_requires_parent_null_map_for_alignment_at(
1711
0
                            map_column.get_keys_ptr(), map_type->get_key_type(), child_row) ||
1712
0
                    _requires_parent_null_map_for_alignment_at(
1713
0
                            map_column.get_values_ptr(), map_type->get_value_type(), child_row)) {
1714
0
                    return true;
1715
0
                }
1716
0
            }
1717
0
            return false;
1718
0
        }
1719
16
        if (const auto* struct_type = typeid_cast<const DataTypeStruct*>(table_type.get())) {
1720
8
            const auto& struct_column = assert_cast<const ColumnStruct&>(*column);
1721
8
            DORIS_CHECK(struct_column.tuple_size() == struct_type->get_elements().size());
1722
12
            for (size_t i = 0; i < struct_column.tuple_size(); ++i) {
1723
8
                if (_requires_parent_null_map_for_alignment_at(struct_column.get_column_ptr(i),
1724
8
                                                               struct_type->get_element(i), row)) {
1725
4
                    return true;
1726
4
                }
1727
8
            }
1728
8
        }
1729
12
        return false;
1730
16
    }
1731
1732
    static bool _requires_collection_parent_null_map(const NullMap* parent_null_map,
1733
                                                     const ColumnPtr& column,
1734
                                                     const DataTypePtr& table_type) {
1735
        // Descendant null maps can be entry-sized. Scan them only when an inherited mask can
1736
        // actually hide a row; absent/all-clear masks cannot authorize any physical child NULL.
1737
        if (parent_null_map == nullptr ||
1738
            std::ranges::none_of(*parent_null_map, [](const auto value) { return value != 0; })) {
1739
            return false;
1740
        }
1741
        return _requires_parent_null_map_for_alignment(column, table_type);
1742
    }
1743
1744
    template <typename Offsets>
1745
    static bool _parent_null_map_hides_collection_entries(const NullMap* container_null_map,
1746
                                                          const NullMap* ancestor_null_map,
1747
                                                          const size_t rows,
1748
43
                                                          const Offsets& offsets) {
1749
43
        if (container_null_map == nullptr && ancestor_null_map == nullptr) {
1750
4
            return false;
1751
4
        }
1752
39
        DORIS_CHECK(container_null_map == nullptr || container_null_map->size() == rows);
1753
39
        DORIS_CHECK(ancestor_null_map == nullptr || ancestor_null_map->size() == rows);
1754
39
        DORIS_CHECK(offsets.size() == rows);
1755
39
        size_t begin = 0;
1756
119
        for (size_t row = 0; row < rows; ++row) {
1757
90
            const size_t end = offsets[row];
1758
90
            const bool hidden = (container_null_map != nullptr && (*container_null_map)[row]) ||
1759
90
                                (ancestor_null_map != nullptr && (*ancestor_null_map)[row]);
1760
            // A hidden collection row protects descendants only when its offset span is nonempty.
1761
90
            if (hidden && end > begin) {
1762
10
                return true;
1763
10
            }
1764
80
            begin = end;
1765
80
        }
1766
29
        return false;
1767
39
    }
1768
1769
    template <typename Offsets>
1770
    static bool _requires_collection_parent_null_map(const NullMap* parent_null_map,
1771
                                                     const ColumnPtr& column,
1772
                                                     const DataTypePtr& table_type,
1773
18
                                                     const size_t rows, const Offsets& offsets) {
1774
18
        if (parent_null_map == nullptr) {
1775
0
            return false;
1776
0
        }
1777
18
        DORIS_CHECK(parent_null_map->size() == rows);
1778
18
        DORIS_CHECK(offsets.size() == rows);
1779
18
        DORIS_CHECK(offsets.empty() || offsets.back() == column->size());
1780
18
        size_t begin = 0;
1781
44
        for (size_t row = 0; row < rows; ++row) {
1782
30
            const size_t end = offsets[row];
1783
30
            if ((*parent_null_map)[row]) {
1784
                // Only ancestor-hidden entries can consume this projection. Restricting the probe
1785
                // to their spans avoids scanning visible payload covered by nearer nullable masks.
1786
24
                for (size_t child_row = begin; child_row < end; ++child_row) {
1787
12
                    if (_requires_parent_null_map_for_alignment_at(column, table_type, child_row)) {
1788
4
                        return true;
1789
4
                    }
1790
12
                }
1791
16
            }
1792
26
            begin = end;
1793
26
        }
1794
14
        return false;
1795
18
    }
1796
1797
    template <typename Offsets>
1798
    static const NullMap* _project_collection_parent_null_map_for_hidden_entries(
1799
            const NullMap* container_null_map, const NullMap* ancestor_null_map, const size_t rows,
1800
43
            const Offsets& offsets, const size_t child_rows, NullMap* const projected_null_map) {
1801
43
        if (!_parent_null_map_hides_collection_entries(container_null_map, ancestor_null_map, rows,
1802
43
                                                       offsets)) {
1803
            // Nullable collection wrappers expose a null-map even when every row is present; avoid
1804
            // allocating entry-coordinate scratch unless a hidden row owns physical entries.
1805
33
            return nullptr;
1806
33
        }
1807
10
        projected_null_map->resize(child_rows);
1808
10
        std::fill(projected_null_map->begin(), projected_null_map->end(), 0);
1809
10
        size_t begin = 0;
1810
30
        for (size_t row = 0; row < rows; ++row) {
1811
20
            const size_t end = offsets[row];
1812
20
            const bool hidden = (container_null_map != nullptr && (*container_null_map)[row]) ||
1813
20
                                (ancestor_null_map != nullptr && (*ancestor_null_map)[row]);
1814
20
            if (hidden) {
1815
                // Collection masks use row coordinates; descendants need the same invariant
1816
                // projected through offsets so hidden physical payload cannot fail validation.
1817
10
                std::fill(projected_null_map->begin() + begin, projected_null_map->begin() + end,
1818
10
                          1);
1819
10
            }
1820
20
            begin = end;
1821
20
        }
1822
10
        DORIS_CHECK(begin == child_rows);
1823
10
        return projected_null_map;
1824
43
    }
1825
1826
    Status _materialize_struct_mapping_column(const ColumnMapping& mapping,
1827
                                              const ColumnPtr& file_column, const size_t rows,
1828
                                              ColumnPtr* column,
1829
64
                                              const NullMap* nullable_parent_null_map = nullptr) {
1830
64
        DORIS_CHECK(mapping.table_type != nullptr);
1831
64
        const auto* table_type =
1832
64
                assert_cast<const DataTypeStruct*>(remove_nullable(mapping.table_type).get());
1833
64
        const auto full_file_column = file_column->convert_to_full_column_if_const();
1834
64
        const NullMap* parent_null_map = nullptr;
1835
64
        const auto* nested_file_column =
1836
64
                _nested_column_if_nullable(full_file_column, &parent_null_map);
1837
64
        const auto* file_struct = assert_cast<const ColumnStruct*>(nested_file_column);
1838
64
        DORIS_CHECK(table_type->get_elements().size() == mapping.child_mappings.size());
1839
1840
64
        NullMap combined_parent_null_map;
1841
64
        const NullMap* descendant_parent_null_map = nullable_parent_null_map;
1842
64
        if (parent_null_map != nullptr) {
1843
60
            DORIS_CHECK(parent_null_map->size() == rows);
1844
60
            if (nullable_parent_null_map != nullptr) {
1845
6
                DORIS_CHECK(nullable_parent_null_map->size() == rows);
1846
6
            }
1847
60
            if (!mapping.table_type->is_nullable()) {
1848
14
                for (size_t i = 0; i < rows; ++i) {
1849
                    // A required nested container may drop its own NULL only when an ancestor
1850
                    // already hides that row; otherwise physical defaults become visible values.
1851
10
                    if ((*parent_null_map)[i] &&
1852
10
                        (nullable_parent_null_map == nullptr || !(*nullable_parent_null_map)[i])) {
1853
2
                        return Status::InternalError(
1854
2
                                "Source struct contains NULL for non-nullable table column");
1855
2
                    }
1856
10
                }
1857
6
            }
1858
58
            combined_parent_null_map.resize(rows);
1859
186
            for (size_t i = 0; i < rows; ++i) {
1860
128
                combined_parent_null_map[i] =
1861
128
                        (*parent_null_map)[i] ||
1862
128
                        (nullable_parent_null_map != nullptr && (*nullable_parent_null_map)[i]);
1863
128
            }
1864
58
            descendant_parent_null_map = &combined_parent_null_map;
1865
58
        }
1866
1867
62
        Columns child_columns;
1868
62
        child_columns.reserve(mapping.child_mappings.size());
1869
62
        const auto file_ordered_children =
1870
62
                _present_child_mappings_in_file_order(mapping.child_mappings);
1871
62
        const auto table_ordered_children =
1872
62
                _child_mappings_in_table_type_order(mapping, *table_type);
1873
98
        for (const auto* child_mapping : table_ordered_children) {
1874
98
            DORIS_CHECK(child_mapping != nullptr);
1875
98
            if (!child_mapping->file_local_id.has_value()) {
1876
26
                ColumnPtr child_column;
1877
26
                RETURN_IF_ERROR(_materialize_default_or_missing_column(
1878
26
                        *child_mapping, nullptr, rows, &child_column, descendant_parent_null_map));
1879
26
                child_column = child_column->convert_to_full_column_if_const();
1880
26
                child_columns.push_back(std::move(child_column));
1881
26
                continue;
1882
26
            }
1883
72
            const auto file_child_idx =
1884
72
                    _file_child_ordinal_for_mapping(mapping, *child_mapping, file_ordered_children);
1885
72
            DORIS_CHECK(file_child_idx < file_struct->get_columns().size());
1886
72
            ColumnPtr child_column = file_struct->get_column_ptr(file_child_idx);
1887
72
            RETURN_IF_ERROR(_materialize_present_child_mapping_column(
1888
72
                    *child_mapping, child_column, rows, &child_column, descendant_parent_null_map));
1889
64
            child_columns.push_back(std::move(child_column));
1890
64
        }
1891
54
        MutableColumns mutable_child_columns;
1892
54
        mutable_child_columns.reserve(child_columns.size());
1893
86
        for (auto& child_column : child_columns) {
1894
86
            mutable_child_columns.push_back(IColumn::mutate(std::move(child_column)));
1895
86
        }
1896
54
        auto result = ColumnStruct::create(std::move(mutable_child_columns));
1897
54
        if (mapping.table_type->is_nullable()) {
1898
46
            auto null_map = ColumnUInt8::create();
1899
46
            auto& null_map_data = null_map->get_data();
1900
46
            null_map_data.resize(rows);
1901
46
            if (parent_null_map != nullptr) {
1902
46
                DORIS_CHECK(parent_null_map->size() == rows);
1903
46
                null_map_data.assign(parent_null_map->begin(), parent_null_map->end());
1904
46
            } else {
1905
0
                std::fill(null_map_data.begin(), null_map_data.end(), 0);
1906
0
            }
1907
46
            *column = ColumnNullable::create(std::move(result), std::move(null_map));
1908
46
        } else {
1909
8
            *column = std::move(result);
1910
8
        }
1911
54
        return Status::OK();
1912
62
    }
1913
1914
    Status _materialize_array_mapping_column(const ColumnMapping& mapping,
1915
                                             const ColumnPtr& file_column, const size_t rows,
1916
                                             ColumnPtr* column,
1917
23
                                             const NullMap* nullable_parent_null_map = nullptr) {
1918
23
        DORIS_CHECK(mapping.child_mappings.size() == 1);
1919
23
        const auto full_file_column = file_column->convert_to_full_column_if_const();
1920
23
        const NullMap* parent_null_map = nullptr;
1921
23
        const auto* nested_file_column =
1922
23
                _nested_column_if_nullable(full_file_column, &parent_null_map);
1923
23
        if (parent_null_map != nullptr && !mapping.table_type->is_nullable()) {
1924
4
            DORIS_CHECK(parent_null_map->size() == rows);
1925
4
            if (nullable_parent_null_map != nullptr) {
1926
2
                DORIS_CHECK(nullable_parent_null_map->size() == rows);
1927
2
            }
1928
8
            for (size_t i = 0; i < rows; ++i) {
1929
                // ARRAY row masks cannot be forwarded to elements because they use different
1930
                // coordinates, so validate the container before dropping its nullable wrapper.
1931
6
                if ((*parent_null_map)[i] &&
1932
6
                    (nullable_parent_null_map == nullptr || !(*nullable_parent_null_map)[i])) {
1933
2
                    return Status::InternalError(
1934
2
                            "Source array contains NULL for non-nullable table column");
1935
2
                }
1936
6
            }
1937
4
        }
1938
21
        const auto* file_array = assert_cast<const ColumnArray*>(nested_file_column);
1939
21
        ColumnPtr nested_column = file_array->get_data_ptr();
1940
21
        auto element_mapping = mapping.child_mappings[0];
1941
        // Keep the descriptor type for schema matching. ARRAY's nullable element wrapper is a
1942
        // storage invariant, so add it only at the materialization boundary.
1943
21
        element_mapping.table_type = make_nullable(element_mapping.table_type);
1944
21
        NullMap descendant_parent_null_map;
1945
21
        const NullMap* descendant_parent_null_map_ptr =
1946
21
                _project_collection_parent_null_map_for_hidden_entries(
1947
21
                        parent_null_map, nullable_parent_null_map, rows, file_array->get_offsets(),
1948
21
                        nested_column->size(), &descendant_parent_null_map);
1949
21
        RETURN_IF_ERROR(_materialize_present_child_mapping_column(
1950
21
                element_mapping, nested_column, nested_column->size(), &nested_column,
1951
21
                descendant_parent_null_map_ptr));
1952
15
        auto offsets_column = file_array->get_offsets_ptr()->convert_to_full_column_if_const();
1953
15
        auto result = ColumnArray::create(IColumn::mutate(std::move(nested_column)),
1954
15
                                          IColumn::mutate(std::move(offsets_column)));
1955
15
        if (mapping.table_type->is_nullable()) {
1956
13
            auto null_map = ColumnUInt8::create();
1957
13
            auto& null_map_data = null_map->get_data();
1958
13
            null_map_data.resize(rows);
1959
13
            if (parent_null_map != nullptr) {
1960
13
                DORIS_CHECK(parent_null_map->size() == rows);
1961
13
                null_map_data.assign(parent_null_map->begin(), parent_null_map->end());
1962
13
            } else {
1963
0
                std::fill(null_map_data.begin(), null_map_data.end(), 0);
1964
0
            }
1965
13
            *column = ColumnNullable::create(std::move(result), std::move(null_map));
1966
13
        } else {
1967
2
            *column = std::move(result);
1968
2
        }
1969
15
        return Status::OK();
1970
21
    }
1971
1972
    Status _materialize_map_mapping_column(const ColumnMapping& mapping,
1973
                                           const ColumnPtr& file_column, const size_t rows,
1974
                                           ColumnPtr* column,
1975
14
                                           const NullMap* nullable_parent_null_map = nullptr) {
1976
14
        const auto full_file_column = file_column->convert_to_full_column_if_const();
1977
14
        const NullMap* parent_null_map = nullptr;
1978
14
        const auto* nested_file_column =
1979
14
                _nested_column_if_nullable(full_file_column, &parent_null_map);
1980
14
        if (parent_null_map != nullptr && !mapping.table_type->is_nullable()) {
1981
0
            DORIS_CHECK(parent_null_map->size() == rows);
1982
0
            if (nullable_parent_null_map != nullptr) {
1983
0
                DORIS_CHECK(nullable_parent_null_map->size() == rows);
1984
0
            }
1985
0
            for (size_t i = 0; i < rows; ++i) {
1986
                // MAP row masks cannot be forwarded to entries because they use different
1987
                // coordinates, so validate the container before dropping its nullable wrapper.
1988
0
                if ((*parent_null_map)[i] &&
1989
0
                    (nullable_parent_null_map == nullptr || !(*nullable_parent_null_map)[i])) {
1990
0
                    return Status::InternalError(
1991
0
                            "Source map contains NULL for non-nullable table column");
1992
0
                }
1993
0
            }
1994
0
        }
1995
14
        const auto* file_map = assert_cast<const ColumnMap*>(nested_file_column);
1996
14
        ColumnPtr key_column = file_map->get_keys_ptr();
1997
14
        ColumnPtr value_column = file_map->get_values_ptr();
1998
14
        DORIS_CHECK(key_column->size() == value_column->size());
1999
14
        NullMap descendant_parent_null_map;
2000
14
        const NullMap* descendant_parent_null_map_ptr =
2001
14
                _project_collection_parent_null_map_for_hidden_entries(
2002
14
                        parent_null_map, nullable_parent_null_map, rows, file_map->get_offsets(),
2003
14
                        key_column->size(), &descendant_parent_null_map);
2004
2005
14
        const ColumnMapping* key_mapping = nullptr;
2006
14
        const ColumnMapping* value_mapping = nullptr;
2007
22
        for (const auto& child_mapping : mapping.child_mappings) {
2008
22
            if (!child_mapping.file_local_id.has_value()) {
2009
0
                continue;
2010
0
            }
2011
22
            if (*child_mapping.file_local_id == 0) {
2012
8
                key_mapping = &child_mapping;
2013
14
            } else if (*child_mapping.file_local_id == 1) {
2014
14
                value_mapping = &child_mapping;
2015
14
            }
2016
22
        }
2017
2018
14
        if (key_mapping != nullptr) {
2019
8
            RETURN_IF_ERROR(_materialize_present_child_mapping_column(
2020
8
                    *key_mapping, key_column, key_column->size(), &key_column,
2021
8
                    descendant_parent_null_map_ptr));
2022
8
        } else {
2023
6
            const auto* table_map =
2024
6
                    assert_cast<const DataTypeMap*>(remove_nullable(mapping.table_type).get());
2025
            // Value-only projection retains the physical key stream to preserve entry offsets;
2026
            // align it under the entry mask so NULL placeholders from hidden Map rows stay hidden.
2027
6
            RETURN_IF_ERROR(_align_column_nullability(&key_column, table_map->get_key_type(),
2028
6
                                                      descendant_parent_null_map_ptr));
2029
6
        }
2030
14
        if (value_mapping != nullptr) {
2031
14
            RETURN_IF_ERROR(_materialize_present_child_mapping_column(
2032
14
                    *value_mapping, value_column, value_column->size(), &value_column,
2033
14
                    descendant_parent_null_map_ptr));
2034
14
        } else {
2035
0
            const auto* table_map =
2036
0
                    assert_cast<const DataTypeMap*>(remove_nullable(mapping.table_type).get());
2037
            // A retained structural value stream follows the same hidden-entry invariant as keys.
2038
0
            RETURN_IF_ERROR(_align_column_nullability(&value_column, table_map->get_value_type(),
2039
0
                                                      descendant_parent_null_map_ptr));
2040
0
        }
2041
14
        auto offsets_column = file_map->get_offsets_ptr()->convert_to_full_column_if_const();
2042
14
        auto result = ColumnMap::create(IColumn::mutate(std::move(key_column)),
2043
14
                                        IColumn::mutate(std::move(value_column)),
2044
14
                                        IColumn::mutate(std::move(offsets_column)));
2045
14
        if (mapping.table_type->is_nullable()) {
2046
10
            auto null_map = ColumnUInt8::create();
2047
10
            auto& null_map_data = null_map->get_data();
2048
10
            null_map_data.resize(rows);
2049
10
            if (parent_null_map != nullptr) {
2050
10
                DORIS_CHECK(parent_null_map->size() == rows);
2051
10
                null_map_data.assign(parent_null_map->begin(), parent_null_map->end());
2052
10
            } else {
2053
0
                std::fill(null_map_data.begin(), null_map_data.end(), 0);
2054
0
            }
2055
10
            *column = ColumnNullable::create(std::move(result), std::move(null_map));
2056
10
        } else {
2057
4
            *column = std::move(result);
2058
4
        }
2059
14
        return Status::OK();
2060
14
    }
2061
2062
6.40k
    Status _open_mapping_expr_tree(const ColumnMapping& mapping, const RowDescriptor& row_desc) {
2063
6.40k
        if (mapping.projection != nullptr) {
2064
4.09k
            RETURN_IF_ERROR(mapping.projection->prepare(_runtime_state, row_desc));
2065
4.09k
            RETURN_IF_ERROR(mapping.projection->open(_runtime_state));
2066
4.09k
        }
2067
6.40k
        if (mapping.default_expr != nullptr) {
2068
34
            RETURN_IF_ERROR(mapping.default_expr->prepare(_runtime_state, row_desc));
2069
34
            RETURN_IF_ERROR(mapping.default_expr->open(_runtime_state));
2070
34
        }
2071
6.40k
        for (const auto& child_mapping : mapping.child_mappings) {
2072
2.23k
            RETURN_IF_ERROR(_open_mapping_expr_tree(child_mapping, row_desc));
2073
2.23k
        }
2074
6.40k
        return Status::OK();
2075
6.40k
    }
2076
2077
949
    Status _open_mapping_exprs() {
2078
949
        RowDescriptor row_desc;
2079
4.16k
        for (const auto& mapping : _data_reader.column_mapper->mappings()) {
2080
4.16k
            RETURN_IF_ERROR(_open_mapping_expr_tree(mapping, row_desc));
2081
4.16k
        }
2082
949
        return Status::OK();
2083
949
    }
2084
2085
    Status _build_file_aggregate_request(TPushAggOp::type agg_type,
2086
26
                                         FileAggregateRequest* request) const {
2087
26
        DORIS_CHECK(request != nullptr);
2088
26
        DORIS_CHECK(_supports_aggregate_pushdown(agg_type));
2089
26
        request->agg_type = agg_type;
2090
26
        request->columns.clear();
2091
26
        if (agg_type == TPushAggOp::type::COUNT) {
2092
18
            DORIS_CHECK(_push_down_count_columns.has_value());
2093
            // An empty explicit list is the semantic signal for COUNT(*). Do not inspect the
2094
            // mapping count: `SELECT COUNT(*) FROM t` may still project one nullable column because
2095
            // the planner keeps a placeholder slot. In a 10,000-row file where that arbitrary slot
2096
            // has 9,015 non-null values, passing the slot would ask Parquet/ORC metadata for
2097
            // COUNT(slot)=9,015 instead of the required row count 10,000.
2098
18
            if (!_push_down_count_columns->empty()) {
2099
8
                const auto& mapping = _push_down_count_mapping();
2100
8
                DORIS_CHECK(mapping.file_local_id.has_value());
2101
8
                FileAggregateRequest::Column column;
2102
8
                column.projection =
2103
8
                        LocalColumnIndex::top_level(LocalColumnId(*mapping.file_local_id));
2104
8
                request->columns.push_back(std::move(column));
2105
8
            }
2106
18
            return Status::OK();
2107
18
        }
2108
8
        request->columns.reserve(_data_reader.column_mapper->mappings().size());
2109
10
        for (const auto& mapping : _data_reader.column_mapper->mappings()) {
2110
10
            DORIS_CHECK(mapping.file_local_id.has_value());
2111
10
            FileAggregateRequest::Column column;
2112
10
            column.projection = LocalColumnIndex::top_level(LocalColumnId(*mapping.file_local_id));
2113
10
            if (!mapping.child_mappings.empty()) {
2114
2
                RETURN_IF_ERROR(build_aggregate_projection(mapping, &column.projection));
2115
2
            }
2116
10
            request->columns.push_back(std::move(column));
2117
10
        }
2118
8
        return Status::OK();
2119
8
    }
2120
2121
28
    const ColumnMapping& _push_down_count_mapping() const {
2122
28
        DORIS_CHECK(_push_down_count_columns.has_value());
2123
28
        DORIS_CHECK(_push_down_count_columns->size() == 1);
2124
28
        const auto mapping_it =
2125
28
                std::ranges::find(_data_reader.column_mapper->mappings(),
2126
28
                                  _push_down_count_columns->front(), &ColumnMapping::global_index);
2127
        // FileScannerV2 translates FE SlotIds through the same projected-column list used to build
2128
        // the mapper, so a missing mapping is an FE/BE contract violation rather than a fallback.
2129
28
        DORIS_CHECK(mapping_it != _data_reader.column_mapper->mappings().end());
2130
28
        return *mapping_it;
2131
28
    }
2132
2133
    Status _materialize_aggregate_pushdown_rows(TPushAggOp::type agg_type,
2134
                                                const FileAggregateResult& file_result,
2135
6
                                                Block* block) {
2136
6
        DORIS_CHECK(agg_type == TPushAggOp::type::MINMAX);
2137
        // MIN/MAX pushdown emits two rows, min first and max second, for each projected column.
2138
        // The upper MIN/MAX aggregate consumes those two rows to produce the final aggregate value.
2139
6
        DORIS_CHECK(file_result.columns.size() == _data_reader.column_mapper->mappings().size());
2140
6
        DORIS_CHECK(block->columns() == _data_reader.column_mapper->mappings().size());
2141
6
        Block file_block;
2142
6
        file_block.reserve(_data_reader.file_block_layout.size());
2143
8
        for (const auto& column : _data_reader.file_block_layout) {
2144
8
            file_block.insert({column.type->create_column(), column.type, column.name});
2145
8
        }
2146
14
        for (size_t column_idx = 0; column_idx < file_result.columns.size(); ++column_idx) {
2147
8
            const auto& result_column = file_result.columns[column_idx];
2148
8
            if (!result_column.has_min || !result_column.has_max) {
2149
0
                return Status::NotSupported("Missing min/max aggregate result for column {}",
2150
0
                                            _projected_columns[column_idx].name);
2151
0
            }
2152
8
            bool found_file_column = false;
2153
10
            for (size_t block_position = 0; block_position < _data_reader.file_block_layout.size();
2154
10
                 ++block_position) {
2155
10
                if (_data_reader.file_block_layout[block_position].file_column_id ==
2156
10
                    file_result.columns[column_idx].projection.column_id()) {
2157
8
                    found_file_column = true;
2158
8
                    auto column = file_block.get_by_position(block_position)
2159
8
                                          .type->create_column()
2160
8
                                          ->assert_mutable();
2161
8
                    RETURN_IF_ERROR(_insert_aggregate_projection_value(
2162
8
                            file_result.columns[column_idx].projection, result_column.min_value,
2163
8
                            column.get()));
2164
8
                    RETURN_IF_ERROR(_insert_aggregate_projection_value(
2165
8
                            file_result.columns[column_idx].projection, result_column.max_value,
2166
8
                            column.get()));
2167
8
                    file_block.replace_by_position(block_position, std::move(column));
2168
8
                    break;
2169
8
                }
2170
10
            }
2171
8
            DORIS_CHECK(found_file_column);
2172
8
        }
2173
14
        for (size_t column_idx = 0; column_idx < _data_reader.column_mapper->mappings().size();
2174
8
             ++column_idx) {
2175
8
            ColumnPtr table_column;
2176
8
            RETURN_IF_ERROR(_materialize_mapping_column(
2177
8
                    _data_reader.column_mapper->mappings()[column_idx], &file_block, 2,
2178
8
                    &table_column,
2179
8
                    column_idx + 1 == _data_reader.column_mapper->mappings().size()));
2180
8
            block->replace_by_position(column_idx, std::move(table_column));
2181
8
        }
2182
6
        return Status::OK();
2183
6
    }
2184
2185
    struct FileBlockColumn {
2186
        LocalColumnId file_column_id = LocalColumnId::invalid();
2187
        std::string name;
2188
        DataTypePtr type;
2189
    };
2190
2191
    struct DataReader {
2192
        std::unique_ptr<FileReader> reader;
2193
        std::unique_ptr<TableColumnMapper> column_mapper;
2194
        // Schema of the data file, also including virtual column (row position).
2195
        std::vector<ColumnDefinition> file_schema;
2196
        // Layout of the block returned by file reader, determined by column mapping and file
2197
        // schema. It is used for file reader to materialize columns into correct type and position.
2198
        std::vector<FileBlockColumn> file_block_layout;
2199
        Block block_template;
2200
    };
2201
    DataReader _data_reader;
2202
    // Latest immutable request queued to the physical reader. The file-block layout remains fixed
2203
    // for the split even while predicates are refreshed at a reader-defined granule boundary.
2204
    std::shared_ptr<FileScanRequest> _file_scan_request;
2205
    std::vector<ColumnDefinition> _projected_columns;
2206
    std::unique_ptr<ScanTask> _current_task;
2207
    std::optional<io::FileDescription> _current_file_description;
2208
    // Range-level compression has higher priority than scan-param compression. TVF/load can keep
2209
    // the logical format as CSV/TEXT while carrying the concrete compression such as GZ or LZO on
2210
    // each TFileRangeDesc, matching the old FileScanner reader contract.
2211
    TFileCompressType::type _current_range_compress_type = TFileCompressType::UNKNOWN;
2212
    std::optional<TUniqueId> _current_range_load_id;
2213
    TFileRangeDesc _current_file_range_desc;
2214
    std::shared_ptr<io::FileSystemProperties> _system_properties;
2215
    // partition key -> value
2216
    std::map<std::string, Field> _partition_values;
2217
    // Predicates built from scan conjuncts before file-level localization.
2218
    std::vector<TableFilter> _table_filters;
2219
    // Number of localized filters before the first unsafe conjunct in the original row-level
2220
    // order. This differs from scanning `_table_filters` for safety because slotless predicates are
2221
    // intentionally absent from that vector but must still act as ordering barriers.
2222
    size_t _constant_pruning_safe_filter_count = 0;
2223
    VExprContextSPtrs _conjuncts;
2224
    ReadProfile _profile;
2225
    // Parsed from row-position based delete files, including position delete and deletion vector.
2226
    DeleteRows* _delete_rows = nullptr;
2227
    DeletionVector* _deletion_vector = nullptr;
2228
    TFileScanRangeParams* _scan_params;
2229
    std::shared_ptr<io::IOContext> _io_ctx;
2230
    RuntimeState* _runtime_state;
2231
    RuntimeProfile* _scanner_profile;
2232
    const std::vector<SlotDescriptor*>* _file_slot_descs = nullptr;
2233
    FileFormat _format;
2234
    TPushAggOp::type _push_down_agg_type = TPushAggOp::type::NONE;
2235
    std::optional<std::vector<GlobalIndex>> _push_down_count_columns;
2236
    size_t _batch_size = 0;
2237
    uint64_t _initial_condition_cache_digest = 0;
2238
    uint64_t _condition_cache_digest = 0;
2239
    // True only when prepare_split() received a digest for the exact conjunct snapshot used by
2240
    // this split. Standalone callers that only supplied TableReadOptions::condition_cache_digest
2241
    // keep the conservative runtime-filter guard.
2242
    bool _condition_cache_digest_covers_current_split = false;
2243
    segment_v2::ConditionCache::ExternalCacheKey _condition_cache_key;
2244
    std::shared_ptr<std::vector<bool>> _condition_cache;
2245
    std::shared_ptr<ConditionCacheContext> _condition_cache_ctx;
2246
    int64_t _condition_cache_hit_count = 0;
2247
    bool _current_reader_reached_eof = false;
2248
    int64_t _remaining_table_level_count = -1;
2249
    int64_t _remaining_file_level_count = -1;
2250
    // True only after the active split selects a table-level row-count shortcut or successfully
2251
    // materializes COUNT rows from file metadata. FileScannerV2 uses this result, rather than the
2252
    // raw aggregate opcode, to keep adaptive batching enabled for normal row-scan fallbacks.
2253
    bool _current_split_uses_metadata_count = false;
2254
    // Snapshot supplied by FileScannerV2 for the active split. It gates every shortcut that emits
2255
    // irreversible aggregate rows, not only the table-level row-count shortcut in prepare_split().
2256
    bool _all_runtime_filters_applied_for_split = true;
2257
    std::optional<GlobalRowIdContext> _global_rowid_context;
2258
    std::optional<std::vector<int64_t>> _row_ids;
2259
    bool _aggregate_pushdown_tried = false;
2260
    bool _current_split_pruned = false;
2261
    TableColumnMapperOptions _mapper_options;
2262
2263
private:
2264
    static std::optional<std::string> _get_int96_timezone_override(
2265
            const TFileScanRangeParams* params);
2266
2267
    static const ColumnDefinition* _find_column_definition(
2268
4.31k
            const std::vector<ColumnDefinition>& schema, LocalColumnId column_id) {
2269
28.1k
        for (const auto& field : schema) {
2270
28.1k
            if (field.file_local_id() == column_id.value()) {
2271
4.22k
                return &field;
2272
4.22k
            }
2273
28.1k
        }
2274
86
        return nullptr;
2275
4.31k
    }
2276
2277
28
    static bool _can_push_down_minmax_for_mapping(const ColumnMapping& mapping) {
2278
28
        if (mapping.child_mappings.empty()) {
2279
            // Direct mappings use a slot-ref projection to materialize the file column. The
2280
            // projection does not transform ordering; casts and other conversions are already
2281
            // represented by a non-trivial mapping and must fall back to row scanning.
2282
22
            return mapping.is_trivial;
2283
22
        }
2284
6
        const auto primitive_type = remove_nullable(mapping.file_type)->get_primitive_type();
2285
6
        if (primitive_type != TYPE_STRUCT) {
2286
2
            return false;
2287
2
        }
2288
4
        size_t mapped_children = 0;
2289
4
        const ColumnMapping* mapped_child = nullptr;
2290
4
        for (const auto& child_mapping : mapping.child_mappings) {
2291
4
            if (!child_mapping.file_local_id.has_value()) {
2292
0
                continue;
2293
0
            }
2294
4
            ++mapped_children;
2295
4
            mapped_child = &child_mapping;
2296
4
        }
2297
4
        return mapped_children == 1 && mapped_child != nullptr &&
2298
4
               _can_push_down_minmax_for_mapping(*mapped_child);
2299
6
    }
2300
2301
    static Status build_aggregate_projection(const ColumnMapping& mapping,
2302
4
                                             LocalColumnIndex* projection) {
2303
4
        DORIS_CHECK(projection != nullptr);
2304
4
        DORIS_CHECK(mapping.file_local_id.has_value());
2305
4
        *projection = LocalColumnIndex::local(*mapping.file_local_id);
2306
4
        projection->children.clear();
2307
4
        projection->project_all_children = true;
2308
4
        if (mapping.child_mappings.empty()) {
2309
2
            return Status::OK();
2310
2
        }
2311
2
        projection->project_all_children = false;
2312
2
        for (const auto& child_mapping : mapping.child_mappings) {
2313
2
            if (!child_mapping.file_local_id.has_value()) {
2314
0
                continue;
2315
0
            }
2316
2
            LocalColumnIndex child_projection;
2317
2
            RETURN_IF_ERROR(build_aggregate_projection(child_mapping, &child_projection));
2318
2
            projection->children.push_back(std::move(child_projection));
2319
2
        }
2320
2
        DORIS_CHECK(projection->children.size() == 1);
2321
2
        return Status::OK();
2322
2
    }
2323
2324
    static Status _insert_aggregate_projection_value(const LocalColumnIndex& projection,
2325
40
                                                     const Field& value, IColumn* column) {
2326
40
        DORIS_CHECK(column != nullptr);
2327
40
        if (auto* nullable_column = check_and_get_column<ColumnNullable>(*column)) {
2328
20
            RETURN_IF_ERROR(_insert_aggregate_projection_value(
2329
20
                    projection, value, &nullable_column->get_nested_column()));
2330
20
            nullable_column->get_null_map_data().push_back(0);
2331
20
            return Status::OK();
2332
20
        }
2333
20
        if (projection.project_all_children || projection.children.empty()) {
2334
16
            column->insert(value);
2335
16
            return Status::OK();
2336
16
        }
2337
4
        auto* struct_column = assert_cast<ColumnStruct*>(column);
2338
4
        DORIS_CHECK(projection.children.size() == 1);
2339
4
        const auto& child_projection = projection.children[0];
2340
4
        DORIS_CHECK(struct_column->get_columns().size() == 1);
2341
4
        RETURN_IF_ERROR(_insert_aggregate_projection_value(child_projection, value,
2342
4
                                                           &struct_column->get_column(0)));
2343
4
        return Status::OK();
2344
4
    }
2345
2346
    // Parse a DV into its compressed bitmap. Position delete files continue to use _delete_rows.
2347
    Status _parse_delete_predicates(const SplitReadOptions& options);
2348
};
2349
2350
} // namespace doris::format