I had thought that I was using the default class, however in the end I found an override that was using the round-robin partitioner. After dropping that and reverting to the defaults the distribution of messages was pretty even across the partitions. 