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; 这行注释去掉，引起异常。这样你就可以看到事务被回退了。