source: java/main/src/main/java/com/framsticks/params/Registry.java @ 86

package com.framsticks.params;

import org.apache.log4j.Logger;

import com.framsticks.params.annotations.FramsClassAnnotation;

import java.util.HashMap;
import java.util.Map;

/**
 * Author: Piotr Śniegowski
 */
public class Registry {
        private static final Logger log = Logger.getLogger(Registry.class.getName());

        protected final Map<String, Class<?>> reflectedClasses = new HashMap<String, Class<?>>();
        protected final Map<String, String> invertedReflectedClasses = new HashMap<String, String>();
        protected final Map<String, FramsClass> infoCache = new HashMap<String, FramsClass>();

        public final void registerReflectedClass(String name, String id, String className) {
                try {
                        registerReflectedClass(name, id, Class.forName(className));
                } catch (ClassNotFoundException e) {
                        log.fatal("class not found during registration: " + e);
                }
        }

        public void registerReflectedClass(String name, String id,
                        Class<?> reflectedClass) {
                if (name != null) {
                        reflectedClasses.put(name, reflectedClass);
                        invertedReflectedClasses.put(reflectedClass.getCanonicalName(), name);
                }
                if (id != null) {
                        reflectedClasses.put(id, reflectedClass);
                }
        }

        public Registry register(Class<?> reflectedClass) {
                FramsClassAnnotation a = reflectedClass.getAnnotation(FramsClassAnnotation.class);
                if (a == null) {
                        log.error("class is not annotated: " + reflectedClass);
                        return this;
                }

                registerReflectedClass(FramsClassBuilder.getName(a, reflectedClass), FramsClassBuilder.getId(a, reflectedClass), reflectedClass);

                return this;
        }

        public AccessInterface createAccess(String name, FramsClass framsClass) throws ConstructionException {
                if (reflectedClasses.containsKey(name)) {
                        return new ReflectionAccess(reflectedClasses.get(name), framsClass);
                }
                return new PropertiesAccess(framsClass);
        }

        public void putInfoIntoCache(FramsClass framsClass) {
                if (infoCache.containsKey(framsClass.getId())) {
                        log.info("already cached " + framsClass);
                        return;
                }
                log.debug("caching info for " + framsClass);
                infoCache.put(framsClass.getId(), framsClass);
                infoCache.put(framsClass.getName(), framsClass);
        }

        public FramsClass getInfoFromCache(String id) {
                if (id == null) {
                        return null;
                }
                if (infoCache.containsKey(id)) {
                        return infoCache.get(id);
                }
                if (invertedReflectedClasses.containsKey(id)) {
                        return getInfoFromCache(invertedReflectedClasses.get(id));
                }
                return null;
        }

        public static AccessInterface wrapAccessWithListIfNeeded(CompositeParam param, AccessInterface access) {
                if (access == null) {
                        return null;
                }
                return param.prepareAccessInterface(access);
        }

        public AccessInterface prepareAccess(CompositeParam param) throws ConstructionException {
                return wrapAccessWithListIfNeeded(param, createAccess(param.getContainedTypeName()));
        }

        public AccessInterface createAccess(String name) throws ConstructionException {
                if (name == null) {
                        return null;
                }
                FramsClass framsClass = getInfoFromCache(name);
                if (framsClass == null) {
                        return null;
                }

                return createAccess(name, framsClass);
        }

}