一个简单的Web服务器-支持Servlet请求

上接 一个简单的Web服务器-支持静态资源请求，这个服务器可以处理静态资源的请求，那么如何处理Servlet请求的呢？

判断是否是Servlet请求

首先Web服务器需要判断当前请求是否是Servlet请求。

像Tomcat，通过解析HTTP报文拿到请求url后，就可以根据web.xml来查找是否有匹配的Servlet，如果有匹配则认定为是一个有效的Servlet请求，然后将request，response传给对应的servlet的service()方法。

这里既然要实现一个简单的Web服务器，就怎么简单怎么来。认定以“/servlet”为前缀的url为Servlet请求，比如http://localhost:8080/servlet/cn.edu.jxau.tomcat.PrimitiveServlet，则是一个请求cn.edu.jxau.tomcat.PrimitiveServlet的Servlet请求。判定逻辑如下

// HttpServer 类的service()方法，是整个服务器的入口
        while (true) {
            try (Socket socket = serverSocket.accept()) {
                System.out.println("客户端建立连接：" + socket);
                Request request = new Request(socket.getInputStream());
                if (Objects.isNull(request.getUri())) {
                    continue;
                }
                if (isShutdownComment(request)) { //如果是shutdown命令，则关闭服务器
                    break;
                }
                Response response = new Response(request, socket.getOutputStream());
                if (request.getUri().startsWith("/servlet/")) { // 请求Servlet资源
                    new ServletProcessor().process(request, response);
                } else { // 请求静态资源
                    response.sendStaticResource();
                    // 书上使用了StaticResourceProcessor new StaticResourceProcessor().process(request,response);
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        }

加载Servlet

如果确实是Servlet请求，下一步就是实例化Servlet并调用service()方法。

首先需要知道请求是哪个Servlet：根据url拿到Servlet的名称

然后需要调用Servlet的service()方法：使用ClassLoader加载对应的Servlet字节码，然后实例化Servlet，调用service()方法，传入request，response。

// ServletProcessor 类的process()方法，用于处理Servlet请求
public void process(Request request, Response response) throws IOException, ClassNotFoundException, IllegalAccessException, InstantiationException, ServletException {

        String uri = request.getUri();
        String servletName = uri.substring(uri.lastIndexOf("/") + 1);
        File file = new File(Constants.WEB_ROOT);
        String urlStr = new URL("file", null, file.getCanonicalPath() + File.separator).toString(); //使用File协议
        URL[] urlArr = new URL[1];
        urlArr[0] = new URL(urlStr);
        System.out.println(Arrays.toString(urlArr));
        URLClassLoader loader = new URLClassLoader(urlArr);
        Class clazz = loader.loadClass(servletName); // 根据Servlet的全路径名加载Servlet字节码
        Servlet servlet = (Servlet) clazz.newInstance(); // 实例化Servlet
        servlet.service(request, response); //调用Servlet.service()
    }

PrimitiveServlet的实现如下，实现了Servlet接口，service()方法的实现是重点。

public class PrimitiveServlet implements Servlet {

    public void init(ServletConfig config) throws ServletException {
        System.out.println("-----init");
    }

    public void service(ServletRequest request, ServletResponse response)
            throws ServletException, IOException {

        PrintWriter out = response.getWriter();
        out.println("HTTP/1.1 200 Status OK\r\n");
        out.println("Content-Type: text/html\r\n");
        out.println("Content-Length: 36\r\n");
        out.println("\r\n");
        out.println("<h1>Status OK</h1>");
        out.print("<h1>Status OK</h1>");
    }

    public void destroy() {
        System.out.println("-------destroy");
    }

    public String getServletInfo() {
        return null;
    }

    public ServletConfig getServletConfig() {
        return null;
    }
}

参考

1.《How Tomcat Works》 - Budi Kurniawan