动态代理工具类

2022-11-30 17:02:36 浏览数 (3)

我刚刚想着每次写动态代理的时候都会写很多的代码,有点不值得,所以我写了一个实现动态代理的工具类,用户可以用JDK动态代理也可以使用CGLIB动态代理,我的ProxyUtils中提供了三种开发动态代理的方式,在实际用的过程中,可以继承Intercepter这个抽象类实现在逻辑代码前后添加控制代码,如果控制代码返回true那么证明可以通过,如果控制代码返回false说明验证不通过,如果不通过那么就返回你逻辑代码中返回的“0”值,如果你逻辑代码返回对象,那么会返回null,如果是其他类型则返回各种类型的零值,用的时候可以实现Intercepter类,也可以用匿名内部类,随你喜欢,下面是我的代码,一段代码代表一个文件,过后我会将代码再好好,封装然后打成一个jar包:

代码语言:javascript复制
package com.utils.AOPInteface;

public enum AOPMethod {
   JDKAOP,CGLIBAOP
}
代码语言:javascript复制
package com.utils.AOPInteface;

public interface CheckCode {
    boolean intercepterBefore();
    boolean intercepterAfter();
}
代码语言:javascript复制
package com.utils.AOPInteface;

import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;

import net.sf.cglib.proxy.Enhancer;
import net.sf.cglib.proxy.MethodInterceptor;
import net.sf.cglib.proxy.MethodProxy;

abstract class Intercepter implements CheckCode,InvocationHandler,MethodInterceptor{
    private Object obj; 
	public Intercepter(Object obj){
		this.obj=obj;
	}
	final public Object getTarget(){
		return obj;
	}
	@Override
	final public Object intercept(Object proxy, Method method, Object[] args,
			MethodProxy methodProxy) throws Throwable {
		Object o=null;
		boolean bef=intercepterBefore();
		if(bef){
			o=method.invoke(obj, args);
		}else{
			return null;
		}
		boolean aft=intercepterAfter();
		if(aft){
			return o;
		}else{
			return null;
		}
		
	}
	
	@Override
	final public Object invoke(Object proxy, Method method, Object[] args)
			throws Throwable {
		return intercept(proxy, method, args, null);
	}
	
	
	
}
代码语言:javascript复制
package com.utils.AOPInteface;

import java.lang.reflect.Proxy;

import net.sf.cglib.proxy.Enhancer;

public class ProxyUtils {
	public static Object getJDKProxy(Intercepter intercepter){
		if(intercepter.getTarget().getClass().getInterfaces().length==0){
			return getCGLIBProxy(intercepter);
		}
		return Proxy.newProxyInstance(intercepter.getTarget().getClass().getClassLoader(), intercepter.getTarget().getClass().getInterfaces(), intercepter);
	}
	
	public static Object getCGLIBProxy(Intercepter intercepter){
		Enhancer enhancer=new Enhancer();
		enhancer.setSuperclass(intercepter.getTarget().getClass());
		enhancer.setCallback(intercepter);
		return enhancer.create();
	}
    public static Object getProxy(AOPMethod amethod,Intercepter intercepter){
    	if(amethod==AOPMethod.JDKAOP){
    		return getJDKProxy(intercepter);
    	}else{
    		return getCGLIBProxy(intercepter);
    	}
    }
    static class Print{
    	public int print(){
    	System.out.println("Hello AOP!!!");
    	return 20;
    	}
    }
   
}

0 人点赞