Class FetchTask

server/src/main/java/io/crate/execution/engine/fetch/FetchTask.java:66–347

import io.crate.metadata.doc.DocTableInfo;

public class FetchTask implements Task {

    private final IntObjectHashMap<RefCountedItem<? extends IndexSearcher>> searchers = new IntObjectHashMap<>();
    private final IntObjectHashMap<SharedShardContext> shardContexts = new IntObjectHashMap<>();
    private final FetchPhase phase;
    private final int memoryLimitInBytes;
    private final String localNodeId;
    private final SharedShardContexts sharedShardContexts;
    private final TreeMap<Integer, PartitionName> tableIdents = new TreeMap<>();
    private final Metadata metadata;
    private final List<? extends Routing> routings;
    private final Map<RelationName, Collection<Reference>> toFetch;
    private final UUID jobId;
    private final Function<RelationName, DocTableInfo> getTableInfo;
    private final CompletableFuture<Void> result = new CompletableFuture<>();


    @GuardedBy("jobId")
    private int borrowed = 0;

    @GuardedBy("jobId")
    private Throwable killed;

    public FetchTask(UUID jobId,
                     FetchPhase phase,
                     int memoryLimitInBytes,
                     String localNodeId,
                     SharedShardContexts sharedShardContexts,
                     Metadata metadata,
                     Function<RelationName, DocTableInfo> getTableInfo,
                     List<? extends Routing> routings) {
        this.jobId = jobId;
        this.phase = phase;
        this.memoryLimitInBytes = memoryLimitInBytes;
        this.localNodeId = localNodeId;
        this.sharedShardContexts = sharedShardContexts;
        this.metadata = metadata;
        this.routings = routings;
        this.toFetch = HashMap.newHashMap(phase.tableIndices().size());
        this.getTableInfo = getTableInfo;
        this.result.whenComplete((_, _) -> {
            synchronized (jobId) {
                for (var cursor : searchers) {
                    cursor.value.close();
                }
                searchers.clear();
            }
        });
    }

    /**
     * From the memory.operation_limit session setting.
     */
    public int memoryLimitInBytes() {
        return memoryLimitInBytes;
    }

    public Map<RelationName, Collection<Reference>> toFetch() {