动态代理工具类

Posted 2020-09-20

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

package com.utils.AOPInteface;

public enum AOPMethod {
   JDKAOP,CGLIBAOP
}

package com.utils.AOPInteface;

public interface CheckCode {
    boolean intercepterBefore();
    boolean intercepterAfter();
}

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);
	}
	
	
	
}

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