Javaweb 自定义 Servlet 实现按照访问路径转发

HttpServlet Service方法 提供的 doGet() 和 doPost() 方法只能根据请求方式进行分发，假设我们现在有多个服务（添加、删除、查询...），每个服务都要对应一个 Servlet（AddServlet、DeleteServlet、SelectServlet），这样管理起来是很不方便的，我们希望使用一个 Servlet 一个实体来处理其业务，比如 BrandServlet 处理 Brand 相关的业务（增删改查），UserServlet 处理 User 相关的业务
这里我们的解决方案有两个 - 方案一：在访问是携带参数，然后通过 request.getParameter 获取参数，通过 if 判断进行分发，如：访问链接为 http://localhost/BrandServlet?method=add，然后通过 request.getParameter("method") 即可获取到 add 参数，再通过 if 进行判断，调用 add() 方法即可 - 方案二：重写 HttpServlet 的 Service 方法，自定义分发规则，直接通过访问路径进行业务处理
这里要介绍的是第二种方法

新建一个 BaseServlet，继承 HttpServlet，重写 Service 方法 ```java package com.ruochen.web.servlet;

代码语言：txt复制

import javax.servlet.ServletException;

代码语言：txt复制

import javax.servlet.http.HttpServlet;

代码语言：txt复制

import javax.servlet.http.HttpServletRequest;

代码语言：txt复制

import javax.servlet.http.HttpServletResponse;

代码语言：txt复制

import java.io.IOException;

代码语言：txt复制

import java.lang.reflect.Method;

代码语言：txt复制

/**

代码语言：txt复制

 * 替换 HttpServlet，根据请求的最后一段路径来进行方法分发
 */
public class BaseServlet extends HttpServlet {

代码语言：txt复制

    // 根据请求的最后一段路径来进行方法分发

代码语言：txt复制

    @Override

代码语言：txt复制

    protected void service(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {

代码语言：txt复制

        // 1. 获取请求路径 "/brand-case/brand/selectAll"

代码语言：txt复制

        String uri = req.getRequestURI();

代码语言：txt复制

        // 2. 获取最后一段路径，方法名

代码语言：txt复制

        int index = uri.lastIndexOf("/");

代码语言：txt复制

        String methodName = uri.substring(index   1);  // selectAll

代码语言：txt复制

        // 3. 执行方法

代码语言：txt复制

        // 3.1 获取 BrandServlet / UserServlet 字节码对象 Class

代码语言：txt复制

        // 谁调用我(this 所在的方法)，我(this)代表谁

代码语言：txt复制

        // System.out.println(this);  // BrandServlet

代码语言：txt复制

        Class<? extends BaseServlet> cls = this.getClass();

代码语言：txt复制

        // 3.2 获取方法 Method 对象，执行方法

代码语言：txt复制

        try {

代码语言：txt复制

            Method method = cls.getMethod(methodName, HttpServletRequest.class, HttpServletResponse.class);

代码语言：txt复制

            method.invoke(this, req, resp);

代码语言：txt复制

        } catch (Exception e) {

代码语言：txt复制

            e.printStackTrace();

代码语言：txt复制

```

然后我们再写一个 UserServlet 测试一下，UserServlet 直接继承 BaseServlet 类就可以 ```java package com.ruochen.web.servlet;

代码语言：txt复制

import javax.servlet.ServletException;

代码语言：txt复制

import javax.servlet.annotation.WebServlet;

代码语言：txt复制

import javax.servlet.http.HttpServletRequest;

代码语言：txt复制

import javax.servlet.http.HttpServletResponse;

代码语言：txt复制

import java.io.IOException;

代码语言：txt复制

@WebServlet("/user/*")

代码语言：txt复制

public class UserServlet extends BaseServlet{

代码语言：txt复制

    public void selectAll(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {

代码语言：txt复制

        System.out.println("user selectAll...");

代码语言：txt复制

    public void add(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {

代码语言：txt复制

        System.out.println("user add...");

代码语言：txt复制

```

0 人点赞