presto-main/src/main/java/io/prestosql/operator/TaskContext.java/TaskContext/getTotalPartitions
class TaskContext:    public OptionalInt getTotalPartitions()
    {
        return totalPartitions;
    }