Function SortStructArray

cpp/src/arrow/compute/kernels/vector_sort.cc:1164–1191

}

Result<NullPartitionResult> SortStructArray(ExecContext* ctx, uint64_t* indices_begin,
                                            uint64_t* indices_end,
                                            const StructArray& array,
                                            SortOrder sort_order,
                                            NullPlacement null_placement) {
  ARROW_ASSIGN_OR_RAISE(auto columns, array.Flatten());
  auto batch = RecordBatch::Make(schema(array.type()->fields()), array.length(),
                                 std::move(columns));

  auto options = SortOptions::Defaults();
  options.null_placement = null_placement;
  options.sort_keys.reserve(array.num_fields());
  for (int i = 0; i < array.num_fields(); ++i) {
    options.sort_keys.push_back(SortKey(FieldRef(i), sort_order));
  }

  ARROW_ASSIGN_OR_RAISE(auto sort_keys,
                        ResolveRecordBatchSortKeys(*batch, options.sort_keys));
  if (sort_keys.size() <= kMaxRadixSortKeys) {
    RadixRecordBatchSorter sorter(indices_begin, indices_end, std::move(sort_keys),
                                  options);
    return sorter.Sort();
  } else {
    MultipleKeyRecordBatchSorter sorter(indices_begin, indices_end, std::move(sort_keys),
                                        options);
    return sorter.Sort();
  }
}

void RegisterVectorSort(FunctionRegistry* registry) {
  DCHECK_OK(registry->AddFunction(std::make_shared<SortIndicesMetaFunction>()));