kikaha.core.cdi.ImplementedClassesContext Maven / Gradle / Ivy

Go to download
package kikaha.core.cdi;

import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import kikaha.core.cdi.helpers.DefaultFieldQualifierExtractor;
import kikaha.core.cdi.helpers.ProvidableClass;
import kikaha.core.cdi.helpers.QualifierExtractor;
import kikaha.core.cdi.helpers.ServiceLoader;

@SuppressWarnings( { "rawtypes", "unchecked" } )
public class ImplementedClassesContext {

	final Map, Iterable>> implementedClasses = new HashMap<>();
	final Map, ProvidableClass> providableClassCache = new HashMap<>();

	// TODO: Make it settable
	final QualifierExtractor qualifierExtractor = new QualifierExtractor( Arrays.asList( new DefaultFieldQualifierExtractor() ) );

	public  List> loadClassesImplementing( final Class interfaceClazz ) {
		List> implementations = (List)implementedClasses.get( interfaceClazz );
		if ( implementations == null )
			synchronized ( implementedClasses ) {
				implementations = (List)implementedClasses.get( interfaceClazz );
				if ( implementations == null ) {
					implementations = ServiceLoader.loadImplementationsFor( interfaceClazz );
					implementedClasses.put( (Class)interfaceClazz, (Iterable)implementations );
				}
			}
		return implementations;
	}

	public ProvidableClass retrieveProvidableClass( final Class targetClazz ) {
		ProvidableClass providableClass = providableClassCache.get( targetClazz );
		if ( providableClass == null )
			synchronized ( providableClassCache ) {
				providableClass = providableClassCache.get( targetClazz );
				if ( providableClass == null ) {
					providableClass = ProvidableClass.wrap( qualifierExtractor, targetClazz );
					providableClassCache.put( targetClazz, providableClass );
				}
			}
		return providableClass;
	}
}