artoria.reflect.ReflectAutoConfiguration Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of artoria-extend Show documentation
Artoria is a java technology framework based on the facade pattern.
The newest version!
package artoria.reflect;

import artoria.aop.AbstractInterceptor;
import artoria.aop.ProxyUtils;
import artoria.data.bean.support.BeanToolsAutoConfiguration;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.DisposableBean;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.boot.autoconfigure.AutoConfigureAfter;
import org.springframework.context.annotation.Configuration;
import org.springframework.util.ConcurrentReferenceHashMap;

import java.lang.reflect.Method;
import java.util.*;

import static org.springframework.util.ConcurrentReferenceHashMap.ReferenceType.WEAK;

/**
 * Reflect auto configuration.
 * @author Kahle
 */
@Configuration
@AutoConfigureAfter(BeanToolsAutoConfiguration.class)
public class ReflectAutoConfiguration implements InitializingBean, DisposableBean {
    private static final Logger log = LoggerFactory.getLogger(ReflectAutoConfiguration.class);
    private static final List METHOD_NAMES;

    static {
        // TODO: Do cache optimize by CacheUtils
        List list = new ArrayList();
        Collections.addAll(list, "findConstructors"
                , "findConstructor"
                , "findFields", "findDeclaredFields"
                , "findAccessFields", "findField"
                , "findMethods", "findDeclaredMethods"
                , "findAccessMethods", "findReadMethods"
                , "findWriteMethods", "findMethod"
                , "findSimilarMethod");
        METHOD_NAMES = Collections.unmodifiableList(list);
    }

    @Override
    public void afterPropertiesSet() throws Exception {
        ReflectProvider reflectProvider = ReflectUtils.getReflectProvider();
        if (reflectProvider != null) {
            ReflectProvider instance = ProxyUtils.proxy(new ReflectProviderInterceptor(reflectProvider));
            ReflectUtils.setReflectProvider(instance);
            log.info("Add cache to reflect provider success. ");
        }
    }

    @Override
    public void destroy() throws Exception {
    }

    private static class ReflectProviderInterceptor extends AbstractInterceptor {
        private final Map cache;

        public ReflectProviderInterceptor(ReflectProvider originalObject) {
            super(originalObject);
            this.cache = new ConcurrentReferenceHashMap(64, WEAK);
        }


        @Override
        public Object intercept(Object proxyObject, Method method, Object[] args) throws Throwable {
            if (METHOD_NAMES.contains(method.getName())) {
                String key = method.getName() + Arrays.toString(args);
                Object val = this.cache.get(key);
                if (val == null) {
                    val = method.invoke(getOriginalObject(), args);
                    this.cache.put(key, val);
                }
                return val;
            }
            else {
                return method.invoke(getOriginalObject(), args);
            }
        }

    }

}