图书管理系统之过滤器--处理用户信息的过滤和订单数据的处理

1.AjaxGetOrdersFilter

package com.chinasoft.filter;

import java.io.IOException;
import java.sql.SQLException;
import java.util.List;

import javax.servlet.Filter;
import javax.servlet.FilterChain;
import javax.servlet.FilterConfig;
import javax.servlet.ServletException;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.http.HttpServletRequest;

import com.chinasoft.dao.impl.ImplOrderDao;
import com.chinasoft.entity.Order;
import com.chinasoft.entity.User;

/***
 * 获取订单数据的过滤器
 * 每当user.jsp页面将要显示的时候都会执行这个过滤器
 * 需要在web.xml中进行配置 (dispater的信息)
 * @author mazg1
 *
 */
public class AjaxGetOrdersFilter implements Filter {

	public AjaxGetOrdersFilter() {
	}

	public void destroy() {
	}

	public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain)
			throws IOException, ServletException {
		HttpServletRequest req = (HttpServletRequest)request;
		// 查询该用户对应的所有的订单信息
		ImplOrderDao implOrderDao = new ImplOrderDao();
		User user = (User) req.getSession().getAttribute("user");
		List<Order> orders;
		try {
			orders = implOrderDao.findOrdersByUserId(user.getId());
			// 把用户订单信息保存到域对象中
			req.getSession().setAttribute("orders", orders);
		} catch (SQLException e) {
			e.printStackTrace();
		}
		chain.doFilter(request, response);
	}

	public void init(FilterConfig fConfig) throws ServletException {

	}

}

2.自动登录过滤器

package com.chinasoft.filter;

import java.io.IOException;
import javax.servlet.Filter;
import javax.servlet.FilterChain;
import javax.servlet.FilterConfig;
import javax.servlet.ServletException;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import com.chinasoft.entity.User;


public class DoAutoLoginFilter implements Filter {

    public DoAutoLoginFilter() {
        // TODO Auto-generated constructor stub
    }

	public void destroy() {
		// TODO Auto-generated method stub
	}


	/**
	 * 自动登录的过滤器
	 */
	public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {
		HttpServletRequest req = (HttpServletRequest)request;
		HttpServletResponse res = (HttpServletResponse)response;
		
		User user = (User)req.getSession().getAttribute("user");
		//如果用户已经登录，则放行
		if (user!=null) {
			chain.doFilter(request, response);
		}
		//如果没有登录，则获取Cookie中的用户名密码进行自动登录
		else{
			Cookie[] cookies = req.getCookies();
			
			if (cookies!=null) {
				for(Cookie c : cookies){
					String name = c.getName();
					//执行自动登录
					if ("autologin".equals(name)) {
						String[] vals = c.getValue().split("&");
						req.getRequestDispatcher("/DoLoginServlet?name="+vals[0]+"&pass="+vals[1]).forward(req, res);
						return;
					}
				}
			}
			//没有获取到指定的cookie信息，也放行
			chain.doFilter(request, response);
		}
	}


	public void init(FilterConfig fConfig) throws ServletException {
		// TODO Auto-generated method stub
	}

}

3.编码过滤器，解决乱码问题

package com.chinasoft.filter;

import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.util.Map;

import javax.servlet.Filter;
import javax.servlet.FilterChain;
import javax.servlet.FilterConfig;
import javax.servlet.ServletException;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.annotation.WebFilter;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;


@WebFilter("/*")
public class EncodingFilter implements Filter {

	@Override
	public void destroy() {
	}

	/**
	 * 全局编码过滤器
	 */
	@Override
	public void doFilter(ServletRequest request, ServletResponse response,
			FilterChain chain) throws IOException, ServletException {
		// 处理请求乱码
		HttpServletRequest httpServletRequest = (HttpServletRequest) request;
		HttpServletRequest myRequest = new MyRequest(httpServletRequest);

		// 处理响应乱码
		response.setContentType("text/html;charset=utf-8");

		chain.doFilter(myRequest, response);
	}

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

}

//自定义request对象
class MyRequest extends HttpServletRequestWrapper {

	private HttpServletRequest request;

	private boolean hasEncode;

	public MyRequest(HttpServletRequest request) {
		super(request);// super必须写
		this.request = request;
	}

	// 对需要增强方法 进行覆盖
	@Override
	public Map<String, String[]> getParameterMap() {
		// 先获得请求方式
		String method = request.getMethod();
		if (method.equalsIgnoreCase("post")) {
			// post请求
			try {
				// 处理post乱码
				request.setCharacterEncoding("utf-8");
				return request.getParameterMap();
			} catch (UnsupportedEncodingException e) {
				e.printStackTrace();
			}
		} else if (method.equalsIgnoreCase("get")) {
			// get请求
			Map<String, String[]> parameterMap = request.getParameterMap();
			if (!hasEncode) { // 确保get手动编码逻辑只运行一次
				for (String parameterName : parameterMap.keySet()) {
					String[] values = parameterMap.get(parameterName);
					if (values != null) {
						for (int i = 0; i < values.length; i++) {
							try {
								// 处理get乱码
								values[i] = new String(values[i]
										.getBytes("ISO-8859-1"), "utf-8");
							} catch (UnsupportedEncodingException e) {
								e.printStackTrace();
							}
						}
					}
				}
				hasEncode = true;
			}
			return parameterMap;
		}

		return super.getParameterMap();
	}

	@Override
	public String getParameter(String name) {
		Map<String, String[]> parameterMap = getParameterMap();
		String[] values = parameterMap.get(name);
		if (values == null) {
			return null;
		}
		return values[0];  // 取回参数的第一个值ֵ
	}

	@Override
	public String[] getParameterValues(String name) {
		Map<String, String[]> parameterMap = getParameterMap();
		String[] values = parameterMap.get(name);
		return values;
	}
}