Add document count limit

[tokee]

Currently we only limits shards on byte size. This has historically produced uneven shards, where the average count is ~300M docs and we have seen one shard with 380M docs.

We should introduce a max document count limit.