Java利用动态代理模拟实现Spring的AOP（面向切面编程）功能的小实践

本文主要是带领读者实现一个小例子，来模拟 Spring 框架的面向切面编程（AOP）。实现过程中用到了 Java 的动态代理。

整个项目由maven管理。

pom.xml

<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd"> <modelVersion>4.0.0</modelVersion>
<groupId>zhangchao</groupId>
<artifactId>AOP</artifactId>
<version>0.0.1-SNAPSHOT</version>

<properties>
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
<project.reporting.outputEncoding>UTF-8</project.reporting.outputEncoding>
<java.version>1.8</java.version>
</properties>
</project>

目录结构

src/main/java
│
└─ zhangchao
│
├─ dao
│    └ UserDao.java
│
├─ framework
│    ├ Factory.java
│    ├ TransactionInterceptor.java
│    └ ZcInvocationHandler.java
│
├─ po
│    └ User.java
│
├─ service
│    ├─ impl
│    │    └ UserServiceImpl.java
│    │
│    └─ UserService.java
│
└─ Main.java

实体类User

package zhangchao.po;

public class User {
String id;
String name;
Integer age;
}

UserDao封装数据库操作

package zhangchao.dao;

public class UserDao {
public void save() {
System.out.println("save user");
}
}

工厂类Factory。这里使用了 Java 动态代理。表面上看是根据className返回每个类对应生成的对象，事实上是返回的是代理对象。每当生成代理对象的时候，就会调用接口InvocationHandler的invoke方法。

package zhangchao.framework;

import java.lang.reflect.Proxy;

public final class Factory {
public final static Object getBean(String className) {
Object bean = null;
try {
Class<?> c = Class.forName(className);
Object realBean = c.newInstance();
bean = Proxy.newProxyInstance(c.getClassLoader(), c.getInterfaces(),
new ZcInvocationHandler(realBean));
} catch (ClassNotFoundException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (InstantiationException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (IllegalAccessException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
return bean;
}
}

ZcInvocationHandler类实现了InvocationHandler接口。在继承的invoke方法里，所有save字符串开头的方法都要做事务处理。事务处理的代码在TransactionInterceptor类中。

package zhangchao.framework;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;

public class ZcInvocationHandler implements InvocationHandler {
private Object bean;

TransactionInterceptor ti = new TransactionInterceptor();

ZcInvocationHandler(Object bean) {
this.bean = bean;
}

public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
String methodName = method.getName();
if (methodName.startsWith("save")) {
ti.begin();
try {
method.invoke(bean, args);
ti.commit();
} catch (Exception e) {
ti.rollBack();
}
} else {
method.invoke(bean, args);
}
return null;
}

}

TransactionInterceptor类用来处理事务。

package zhangchao.framework;

public class TransactionInterceptor {
public void begin () {
System.out.println("开始事务");
}

public void commit() {
System.out.println("提交事务");
}

public void rollBack() {
System.out.println("回滚事务");
}
}

服务类的接口：

package zhangchao.service;

import zhangchao.po.User;

public interface UserService {
void save(User user);
}

服务类的实现：

package zhangchao.service.impl;

import zhangchao.dao.UserDao;
import zhangchao.po.User;
import zhangchao.service.UserService;

public class UserServiceImpl implements UserService {

private UserDao userDao = new UserDao();

public void save(User user) {
userDao.save();
//      int r = 4/0;
}
}

Main.java

package zhangchao;

import zhangchao.framework.Factory;
import zhangchao.po.User;
import zhangchao.service.UserService;

public class Main {

public static void main(String[] args) {
UserService userDao = (UserService)Factory.getBean("zhangchao.service.impl.UserServiceImpl");
userDao.save(new User());
}

}

在 UserServiceImpl 类中，可以把

int r = 4/0;

这行注释去掉，引起异常。这样你就可以看到事务被回退了。