com.automationrockstars.gir.data.impl.TestDataServiceRegistry Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of gir-data Show documentation
The newest version!
/*
 * 
 */

package com.automationrockstars.gir.data.impl;

import com.automationrockstars.base.ConfigLoader;
import com.automationrockstars.gir.data.TestDataService;
import com.automationrockstars.gir.data.TestDataServices;
import com.google.common.base.Function;
import com.google.common.base.Predicate;
import com.google.common.collect.FluentIterable;
import com.google.common.collect.Lists;
import org.reflections.Reflections;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.Arrays;
import java.util.List;
import java.util.concurrent.atomic.AtomicInteger;

public class TestDataServiceRegistry {

    private static final List> classRegistry = Lists.newArrayList();
    private static final Logger LOG = LoggerFactory.getLogger(TestDataServiceRegistry.class);
    private static String[] packagesToScan = null;

    static {
        loadFromClasspath();
    }

    public static synchronized void register(Class serviceClass) {
        if (!classRegistry.contains(serviceClass)) {
            classRegistry.add(serviceClass);
        }
    }

    public static List services() {
        List result = Lists.newArrayList();
        for (Class serviceClass : classRegistry) {
            try {
                result.add(serviceClass.newInstance());
            } catch (IllegalAccessException | InstantiationException e) {
                LOG.error("Registering test data service {} failed", serviceClass, e);
            }
        }
        return result;
    }

    public static String[] packagesToScan() {
        if (packagesToScan == null) {
            String[] result = ConfigLoader.config().getStringArray("data.packages");

            if (result == null || result.length == 0) {
                result = stackTracePackages();
            }
            packagesToScan = result;
            LOG.info("Using {} to scan for data record classes", Arrays.toString(packagesToScan));

        }
        return packagesToScan;
    }

    public static String[] stackTracePackages() {
        String[] result = null;
        final AtomicInteger lastValidClass = new AtomicInteger(0);
        FluentIterable classes = FluentIterable.from(Lists.reverse(Lists.newArrayList(Thread.currentThread().getStackTrace()))).transform(new Function() {

            @Override
            public String apply(StackTraceElement input) {
                return input.getClassName();
            }
        });
        classes.firstMatch(new Predicate() {

            @Override
            public boolean apply(String input) {
                lastValidClass.incrementAndGet();
                return input.startsWith(TestDataServices.class.getName());
            }
        });
        result = classes.limit(lastValidClass.get()).transform(new Function() {

            @Override
            public String apply(String input) {
                return String.format("%s.%s", input.split("\\.")[0], input.split("\\.")[1]);
            }
        }).filter(new Predicate() {

            @Override
            public boolean apply(String input) {
                return !input.startsWith("java.") && !input.startsWith("sun.");
            }
        }).toSet().toArray(new String[0]);

        return result;
    }

    private static void loadFromClasspath() {
        for (Class serviceClass : new Reflections((Object[]) stackTracePackages()).getSubTypesOf(TestDataService.class)) {
            register(serviceClass);
        }
        ;
    }
}