io.katharsis.resource.registry.DefaultResourceLookup Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of katharsis-core Show documentation

Core Katharsis library

There is a newer version: 3.0.2

package io.katharsis.resource.registry;

import io.katharsis.repository.RelationshipRepository;
import io.katharsis.repository.ResourceRepository;
import io.katharsis.repository.annotations.JsonApiRelationshipRepository;
import io.katharsis.repository.annotations.JsonApiResourceRepository;
import io.katharsis.resource.annotations.JsonApiResource;
import org.reflections.Reflections;

import java.util.HashSet;
import java.util.Set;

/**
 * Scans all classes in provided package and finds all resources and repositories associated with found resource.
 */
public class DefaultResourceLookup implements ResourceLookup {

	private String packageName;
	private Reflections reflections;

	public DefaultResourceLookup(String packageName) {
		this.packageName = packageName;
        if (packageName != null) {
            String[] packageNames = packageName.split(",");
            reflections = new Reflections(packageNames);
        } else {
            reflections = new Reflections(packageName);
        }
	}

	@Override
	public Set> getResourceClasses() {
        return reflections.getTypesAnnotatedWith(JsonApiResource.class);
	}
	
	@Override
	public Set> getResourceRepositoryClasses() {
		Set> annotatedResourceRepositories = reflections.getTypesAnnotatedWith(JsonApiResourceRepository.class);
		Set> annotatedRelationshipRepositories = reflections.getTypesAnnotatedWith(JsonApiRelationshipRepository.class);
		Set> resourceRepositories = reflections.getSubTypesOf(ResourceRepository.class);
		Set> relationshipRepositories = reflections.getSubTypesOf(RelationshipRepository.class);
		
		Set> result = new HashSet<>();
		result.addAll(annotatedResourceRepositories);
		result.addAll(annotatedRelationshipRepositories);
		result.addAll(resourceRepositories);
		result.addAll(relationshipRepositories);
		return result;
	}
}