AbstractStreamParser.java

/* AWE - Amanzi Wireless Explorer
 * http://awe.amanzi.org
 * (C) 2008-2009, AmanziTel AB
 *
 * This library is provided under the terms of the Eclipse Public License
 * as described at http://www.eclipse.org/legal/epl-v10.html. Any use,
 * reproduction or distribution of the library constitutes recipient's
 * acceptance of this agreement.
 *
 * This library is distributed WITHOUT ANY WARRANTY; without even the
 * implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 */

package org.amanzi.neo.loader.core.parser.impl.internal;

import java.io.File;
import java.io.FileInputStream;
import java.io.InputStream;
import java.io.InputStreamReader;

import org.amanzi.neo.loader.core.IData;
import org.amanzi.neo.loader.core.ISingleFileConfiguration;
import org.amanzi.neo.loader.core.exception.impl.FileNotFoundException;
import org.apache.commons.io.IOUtils;
import org.apache.commons.io.input.CountingInputStream;
import org.eclipse.core.runtime.IProgressMonitor;

/**
 * TODO Purpose of
 * <p>
 * </p>
 * 
 * @author Nikolay Lagutko (nikolay.lagutko@amanzitel.com)
 * @since 1.0.0
 */
public abstract class AbstractStreamParser<C extends ISingleFileConfiguration, D extends IData> extends AbstractParser<C, D> {

    private CountingInputStream stream;

    private InputStreamReader reader;

    protected CountingInputStream getStream() {
        if (stream == null) {
            stream = initializeStream(getConfiguration());
        }
        return stream;
    }

    protected InputStreamReader getReader() {
        if (reader == null) {
            reader = initializeReader(getStream());
        }
        return reader;
    }

    protected CountingInputStream initializeStream(final C configuration) {
        try {
            return new CountingInputStream(new FileInputStream(configuration.getFile()));
        } catch (java.io.FileNotFoundException e) {
            throw new FileNotFoundException(configuration.getFile(), e);
        }
    }

    protected InputStreamReader initializeReader(final InputStream stream) {
        return new InputStreamReader(stream);
    }

    @Override
    public D next() {
        D result = super.next();

        work(stream.getCount());

        return result;
    }

    @Override
    public void setProgressMonitor(final String monitorName, final IProgressMonitor monitor) {
        monitor.beginTask(monitorName, (int)getFileFromConfiguration(getConfiguration()).length());

        super.setProgressMonitor(monitorName, monitor);
    }

    @Override
    public void finishUp() {
        IOUtils.closeQuietly(reader);
        IOUtils.closeQuietly(stream);
    }

    @Override
    protected File getFileFromConfiguration(final ISingleFileConfiguration configuration) {
        return configuration.getFile();
    }
}