Coverage Report

Created: 2026-10-30 14:13

next uncovered line (L), next uncovered region (R), next uncovered branch (B)
be/src/storage/segment/segment_iterator.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 <gen_cpp/Exprs_types.h>
21
22
#include <cstddef>
23
#include <cstdint>
24
#include <map>
25
#include <memory>
26
#include <roaring/roaring.hh>
27
#include <set>
28
#include <string>
29
#include <unordered_map>
30
#include <utility>
31
#include <vector>
32
33
#include "common/status.h"
34
#include "core/block/adaptive_block_size_predictor.h"
35
#include "core/block/block.h"
36
#include "core/block/column_with_type_and_name.h"
37
#include "core/block/columns_with_type_and_name.h"
38
#include "core/column/column.h"
39
#include "core/data_type/data_type.h"
40
#include "core/data_type/primitive_type.h"
41
#include "core/field.h"
42
#include "exec/common/variant_util.h"
43
#include "exprs/score_runtime.h"
44
#include "exprs/vexpr_fwd.h"
45
#include "io/fs/file_reader_writer_fwd.h"
46
#include "runtime/runtime_profile.h"
47
#include "storage/index/ann/ann_topn_runtime.h"
48
#include "storage/index/index_iterator.h"
49
#include "storage/iterators.h"
50
#include "storage/olap_common.h"
51
#include "storage/predicate/block_column_predicate.h"
52
#include "storage/predicate/column_predicate.h"
53
#include "storage/row_cursor.h"
54
#include "storage/schema.h"
55
#include "storage/segment/common.h"
56
#include "storage/segment/segment.h"
57
#include "util/json/path_in_data.h"
58
#include "util/slice.h"
59
60
namespace doris {
61
62
class VExpr;
63
class VExprContext;
64
65
namespace segment_v2 {
66
67
class ColumnIterator;
68
struct ColumnReadAheadContext;
69
struct ColumnReadAheadPlan;
70
class RowRanges;
71
class IndexIterator;
72
class SegmentReadAhead;
73
74
class SegmentIterator : public RowwiseIterator {
75
public:
76
    // Within SegmentIterator, ColumnId means an ordinal in the read schema.
77
    // Storage UIDs and caller-visible Block positions are named explicitly.
78
    SegmentIterator(std::shared_ptr<Segment> segment, ReadSchemaSPtr schema,
79
                    const StorageReadOptions& opts);
80
    ~SegmentIterator() override;
81
82
    [[nodiscard]] Status init_iterators();
83
    [[nodiscard]] Status init(const StorageReadOptions& opts) override;
84
    [[nodiscard]] Status next_batch(Block* block) override;
85
86
314
    const ReadSchema& schema() const override { return *_schema; }
87
26
    uint64_t data_id() const override { return _segment->id(); }
88
89
0
    void update_profile(RuntimeProfile* profile) override {
90
0
        _update_profile(profile, _short_cir_eval_predicate, "ShortCircuitPredicates");
91
0
        _update_profile(profile, _pre_eval_block_predicate, "PreEvaluatePredicates");
92
93
0
        if (_opts.delete_condition_predicates != nullptr) {
94
0
            std::set<std::shared_ptr<const ColumnPredicate>> delete_predicate_set;
95
0
            _opts.delete_condition_predicates->get_all_column_predicate(delete_predicate_set);
96
0
            _update_profile(profile, delete_predicate_set, "DeleteConditionPredicates");
97
0
        }
98
0
    }
99
100
218
    bool has_index_in_iterators() const {
101
218
        return std::any_of(_index_iterators.begin(), _index_iterators.end(),
102
400
                           [](const auto& iterator) { return iterator != nullptr; });
103
218
    }
104
105
private:
106
    Status _next_batch_internal(Block* block);
107
108
    Status _check_output_block(Block* block);
109
110
    template <typename Container>
111
    void _update_profile(RuntimeProfile* profile, const Container& predicates,
112
0
                         const std::string& title) {
113
0
        if (predicates.empty()) {
114
0
            return;
115
0
        }
116
0
        std::string info;
117
0
        for (auto pred : predicates) {
118
0
            info += "\n" + pred->debug_string();
119
0
        }
120
0
        profile->add_info_string(title, info);
121
0
    }
Unexecuted instantiation: _ZN5doris10segment_v215SegmentIterator15_update_profileISt6vectorISt10shared_ptrINS_15ColumnPredicateEESaIS6_EEEEvPNS_14RuntimeProfileERKT_RKNSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEE
Unexecuted instantiation: _ZN5doris10segment_v215SegmentIterator15_update_profileISt3setISt10shared_ptrIKNS_15ColumnPredicateEESt4lessIS7_ESaIS7_EEEEvPNS_14RuntimeProfileERKT_RKNSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEE
122
123
    [[nodiscard]] Status _lazy_init(Block* block);
124
    [[nodiscard]] Status _init_impl(const StorageReadOptions& opts);
125
    // Create the scalar read-ahead context before column iterators capture their FileReader.
126
    [[nodiscard]] Status _init_query_read_ahead();
127
    // Complex columns retain their original reader and segment prefetcher.
128
    bool _use_read_ahead_for_column(ColumnId cid) const;
129
    [[nodiscard]] Status _init_column_iterators();
130
    [[nodiscard]] Status _init_index_iterators();
131
    // Initialize eager and lazy page sequences after pruning and submit their first windows
132
    // together. Subsequent data-page reads advance each physical column's window.
133
    void _prepare_scan_read_ahead();
134
    // Classify physical iterators by filtering dependency and plan their initial windows.
135
    std::vector<ColumnReadAheadPlan> _plan_scan_read_ahead();
136
137
    // calculate row ranges that fall into requested key ranges using short key index
138
    [[nodiscard]] Status _get_row_ranges_by_keys();
139
    [[nodiscard]] Status _prepare_seek(const StorageReadOptions::KeyRange& key_range);
140
    [[nodiscard]] Status _lookup_ordinal(const RowCursor& key, bool is_include, rowid_t upper_bound,
141
                                         rowid_t* rowid);
142
    // lookup the ordinal of given key from short key index
143
    // the returned rowid is rowid in primary index, not the rowid encoded in primary key
144
    [[nodiscard]] Status _lookup_ordinal_from_sk_index(const RowCursor& key, bool is_include,
145
                                                       rowid_t upper_bound, rowid_t* rowid);
146
    // lookup the ordinal of given key from primary key index
147
    [[nodiscard]] Status _lookup_ordinal_from_pk_index(const RowCursor& key, bool is_include,
148
                                                       rowid_t* rowid);
149
    [[nodiscard]] Status _seek_and_peek(rowid_t rowid);
150
151
    // calculate row ranges that satisfy requested column conditions using various column index
152
    [[nodiscard]] Status _get_row_ranges_by_column_conditions();
153
    [[nodiscard]] Status _apply_scan_restrictions();
154
    [[nodiscard]] Status _get_row_ranges_from_conditions(RowRanges* condition_row_ranges);
155
    [[nodiscard]] Status _apply_expr_zonemap_to_row_ranges(const VExprContextSPtrs& conjuncts,
156
                                                           rowid_t min_rowid,
157
                                                           RowRanges* row_ranges);
158
    [[nodiscard]] Status _apply_inverted_index();
159
    [[nodiscard]] Status _apply_inverted_index_on_column_predicate(
160
            std::shared_ptr<ColumnPredicate> pred,
161
            std::vector<std::shared_ptr<ColumnPredicate>>& remaining_predicates,
162
            bool* continue_apply);
163
    [[nodiscard]] Status _apply_ann_topn_predicate();
164
    [[nodiscard]] Status _apply_index_expr();
165
    // Publish _row_bitmap as IndexQueryContext::candidate_rows when it is
166
    // below the configured engage ratio; refreshed at conjunct boundaries as
167
    // earlier index conjuncts shrink the bitmap. No-op once engaged.
168
    void _refresh_candidate_pushdown();
169
    // Intersect an approximate result and keep its expression for row evaluation.
170
    bool _apply_approx_index_result(VExprContext* expr_ctx);
171
    // G02: true iff answering the single pushed-down MATCH predicate by its
172
    // match COUNT alone is indistinguishable from the row-accurate bitmap for
173
    // this COUNT_ON_INDEX scan (no deletes, no other filters, full row bitmap,
174
    // no row-id consumers). Gates IndexQueryContext::count_on_index_fastpath;
175
    // the decision predicate itself lives in count_on_index_fastpath.h.
176
    bool _count_on_index_fastpath_safe() const;
177
    // G03: teardown of the G02 handshake. Captures whether the reader answered
178
    // with a fabricated count bitmap into _count_fastpath_hit and clears both
179
    // context flags so no later read_from_index call can observe or forge
180
    // them. Runs on every exit of the index-apply scope.
181
    void _capture_count_fastpath_hit();
182
    // G03: true iff the per-batch defaults fill of _read_columns_by_index
183
    // would apply to `cid` (the _no_need_read_key_data or _prune_column
184
    // branch) AND the block column needs no storage->schema cast, i.e. the
185
    // emission shortcut can reproduce the column's batch content exactly.
186
    bool _column_emits_defaults_for_count(ColumnId cid);
187
    // G03: fills CountEmitShortcutFacts from live iterator state at the end of
188
    // _lazy_init and returns the pure-guard verdict; the decision predicate
189
    // itself lives in count_on_index_fastpath.h.
190
    bool _should_engage_count_emit_shortcut(const Block* block);
191
    // G03: one emission-shortcut batch: min(remaining, kCountEmitBatchRows)
192
    // default rows filled straight into the block (NOT-NULL defaults for
193
    // nullable columns, mirroring _prune_column), then EOF once the countdown
194
    // reaches zero. Replaces the whole per-rowid _next_batch_internal body for
195
    // engaged scans.
196
    Status _emit_count_shortcut_batch(Block* block);
197
198
    bool _column_has_fulltext_index(int32_t cid);
199
    bool _column_has_ann_index(int32_t cid);
200
    bool _downgrade_without_index(Status res, bool need_remaining = false);
201
    inline bool _inverted_index_not_support_pred_type(const PredicateType& type);
202
203
    void _init_column_states();
204
    void _rebuild_scan_predicate_states();
205
    void _mark_common_expr_states(const VExprSPtr& expr, bool runtime_generated);
206
    void _mark_common_expr_column(ColumnId ordinal, bool runtime_generated);
207
    Status _vec_init_lazy_materialization();
208
209
13.8k
    uint32_t segment_id() const { return _segment->id(); }
210
15.4k
    uint32_t num_rows() const { return _segment->num_rows(); }
211
212
    [[nodiscard]] Status _read_columns_by_index(const std::vector<ColumnId>& read_ordinals,
213
                                                uint32_t nrows_read_limit, uint16_t& nrows_read);
214
    Status _init_current_block(Block* block, std::vector<MutableColumnPtr>& non_pred_vector,
215
                               uint32_t nrows_read_limit);
216
    uint16_t _evaluate_vectorization_predicate(uint16_t* sel_rowid_idx, uint16_t selected_size);
217
    uint16_t _evaluate_short_circuit_predicate(uint16_t* sel_rowid_idx, uint16_t selected_size);
218
    Status _apply_read_limit_to_selected_rows(Block* block, uint16_t& selected_size);
219
    Status _output_columns_to_block(Block* block);
220
    [[nodiscard]] Status _read_columns_by_rowids(const std::vector<ColumnId>& read_ordinals,
221
                                                 std::vector<rowid_t>& rowid_vector,
222
                                                 uint16_t* sel_rowid_idx, size_t select_size,
223
                                                 MutableColumns* mutable_columns,
224
                                                 bool init_condition_cache = false,
225
                                                 bool read_for_predicate = false);
226
    [[nodiscard]] Status _read_lazy_pruned_columns(Block* block);
227
228
    Status copy_column_data_by_selector(IColumn* input_col_ptr, MutableColumnPtr& output_col,
229
                                        uint16_t* sel_rowid_idx, uint16_t select_size,
230
                                        size_t batch_size);
231
232
    template <class Container>
233
    [[nodiscard]] Status _output_column_by_sel_idx(Block* block, const Container& ordinals,
234
1.68k
                                                   uint16_t* sel_rowid_idx, uint16_t select_size) {
235
1.68k
        SCOPED_RAW_TIMER(&_opts.stats->output_col_ns);
236
1.68k
        for (auto ordinal : ordinals) {
237
1.68k
            if (ordinal >= _schema->num_block_columns()) {
238
510
                continue;
239
510
            }
240
1.17k
            const auto& file_column_type = _storage_name_and_type[ordinal].second;
241
1.17k
            if (!file_column_type->equals(*block->get_by_position(ordinal).type)) {
242
                // Do additional cast
243
0
                MutableColumnPtr tmp = file_column_type->create_column();
244
0
                RETURN_IF_ERROR(copy_column_data_by_selector(_current_columns[ordinal].get(), tmp,
245
0
                                                             sel_rowid_idx, select_size,
246
0
                                                             _opts.block_row_max));
247
0
                RETURN_IF_ERROR(variant_util::cast_column({tmp->get_ptr(), file_column_type, ""},
248
0
                                                          block->get_by_position(ordinal).type,
249
0
                                                          &block->get_by_position(ordinal).column));
250
1.17k
            } else {
251
1.17k
                MutableColumnPtr output_column =
252
1.17k
                        block->get_by_position(ordinal).column->assert_mutable();
253
1.17k
                RETURN_IF_ERROR(copy_column_data_by_selector(_current_columns[ordinal].get(),
254
1.17k
                                                             output_column, sel_rowid_idx,
255
1.17k
                                                             select_size, _opts.block_row_max));
256
1.17k
            }
257
1.17k
        }
258
1.68k
        return Status::OK();
259
1.68k
    }
260
261
    bool _can_evaluated_by_vectorized(std::shared_ptr<ColumnPredicate> predicate);
262
263
    [[nodiscard]] Status _execute_common_expr(uint16_t* sel_rowid_idx, uint16_t& selected_size,
264
                                              Block* block);
265
    Status _process_common_expr(uint16_t* sel_rowid_idx, uint16_t& selected_size, Block* block);
266
267
    uint16_t _evaluate_common_expr_filter(uint16_t* sel_rowid_idx, uint16_t selected_size,
268
                                          const IColumn::Filter& filter);
269
270
    // Dictionary column should do something to initial.
271
    void _convert_dict_code_for_predicate_if_necessary();
272
273
    void _convert_dict_code_for_predicate_if_necessary_impl(const ColumnPredicate& predicate);
274
275
    bool _check_apply_by_inverted_index(std::shared_ptr<ColumnPredicate> pred);
276
277
    void _output_index_result_column(const VExprContextSPtrs& expr_ctxs, uint16_t* sel_rowid_idx,
278
                                     uint16_t select_size);
279
280
    // False for MoR and AGG keys: the merge above this iterator needs the real values.
281
    bool _keys_type_allows_skipping_data() const;
282
283
    bool _need_read_data(ColumnId cid);
284
    bool _prune_column(ColumnId cid, MutableColumnPtr& column, size_t num_of_defaults);
285
286
    Status _construct_compound_expr_context();
287
288
    // Both the key cursor and _seek_block lay out the leading tablet key
289
    // columns densely, so position i addresses the same column in both.
290
45
    int _compare_short_key_with_seek_block(const RowCursor& key, size_t num_key_cols) {
291
57
        for (uint32_t i = 0; i < num_key_cols; ++i) {
292
45
            auto ord = key.field(i) <=> (*_seek_block[i])[0];
293
45
            if (ord != std::strong_ordering::equal) {
294
33
                return ord == std::strong_ordering::less ? -1 : 1;
295
33
            }
296
45
        }
297
12
        return 0;
298
45
    }
299
300
    Status _convert_column_to_expected_type(ColumnId column_id);
301
    Status _convert_to_expected_type(const std::vector<ColumnId>& ordinals);
302
303
    bool _no_need_read_key_data(ColumnId cid, MutableColumnPtr& column, size_t nrows_read);
304
    // Side-effect-free eligibility half of _no_need_read_key_data (no column
305
    // fill); shared by the per-batch fill and the G03 engage-time per-column
306
    // proof so the two can never drift.
307
    bool _no_need_read_key_data_eligible(ColumnId cid);
308
309
    bool _has_delete_pred(ColumnId cid) const;
310
    bool _has_lazy_pruned_children(ColumnId cid) const;
311
    bool _can_skip_reading_extra_column(ColumnId cid);
312
313
    bool _can_opt_limit_reads();
314
315
    void _initialize_predicate_results();
316
    bool _check_all_conditions_passed_inverted_index_for_column(ColumnId cid,
317
                                                                bool default_return = false);
318
319
    void _calculate_common_expr_index_exec_status();
320
321
    Status _process_eof(Block* block);
322
323
    void _fill_column_nothing();
324
325
    Status _process_columns(const std::vector<ColumnId>& ordinals, Block* block);
326
327
    // Initialize virtual columns in the block, set all virtual columns in the block to ColumnNothing
328
    void _init_virtual_columns(Block* block);
329
    // Fallback logic for virtual column materialization, materializing all unmaterialized virtual columns through expressions
330
    Status _materialization_of_virtual_column(Block* block);
331
    void _prepare_score_column_materialization();
332
333
    void _init_row_bitmap_by_condition_cache();
334
335
    void _init_segment_prefetchers();
336
337
    class BitmapRangeIterator;
338
    class BackwardBitmapRangeIterator;
339
340
    // Example:
341
    //   SELECT k, s.b, o FROM t
342
    //   WHERE k > 1 AND abs(k) < 10 AND abs(s.a) < 5;
343
    //   ReadSchema ordinals: [0:k, 1:s STRUCT<a,b>, 2:o]
344
    // When no filter is fully evaluated by an index:
345
    //   state[0:k] = {has_delete_pred=false, has_scan_pred=true,
346
    //                 has_common_expr=true, need_read_data=true}
347
    //   state[1:s] = {has_delete_pred=false, has_scan_pred=false,
348
    //                 has_common_expr=true, need_read_data=true}
349
    //   state[2:o] = {has_delete_pred=false, has_scan_pred=false,
350
    //                 has_common_expr=false, need_read_data=true}
351
    // A storage-only column appended for a delete condition would have
352
    // has_delete_pred=true.
353
    struct ColumnReadState {
354
        bool has_delete_pred = false;
355
        // Mirrors the mutable _col_predicates list: initially all safe scan
356
        // predicates, then only residual predicates after index evaluation.
357
        bool has_scan_pred = false;
358
        bool has_common_expr = false;
359
        // Set when a pushed-down common expression was generated on BE at runtime
360
        // (TopN filter or runtime filter) rather than by the FE planner. FE computes
361
        // predicate access paths only from planner-visible predicates, so such an
362
        // expression may touch nested fields that are not predicate paths. The column
363
        // must then read all of its access paths before filtering instead of splitting
364
        // lazy nested-column recovery.
365
        bool has_runtime_common_expr = false;
366
        // Index evaluation sets this to false when it fully supplies the column result.
367
        // _need_read_data() applies the remaining read constraints.
368
        bool need_read_data = true;
369
370
123k
        bool has_predicate() const { return has_delete_pred || has_scan_pred; }
371
    };
372
373
    std::shared_ptr<Segment> _segment;
374
    // read schema from scanner
375
    ReadSchemaSPtr _schema;
376
    // Inverted-index field name and storage/materialization type for each ReadSchema column.
377
    std::vector<IndexFieldNameAndTypePair> _storage_name_and_type;
378
    // vector idx -> column iterarator
379
    std::vector<std::unique_ptr<ColumnIterator>> _column_iterators;
380
    std::vector<std::unique_ptr<IndexIterator>> _index_iterators;
381
    // after init(), `_row_bitmap` contains all rowid to scan
382
    roaring::Roaring _row_bitmap;
383
    // Cardinality cached after all row bitmap filters are applied in _lazy_init().
384
    uint64_t _row_bitmap_cardinality = 0;
385
    // an iterator for `_row_bitmap` that can be used to extract row range to scan
386
    std::unique_ptr<BitmapRangeIterator> _range_iter;
387
    // the next rowid to read
388
    rowid_t _cur_rowid;
389
    // members related to lazy materialization read
390
    // --------------------------------------------
391
    // remember the rowids we've read for the current row block.
392
    // could be a local variable of next_batch(), kept here to reuse vector memory
393
    std::vector<rowid_t> _block_rowids;
394
    bool _is_need_vec_eval = false;
395
    bool _is_need_short_eval = false;
396
    bool _is_need_expr_eval = false;
397
398
    bool _enable_prune_nested_column = false;
399
400
    // Per-column state indexed by read schema ordinal. Ordered column lists
401
    // below are execution plans rather than additional column membership sets.
402
    std::vector<ColumnReadState> _column_states;
403
    // Columns of the current batch, indexed by read schema ordinal.
404
    MutableColumns _current_columns;
405
    std::vector<std::shared_ptr<ColumnPredicate>> _pre_eval_block_predicate;
406
    std::vector<std::shared_ptr<ColumnPredicate>> _short_cir_eval_predicate;
407
    // Example:
408
    //   SELECT k, s.b, o FROM t
409
    //   WHERE k > 1 AND abs(k) < 10 AND abs(s.a) < 5;
410
    //   ReadSchema ordinals: [0:k, 1:s STRUCT<a,b>, 2:o]
411
    //
412
    // The first three lists assign each active column to its earliest materialization stage:
413
    //   _predicate_ordinals   = [0] // k is used by both k > 1 and abs(k) < 10; predicate wins.
414
    //   _common_expr_ordinals = [1] // s is read for the abs(s.a) < 5 expression.
415
    //   _output_ordinals      = [2] // o is needed only by output.
416
    std::vector<ColumnId> _predicate_ordinals;
417
    std::vector<ColumnId> _common_expr_ordinals;
418
    std::vector<ColumnId> _output_ordinals;
419
    //   _lazy_pruned_ordinals = [1] // After filtering on s.a, read s.b for surviving rows.
420
    // Unlike the first three disjoint lists, this recovery list may contain the same ordinal.
421
    std::vector<ColumnId> _lazy_pruned_ordinals;
422
423
    // the actual init process is delayed to the first call to next_batch()
424
    bool _lazy_inited;
425
    bool _inited;
426
427
    StorageReadOptions _opts;
428
    // Adaptive batch size predictor; null when the feature is disabled.
429
    std::unique_ptr<AdaptiveBlockSizePredictor> _block_size_predictor;
430
    // Build the AdaptiveBlockSizePredictor for this segment based on segment footer
431
    // metadata for the projected output columns. Returns nullptr if the feature is
432
    // disabled or the byte budget is non-positive.
433
    std::unique_ptr<AdaptiveBlockSizePredictor> _make_block_size_predictor() const;
434
    // Snapshot of _opts.block_row_max at init time; used as the hard upper bound so that
435
    // dynamic adjustments never exceed the capacity of pre-allocated buffers.
436
    uint32_t _initial_block_row_max = 0;
437
    // make a copy of `_opts.column_predicates` in order to make local changes
438
    std::vector<std::shared_ptr<ColumnPredicate>> _col_predicates;
439
    VExprContextSPtrs _common_expr_ctxs_push_down;
440
    // row schema of the key to seek
441
    // only used in `_get_row_ranges_by_keys`
442
    std::unique_ptr<ReadSchema> _seek_schema;
443
    // used to binary search the rowid for a given key
444
    // only used in `_get_row_ranges_by_keys`
445
    MutableColumns _seek_block;
446
    // Per-seek-schema-ordinal column iterators for the short-key seek path.
447
    // Points into _column_iterators when the key column is also read, otherwise
448
    // into _owned_seek_column_iterators (a seek key column may not be part of
449
    // the read schema at all).
450
    std::vector<ColumnIterator*> _seek_column_iterators;
451
    std::vector<std::unique_ptr<ColumnIterator>> _owned_seek_column_iterators;
452
453
    io::FileReaderSPtr _file_reader;
454
    std::unique_ptr<SegmentReadAhead> _segment_read_ahead;
455
    std::unique_ptr<ColumnReadAheadContext> _column_read_ahead_context;
456
457
    // Number of selected rows in the current batch.
458
    uint16_t _selected_size;
459
    // Positions of selected rows within the current batch.
460
    std::vector<uint16_t> _sel_rowid_idx;
461
462
    // Rows already produced by this iterator. Used together with
463
    // _opts.read_limit to compute the remaining per-batch budget.
464
    size_t _rows_returned = 0;
465
466
    int64_t _tablet_id = 0;
467
    // Column UIDs requested by the caller. A -1 entry means light schema change is disabled and
468
    // the column has no UID, so the _need_read_data() optimization is disabled.
469
    std::set<int32_t> _output_column_uids;
470
471
    std::vector<uint8_t> _ret_flags;
472
473
    /*
474
    * column and column_predicates on it.
475
    * a boolean value to indicate whether the column has been read by the index.
476
    */
477
    std::unordered_map<ColumnId, std::unordered_map<std::shared_ptr<ColumnPredicate>, bool>>
478
            _column_predicate_index_exec_status;
479
480
    /*
481
    * column and common expr on it.
482
    * a boolean value to indicate whether the column has been read by the index.
483
    */
484
    std::unordered_map<ColumnId, std::unordered_map<const VExpr*, bool>>
485
            _common_expr_index_exec_status;
486
487
    /*
488
    * common expr context to slotref map
489
    * slot ref map is used to get slot ref expr by using column id.
490
    */
491
    std::unordered_map<VExprContext*, std::unordered_map<ColumnId, VExpr*>>
492
            _common_expr_to_slotref_map;
493
494
    ScoreRuntimeSPtr _score_runtime;
495
496
    std::shared_ptr<segment_v2::AnnTopNRuntime> _ann_topn_runtime;
497
498
    // cid to virtual column expr
499
    std::map<ColumnId, VExprContextSPtr> _virtual_column_exprs;
500
501
    IndexQueryContextPtr _index_query_context;
502
503
    // G03 count-emission shortcut state (see count_on_index_fastpath.h).
504
    // _count_fastpath_hit: the reader answered the single MATCH predicate with
505
    // a fabricated count bitmap (captured from the G02 handshake reply).
506
    // _count_emit_shortcut: engaged at the end of _lazy_init when
507
    // count_emit_shortcut_safe holds; every subsequent batch is emitted by
508
    // _emit_count_shortcut_batch from _count_emit_rows_remaining (initialized
509
    // to the post-apply _row_bitmap cardinality) without touching the row
510
    // bitmap iterator.
511
    bool _count_fastpath_hit = false;
512
    bool _count_emit_shortcut = false;
513
    uint64_t _count_emit_rows_remaining = 0;
514
515
    // An indexed conjunct prefix emptied _row_bitmap, proving the WHOLE
516
    // pushed-down conjunction false. Set by the _apply_index_expr short
517
    // circuit when it consumes (clears) the remaining conjuncts, and read
518
    // where an empty conjunct list would otherwise zero the condition-cache
519
    // digest: the all-false result stays valid for the full conjunction, so
520
    // it must remain cacheable.
521
    bool _index_conjuncts_proved_empty = false;
522
    // Batch size for shortcut emission: VStatisticsIterator's
523
    // MAX_ROW_SIZE_IN_COUNT, the largest default-rows block shape already
524
    // proven through every consumer above the segment iterator by the plain
525
    // COUNT pushdown (rowset reader, collect iterator, block reader, scanner).
526
    static constexpr uint64_t kCountEmitBatchRows = 65535;
527
528
    // key is column uid, value is the sparse column cache
529
    std::unordered_map<int32_t, PathToBinaryColumnCacheUPtr> _variant_sparse_column_cache;
530
531
    bool _find_condition_cache = false;
532
    std::shared_ptr<std::vector<bool>> _condition_cache;
533
    static constexpr int CONDITION_CACHE_OFFSET = 2048;
534
};
535
536
} // namespace segment_v2
537
} // namespace doris