servlet-url-map

最新推荐文章于 2024-03-25 17:10:40 发布

原创最新推荐文章于 2024-03-25 17:10:40 发布 · 157 阅读

0 ·

CC 4.0 BY-SA版权

帮助代码专栏收录该内容

40 篇文章

订阅专栏

本文介绍了一个自定义Servlet（AServlet）的实现，该Servlet通过注解动态映射URL到具体的方法，支持GET和POST请求。文章详细展示了如何在Servlet初始化时读取注解并建立URL到方法名的映射，以及如何在请求到达时调用相应的方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >


import javax.servlet.ServletConfig;
import javax.servlet.ServletContext;
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.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
import java.lang.reflect.Method;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

public class AServlet extends HttpServlet {
	private Map<String, String> urlHandlerMethods = new ConcurrentHashMap<String, String>();

	private String baseUrl = "";

    private Object methodInvoke(Object object, String methodName,Class<?>... parameterTypes) {
        try {
            Class<? extends Object> classInfo = object.getClass();
            Method method = classInfo.getMethod(methodName,parameterTypes);
            Object result = method.invoke(object);
            return result;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return "error";
    }


	@Override
	public void init() throws ServletException {
        String baseUrlTmp = this.getInitParameter("baseUrl");
        if(baseUrlTmp != null){
            this.baseUrl = baseUrlTmp;
        }
        ServletContext servletContext = this.getServletContext();
        String contextPath = servletContext.getContextPath();
        String servletInfo = this.getServletInfo();
         ServletConfig config = getServletConfig();
        Class<? extends EleDocServlet> aClass = this.getClass();
        Method[] declaredMethods = aClass.getDeclaredMethods();
        for (Method method : declaredMethods) {
            UrlMapping methodUrlMapping = method.getDeclaredAnnotation(UrlMapping.class);
            if (methodUrlMapping != null) {
                String methodUrl = this.baseUrl + methodUrlMapping.value();
                urlHandlerMethods.put(methodUrl, method.getName());
            }
        }
	}

	@Override
	protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
		doPost(request, response);
	}

	@Override
	protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        String requestURI = request.getRequestURI();
        String methodName = urlHandlerMethods.get(requestURI);
		if (methodName == null) {
            response.getWriter().println("404_not_found");
			return;
		}

        Object result = methodInvoke(this, methodName, request.getClass(), response.getClass());
        response.getWriter().println(result);
	}

	@UrlMapping("/index")
	public String test(HttpServletRequest request, HttpServletResponse response){
        return "ok";
    }

}

@Target({ ElementType.METHOD, ElementType.TYPE })
@Retention(RetentionPolicy.RUNTIME)
@interface UrlMapping {
    String value() default "";
}