Skip to content

Add ability to disable DiskManager - #4330

Merged
alamb merged 1 commit into
apache:masterfrom
tustvold:disk-manager-disable
Nov 23, 2022
Merged

Add ability to disable DiskManager#4330
alamb merged 1 commit into
apache:masterfrom
tustvold:disk-manager-disable

Conversation

@tustvold

@tustvoldtustvold commented Nov 22, 2022

Copy link
Copy Markdown
Contributor

Which issue does this PR close?

re #3941

Rationale for this change

I want to be able to enable a memory limit, but instead of spilling to disk return an error. The simplest way I could see to support this was to add the ability to create a DiskManager that is disabled.

What changes are included in this PR?

Are these changes tested?

Are there any user-facing changes?

@tustvold
tustvold requested a review from alambNovember 22, 2022 17:17
@github-actionsgithub-actionsBot added the core Core DataFusion crate label Nov 22, 2022

@alambalamb left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks great -- thank you @tustvold

I really like that this should allow unified implementations of spilling operators as well as "error if we run out"

cc @crepererum

/// TempDirs to put temporary files in.
///
/// If `Some(vec![])` a new OS specified temporary directory will be created
/// If `None` an error will be returned

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
/// If `None` an error will be returned
/// If `None` an error will be returned (configured not to spill)

let mut guard = self.local_dirs.lock();
let local_dirs = guard.as_mut().ok_or_else(|| {
DataFusionError::ResourcesExhausted(
"Cannot spill to temporary file as DiskManager is disabled".to_string(),

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

If a user sees this it may not be clear what is happening or what they can do about it.

What do you think about changing the message to something like

Suggested change
"Cannot spill to temporary file as DiskManager is disabled".to_string(),
"Memory Exhausted (DiskManager is disabled, cannot spill to temporary file)".to_string(),

As a follow up (which I will do), I think it would be excellent to pass in a context string to create_tmp_file to make the error more specific.

pubfn create_tmp_file(&self,context:&str) -> Result<NamedTempFile>{

@alamb

Copy link
Copy Markdown
Contributor

I will improve the error messages in a follow on PR

@alamb
alamb merged commit df8aa7a into apache:masterNov 23, 2022
@ursabot

Copy link
Copy Markdown

Benchmark runs are scheduled for baseline = 07f65bf and contender = df8aa7a. df8aa7a is a master commit associated with this PR. Results will be available as each benchmark for each run completes.
Conbench compare runs links:
[Skipped ⚠️ Benchmarking of arrow-datafusion-commits is not supported on ec2-t3-xlarge-us-east-2] ec2-t3-xlarge-us-east-2
[Skipped ⚠️ Benchmarking of arrow-datafusion-commits is not supported on test-mac-arm] test-mac-arm
[Skipped ⚠️ Benchmarking of arrow-datafusion-commits is not supported on ursa-i9-9960x] ursa-i9-9960x
[Skipped ⚠️ Benchmarking of arrow-datafusion-commits is not supported on ursa-thinkcentre-m75q] ursa-thinkcentre-m75q
Buildkite builds:
Supported benchmarks:
ec2-t3-xlarge-us-east-2: Supported benchmark langs: Python, R. Runs only benchmarks with cloud = True
test-mac-arm: Supported benchmark langs: C++, Python, R
ursa-i9-9960x: Supported benchmark langs: Python, R, JavaScript
ursa-thinkcentre-m75q: Supported benchmark langs: C++, Java

@alamb

Copy link
Copy Markdown
Contributor

Follow on in #4348

Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Labels

coreCore DataFusion crate

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants

@tustvold@alamb@ursabot@crepererum