Coverage Report

Created: 2026-07-31 13:05

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