docs: private boolean receivedLastMessageForPartition(PartitionReplyMsg msg, List<Integer> finishedPartitions) {
  
          switch (msg.getMsgType()) {
  
              case PARTITION_FINAL_STATUS:
                  if (msg.getPartitionPlanConfig() != null) {
                      partitionFinished(msg);
                      finishedPartitions.add(msg.getPartitionPlanConfig().getPartitionNumber());
                      return true;
                  } else {
                      return false;
                  }
              case PARTITION_THREAD_COMPLETE:
                  partitionFinished(msg);
                  finishedPartitions.add(msg.getPartitionPlanConfig().getPartitionNumber());
                  return true;
              default:
                  return false;
          }
      }