项目阶段七:使用 Filter 和 ThreadLocal 组合管理事务

1、使用 ThreadLocal 来确保所有 dao 操作都在同一个 Connection 连接对象中完成

原理分析图：

JdbcUtils 工具类的修改：

public class JdbcUtils {

    private  static DruidDataSource dataSource;
    private static ThreadLocal<Connection> conns = new ThreadLocal<Connection>();


    /**
     * 静态代码块，类加载的时候执行
     * 把注册驱动程序的代码放在静态代码块中，避免多次获取连接对象时重复调用
     */
    static{
        try {
            Properties pros = new Properties();

            // 读取 jdbc.properties属性配置文件
            InputStream inputStream = JdbcUtils.class.getClassLoader().getResourceAsStream("jdbc.properties");

            // 从流中加载数据
            pros.load(inputStream);

            // 创建 数据库连接 池
            dataSource = (DruidDataSource) DruidDataSourceFactory.createDataSource(pros);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 获取数据库连接池中的连接
     * @return 如果返回null,说明获取连接失败<br/>值就是获取连接成功
     */
    public static Connection getConnection(){

        Connection conn = conns.get();

        if(conn == null){
            try {
                conn = dataSource.getConnection();//从数据库连接池中获取连接
                conns.set(conn);// 保存到ThreadLocal对象中，供后面的jdbc操作使用
                conn.setAutoCommit(false);// 设置为手动管理事务
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }

        return conn;

    }

    /**
     *  提交事务，并关闭释放连接
     */
    public static void commitAndClose(){
        Connection conn = conns.get();

        if(conn != null){// 如果不等于null，说明 之前使用过连接，操作过数据库
            try {
                conn.commit();// 提交 事务
            } catch (SQLException e) {
                e.printStackTrace();
            }finally {
                try {
                    conn.close(); // 关闭连接，资源资源
                } catch (SQLException e) {
                    e.printStackTrace();
                }
            }
        }
        // 一定要执行remove操作，否则就会出错。（因为Tomcat服务器底层使用了线程池技术
        conns.remove();
    }

    /**
     * 回滚事务，并关闭释放连接
     */
    public static void rollbackAndClose(){

        Connection conn = conns.get();

        if(conn != null){// 如果不等于null，说明 之前使用过连接，操作过数据库
            try {
                conn.rollback();//回滚事务
            } catch (SQLException e) {
                e.printStackTrace();
            }finally {
                try {
                    conn.close();// 关闭连接，资源资源
                } catch (SQLException e) {
                    e.printStackTrace();
                }
            }
        }
        // 一定要执行remove操作，否则就会出错。（因为Tomcat服务器底层使用了线程池技术
        conns.remove();
    }

//    /**
//     * 关闭连接，放回数据库连接池
//     * @param conn
//     */
//    public static void close(Connection conn){
//        DbUtils.closeQuietly(conn);
//    }
}

修改 BaseDao：

public abstract class BaseDao {
    //使用DbUtils操作数据库
    private QueryRunner queryRunner = new QueryRunner();

    /**
     * update() 方法用来执行：Insert\Update\Delete语句
     * @param sql
     * @param args
     * @return 如果返回-1,说明执行失败<br/>返回其他表示影响的行数
     */
    public  int upDate(String sql,Object...args){
        //与数据库获取连接
        Connection conn = JdbcUtils.getConnection();

        try {
            return queryRunner.update(conn,sql,args);
        } catch (SQLException e) {
            e.printStackTrace();
            throw new RuntimeException(e);
        }
    }

    /**
     * 查询返回一个javaBean的sql语句
     * @param type 返回的对象类型
     * @param sql 执行的sql语句
     * @param args sql对应的参数值
     * @param <T> 返回的类型的泛型
     * @return null:表示未查询到结果，其它则查询到结果
     */
    public <T> T queryForOne(Class<T> type ,String sql,Object...args){
        Connection conn = JdbcUtils.getConnection();

        //BeanHander:是ResultSetHandler接口的实现类，用于封装表中的一条记录。
        try {
            return queryRunner.query(conn, sql, new BeanHandler<T>(type),args);
        } catch (SQLException e) {
            e.printStackTrace();
            throw new RuntimeException(e);
        }
    }

    /**
     * 查询返回多个javaBean的sql语句
     * @param type 返回的对象类型
     * @param sql 执行的sql语句
     * @param args sql对应的参数值
     * @param <T> 返回的类型的泛型
     * @return null:表示未查询到结果，其它则查询到结果
     */
    public <T> List<T> queryForList(Class<T> type , String sql, Object...args){
        Connection conn = JdbcUtils.getConnection();

        //BeanListHandler:是ResultSetHandler接口的实现类，用于封装表中的多条记录构成的集合。
        try {
            return queryRunner.query(conn, sql, new BeanListHandler<T>(type),args);
        } catch (SQLException e) {
            e.printStackTrace();
            throw new RuntimeException(e);
        }
    }

    /**
     * 执行返回一行一列的sql语句
     * @param sql   执行的sql语句
     * @param args  sql对应的参数值
     * @return null:表示未查询到结果，其它则表示结果
     */
    public Object queryForSingleValue(String sql, Object... args){
        Connection conn = JdbcUtils.getConnection();

        //ScalarHandler:用于查询特殊值
        try {
            return queryRunner.query(conn,sql,new ScalarHandler(),args);
        } catch (SQLException e) {
            e.printStackTrace();
            throw new RuntimeException(e);
        }
    }
}

2、使用 Filter 过滤器统一给所有的 Service 方法都加上 try-catch。来进行实现的管理

原理分析图：

Filter 类代码：

@Override
public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {

    try {
        chain.doFilter(request,response);
        JdbcUtils.commitAndClose();// 提交事务
    } catch (Exception e) {
        JdbcUtils.rollbackAndClose();//回滚事务
        e.printStackTrace();
        //把异常抛给Tomcat管理展示友好的错误页面
    }
}

在 web.xml 中的配置：

<!--    对事务处理进行过滤进行配置-->
<filter>
    <filter-name>TransactionFilter</filter-name>
    <filter-class>loey.filter.TransactionFilter</filter-class>
</filter>
<filter-mapping>
    <filter-name>TransactionFilter</filter-name>
    <url-pattern>/*</url-pattern>
</filter-mapping>

一定要记得把 BaseServlet 中的异常往外抛给 Filter 过滤器:

@Override
protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
    // 设置请求体的字符集为UTF-8，从而解决post请求的中文乱码问题
    // 也要在获取请求参数之前调用才效
    req.setCharacterEncoding("UTF-8");
    // 它会同时设置服务器和客户端都使用UTF-8字符集，还设置了响应头
    // 此方法一定要在获取流对象之前调用才有效
    resp.setContentType("text/html;charset=UTF-8");
    String action = req.getParameter("action");

    try {
        //获 取 action 业 务 鉴 别 字 符 串 ， 获 取 相 应 的 业 务 方 法 反 射 对 象
        Method method = this.getClass().getDeclaredMethod(action,HttpServletRequest.class,HttpServletResponse.class);
        //调用目标业务方法
        method.invoke(this,req,resp);
    } catch (Exception e) {
        e.printStackTrace();
        throw new RuntimeException(e);// 把异常抛给Filter过滤器
    }
}