name.nkonev.r2dbc.migrate.reader.ReflectionsClasspathResourceReader Maven / Gradle / Ivy

Go to download
package name.nkonev.r2dbc.migrate.reader;

import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;

import org.reflections.Reflections;
import org.reflections.scanners.Scanners;
import org.reflections.util.ClasspathHelper;
import org.reflections.util.ConfigurationBuilder;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class ReflectionsClasspathResourceReader implements MigrateResourceReader {

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

    private final String pattern;

    public ReflectionsClasspathResourceReader(String pattern) {
        this.pattern = pattern;
    }

    public ReflectionsClasspathResourceReader() {
        this(".*\\.sql");
    }

    @Override
    public List getResources(String locationPattern) {
        Reflections reflections = new Reflections(
            new ConfigurationBuilder()
                .setScanners(Scanners.Resources)
                .setUrls(ClasspathHelper.forJavaClassPath())
                .filterInputsBy(s -> s.startsWith(locationPattern))
        );
        Set resources = reflections.getResources(pattern);
        resources.forEach(s -> LOGGER.debug("Got resource {}", s));
        return resources.stream().map(ReflectionsClasspathResource::new).collect(Collectors.toList());
    }

}