presto-main/src/main/java/io/prestosql/memory/QueryContext.java/QueryContext/getTaskContextByTaskId
class QueryContext:    public TaskContext getTaskContextByTaskId(TaskId taskId)
    {
        TaskContext taskContext = taskContexts.get(taskId);
        return verifyNotNull(taskContext, "task does not exist");
    }