第二次改进这个东西,第一次写的发现不支持带有参数和返回类型的方法的类。额,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