手撸一个java网关(二)

要在我们的简单Java Gateway框架中添加请求过滤功能，我们可以在GatewayServlet中添加一个过滤链（Filter Chain）的概念。这个过滤链将包含多个过滤器（Filter），每个过滤器都可以对请求或响应进行一些操作，例如验证、修改头信息、记录日志等。
首先，我们定义一个过滤器接口：

package com.example.gateway.filter;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
public interface Filter {
    boolean process(HttpServletRequest request, HttpServletResponse response) throws IOException;
}

然后，我们可以创建一些具体的过滤器实现：

package com.example.gateway.filter;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
public class LoggingFilter implements Filter {
    @Override
    public boolean process(HttpServletRequest request, HttpServletResponse response) throws IOException {
        System.out.println("Logging request: " + request.getRequestURI());
        return true; // 继续执行后续过滤器
    }
}
package com.example.gateway.filter;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
public class AuthenticationFilter implements Filter {
    @Override
    public boolean process(HttpServletRequest request, HttpServletResponse response) throws IOException {
        // 这里可以添加认证逻辑
        String authToken = request.getHeader("Authorization");
        if (authToken == null || !authToken.equals("SecretToken")) {
            response.setStatus(HttpServletResponse.SC_UNAUTHORIZED);
            response.getWriter().write("Unauthorized");
            return false; // 不继续执行后续过滤器
        }
        return true; // 认证通过，继续执行后续过滤器
    }
}

接下来，我们修改GatewayServlet来使用这些过滤器：

package com.example.gateway;
import com.example.gateway.filter.AuthenticationFilter;
import com.example.gateway.filter.Filter;
import com.example.gateway.filter.LoggingFilter;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
public class GatewayServlet extends HttpServlet {
    private final List<Filter> filters = new ArrayList<>();
    public GatewayServlet() {
        // 初始化过滤器链
        filters.add(new LoggingFilter());
        filters.add(new AuthenticationFilter());
    }
    @Override
    protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        routeRequest(request, response);
    }
    @Override
    protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        routeRequest(request, response);
    }
    private void routeRequest(HttpServletRequest request, HttpServletResponse response) throws IOException {
        // 遍历过滤器链
        for (Filter filter : filters) {
            if (!filter.process(request, response)) {
                // 如果过滤器返回false，则停止执行后续过滤器
                return;
            }
        }
        // 过滤器链执行完毕后，进行请求路由
        String path = request.getRequestURI();
        if (path.startsWith("/service1")) {
            response.getWriter().write("Service 1 response");
        } else if (path.startsWith("/service2")) {
            response.getWriter().write("Service 2 response");
        } else {
            response.setStatus(HttpServletResponse.SC_NOT_FOUND);
            response.getWriter().write("Not Found");
        }
    }
}

现在，GatewayServlet会在路由请求之前先执行过滤器链中的所有过滤器。如果任何过滤器返回false，则请求处理会停止，并且不会执行后续的过滤器或请求路由。
这个简单的过滤机制可以根据需要进一步扩展，例如添加异常处理、动态过滤器加载、更复杂的过滤逻辑等。在实际应用中，你可能需要使用更成熟的框架来处理这些问题，但是这个示例提供了一个基本的实现框架。