ru.yandex.qatools.allure.io.AbstractResultIterator Maven / Gradle / Ivy

Go to download
package ru.yandex.qatools.allure.io;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.IOException;
import java.nio.file.Path;
import java.util.Iterator;
import java.util.NoSuchElementException;

import static ru.yandex.qatools.allure.AllureUtils.listFiles;

/**
 * @author Dmitry Baev [email protected]
 *         Date: 08.10.15
 */
public abstract class AbstractResultIterator implements Iterator {

    private static final Logger LOGGER = LoggerFactory.getLogger(AbstractResultIterator.class);

    private final Iterator files;

    /**
     * Creates an instance of iterator.
     */
    public AbstractResultIterator(Path... resultDirectories) throws IOException {
        files = listFiles(getFilesGlob(), resultDirectories).iterator();
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public boolean hasNext() {
        return files.hasNext();
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public T next() {
        if (!files.hasNext()) {
            throw new NoSuchElementException();
        }
        Path next = files.next();
        try {
            return readResult(next);
        } catch (IOException e) {
            LOGGER.warn(String.format("Could not read <%s> file", next.toAbsolutePath().toString()), e);
            return next();
        }
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public void remove() {
        throw new UnsupportedOperationException();
    }

    /**
     * Read result from given path.
     *
     * @throws IOException if any occurs.
     */
    protected abstract T readResult(Path path) throws IOException;

    /**
     * Returns the glob for files to read.
     */
    protected abstract String getFilesGlob();
}