[BugFix] Fix get struct ordinal index error (backport #44608) (#44667)

Signed-off-by: Seaven <[email protected]>

be/src/storage/rowset/struct_column_iterator.cpp

@@ -18,6 +18,7 @@
 #include "column/nullable_column.h"
 #include "column/struct_column.h"
 #include "storage/rowset/column_reader.h"
+#include "storage/rowset/common.h"
 #include "storage/rowset/scalar_column_iterator.h"
 
 namespace starrocks {
@@ -39,7 +40,7 @@ class StructColumnIterator final : public ColumnIterator {
 
  Status seek_to_ordinal(ordinal_t ord) override;
 
- ordinal_t get_current_ordinal() const override { return _field_iters[0]->get_current_ordinal(); }
+ ordinal_t get_current_ordinal() const override { return _current_ordinal; }
 
  /// for vectorized engine
  Status get_row_ranges_by_zone_map(const std::vector<const ColumnPredicate*>& predicates,
@@ -55,6 +56,7 @@ class StructColumnIterator final : public ColumnIterator {
  const ColumnAccessPath* _path;
 
  std::vector<uint8_t> _access_flags;
+ ordinal_t _current_ordinal = 0;
 };
 
 StatusOr<std::unique_ptr<ColumnIterator>> create_struct_iter(ColumnReader* _reader,
@@ -117,6 +119,7 @@ Status StructColumnIterator::next_batch(size_t* n, Column* dst) {
  if (_access_flags[i]) {
  auto num_to_read = *n;
  RETURN_IF_ERROR(_field_iters[i]->next_batch(&num_to_read, fields[i].get()));
+ _current_ordinal = _field_iters[i]->get_current_ordinal();
  }
  }
 
@@ -153,6 +156,7 @@ Status StructColumnIterator::next_batch(const SparseRange& range, Column* dst) {
  if (_access_flags[i]) {
  RETURN_IF_ERROR(_field_iters[i]->next_batch(range, fields[i].get()));
  row_count = fields[i]->size();
+ _current_ordinal = _field_iters[i]->get_current_ordinal();
  }
  }
 
@@ -183,6 +187,7 @@ Status StructColumnIterator::fetch_values_by_rowid(const rowid_t* rowids, size_t
  for (int i = 0; i < _field_iters.size(); ++i) {
  if (_access_flags[i]) {
  RETURN_IF_ERROR(_field_iters[i]->fetch_values_by_rowid(rowids, size, fields[i].get()));
+ _current_ordinal = _field_iters[i]->get_current_ordinal();
  }
  }
 
@@ -201,6 +206,7 @@ Status StructColumnIterator::seek_to_first() {
  for (auto& iter : _field_iters) {
  RETURN_IF_ERROR(iter->seek_to_first());
  }
+ _current_ordinal = _field_iters[0]->get_current_ordinal();
  return Status::OK();
 }
 
@@ -211,6 +217,7 @@ Status StructColumnIterator::seek_to_ordinal(ordinal_t ord) {
  for (auto& iter : _field_iters) {
  RETURN_IF_ERROR(iter->seek_to_ordinal(ord));
  }
+ _current_ordinal = _field_iters[0]->get_current_ordinal();
  return Status::OK();
 }
 
@@ -219,5 +226,4 @@ Status StructColumnIterator::get_row_ranges_by_zone_map(const std::vector<const
  row_ranges->add({0, static_cast<rowid_t>(_reader->num_rows())});
  return Status::OK();
 }
-
 } // namespace starrocks