presto-main/src/main/java/io/prestosql/execution/scheduler/NodeAssignmentStats.java/NodeAssignmentStats/getTotalSplitCount
class NodeAssignmentStats:    public int getTotalSplitCount(InternalNode node)
    {
        return assignmentCount.getOrDefault(node, 0) + splitCountByNode.computeIfAbsent(node, nodeTaskMap::getPartitionedSplitsOnNode);
    }