We need to make sure the publisher supports partition handling as well, similar to DistributedStreams

Attached you can find a java file that should work - although it has no testing done for it yet.