/**
 * DataCleaner (community edition)
 * Copyright (C) 2014 Neopost - Customer Information Management
 *
 * This copyrighted material is made available to anyone wishing to use, modify,
 * copy, or redistribute it subject to the terms and conditions of the GNU
 * Lesser General Public License, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
 * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License
 * for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this distribution; if not, write to:
 * Free Software Foundation, Inc.
 * 51 Franklin Street, Fifth Floor
 * Boston, MA  02110-1301  USA
 */
package org.datacleaner.job.tasks;

import java.util.Collection;

import org.datacleaner.api.AnalyzerResult;
import org.datacleaner.api.AnalyzerResultFuture;
import org.datacleaner.api.HasAnalyzerResult;
import org.datacleaner.job.AnalysisJob;
import org.datacleaner.job.ComponentJob;
import org.datacleaner.job.runner.AnalysisListener;
import org.datacleaner.job.runner.JobAndResult;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Task that collects the {@link AnalyzerResult} from an Analyzer.
 */
public final class CollectResultsTask implements Task {

    private static final Logger logger = LoggerFactory.getLogger(CollectResultsTask.class);

    private final HasAnalyzerResult<?> _hasResult;
    private final Collection<JobAndResult> _results;
    private final AnalysisListener _analysisListener;
    private final AnalysisJob _job;
    private final ComponentJob _componentJob;

    public CollectResultsTask(final HasAnalyzerResult<?> hasResult, final AnalysisJob job,
            final ComponentJob componentJob, final Collection<JobAndResult> results,
            final AnalysisListener analysisListener) {
        _hasResult = hasResult;
        _job = job;
        _componentJob = componentJob;
        _results = results;
        _analysisListener = analysisListener;
    }

    @Override
    public void execute() throws Exception {
        logger.debug("execute()");

        final AnalyzerResult result = _hasResult.getResult();
        _analysisListener.componentSuccess(_job, _componentJob, result);
        if (result == null) {
            logger.warn("Result (from {}) was null", _hasResult);
        } else {
            _results.add(new JobAndResult(_componentJob, result));

            if (result instanceof AnalyzerResultFuture) {
                // block the task from finishing
                final AnalyzerResultFuture<?> analyzerResult = (AnalyzerResultFuture<?>) result;
                analyzerResult.get();
            }
        }
    }

}