presto-main/src/main/java/io/prestosql/operator/index/IndexLookupSourceFactory.java/IndexLookupSourceFactory/setTaskContext
class IndexLookupSourceFactory:    @Override
    public void setTaskContext(TaskContext taskContext)
    {
        this.taskContext = taskContext;
        whenTaskContextSet.set(null);
    }