Method SpillWriteFile

bolt/exec/SpillFile.cpp:117–153

}

SpillWriteFile::SpillWriteFile(
    uint32_t id,
    const std::string& pathPrefix,
    const std::string& fileCreateConfig,
    const bool spillUringEnabled)
    : id_(id),
      path_(fmt::format("{}-{}", pathPrefix, ordinalCounter_++)),
      spillUringEnabled_(spillUringEnabled) {
  if (spillUringEnabled_) {
    // TODO: it is better to let memory pool to set the ringSize @Zhongjun
    // A smaller number should be set (e.g. 8) when row based spill is enabled
    const size_t ringSize =
        64; // should be same with ringDepth in LocalWriteFile
    writeBuffers_ = std::make_shared<WriteBuffers>(ringSize);
  }
  auto fs = filesystems::getFileSystem(path_, nullptr);
#ifdef IO_URING_SUPPORTED
  if (spillUringEnabled_) {
    file_ = fs->openAsyncFileForWrite(
        path_,
        filesystems::FileOptions{
            {{filesystems::FileOptions::kFileCreateConfig.toString(),
              fileCreateConfig}},
            0,
            nullptr});
    return;
  }
#endif

  file_ = fs->openFileForWrite(
      path_,
      filesystems::FileOptions{
          {{filesystems::FileOptions::kFileCreateConfig.toString(),
            fileCreateConfig}},
          0,
          nullptr});
}

void SpillWriteFile::finish() {
  BOLT_CHECK_NOT_NULL(file_);