}
func (b *tqLoadBalancer) forceReadPartition(partitionCount, partitionID int) *pollToken {
loadbalancer.go
b.lock.Lock()
defer b.lock.Unlock()
b.ensurePartitionCountLocked(max(partitionCount, partitionID+1))
b.pollerCounts[partitionID]++
return &pollToken{
TQPartition: b.taskQueue.NormalPartition(partitionID),
balancer: b,
}
}
// caller to ensure that lock is obtained before call this function