com.jfinal.ext.proxy.JavassistProxyFactory Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jfinal Show documentation

JFinal is a simple, light, rapid,independent, extensible Java WEB + ORM framework. The feature of JFinal looks like ruby on rails especially ActiveRecord.

There is a newer version: 5.2.3

Show newest version

/**
 * Copyright (c) 2011-2023, James Zhan 詹波 ([email protected]).
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.jfinal.ext.proxy;

import java.lang.reflect.Method;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Set;
import com.jfinal.proxy.ProxyFactory;
import javassist.util.proxy.MethodFilter;
import javassist.util.proxy.ProxyObject;

/**
 * JavassistProxyFactory 用于扩展 javassist 的代理模式，默认不使用
 * 
 *  * 配置方法：
 * public void configConstant(Constants me) {
 *     ProxyManager.me().setProxyFactory(new JavassistProxyFactory());
 * }
 * 
 */
public class JavassistProxyFactory extends ProxyFactory {
    
    protected HashMap, Class> cache = new HashMap<>(1024, 0.25F);
    protected JavassistCallback callback = new JavassistCallback();
    protected JavassistMethodFilter methodFilter = new JavassistMethodFilter();
    
    @SuppressWarnings("unchecked")
    public  T get(Class target) {
        // 被 javassist 代理过的类名包含 "_$$_"。不存在代理过两层的情况，仅需调用一次 getSuperclass() 即可
        if (target.getName().indexOf("_$$_") > -1) {
            target = (Class) target.getSuperclass();
        }
        
        try {
            Class clazz = (Class) cache.get(target);
            if (clazz == null) {
                clazz = getProxyClass(target);
            }
            
            T ret = clazz.newInstance();
            ((ProxyObject) ret).setHandler(callback);
            return ret;
            
        } catch (ReflectiveOperationException e) {
            throw new RuntimeException(e);
        }
    }
    
    @SuppressWarnings("unchecked")
    protected  Class getProxyClass(Class target) throws ReflectiveOperationException {
        synchronized(this) {
            return (Class) cache.computeIfAbsent(target, key -> {
                javassist.util.proxy.ProxyFactory factory = new javassist.util.proxy.ProxyFactory();
                factory.setSuperclass(key);
                factory.setFilter(methodFilter);
                return factory.createClass();
            });
        }
    }
    
    /**
     * 过滤不需要代理的方法，参考资料：
     * https://github.com/jboss-javassist/javassist/blob/master/src/test/testproxy/ProxyTester.java
     */
    private static class JavassistMethodFilter implements MethodFilter {
        
        private static final Set excludedMethodName = buildExcludedMethodName();
        
        @Override
        public boolean isHandled(Method method) {
            return ! excludedMethodName.contains(method.getName());
        }
        
        private static final Set buildExcludedMethodName() {
            Set excludedMethodName = new HashSet(64, 0.25F);
            Method[] methods = Object.class.getDeclaredMethods();
            for (Method m : methods) {
                excludedMethodName.add(m.getName());
            }
            // getClass() registerNatives() can not be enhanced
            // excludedMethodName.remove("getClass");   
            // excludedMethodName.remove("registerNatives");
            return excludedMethodName;
        }
    }
}