Method TestTmpFileBufferPoolHelper

be/src/runtime/tmp-file-mgr-test.cc:1885–1930

}

void TmpFileMgrTest::TestTmpFileBufferPoolHelper(TmpFileMgr& tmp_file_mgr,
    unique_ptr<TmpFileGroup>* tmp_file_group,
    boost::scoped_ptr<io::WriteRange>& write_range,
    WriteRange::WriteDoneCallback& callback) {
  TUniqueId id;
  vector<string> tmp_create_dirs{{LOCAL_BUFFER_PATH}};
  vector<string> tmp_dirs{{Substitute(LOCAL_BUFFER_PATH + ":$0", 2048)}};
  RemoveAndCreateDirs(tmp_create_dirs);
  tmp_dirs.push_back(remote_url_);
  int64_t alloc_size = 1024;
  int64_t file_size = 1024;
  FLAGS_remote_tmp_file_size = "1KB";
  int64_t offset;
  TmpFile *file1, *file2, *file3;
  cb_counter_ = 0;

  ASSERT_OK(tmp_file_mgr.InitCustom(tmp_dirs, false, "", false, metrics_.get()));
  unique_ptr<TmpFileGroup> unique_file_group =
      make_unique<TmpFileGroup>(&tmp_file_mgr, io_mgr(), profile_, id);
  auto file_group = unique_file_group.get();
  EXPECT_EQ(tmp_file_mgr.GetRemoteTmpFileSize(), file_size);
  IntGauge* dir_usage_local = metrics_->FindMetricForTesting<IntGauge>(
      "tmp-file-mgr.scratch-space-bytes-used.dir-0");
  ASSERT_TRUE(dir_usage_local != nullptr);
  ASSERT_OK(GroupAllocateSpace(file_group, alloc_size, &file1, &offset));
  EXPECT_EQ(file_size, dir_usage_local->GetValue());
  ASSERT_OK(GroupAllocateSpace(file_group, alloc_size, &file2, &offset));
  EXPECT_NE(file1, file2);
  EXPECT_TRUE(file1->GetWriteFile()->IsSpaceReserved());
  EXPECT_TRUE(file2->GetWriteFile()->IsSpaceReserved());
  ASSERT_OK(GroupAllocateSpace(file_group, alloc_size, &file3, &offset));
  EXPECT_NE(file2, file3);

  write_range.reset(new WriteRange(
      file3->path(), offset, file3->AssignDiskQueue(!file3->is_local()), callback));
  const char* dummy_data = "dummy data";
  write_range->SetData((const uint8_t*)dummy_data, strlen(dummy_data));
  write_range->SetDiskFile(file3->GetWriteFile());
  SetWriteRangeContext(write_range.get(), file_group);
  // The space hasn't been reserved for file3 because the pool size is 2K, and there are
  // two files in use, we use the AsyncWriteRange() function to wait until a file is
  // available to be evicted, then finish the writing of file3.
  EXPECT_FALSE(file3->GetWriteFile()->IsSpaceReserved());
  ASSERT_OK(tmp_file_mgr.AsyncWriteRange(write_range.get(), file3));
  *tmp_file_group = move(unique_file_group);
}

void TmpFileMgrTest::TestTmpFileBufferPoolTearDown(TmpFileMgr& tmp_file_mgr) {
  auto tmp_file_pool = tmp_file_mgr.tmp_dirs_remote_ctrl_.tmp_file_pool_.get();