Add a config to limit the number of rows per shard

This imposes an approximate limit on the number of rows per shard.
This helps ensure that the shards are not too big, we do not care about
each shard having exactly the same number of rows.