HLTech / pact-gen

import com.google.common.collect.Lists;

import com.hltech.pact.gen.domain.client.annotation.handlers.AnnotatedMethodHandler;

import io.github.classgraph.ClassGraph;

import java.util.List;

public class MappingHandlerFactory {

    private static final String PACKAGE = "com.hltech.pact.gen";

    public List<AnnotatedMethodHandler> createAll() {

        try {

            return createHandlers(findClasses());

        } catch (IllegalAccessException | InstantiationException exception) {

            throw new MappingMethodCreationException("Cannot create mapping handler", exception);

        }

    }

    private List<Class<?>> findClasses() {

        return new ClassGraph()

            .whitelistPackages(PACKAGE)

            .enableAnnotationInfo()

            .scan()

            .getClassesWithAnnotation(MappingMethodHandler.class.getName())

            .loadClasses();

    }

}