TsvCallable.java

/**
 *  Copyright 2012 Universitat Pompeu Fabra.
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 *
 *
 */
package org.onexus.collection.loader.tsv.internal;

import org.onexus.collection.api.Collection;
import org.onexus.collection.api.IEntitySet;
import org.onexus.collection.api.utils.EmptyEntitySet;
import org.onexus.data.api.IDataManager;
import org.onexus.data.api.IDataStreams;
import org.onexus.data.api.utils.EmptyDataStreams;
import org.onexus.resource.api.ORI;
import org.onexus.resource.api.Progress;
import org.onexus.resource.api.exceptions.ResourceNotFoundException;

import java.util.concurrent.Callable;

public class TsvCallable implements Callable<IEntitySet> {

    private static String PARAMETER_DATA_URI = "data";

    private IDataManager dataManager;
    private Progress progress;
    private Collection collection;

    public TsvCallable(Progress progress, IDataManager dataManager, Collection collection) {
        super();

        this.dataManager = dataManager;
        this.progress = progress;
        this.collection = collection;
    }

    @Override
    public IEntitySet call() throws Exception {

        String dataUri = collection.getLoader().getParameter(PARAMETER_DATA_URI);

        if (dataUri == null) {
            String errMsg = "Required parameter '" + PARAMETER_DATA_URI + "' not found in '" + collection.getORI() + "'.";
            progress.error(errMsg);
            progress.fail();
            return new EmptyEntitySet(collection);
        }

        ORI absDataUri = new ORI(dataUri).toAbsolute(collection.getORI());

        IDataStreams dataStreams;
        try {

            dataStreams = dataManager.load(absDataUri);

            if (dataStreams.getProgress() != null) {
                Progress subProgress = dataStreams.getProgress();
                progress.addSubTask(subProgress);

                // Wait sub task finish
                while (!subProgress.isDone()) {
                    try {
                        Thread.sleep(1000);
                    } catch (InterruptedException e) {
                    }
                }

                if (subProgress.isAborted()) {
                    progress.setStatus(subProgress.getStatus());
                } else {
                    dataStreams = dataManager.load(absDataUri);
                }
            }

        } catch (ResourceNotFoundException e) {
            dataStreams = new EmptyDataStreams(progress);
            progress.warning(e.getMessage());
        }

        return new TsvEntitySet(dataStreams, collection);

    }

}