Extract Reusable BatchPartitioner

[tustvold]

Is your feature request related to a problem or challenge? Please describe what you are trying to do.

See comment by @Dandandan #2226 (comment)

Describe the solution you'd like

Extract a [BatchPartitioner] that can factor out this logic for reuse

Describe alternatives you've considered

We could not do this