IngestModuleTier.java

Go to the documentation of this file.

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package org.sleuthkit.autopsy.ingest;

import java.util.ArrayList;
import java.util.List;
import java.util.Optional;
import java.util.concurrent.LinkedBlockingQueue;
import org.python.google.common.collect.ImmutableList;

class IngestModuleTier {

    private DataSourceIngestPipeline dataSourceIngestPipeline;
    private final LinkedBlockingQueue<FileIngestPipeline> fileIngestPipelinesQueue = new LinkedBlockingQueue<>();
    private final List<FileIngestPipeline> fileIngestPipelines = new ArrayList<>();
    private DataArtifactIngestPipeline dataArtifactIngestPipeline;
    private AnalysisResultIngestPipeline analysisResultIngestPipeline;

    void setDataSourceIngestPipeline(DataSourceIngestPipeline pipeline) {
        dataSourceIngestPipeline = pipeline;
    }    
    
    boolean hasDataSourceIngestModules() {
        return (dataSourceIngestPipeline != null && dataSourceIngestPipeline.isEmpty() == false);
    }

    Optional<DataSourceIngestPipeline> getDataSourceIngestPipeline() {
        return Optional.ofNullable(dataSourceIngestPipeline);
    }    

    void setsFileIngestPipelines(List<FileIngestPipeline> pipelines) throws InterruptedException {
        fileIngestPipelines.addAll(pipelines);
        for (FileIngestPipeline pipeline : pipelines) {
            fileIngestPipelinesQueue.put(pipeline);
        }
    }

    boolean hasFileIngestModules() {
        return (!fileIngestPipelines.isEmpty() && !fileIngestPipelines.get(0).isEmpty());
    }

    List<FileIngestPipeline> getFileIngestPipelines() {
        return ImmutableList.copyOf(fileIngestPipelines);
    }

    FileIngestPipeline takeFileIngestPipeline() throws InterruptedException {
        return fileIngestPipelinesQueue.take();
    }

    void returnFileIngestPipeleine(FileIngestPipeline pipeline) throws InterruptedException {
        fileIngestPipelinesQueue.put(pipeline);
    }

    void setDataArtifactIngestPipeline(DataArtifactIngestPipeline pipeline) {
        dataArtifactIngestPipeline = pipeline;
    }

    boolean hasDataArtifactIngestModules() {
        return (dataArtifactIngestPipeline != null && dataArtifactIngestPipeline.isEmpty() == false);
    }

    Optional<DataArtifactIngestPipeline> getDataArtifactIngestPipeline() {
        return Optional.ofNullable(dataArtifactIngestPipeline);
    }

    void setAnalysisResultIngestPipeline(AnalysisResultIngestPipeline pipeline) {
        analysisResultIngestPipeline = pipeline;
    }

    boolean hasAnalysisResultIngestModules() {
        return (analysisResultIngestPipeline != null && analysisResultIngestPipeline.isEmpty() == false);
    }

    Optional<AnalysisResultIngestPipeline> getAnalysisResultIngestPipeline() {
        return Optional.ofNullable(analysisResultIngestPipeline);
    }

}