Coverage Report

Created: 2026-11-03 11:39

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