从0搭建Tomcat第二天：深入理解Servlet容器与反射机制

在上一篇博客中，我们从0开始搭建了一个简易的Tomcat服务器，并实现了基本的HTTP请求处理。今天，我们将继续深入探讨Tomcat的核心组件之一——Servlet容器，并介绍如何使用反射机制动态加载和管理Servlet。

1. Servlet容器的作用

Servlet容器是Tomcat的核心组件之一，负责管理Servlet的生命周期，并将HTTP请求分发给相应的Servlet进行处理。Servlet容器的主要功能包括：

加载和初始化Servlet：在Tomcat启动时，Servlet容器会加载所有的Servlet类，并调用它们的init方法进行初始化。
处理HTTP请求：当客户端发送HTTP请求时，Servlet容器会根据请求的URL找到对应的Servlet，并调用其service方法处理请求。
销毁Servlet：在Tomcat关闭时，Servlet容器会调用Servlet的destroy方法进行资源释放。

2. 实现Servlet容器

2.1 使用Map管理Servlet

我们可以使用一个Map来管理Servlet对象，其中key是Servlet的访问路径，value是对应的Servlet对象。以下是一个简单的实现：

Map<String, Servlet> servletContainer = new HashMap<>();// 在Tomcat启动时加载Servlet
servletContainer.put("/login", new LoginServlet());
servletContainer.put("/show", new ShowServlet());// 根据请求路径获取相应的Servlet
Servlet servlet = servletContainer.get(request.getPath());
if (servlet != null) {servlet.service(request, response);
}

在这个实现中，我们手动将LoginServlet和ShowServlet对象放入servletContainer中。然而，在实际的Tomcat中，Servlet的加载和管理是自动化的，通常通过反射机制来实现。

2.2 使用反射动态加载Servlet

为了动态加载Servlet，我们可以使用Java的反射机制。反射允许我们在运行时获取类的信息，并动态创建对象。以下是一个使用反射加载Servlet的示例：

package com.qcby.Util;import java.io.File;
import java.util.ArrayList;
import java.util.List;public class SearchClassUtil {public static List<String> classPaths = new ArrayList<String>();/*** 扫描指定包下的所有类，并获取全路径名*/public static List<String> searchClass() {String basePack = "com.qcby.webapps";String classPath = SearchClassUtil.class.getResource("/").getPath();basePack = basePack.replace(".", File.separator);String searchPath = classPath + basePack;doPath(new File(searchPath), classPath);return classPaths;}/*** 递归扫描目录，获取所有类的全路径名*/private static void doPath(File file, String classpath) {if (file.isDirectory()) {File[] files = file.listFiles();if (files != null) {for (File f1 : files) {doPath(f1, classpath);}}} else {if (file.getName().endsWith(".class")) {String path = file.getPath().replace(classpath.replace("/", "\\").replaceFirst("\\\\", ""), "").replace("\\", ".").replace(".class", "");classPaths.add(path);}}}/*** 根据全路径名加载类对象*/public static List<Class<?>> loadClasses(List<String> classPaths) {List<Class<?>> classes = new ArrayList<>();for (String className : classPaths) {try {Class<?> clazz = Class.forName(className);classes.add(clazz);} catch (ClassNotFoundException e) {System.err.println("无法加载类: " + className);e.printStackTrace();}}return classes;}/*** 获取类中的 @WebServlet 注解值*/public static void getWebServletAnnotation(List<Class<?>> classes) {for (Class<?> clazz : classes) {if (clazz.isAnnotationPresent(WebServlet.class)) {WebServlet webServletAnnotation = clazz.getAnnotation(WebServlet.class);String urlMapping = webServletAnnotation.urlMapping();System.out.println("类 " + clazz.getName() + " 的 @WebServlet 注解值: " + urlMapping);}}}public static void main(String[] args) {List<String> classPaths = searchClass();List<Class<?>> classes = loadClasses(classPaths);getWebServletAnnotation(classes);}
}

在这个代码中，我们使用SearchClassUtil类扫描指定包下的所有类，并加载带有@WebServlet注解的类。通过反射，我们可以动态获取这些类的注解信息，并根据注解中的urlMapping值将Servlet对象放入servletContainer中。

2.3 自定义`@WebServlet`注解

为了标识哪些类是Servlet，我们可以定义一个自定义注解@WebServlet：

package com.qcby.Util;import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.TYPE)
public @interface WebServlet {String urlMapping() default "";
}

在Servlet类中，我们可以使用这个注解来指定Servlet的访问路径：

package com.qcby.webapps.myweb;import com.qcby.Util.WebServlet;
import com.qcby.servlet.HttpServlet;
import com.qcby.servlet.req.HttpServletRequest;
import com.qcby.servlet.req.HttpServletResponse;@WebServlet(urlMapping = "/login")
public class LoginServlet extends HttpServlet {@Overrideprotected void doGet(HttpServletRequest request, HttpServletResponse response) {System.out.println("处理登录的GET请求");}@Overrideprotected void doPost(HttpServletRequest request, HttpServletResponse response) {System.out.println("处理登录的POST请求");}
}