Method SortWindowBuild

bolt/exec/SortWindowBuild.cpp:43–100

namespace bytedance::bolt::exec {

SortWindowBuild::SortWindowBuild(
    const std::shared_ptr<const core::WindowNode>& node,
    bolt::memory::MemoryPool* pool,
    const common::SpillConfig* spillConfig,
    tsan_atomic<bool>* nonReclaimableSection,
    uint64_t spillMemoryThreshold,
    bool enableJit)
    : WindowBuild(
          node,
          pool,
          spillConfig,
          nonReclaimableSection,
          spillMemoryThreshold,
          enableJit),
      pool_(pool),
      comparator_(sortKeyInfo_, data_.get()) {
  BOLT_CHECK_NOT_NULL(pool_);
  allKeyInfo_.reserve(partitionKeyInfo_.size() + sortKeyInfo_.size());
  allKeyInfo_.insert(
      allKeyInfo_.cend(), partitionKeyInfo_.begin(), partitionKeyInfo_.end());
  allKeyInfo_.insert(
      allKeyInfo_.cend(), sortKeyInfo_.begin(), sortKeyInfo_.end());
  partitionStartRows_.resize(0);
  followedTopNum_ = node->followedTopNum();
  if (followedTopNum_ > 0) {
    if (numPartitionKeys_ > 0) {
      Accumulator accumulator{
          true,
          sizeof(TopRows),
          false,
          1,
          nullptr,
          nullptr,
          [](auto, auto) { BOLT_UNREACHABLE(); },
          [](auto, auto) { BOLT_UNREACHABLE(); },
          [](auto) {}};

      table_ = std::make_unique<HashTable<false>>(
          createVectorHashers(node->inputType(), node->partitionKeys()),
          std::vector<Accumulator>{accumulator},
          std::vector<TypePtr>{},
          false, // allowDuplicates
          false, // isJoinBuild
          false, // hasProbedFlag
          0, // minTableSizeForParallelJoinBuild
          pool_,
          nullptr,
          false);
      partitionOffset_ =
          table_->rows()->columnAt(node->partitionKeys().size()).offset();
      lookup_ = std::make_unique<HashLookup>(table_->hashers());
    } else {
      allocator_ = std::make_unique<HashStringAllocator>(pool_);
      singlePartition_ =
          std::make_unique<TopRows>(allocator_.get(), comparator_);
    }
  }
}