模仿JDK实现动态代理

第二次改进这个东西，第一次写的发现不支持带有参数和返回类型的方法的类。额，Java代码拼接简直搞死人。

代理模式是一个非常常用的设计模式，而动态代理更是魅力无穷。额，不熟悉代理模式的请自行了解，一直想写一篇总结设计模式的博文，但一直拖沓着。。。日后再说。

这是类目录：

Moveable是一个接口，Car实现这个接口。

InvocationHandle是一个接口，TimeHandler实现这个接口。

与JDK类似的实现的好处是可以将增强方法单独写在一个类里，增强了灵活性，降低了耦合度。但有一个缺点是被代理的类必须提供一个接口。

基本原理其实就是在InvocationHandle接口实现类里实现你的”加强方法“，即代理时要增加的方法，该样例是日志功能。然后利用Java反射在Proxy类里生成Java代码字符串，生成文件并编译载入。获得新载入的类的构造方法生成一个实例，即代理类。最后删除生成的多余文件。

上代码：

Moveable接口

package proxy.betterbyself;

public interface Moveable {
	void move();
	//加入一个带参数和返回类型的方法
	String fun(String s, int a);
}

Car类

package proxy.betterbyself;

import java.util.Random;

public class Car implements Moveable {

	@Override
	public void move() {
		//实现开车
		try {
			Thread.sleep(new Random().nextInt(1000));
			System.out.println("汽车行驶中....");
		} catch (InterruptedException e) {
			e.printStackTrace();
		}
	}

	@Override
	public String fun(String s, int a) {
		// TODO Auto-generated method stub
		System.out.println("运行fun中");
		return s+a;
	}

}

InvocationHandler接口

package proxy.betterbyself;

import java.lang.reflect.Method;

public interface InvocationHandler {

//	public void invoke(Object o,Method m);
	public Object invoke(Object o,Method m, Object... obj);

}

TimeHandler类

package proxy.betterbyself;

import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;

/**
 * 在这个类实现加强的方法
 */
public class TimeHandler implements InvocationHandler {

	private Object target;
	
	public TimeHandler(Object target) {
		super();
		this.target = target;
	}

	@Override
	public Object invoke(Object o, Method m, Object... obj) {
		Object object=null;
		try {
			long starttime = System.currentTimeMillis();
			System.out.println("汽车开始行驶....");
			if("void".equals(m.getReturnType()))
				m.invoke(target, obj);
			else {
				object=m.invoke(target, obj);
			}
			long endtime = System.currentTimeMillis();
			System.out.println("汽车结束行驶....  汽车行驶时间："
							+ (endtime - starttime) + "毫秒！");
		} catch (Exception e) {
			e.printStackTrace();
		}
		return object;
	}

}

Proxy类

package proxy.betterbyself;

import java.io.File;
import java.lang.reflect.Constructor;
import java.lang.reflect.Method;

import javax.tools.JavaCompiler;
import javax.tools.StandardJavaFileManager;
import javax.tools.ToolProvider;
import javax.tools.JavaCompiler.CompilationTask;

import org.apache.commons.io.FileUtils;

public class Proxy {
	
	@SuppressWarnings("unchecked")
	public static Object newProxyInstance(Class infce,InvocationHandler h) throws Exception{
		String rt = "\r\n";
		String methodStr = "";
		for(Method m : infce.getMethods()){
			Class[] pras=m.getParameterTypes();
			String pra1="";
			String pra2="";
			String argString=" arg";
			String args="";
			int II=0;
			//构造方法参数
			for(Class pra : pras) {
				pra1+=","+pra.getName() + argString+II;
				pra2+=","+pra.getName()+".class";
				args+=","+argString +(II++);
			}
			if(pra1.length()>=1) {
				pra1=pra1.substring(1);
			}
			else {
				pra1=pra2="";
			}
			//获取并构造返回值
			String returnType=(m.getReturnType()).toString();
			
			if(returnType.length()>5 && "class".equals(returnType.substring(0, 5))) {
				returnType=returnType.substring(6);
			}
			//拼接Java代码
			methodStr += "	@Override" + rt +
			"	public "+ returnType +' '+ m.getName() + "("+pra1+") {" + rt +
			"Object obj=null;"+rt+
			"  try{" + rt +
			"  Method md = " + infce.getName() + ".class.getMethod(\"" 
										+ m.getName() + "\"" + pra2+");" + rt +
			"obj =  h.invoke(this,md"+ args +");" +rt+  //在外部定义动作 解耦
			"  }catch(Exception e){ e.printStackTrace();}" + rt;
			
			//返回值是否为void的分类讨论
			if("void".equals(returnType)) {
				methodStr+="return;"+rt+"}"+rt;
			} else {
				methodStr += "return ("+returnType+")" +"obj;"+rt+"}"+rt;
			}
			
		}
		
		String str =
		"package proxy.betterbyself;" + rt +
		"import java.lang.reflect.Method;" + rt +
		"import proxy.betterbyself.InvocationHandler;" +  rt+
		"public class $Proxy0 implements " + infce.getName() + " {" + rt +
		"	public $Proxy0(InvocationHandler h) {" + rt +
		"		this.h = h;" + rt +
		"	}" + rt +
		"  private InvocationHandler h;" + rt+ 
		methodStr + rt +
		"}" ;
		
		String pac=infce.getPackage().toString();
		pac=pac.substring(8);
		pac=pac.replace(".", "/");
//		System.out.println(pac);
		//产生代理类的java文件
		String filename = System.getProperty("user.dir") +"/bin/"+pac+"/$Proxy0.java";
		File file = new File(filename);
		FileUtils.writeStringToFile(file, str);
		
		//编译
		//拿到编译器
		JavaCompiler complier = ToolProvider.getSystemJavaCompiler();
		//文件管理者
		StandardJavaFileManager fileMgr = 
				complier.getStandardFileManager(null, null, null);
		//获取文件
		Iterable units = fileMgr.getJavaFileObjects(filename);
		//编译任务
		CompilationTask t = complier.getTask(null, fileMgr, null, null, null, units);
		//进行编译
		t.call();
		fileMgr.close();
		
		//load 到内存
		ClassLoader cl = ClassLoader.getSystemClassLoader();
		Class c = cl.loadClass("proxy.betterbyself.$Proxy0");
		
		//获得该类构造方法
		Constructor ctr = c.getConstructor(InvocationHandler.class);
		
		//删除多余文件
		if(null!=file && file.exists()) {
			file.delete();
		}
		filename = System.getProperty("user.dir") +"/bin/"+pac+"/$Proxy0.class";
		file = new File(filename);
		if(null!=file && file.exists()) {
			file.delete();
		}
		//创建实例
		return ctr.newInstance(h);//传入h实例
	}

	
	
	
}

Client类

package proxy.betterbyself;

public class Client {

	/**
	 * 测试类
	 * @throws Exception 
	 */
	public static void main(String[] args) throws Exception {
		Car car = new Car();
		InvocationHandler h = new TimeHandler(car);
		Moveable m = (Moveable)Proxy.newProxyInstance(Moveable.class,h);
		m.move();
		System.out.println();
		System.out.println("返回值："+m.fun("ss", 10));
		
	}

}

执行结果：

汽车开始行驶....
汽车行驶中....
汽车结束行驶....  汽车行驶时间：747毫秒！


汽车开始行驶....
运行fun中
汽车结束行驶....  汽车行驶时间：0毫秒！
返回值：ss10