Servlet中的方法分发

Servlet中的方法分发

在通过servlet处理前端各类业务时，常常要创建多个servlet，比如对于一个User，可能就要写LoginServlet、ExitServlet、RegisterServlet，使得创建的包中目录繁多，而且不便于统一管理。
这时可以利用java的高级特性——反射定义一个BaseSevlet，在其中进行方法的分发。

定义BaseSevlet（方法分发器）

import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.lang.reflect.Method;

public class BaseServlet extends HttpServlet {
    @Override
    protected void service(HttpServletRequest req, HttpServletResponse resp) {

        String uri = req.getRequestURI();   // /虚拟路径/user/(method)
        // 最后一个'/'后的内容
        String methodName = uri.substring(uri.lastIndexOf('/') + 1);

        // 获取方法
        try {
        	// 此处的this不是BaseServlet，而是调用者本身（Base的子类）
            Method method = this.getClass().getMethod(methodName, HttpServletRequest.class, HttpServletResponse.class);
            // 执行方法
            method.invoke(this, req, resp);
        } catch (Exception e) {
            e.printStackTrace();
        }

    }
}

在BaseServlet中覆盖重写HttpServlet中的service方法（每次客户向服务器发出请求时，服务器就会调用这个方法）。

继承BaseServlet

将面向同一业务对象（如User）的方法写到一个继承BaseServlet的子类中，在注解中配置路径处理客户端发起的/user下的所有请求：

@WebServlet("/user/*")
public class UserServlet extends BaseServlet {

    private UserServiceImpl service = new UserServiceImpl();

    public void regist(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException{
        request.setCharacterEncoding("utf-8");
        response.setContentType("application/json;charset=utf-8");

        //获取填写的验证码
        String post_code = request.getParameter("check");
        HttpSession session = request.getSession();
        String real_code = (String) session.getAttribute("CHECKCODE_SERVER");
        session.removeAttribute("CHECKCODE_SERVER");    // 保证验证码只能使用一次

        if (real_code == null || !post_code.equalsIgnoreCase(real_code)){
            ResultInfo info = new ResultInfo();
            info.setFlag(false);
            info.setErrorMsg("验证码错误！");
            //info对象序列化为json
            ObjectMapper mapper = new ObjectMapper();
            String json = mapper.writeValueAsString(info);
            //写回客户端
            response.getWriter().write(json);
            return;
        }

        // 封装Bean对象
        Map<String, String[]> pMap = request.getParameterMap();
        User user = new User();
        try {
            BeanUtils.populate(user, pMap);
        } catch (Exception e) {
            e.printStackTrace();
        }

        boolean flag = service.regist(user);
        ResultInfo info = new ResultInfo();
        if (flag){
            info.setFlag(true);
        }else{
            info.setFlag(false);
            info.setErrorMsg("注册失败!");
        }
        //info对象序列化为json
        ObjectMapper mapper = new ObjectMapper();
        String json = mapper.writeValueAsString(info);
        //写回客户端
        response.getWriter().write(json);
    }

    public void login(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException{
        request.setCharacterEncoding("utf-8");

        //封装用户对象
        Map<String, String[]> pMap = request.getParameterMap();
        User user = new User();
        try {
            BeanUtils.populate(user, pMap);
        } catch (IllegalAccessException e) {
            e.printStackTrace();
        } catch (InvocationTargetException e) {
            e.printStackTrace();
        }

        //登录
        User login = service.login(user);

        ResultInfo info = new ResultInfo();

        //用户名或密码错误
        if (login == null){
            info.setFlag(false);
            info.setErrorMsg("用户名或密码错误");
        }

        //用户未激活
        if (login != null && !"Y".equals(login.getStatus())){
            info.setFlag(false);
            info.setErrorMsg("账号尚未激活");
        }

        //登陆成功
        if (login != null && "Y".equals(login.getStatus())){
            info.setFlag(true);
            request.getSession().setAttribute("user", login);
        }

        response.setContentType("application/json;charset=utf-8");
        ObjectMapper mapper = new ObjectMapper();
        mapper.writeValue(response.getOutputStream(), info);
    }

    public void checkMail(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException{
        request.setCharacterEncoding("utf-8");
        String code = request.getParameter("code");

        boolean flag = service.activate(code);

        String msg = null;
        if (flag){
            msg = "激活成功，请<a href='login.html'>登录</a>";
        }else{
            msg = "注册失败，请联系管理员";
        }

        response.setContentType("text/html;charset=utf-8");
        response.getWriter().write(msg);
    }

    public void exit(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException{
        // 获取并销毁session
        request.getSession().removeAttribute("user");

        // 跳转
        response.sendRedirect(request.getContextPath() + "login.html");
    }

关于反射，可以参考JAVA高级特性——反射。