io.jboot.aop.javassist.JbootJavassistHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jboot Show documentation
Jboot is a similar SpringCloud project base on JFinal, Dubbo and Undertow.
There is a newer version: 4.1.9
/**
 * Copyright (c) 2015-2022, Michael Yang 杨福海 ([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 io.jboot.aop.javassist;

import com.jfinal.aop.Interceptor;
import com.jfinal.aop.InterceptorManager;
import com.jfinal.aop.Invocation;
import io.jboot.aop.InterceptorBuilderManager;
import io.jboot.aop.InterceptorCache;
import javassist.util.proxy.MethodHandler;

import java.lang.reflect.Method;
import java.util.HashSet;
import java.util.Set;


public class JbootJavassistHandler implements MethodHandler {

    private static final Set excludedMethodName = buildExcludedMethodName();
    private static final InterceptorManager interManager = InterceptorManager.me();
    private static final InterceptorBuilderManager builderManager = InterceptorBuilderManager.me();


    @Override
    public Object invoke(Object self, Method originalMethod, Method proxyMethod, Object[] args) throws Throwable {

        if (excludedMethodName.contains(originalMethod.getName())) {
            return proxyMethod.invoke(self, args);
        }

        Class targetClass = self.getClass().getSuperclass();
        //ClassUtil.getUsefulClass(self.getClass());

        InterceptorCache.MethodKey key = InterceptorCache.getMethodKey(targetClass, originalMethod);
        Interceptor[] inters = InterceptorCache.get(key);
        if (inters == null) {
            inters = interManager.buildServiceMethodInterceptor(targetClass, originalMethod);
            inters = builderManager.build(targetClass, originalMethod, inters);

            InterceptorCache.put(key, inters);
        }

        if (inters.length == 0) {
            return proxyMethod.invoke(self, args);
        } else {
            Invocation invocation = new Invocation(self, originalMethod, inters,
                    x -> proxyMethod.invoke(self, x), args);
            invocation.invoke();
            return invocation.getReturnValue();
        }
    }


    private static 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;
    }


}