org.geotools.data.store.FeatureWriterIterator Maven / Gradle / Ivy

Go to download
/*
 *    GeoTools - The Open Source Java GIS Toolkit
 *    http://geotools.org
 *
 *    (C) 2004-2008, Open Source Geospatial Foundation (OSGeo)
 *
 *    This library is free software; you can redistribute it and/or
 *    modify it under the terms of the GNU Lesser General Public
 *    License as published by the Free Software Foundation;
 *    version 2.1 of the License.
 *
 *    This library 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.
 */
package org.geotools.data.store;

import java.io.IOException;
import java.util.Iterator;
import java.util.NoSuchElementException;
import org.geotools.data.FeatureWriter;
import org.opengis.feature.simple.SimpleFeature;
import org.opengis.feature.simple.SimpleFeatureType;

/**
 * An iterator wrapper for a FeatureWriter - for use with an AbstractFeatureCollection.
 *
 * There is no reason modify this class, subclasses that wish to work with a custom iterator need
 * just that - a custom iterator.
 *
 * 

 *
 * The use of this class against a SimpleFeatureSource not backed by a Transaction may *really*
 * cut into performance. Consider if you will the overhead involved in writing out each feature into
 * a temporary file (when the user may not even modify anything).
 *
 * @author jgarnett
 * @since 2.1.RC0
 */
final class FeatureWriterIterator implements Iterator {
    FeatureWriter writer;

    public FeatureWriterIterator(FeatureWriter writer) {
        this.writer = writer;
    }

    public boolean hasNext() {
        try {
            if (writer == null) {
                return false;
            }
            writer.write(); // write out any "changes" made
            if (writer.hasNext()) {
                return true;
            } else {
                close();
                return false;
                // auto close because we don't trust client
                // code to call close
            }
        } catch (IOException e) {
            close();
            return false; // failure sounds like lack of next to me
        }
    }

    public SimpleFeature next() {
        if (writer == null) {
            throw new NoSuchElementException("Iterator has been closed");
        }
        try {
            return writer.next();
        } catch (IOException io) {
            NoSuchElementException problem =
                    new NoSuchElementException("Could not obtain the next feature:" + io);
            problem.initCause(io);
            throw problem;
        }
    }

    public void remove() {
        try {
            writer.remove();
        } catch (IOException problem) {
            throw (IllegalStateException)
                    new IllegalStateException("Could not remove feature").initCause(problem);
        }
    }
    /**
     * This method only needs package visability as only AbstractFeatureCollection is trusted enough
     * to call it.
     */
    void close() {
        if (writer != null) {
            try {
                writer.close();
            } catch (IOException e) {
                // sorry but iterators die quitely in the night
            }
            writer = null;
        }
    }
}