code: public void claimWork() throws InterruptedException {
          if (state.get() == NodeState.Started && !waitInProgress.get() && connected.get()) {
              balancingPolicy.claimWork();
          }
      }
docs: Triggers a work-claiming cycle. If smart balancing is enabled, claim work based
  on node and cluster load. If simple balancing is in effect, claim by count.