fix: 여러가지 RequestMethod에 대한 Handler 맵핑 처리 로직 추가

woowacourse · Sep 15, 2023 · 57f1a4c · 57f1a4c
1 parent 7b607f3
commit 57f1a4c
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/...c/main/java/webmvc/org/springframework/web/servlet/mvc/tobe/AnnotationHandlerMapping.java b/...c/main/java/webmvc/org/springframework/web/servlet/mvc/tobe/AnnotationHandlerMapping.java
@@ -45,7 +45,14 @@ private void mapHandlerForAnnotatedController(Class<?> controller) {
     private void mapHandlerForAnnotatedMethod(Method declaredMethod) {
         if (declaredMethod.isAnnotationPresent(RequestMapping.class)) {
             RequestMapping annotation = declaredMethod.getAnnotation(RequestMapping.class);
-            HandlerKey handlerKey = new HandlerKey(annotation.value(), annotation.method()[0]);
+            RequestMethod[] method = annotation.method();
+            mapHandlerForRequestMethods(declaredMethod, annotation, method);
+        }
+    }
+
+    private void mapHandlerForRequestMethods(Method declaredMethod, RequestMapping annotation, RequestMethod[] method) {
+        for (RequestMethod requestMethod : method) {
+            HandlerKey handlerKey = new HandlerKey(annotation.value(), requestMethod);
             HandlerExecution handlerExecution = new HandlerExecution(declaredMethod);
             handlerExecutions.put(handlerKey, handlerExecution);
         }
@@ -54,7 +61,6 @@ private void mapHandlerForAnnotatedMethod(Method declaredMethod) {
     public Object getHandler(final HttpServletRequest request) {
         String requestURI = request.getRequestURI();
         String method = request.getMethod();
-
         return handlerExecutions.get(new HandlerKey(requestURI, RequestMethod.valueOf(method)));
     }
 }