Method Move

cpp/src/arrow/flight/sql/odbc/odbc_impl/flight_sql_result_set.cc:64–205

}

size_t FlightSqlResultSet::Move(size_t rows, size_t bind_offset, size_t bind_type,
                                uint16_t* row_status_array) {
  // Consider it might be the first call to Move() and current_chunk is not
  // populated yet
  assert(rows > 0);
  if (current_chunk_.data == nullptr) {
    if (!chunk_buffer_.GetNext(&current_chunk_)) {
      return 0;
    }

    if (transformer_) {
      current_chunk_.data = transformer_->Transform(current_chunk_.data);
    }

    for (size_t column_num = 0; column_num < columns_.size(); ++column_num) {
      columns_[column_num].ResetAccessor(current_chunk_.data->column(column_num));
    }
  }

  // Reset GetData value offsets.
  if (static_cast<size_t>(num_binding_) != get_data_offsets_.size() && reset_get_data_) {
    std::fill(get_data_offsets_.begin(), get_data_offsets_.end(), 0);
  }

  size_t fetched_rows = 0;
  while (fetched_rows < rows) {
    size_t batch_rows = current_chunk_.data->num_rows();
    size_t rows_to_fetch = std::min(static_cast<size_t>(rows - fetched_rows),
                                    static_cast<size_t>(batch_rows - current_row_));

    if (rows_to_fetch == 0) {
      if (!chunk_buffer_.GetNext(&current_chunk_)) {
        break;
      }

      if (transformer_) {
        current_chunk_.data = transformer_->Transform(current_chunk_.data);
      }

      for (size_t column_num = 0; column_num < columns_.size(); ++column_num) {
        columns_[column_num].ResetAccessor(current_chunk_.data->column(column_num));
      }
      current_row_ = 0;
      continue;
    }

    for (auto& column : columns_) {
      // There can be unbound columns.
      if (!column.is_bound) continue;

      auto* accessor = column.GetAccessorForBinding();
      ColumnBinding shifted_binding = column.binding;
      uint16_t* shifted_row_status_array =
          row_status_array ? &row_status_array[fetched_rows] : nullptr;

      if (shifted_row_status_array) {
        std::fill(shifted_row_status_array, &shifted_row_status_array[rows_to_fetch],
                  RowStatus_SUCCESS);