48 lines
1.5 KiB
Java
48 lines
1.5 KiB
Java
package org.gcube.data.access.storagehub.handlers;
|
|
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
import java.util.Set;
|
|
|
|
import org.gcube.common.storagehub.model.annotations.RootNode;
|
|
import org.gcube.common.storagehub.model.items.Item;
|
|
import org.reflections.Reflections;
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
|
|
public class ClassHandler {
|
|
|
|
private static Logger log = LoggerFactory.getLogger(ClassHandler.class);
|
|
|
|
private Reflections reflection = new Reflections();
|
|
|
|
private Map<String, Class<? extends Item>> classMap = new HashMap<String, Class<? extends Item>>();
|
|
private Map<Class<? extends Item>, String> typeMap = new HashMap<Class<? extends Item>, String>();
|
|
|
|
|
|
public ClassHandler() {
|
|
|
|
Set<Class<?>> classesAnnotated = reflection.getTypesAnnotatedWith(RootNode.class);
|
|
for (Class<?> clazz: classesAnnotated ){
|
|
if (Item.class.isAssignableFrom(clazz))
|
|
for (String value: clazz.getAnnotation(RootNode.class).value()){
|
|
log.debug("loading class {} with value {} ", clazz, value );
|
|
classMap.put(value, (Class<? extends Item>) clazz);
|
|
typeMap.put((Class<? extends Item>) clazz, value);
|
|
}
|
|
}
|
|
}
|
|
|
|
public Class<? extends Item> get(String nodeType){
|
|
if (classMap.containsKey(nodeType)) return classMap.get(nodeType);
|
|
throw new RuntimeException("mapping not found for nodetype "+ nodeType);
|
|
}
|
|
|
|
public String getNodeType(Class<? extends Item> clazz){
|
|
if (typeMap.containsKey(clazz)) return typeMap.get(clazz);
|
|
throw new RuntimeException("mapping not found for nodetype "+ clazz.getSimpleName());
|
|
}
|
|
|
|
|
|
}
|