D18 Sping Boot 入门 Sping框架--Java Web之书城项目（八） 过滤器

一、使用Filter过滤器拦截/pages/manager/所有内容，实现权限检查

新建ManagerFilter

package com.gychen.filter;

import javax.servlet.*;
import javax.servlet.http.HttpServletRequest;
import java.io.IOException;

public class ManagerFilter implements Filter {

    @Override
    public void init(FilterConfig filterConfig) throws ServletException {

    }

    @Override
    public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) throws IOException, ServletException {

        HttpServletRequest httpServletRequest = (HttpServletRequest) servletRequest;

        Object user =httpServletRequest.getSession().getAttribute("user");

        if (user ==null) {
            httpServletRequest.getRequestDispatcher("/pages/user/login.jsp").forward(servletRequest,servletResponse);
        } else {
            filterChain.doFilter(servletRequest,servletResponse);
        }
    }

    @Override
    public void destroy() {

    }
}

二、ThreadLocal的使用

ThreadLocal 的作用，它可以解决多线程的数据安全问题。

ThreadLocal 它可以给当前线程关联一个数据（可以是普通变量，可以是对象，也可以是数组，集合）

ThreadLocal 的特点：

　　1、ThreadLocal 可以为当前线程关联一个数据。（它可以像 Map 一样存取数据，key 为当前线程）

　　2、每一个 ThreadLocal 对象，只能为当前线程关联一个数据，如果要为当前线程关联多个数据，就需要使用多个

　　　　ThreadLocal 对象实例。

　　3、每个 ThreadLocal 对象实例定义的时候，一般都是 static 类型

　　4、ThreadLocal 中保存数据，在线程销毁后。会由 JVM 虚拟自动释放。

三、实现Filter和ThreadLocal组合管理事务

3.1 在JdbcUtils里用ThreadLocal重构

package com.gychen.utils;

import com.alibaba.druid.pool.DruidDataSource;
import com.alibaba.druid.pool.DruidDataSourceFactory;

import java.io.InputStream;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.Properties;

public class JdbcUtils {

    private static DruidDataSource dataSource;
    // 用作单线程事务管理
    private static ThreadLocal<Connection> conns = new ThreadLocal<Connection>();

    static {

        try {
            Properties properties = new Properties();
            //读取jdbc.properties配置文件属性
            InputStream inputStream = JdbcUtils.class.getClassLoader().getResourceAsStream("jdbc.properties");
            //从流中加载数据
            properties.load(inputStream);
            //创建 数据库 连接池
            dataSource = (DruidDataSource) DruidDataSourceFactory.createDataSource(properties);
//            System.out.println(dataSource.getConnection());
        } catch (Exception e) {
            e.printStackTrace();
        }

    }


    /**
     * 获取数据库连接池中的连接
     * @return 如果返回null 说明获取连接失败<br/> 有值就是获取连接成功
     */
//    public static Connection getConnection(){
//        Connection conn = null;
//        try {
//            conn = dataSource.getConnection();
//        } catch (SQLException e) {
//            e.printStackTrace();
//        }
//        return  conn;
//    }

    public static Connection getConnection(){
        Connection conn = conns.get();
        if (conn == null) {
            try {
                conn = dataSource.getConnection(); // 从数据库连接池中获取连接

                conns.set(conn);  // 保存到ThreadLocal对象中，供后面的JBDC操作使用

                conn.setAutoCommit(false);  // 设置为手动管理事务
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
        return conn;
    }

//    /**
//     * 关闭连接，放回数据库连接池
//     * @param conn
//     */
//    public static void close(Connection conn){
//        if(conn != null){
//            try {
//                conn.close();
//            } catch (SQLException e) {
//                e.printStackTrace();
//            }
//        }
//    }

    /**
     * 提交事务并关闭释放连接
     */
    public static void commitAndClose(){
        Connection connection = conns.get();
        if (connection != null) {  // 如果不等于null，说明之前使用过连接，操作过数据库
            try {
                connection.commit();  // 提交事务
            } catch (SQLException e) {
                e.printStackTrace();
            } finally {
                try {
                    connection.close();   // 关闭连接，释放资源
                } catch (SQLException e) {
                    e.printStackTrace();
                }
            }

        }
        // 一定要执行remove操作，否则会出错（因为Tomcat服务器底层使用了线程池技术）
        conns.remove();
    }


    /**
     * 回滚事务并关闭释放连接
     */
    public static void rollbackAndClose(){
        Connection connection = conns.get();
        if (connection != null) {  // 如果不等于null，说明之前使用过连接，操作过数据库
            try {
                connection.rollback();  // 回滚事务
            } catch (SQLException e) {
                e.printStackTrace();
            } finally {
                try {
                    connection.close();   // 关闭连接，释放资源
                } catch (SQLException e) {
                    e.printStackTrace();
                }
            }

        }
        // 一定要执行remove操作，否则会出错（因为Tomcat服务器底层使用了线程池技术）
        conns.remove();
    }


    public static void main(String[] args) {

    }

}

在Dao里删除close操作并抛出异常

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

  在com.gychen.filter里新建TransactionFilter并配置web.xml

package com.gychen.filter;

import com.gychen.utils.JdbcUtils;

import javax.servlet.*;
import java.io.IOException;

public class TransactionFilter implements Filter {

    @Override
    public void init(FilterConfig filterConfig) throws ServletException {

    }

    @Override
    public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) throws IOException, ServletException {

        try {
            filterChain.doFilter(servletRequest,servletResponse);
            JdbcUtils.commitAndClose();  // 提交事务
        } catch (Exception e) {
            JdbcUtils.rollbackAndClose();  // 回滚事务
            e.printStackTrace();

        }
    }

    @Override
    public void destroy() {

    }
}

在BaseServlet中抛出异常 throw new RuntimeException(e); 

3.3 将所有异常都统一交给Tomcat展示友好的错误信息页面

在web.xml中可以通过错误页面配置来进行管理

 <!--错误页面管理配置-->
    <!-- error-page标签配置 服务器出错之后自动跳转的页面 -->
    <error-page>
        <!-- error-code是错误类型 -->
        <error-code>500</error-code>
        <!-- location表示要跳转的页面路径 -->
        <location>/pages/error/error500.jsp</location>
    </error-page>

在pages里新建error文件夹，新建erro500.jsp

<%@ page contentType="text/html;charset=UTF-8" language="java" %>
<html>
<head>
    <title>error500</title>
</head>
<body>
    很抱歉，您访问的页面服务器出现错误
</body>
</html>