Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add ExecutorMetricsCollector interface #2234

Merged
merged 2 commits into from
Apr 15, 2022
Merged

Conversation

thinkharderdev
Copy link
Contributor

Which issue does this PR close?

Closes #2204

Rationale for this change

Currently there is no way of getting execution metrics from Ballista executors (aside from parsing log messages). This adds a trait ExecutorMetricsCollector which can record stage execution metrics. Currently it just defaults to LoggingMetricsCollector which just logs the plan (same as before).

I'm also working on a prometheus metrics collector which we can upstream (under a feature flag) when it's ready.

What changes are included in this PR?

  1. Add ExecutorMetricsCollector trait
  2. Created default LoggingMetricsCollector implementation
  3. Executor needs to own an Arc<dyn ExecutorMetricsCollector> and call record_stage after executing a shuffle write.

Are there any user-facing changes?

No

No

Copy link
Member

@andygrove andygrove left a comment

Choose a reason for hiding this comment

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

This looks like a good improvement to me. Thanks @thinkharderdev

@alamb alamb merged commit 9f2ed42 into apache:master Apr 15, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Improve metrics reporting in Ballista
3 participants