Function TEST_F

bolt/exec/tests/SpillableWindowTest.cpp:234–305

};

TEST_F(SpillableWindowTest, spillAgg) {
  const vector_size_t size = 4096;
  auto data = makeRowVector(
      {"c1", "c2"},
      {
          // aggregate column.
          makeFlatVector<int64_t>(size, [](auto row) { return row; }),
          // partition key.
          makeFlatVector<int32_t>(
              size, [](auto row) { return row < 2048 ? 1 : 2; }),

      });

  createDuckDbTable({data});
  auto spillDirectory = TempDirectoryPath::create();

  core::PlanNodeId windowId;
  {
    auto plan =
        PlanBuilder()
            .values(split(data, 10))
            .streamingWindow(
                {"sum(c1) over (partition by c2 ROWS BETWEEN UNBOUNDED PRECEDING and UNBOUNDED FOLLOWING)"})
            .capturePlanNodeId(windowId)
            .planNode();

    auto task =
        AssertQueryBuilder(plan, duckDbQueryRunner_)
            .config(core::QueryConfig::kPreferredOutputBatchBytes, "1024")
            .config(core::QueryConfig::kMaxOutputBatchRows, "1024")
            .config(core::QueryConfig::kSpillEnabled, "true")
            .config(core::QueryConfig::kWindowSpillEnabled, "true")
            .config(core::QueryConfig::kTestingSpillPct, "100")
            .spillDirectory(spillDirectory->path)
            .assertResults(
                "SELECT *, sum(c1) over (partition by c2 ROWS BETWEEN UNBOUNDED PRECEDING and UNBOUNDED FOLLOWING) FROM tmp ");

    auto taskStats = exec::toPlanStats(task->taskStats());
  }

  TestScopedSpillInjection scopedSpillInjection(100);
  bytedance::bolt::exec::TestWindowInjection windowInjection(
      WindowBuildType::kSpillableWindowBuild);

  auto plan =
      PlanBuilder()
          .values(split(data, 10))
          .window(
              {"sum(c1) over (partition by c2 ROWS BETWEEN UNBOUNDED PRECEDING and UNBOUNDED FOLLOWING)"})
          .capturePlanNodeId(windowId)
          .planNode();

  auto task =
      AssertQueryBuilder(plan, duckDbQueryRunner_)
          .config(core::QueryConfig::kMaxOutputBatchRows, "100")
          .config(core::QueryConfig::kSpillEnabled, "true")
          .config(core::QueryConfig::kWindowSpillEnabled, "true")
          .config(core::QueryConfig::kRowBasedSpillMode, "compression")