借助Javassist实现将方法转为异步执行

找点事情做,写点代码练练手。实现一个简单的功能:将标注了@Async注解的实例方法调用时自动异步执行。
Spring3中已支持此功能, 如:
   @Service
public class MailServiceImpl implements MailService{
@Async
public void send(Mail mail){
//在此实现发送email逻辑
//异步执行的
}
}

至于Spring对它的实现原理在此就不罗嗦了, 反正离不开AOP、动态代理、多线程等技术。Guice 也有AOP支持, 要实现此功能,需要自己写点代码了。

现在我来借助[url=http://www.jboss.org/javassist]Javassist[/url]手工编写个简单山寨版。

直接上源码吧,加了点点注释:
1 注解Async源码(Async.java)
import static java.lang.annotation.RetentionPolicy.RUNTIME;
import static java.lang.annotation.ElementType.METHOD;
import java.lang.annotation.Retention;
import java.lang.annotation.Target;

/**
* 异步执行的方法注解.
*
*/
@Target(METHOD)
@Retention(RUNTIME)
public @interface Async {
}


2 实现和测试代码(Main.java)
import java.lang.reflect.Method;
import java.util.logging.Level;
import java.util.logging.Logger;

import javassist.util.proxy.MethodFilter;
import javassist.util.proxy.MethodHandler;
import javassist.util.proxy.ProxyFactory;
import javassist.util.proxy.ProxyObject;
public class Main {
public static void main(String[] args) throws InstantiationException, IllegalAccessException {
long s = System.currentTimeMillis();
new Main().asynMethod();
System.out.println((System.currentTimeMillis() - s));

Main main = getManagedBean(Main.class, new AsynMethodHandler());

s = System.currentTimeMillis();
main.asynMethod();
System.out.println((System.currentTimeMillis() - s)); //输出的值远远小于2000哦, 达到异步目的
}

/**
* 模拟从"容器"获取管理的Bean.
*/
public static <T> T getManagedBean(Class<T> type, TheMethodHandler mh) {
ProxyFactory f = new ProxyFactory();
f.setSuperclass(type);
f.setFilter(mh.getMethodFilter());
Class c = f.createClass(); //创建代理类
T main = null;
try {
main = (T) c.newInstance(); //使用代理类创建实例
((ProxyObject) main).setHandler(mh.getMethodHandler()); //设置方法拦截器
return main;
} catch (Exception ex) {
Logger.getLogger(Main.class.getName()).log(Level.SEVERE, null, ex);
}
return main;
}

/**
* 想要异步执行的方法, 标注了@Async
*/
@Async
public void asynMethod() {
try {
Thread.sleep(2000);
System.out.println("hello");
} catch (Exception ex) {
Logger.getLogger(Main.class.getName()).log(Level.SEVERE, null, ex);
}
}

public static interface TheMethodHandler {
public MethodFilter getMethodFilter();
public MethodHandler getMethodHandler();
}

static class AsynMethodHandler implements TheMethodHandler {
public MethodFilter getMethodFilter() {
return new MethodFilter() {
public boolean isHandled(Method m) {
boolean isAsyncMethod = m.isAnnotationPresent(Async.class);
//拦截标注为@Async的方法(并排除finalize方法)
return !m.getName().equals("finalize") && isAsyncMethod;
}
};
}

public MethodHandler getMethodHandler() {
return new MethodHandler() {
public Object invoke(final Object self, Method m, final Method proceed,
final Object[] args) throws Throwable {
//用new Thread的方式简单实现
//TODO:用JUC替换
(new Thread() {
@Override
public void run() {
try {
proceed.invoke(self, args);
} catch (Exception e) {
throw new RuntimeException(e);
}
}
}).start();
return null; //TODO: 返回Feture?
}
};
}
}
}
Javassist是一个Java字节码编辑库,可以在运行时对Java字节码进行修改和生成,可以用它来实现空指针异常检测工具。下面是一个简单的实现步骤: 1. 使用Javassist加载需要进行空指针异常检测的类,获取类的所有方法。 2. 遍历每个方法的每个指令,在指令执行之前插入代码进行空指针检测。 3. 插入的代码可以是try-catch语句块,捕获到空指针异常后可以使用日志记录下来,也可以是抛出自定义的异常。 4. 保存修改后的字节码,生成新的类文件。 下面是一个简单的例子,实现了对方法中所有引用类型参数的空指针检测: ```java import javassist.*; public class NullPointerChecker { public static void main(String[] args) throws Exception { // 加载需要修改的类 ClassPool pool = ClassPool.getDefault(); CtClass targetClass = pool.get("com.example.TargetClass"); // 遍历每个方法 for (CtMethod method : targetClass.getDeclaredMethods()) { // 获取方法的参数类型 CtClass[] parameterTypes = method.getParameterTypes(); // 遍历每个参数 for (int i = 0; i < parameterTypes.length; i++) { CtClass parameterType = parameterTypes[i]; // 如果参数是引用类型 if (!parameterType.isPrimitive()) { // 在方法第一行插入代码进行空指针检测 method.insertBefore("if ($" + (i + 1) + " == null) { throw new IllegalArgumentException(\"Argument " + (i + 1) + " is null\"); }"); } } } // 保存修改后的类文件 targetClass.writeFile(); } } ``` 在这个例子中,我们通过Javassist加载了一个名为TargetClass的类,遍历了这个类中的所有方法和参数,对于引用类型的参数,在方法第一行插入了代码进行空指针检测。在方法调用时,如果传入参数为null,则会抛出IllegalArgumentException异常。 当然,这只是一个简单的例子,实际上还需要考虑更多的情况,比如对于方法中的局部变量也需要进行空指针检测等。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值