pers.clare.hisql.method.SQLMethodInterceptor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of hi-sql Show documentation
Have fun using SQL
The newest version!
package pers.clare.hisql.method;

import org.aopalliance.intercept.MethodInterceptor;
import org.aopalliance.intercept.MethodInvocation;
import pers.clare.hisql.exception.HiSqlException;
import pers.clare.hisql.util.ExceptionUtil;

import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.HashMap;
import java.util.Map;

public class SQLMethodInterceptor implements MethodInterceptor {
    private final Object target;
    private final Map methodMaps = new HashMap<>();
    private final Map methodInterceptorMap;

    public SQLMethodInterceptor(
            Class interfaceClass
            , Object target
            , Map methodInterceptorMap
    ) {
        this.methodInterceptorMap = methodInterceptorMap;
        this.target = target;
        Class targetClass = target.getClass();
        Method targetMethod;
        for (Method method : interfaceClass.getMethods()) {
            try {
                targetMethod = targetClass.getMethod(method.getName(), method.getParameterTypes());
                this.methodMaps.put(method, targetMethod);
            } catch (NoSuchMethodException ignored) {
            }
        }
    }

    @Override
    public Object invoke(MethodInvocation methodInvocation) throws Throwable {
        try {
            Method method = this.methodMaps.get(methodInvocation.getMethod());
            if (method == null) {
                MethodInterceptor handler = methodInterceptorMap.get(methodInvocation.getMethod());
                if (handler == null)
                    throw new HiSqlException("%s not found", methodInvocation.getMethod());
                return handler.invoke(methodInvocation);
            } else {
                try {
                    return method.invoke(target, methodInvocation.getArguments());
                } catch (InvocationTargetException e) {
                    throw e.getTargetException();
                }
            }
        } catch (Throwable e) {
            throw ExceptionUtil.insertBefore(methodInvocation.getMethod(), e);
        }
    }
}