[FEATURE] Add metrics to track shuffle data block count and avg block size #2738
Description
Code of Conduct
- I agree to follow this project's Code of Conduct
Search before asking
- I have searched in the issues and found no similar issues.
Describe the feature
Add metrics to track shuffle data block count and avg block size.
Motivation
Currently, we can limit the number of data blocks stored in memory for a single partition, and we can also use the LAB feature to reduce the on-heap memory usage of small blocks. However, the problem of heap memory being filled up has not been completely solved. We need to identify and stop the application that generate large amounts of blocks when heap memory is insufficient.
Describe the solution
No response
Additional context
No response
Are you willing to submit PR?
- Yes I am willing to submit a PR!