HLTech / pact-gen

public class JaxRsClientsFinder {

    public Set<Class<?>> findJaxRsClients(String packageRoot) {

        Set<Class<?>> jaxRsClients = new HashSet<>();

        jaxRsClients.addAll(classAnnotatedClients(packageRoot));

        jaxRsClients.addAll(methodAnnotatedClients(packageRoot));

        return jaxRsClients;

    }

    private Set<Class<?>> classAnnotatedClients(String packageRoot) {

        return new Reflections(packageRoot)

            .getTypesAnnotatedWith(Path.class);

    }

    private Set<Class<?>> methodAnnotatedClients(String packageRoot) {

        return new Reflections(packageRoot, new MethodAnnotationsScanner())

            .getMethodsAnnotatedWith(Path.class).stream()

            .map(Method::getDeclaringClass)

            .collect(Collectors.toSet());