[톰캣 구현하기 - 3, 4단계] 베베(최원용) 미션 제출합니다.

tomcat/src/main/java/org/apache/catalina/connector/Connector.java

@@ -4,6 +4,14 @@
 import java.io.UncheckedIOException;
 import java.net.ServerSocket;
 import java.net.Socket;
+import java.util.Arrays;
+import java.util.List;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.Executors;
+import java.util.concurrent.Future;
+import java.util.concurrent.LinkedBlockingQueue;
+import java.util.concurrent.ThreadPoolExecutor;
+import java.util.concurrent.TimeUnit;
 import org.apache.coyote.http11.Http11Processor;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -14,8 +22,12 @@ public class Connector implements Runnable {
 
     private static final int DEFAULT_PORT = 8080;
     private static final int DEFAULT_ACCEPT_COUNT = 100;
+    private static final int MAX_THREAD_POOL_SIZE = 250;
+    private static final int CORE_POOL_SIZE = 10;
+    private static final Long KEEP_ALIVE_TIME = 60L;
 
     private final ServerSocket serverSocket;
+    private final ThreadPoolExecutor threadPoolExecutor;
     private boolean stopped;
 
     public Connector() {
@@ -24,6 +36,12 @@ public Connector() {
 
     public Connector(final int port, final int acceptCount) {
         this.serverSocket = createServerSocket(port, acceptCount);
+        this.threadPoolExecutor = new ThreadPoolExecutor(
+                CORE_POOL_SIZE,
+                MAX_THREAD_POOL_SIZE,
+                KEEP_ALIVE_TIME,
+                TimeUnit.SECONDS,
+                new LinkedBlockingQueue<>(DEFAULT_ACCEPT_COUNT));
         this.stopped = false;
     }
 
@@ -66,13 +84,14 @@ private void process(final Socket connection) {
             return;
         }
         var processor = new Http11Processor(connection);
-        new Thread(processor).start();
+        threadPoolExecutor.execute(processor);
     }
 
     public void stop() {
         stopped = true;
         try {
             serverSocket.close();
+            threadPoolExecutor.shutdown();
         } catch (IOException e) {
             log.error(e.getMessage(), e);
         }

tomcat/src/main/java/org/apache/coyote/http11/Http11Processor.java

@@ -6,16 +6,13 @@
 import java.io.InputStreamReader;
 import java.io.OutputStream;
 import java.net.Socket;
-import java.util.ArrayList;
-import java.util.List;
 import nextstep.jwp.exception.UncheckedServletException;
 import org.apache.coyote.Processor;
-import org.apache.coyote.http11.request.body.RequestBody;
-import org.apache.coyote.http11.request.handler.RequestHandler;
-import org.apache.coyote.http11.request.header.RequestHeader;
-import org.apache.coyote.http11.request.line.RequestLine;
+import org.apache.coyote.http11.request.HttpRequest;
+import org.apache.coyote.http11.request.HttpRequestGenerator;
+import org.apache.coyote.http11.request.RequestHandler;
+import org.apache.coyote.http11.response.HttpResponse;
 import org.apache.coyote.http11.response.HttpResponseGenerator;
-import org.apache.coyote.http11.response.ResponseEntity;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -24,7 +21,7 @@ public class Http11Processor implements Runnable, Processor {
     private static final Logger LOG = LoggerFactory.getLogger(Http11Processor.class);
 
     private final Socket connection;
-    private final HttpResponseGenerator httpResponseGenerator = new HttpResponseGenerator();
+    private final HttpResponseGenerator responseGenerator = new HttpResponseGenerator();
     private final RequestHandler requestHandler = new RequestHandler();
 
     public Http11Processor(Socket connection) {
@@ -42,44 +39,17 @@ public void process(final Socket connection) {
         try (final InputStream inputStream = connection.getInputStream();
              final OutputStream outputStream = connection.getOutputStream();
              final BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(inputStream))) {
-            final String firstLine = bufferedReader.readLine();
-            if (firstLine == null) {
-                return;
-            }
-            final String response = getResponse(bufferedReader, firstLine);
+
+            final HttpRequest httpRequest = HttpRequestGenerator.generate(bufferedReader);
+            final HttpResponse httpResponse = requestHandler.getResponse(httpRequest);
+            final String response = responseGenerator.generate(httpResponse);
+            System.out.println("response = " + response);
+
             outputStream.write(response.getBytes());
             outputStream.flush();
         } catch (IOException | UncheckedServletException e) {
             LOG.error(e.getMessage(), e);
         }
     }
 
-    private String getResponse(BufferedReader bufferedReader, String firstLine) throws IOException {
-        final RequestLine requestLine = RequestLine.from(firstLine);
-        final RequestHeader requestHeader = getHeader(bufferedReader);
-        final RequestBody requestBody = getBody(bufferedReader, requestHeader);
-        final ResponseEntity responseEntity = requestHandler.getResponse(requestLine, requestHeader, requestBody);
-        return httpResponseGenerator.generate(responseEntity);
-    }
-
-    private RequestHeader getHeader(final BufferedReader bufferedReader) throws IOException {
-        List<String> requestHeaders = new ArrayList<>();
-        for (String line = bufferedReader.readLine(); !"".equals(line); line = bufferedReader.readLine()) {
-            requestHeaders.add(line);
-        }
-        return RequestHeader.from(requestHeaders);
-    }
-
-    private RequestBody getBody(final BufferedReader bufferedReader, final RequestHeader requestHeader)
-            throws IOException {
-        List<String> contentLengths = requestHeader.headers().get("Content-Length");
-        if (contentLengths == null) {
-            return RequestBody.from(null);
-        }
-        int contentLength = Integer.parseInt(contentLengths.get(0));
-        char[] buffer = new char[contentLength];
-        bufferedReader.read(buffer, 0, contentLength);
-        return RequestBody.from(new String(buffer));
-    }
-
 }

tomcat/src/main/java/org/apache/coyote/http11/auth/Cookie.java

@@ -3,6 +3,7 @@
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
+import java.util.Optional;
 
 import static java.util.Collections.EMPTY_MAP;
 import static java.util.stream.Collectors.collectingAndThen;
@@ -32,12 +33,22 @@ public static Cookie from(final List<String> elements) {
                 ));
     }
 
+    public Cookie() {
+    }
+
+    public void setSession(Session session) {
+        elements.put("JSESSIONID", session.getId());
+    }
+
     public void put(final String key, final String value) {
         elements.put(key, value);
     }
 
-    public String get(final String key) {
-        return elements.get(key);
+    public Optional<String> get(final String key) {
+        if (key == null) {
+            return Optional.empty();
+        }
+        return Optional.ofNullable(elements.get(key));
     }
 
 }

tomcat/src/main/java/org/apache/coyote/http11/auth/LoginService.java

@@ -0,0 +1,49 @@
+package org.apache.coyote.http11.auth;
+
+import java.util.Optional;
+import java.util.UUID;
+import nextstep.jwp.db.InMemoryUserRepository;
+import nextstep.jwp.model.User;
+import org.apache.coyote.http11.request.line.Protocol;
+import org.apache.coyote.http11.response.HttpResponse;
+
+import static org.apache.coyote.http11.response.ResponsePage.INDEX_PAGE;
+import static org.apache.coyote.http11.response.ResponsePage.LOGIN_PAGE;
+import static org.apache.coyote.http11.response.ResponsePage.UNAUTHORIZED_PAGE;
+
+public class LoginService {
+
+    public static final String COOKIE_KEY = "JSESSIONID";
+
+    private final SessionRepository sessionRepository = new SessionRepository();
+
+    public HttpResponse getLoginViewResponse(Cookie cookie, Protocol protocol) {
+        Optional<String> cookieOption = cookie.get(COOKIE_KEY);
+        if (cookieOption.isEmpty()) {
+            return HttpResponse.getCookieNullResponseEntity(protocol, LOGIN_PAGE);
+        }
+        final Optional<Session> session = sessionRepository.getSession(cookieOption.get());
+        if (session.isEmpty()) {
+            return HttpResponse.getCookieNullResponseEntity(protocol, LOGIN_PAGE);
+        }
+        return HttpResponse.getCookieNullResponseEntity(protocol, INDEX_PAGE);
+    }
+
+    public HttpResponse getLoginOrElseUnAuthorizedResponse(Protocol protocol, String account, String password) {
+        return InMemoryUserRepository.findByAccount(account)
+                .filter(user -> user.checkPassword(password))
+                .map(user -> getSuccessLoginResponse(user, protocol))
+                .orElseGet(() -> HttpResponse.getCookieNullResponseEntity(protocol, UNAUTHORIZED_PAGE));
+    }
+
+    private HttpResponse getSuccessLoginResponse(final User user, final Protocol protocol) {
+        final String uuid = UUID.randomUUID().toString();
+        final Session session = Session.from(uuid);
+        session.setAttribute("user", user);
+        sessionRepository.create(session);
+        Cookie cookie = new Cookie();
+        cookie.setSession(session);
+        return HttpResponse.getResponseEntity(protocol, cookie, INDEX_PAGE);
+    }
+
+}

tomcat/src/main/java/org/apache/coyote/http11/auth/RegisterService.java

@@ -0,0 +1,22 @@
+package org.apache.coyote.http11.auth;
+
+import nextstep.jwp.db.InMemoryUserRepository;
+import nextstep.jwp.model.User;
+import org.apache.coyote.http11.request.line.Protocol;
+import org.apache.coyote.http11.response.HttpResponse;
+
+import static org.apache.coyote.http11.response.ResponsePage.CONFLICT_PAGE;
+import static org.apache.coyote.http11.response.ResponsePage.INDEX_PAGE;
+
+public class RegisterService {
+
+    public HttpResponse getIndexOrConflictResponse(String account, String password, String email, Protocol protocol) {
+        if (InMemoryUserRepository.findByAccount(account).isPresent()) {
+            return HttpResponse.getCookieNullResponseEntity(protocol, CONFLICT_PAGE);
+        }
+
+        InMemoryUserRepository.save(new User(account, password, email));
+        return HttpResponse.getCookieNullResponseEntity(protocol, INDEX_PAGE);
+    }
+
+}

tomcat/src/main/java/org/apache/coyote/http11/auth/Session.java

@@ -2,11 +2,12 @@
 
 import java.util.HashMap;
 import java.util.Map;
+import nextstep.jwp.model.User;
 
 public class Session {
 
     private final String id;
-    private final Map<String, Object> items = new HashMap<>();
+    private final Map<String, User> items = new HashMap<>();
 
     private Session(final String id) {
         this.id = id;
@@ -16,7 +17,7 @@ public static Session from(String id) {
         return new Session(id);
     }
 
-    public void setAttribute(final String key, final Object value) {
+    public void setAttribute(final String key, final User value) {
         items.put(key, value);
     }
 

tomcat/src/main/java/org/apache/coyote/http11/auth/SessionRepository.java

@@ -1,22 +1,22 @@
 package org.apache.coyote.http11.auth;
 
-import java.util.HashMap;
 import java.util.Map;
+import java.util.Optional;
+import java.util.concurrent.ConcurrentHashMap;
 
 public class SessionRepository {
 
-    private static final Map<String, Session> SESSIONS = new HashMap<>();
+    private static final Map<String, Session> SESSIONS = new ConcurrentHashMap<>();
 
     public static void create(Session session) {
         SESSIONS.put(session.getId(), session);
     }
 
-    public Session getSession(String id) {
-        return SESSIONS.get(id);
-    }
-
-    public static void clearSessions() {
-        SESSIONS.clear();
+    public Optional<Session> getSession(String id) {
+        if (id == null) {
+            return Optional.ofNullable(null);
+        }
+        return Optional.ofNullable(SESSIONS.get(id));
     }
 
 }

tomcat/src/main/java/org/apache/coyote/http11/controller/AbstractController.java

@@ -0,0 +1,24 @@
+package org.apache.coyote.http11.controller;
+
+import org.apache.coyote.http11.request.HttpRequest;
+import org.apache.coyote.http11.request.line.HttpMethod;
+import org.apache.coyote.http11.response.HttpResponse;
+
+public abstract class AbstractController implements Controller {
+
+    @Override
+    public HttpResponse service(final HttpRequest request, final HttpResponse response) {
+        if (request.methodIsEqualTo(HttpMethod.GET)) {
+            return doGet(request, response);
+        }
+        if (request.methodIsEqualTo(HttpMethod.POST)) {
+            return doPost(request, response);
+        }
+        return null;
+    }
+
+    protected abstract HttpResponse doPost(final HttpRequest request, final HttpResponse response);
+
+    protected abstract HttpResponse doGet(final HttpRequest request, final HttpResponse response);
+
+}