balancer: [shard] implement shard draining