CloseableIterator.java

package org.simpleflatmapper.util;

import java.io.Closeable;
import java.io.IOException;
import java.util.Iterator;

public class CloseableIterator<E> implements Iterator<E>, Closeable {
    private final Iterator<E> delegate;
    private final Closeable resource;

    public CloseableIterator(Iterator<E> delegate, Closeable resource) {
        this.resource = resource;
        this.delegate = delegate;
    }

    @Override
    public void close() throws IOException {
        resource.close();
    }

    @Override
    public boolean hasNext() {
        return delegate.hasNext();
    }

    @Override
    public void remove() {
        delegate.remove();
    }

    @Override
    public E next() {
        return delegate.next();
    }
}