[Database] 쿼리 자동생성 기능 개발

build.gradle

@@ -16,6 +16,8 @@ dependencies {
     implementation 'ch.qos.logback:logback-classic:1.2.3'
     testImplementation 'org.assertj:assertj-core:3.16.1'
 
+    runtimeOnly 'com.h2database:h2:2.2.224'
+
     testImplementation 'org.mockito:mockito-core:4.11.0'
     testImplementation('org.mockito:mockito-junit-jupiter:4.11.0') {
         exclude group: 'org.junit.jupiter'

src/main/java/app/db/Database.java

@@ -14,7 +14,7 @@ public class Database {
 
     public static void addUser(User user) {
         long id = sequentialId.getAndIncrement();
-        user.setUserId(id);
+        user.setId(id);
         users.put(id, user);
     }
 

src/main/java/app/handler/LoginWithPost.java

@@ -38,7 +38,7 @@ public HandlerResponse handle(QueryParameters params) {
         }
 
         SessionEntity session = sessionManager.create(
-                user.getUserId(),
+                user.getId(),
                 user.getUserRole(),
                 user.getNickname());
 

src/main/java/app/handler/RegisterWithGet.java

@@ -11,7 +11,7 @@
 import web.filter.authentication.UserRole;
 import web.handler.SingleArgHandler;
 import web.response.HandlerResponse;
-import web.response.StaticViewResponse;
+import web.response.RedirectResponse;
 
 public class RegisterWithGet extends SingleArgHandler<QueryParameters> {
     private static final Logger log = LoggerFactory.getLogger(RegisterWithGet.class);
@@ -28,6 +28,6 @@ public HandlerResponse handle(QueryParameters params) {
         String nickname = params.getQueryValue("nickname").orElseThrow(()-> new ServiceException(ErrorCode.MISSING_REGISTER_TOKEN, "nickname required"));
         Database.addUser(new User(password, nickname, email, UserRole.MEMBER.toString()));
         log.info("Registered - password:{}, nickname:{}, email:{}", password, nickname, email);
-        return StaticViewResponse.of("/login");
+        return RedirectResponse.to("/login");
     }
 }

src/main/java/app/model/User.java

@@ -1,7 +1,7 @@
 package app.model;
 
 public class User {
-    private Long userId;
+    private Long id;
     private String password;
     private String nickname;
     private String email;
@@ -14,12 +14,12 @@ public User(String password, String nickname, String email, String userRole) {
         this.userRole = userRole;
     }
 
-    public Long getUserId() {
-        return userId;
+    public Long getId() {
+        return id;
     }
 
-    public void setUserId(Long userId){
-        this.userId = userId;
+    public void setId(Long id){
+        this.id = id;
     }
 
     public String getPassword() {
@@ -40,6 +40,6 @@ public String getUserRole() {
 
     @Override
     public String toString() {
-        return "User [userId=" + userId + ", password=" + password + ", name=" + nickname + ", email=" + email + "]";
+        return "User [userId=" + id + ", password=" + password + ", name=" + nickname + ", email=" + email + "]";
     }
 }

src/main/java/bootstrap/WebServer.java

@@ -6,6 +6,7 @@
 import java.util.concurrent.Executors;
 
 import config.AppConfig;
+import config.DatabaseConfig;
 import config.FilterConfig;
 import config.SecurityConfig;
 import org.slf4j.Logger;
@@ -18,6 +19,7 @@ public class WebServer {
     private static final AppConfig LOADER = new AppConfig();
     private static final SecurityConfig securityConfig = new SecurityConfig();
     private static final FilterConfig filterConfig = new FilterConfig();
+    private static final DatabaseConfig databaseConfig = new DatabaseConfig();
     private static final ExecutorService executor = Executors.newFixedThreadPool(32);
 
     public static void main(String args[]) throws Exception {
@@ -53,5 +55,6 @@ public static void main(String args[]) throws Exception {
     private static void config(){
         securityConfig.config();
         filterConfig.config();
+        databaseConfig.config();
     }
 }

src/main/java/config/AppConfig.java

@@ -1,6 +1,8 @@
 package config;
 
 import app.handler.*;
+import database.ConnectionManager;
+import database.H2DbManager;
 import exception.ExceptionHandlerMapping;
 import exception.handler.ErrorExceptionHandler;
 import exception.handler.ServiceExceptionHandler;
@@ -94,6 +96,7 @@ public List<WebHandler> webHandlerList() {
         );
     }
 
+    // ===== Handler =====
     public StaticContentHandler staticContentHandler() {
         return getOrCreate(
                 "staticContentHandler",
@@ -298,4 +301,21 @@ public SessionStorage sessionStorage() {
         return getOrCreate("sessionStorage",
                 SessionStorage::new);
     }
+
+    /**
+     * ===== DB =====
+     */
+    public ConnectionManager connectionManager(){
+        return h2DbManager();
+    }
+
+    public H2DbManager h2DbManager(){
+        return getOrCreate(H2DbManager.class.getSimpleName(), H2DbManager::new);
+    }
+
+    public DdlGenerator ddlGenerator(){
+        return getOrCreate(DdlGenerator.class.getSimpleName(), () ->
+                new DdlGenerator(connectionManager()));
+    }
 }
+

src/main/java/config/DatabaseConfig.java

@@ -0,0 +1,28 @@
+package config;
+
+import app.model.User;
+import java.util.List;
+
+public class DatabaseConfig {
+    private final AppConfig appConfig = new AppConfig();
+
+    public static final List<Class<?>> ENTITY_CLASSES = List.of(
+            User.class
+    );
+
+    public static final List<String> RESOLVED_WORD = List.of(
+            "user"
+    );
+
+    public static final String H2_DB_URL = "jdbc:h2:tcp://localhost//Users/apple/h2/testdb";
+    public static final String H2_DB_USER = "sa";
+    public static final String H2_DB_PASSWORD = "";
+
+    public static final boolean CREATE_TABLES = false;
+    public static final boolean DROP_IF_EXISTS = false;
+
+    public void config(){
+        DdlGenerator ddlGenerator = appConfig.ddlGenerator();
+        if(CREATE_TABLES) ddlGenerator.generateTables();
+    }
+}

src/main/java/config/DdlGenerator.java

@@ -0,0 +1,111 @@
+package config;
+
+import database.ConnectionManager;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.lang.reflect.Field;
+import java.lang.reflect.Modifier;
+import java.sql.Connection;
+import java.sql.SQLException;
+import java.sql.Statement;
+import java.time.LocalDateTime;
+import java.util.List;
+
+public class DdlGenerator {
+    private static final Logger log = LoggerFactory.getLogger(DdlGenerator.class);
+    private final List<String> resolvedWord = DatabaseConfig.RESOLVED_WORD;
+    private final ConnectionManager connectionManager;
+    private final boolean dropIfExists = DatabaseConfig.DROP_IF_EXISTS;
+    private final List<Class<?>> entityClasses = DatabaseConfig.ENTITY_CLASSES;
+
+    public DdlGenerator(ConnectionManager connectionManager) {
+        this.connectionManager = connectionManager;
+    }
+
+    public void generateTables() {
+        try (Connection conn = connectionManager.getConnection();
+             Statement stmt = conn.createStatement()) {
+
+            for (Class<?> entityClass : entityClasses) {
+                String tableName = toTableName(entityClass);
+                String ddl = buildDdlForEntity(entityClass, tableName);
+                log.info("DDL for {}:\n{}", tableName, ddl);
+                stmt.execute(ddl);
+            }
+
+        } catch (SQLException e) {
+            throw new RuntimeException("DDL 생성/실행 중 오류", e);
+        }
+    }
+
+    private String buildDdlForEntity(Class<?> entityClass, String tableName) {
+        StringBuilder ddl = new StringBuilder();
+
+        if (dropIfExists) {
+            ddl.append("DROP TABLE IF EXISTS ").append(tableName).append(";\n");
+        }
+
+        ddl.append("CREATE TABLE ").append(tableName).append(" (\n");
+
+        Field[] fields = entityClass.getDeclaredFields();
+
+        boolean firstColumn = true;
+
+        for (Field field : fields) {
+            if (Modifier.isStatic(field.getModifiers())) {
+                continue;
+            }
+
+            String columnName = field.getName();
+            Class<?> fieldType = field.getType();
+
+            if (!firstColumn) {
+                ddl.append(",\n");
+            }
+            firstColumn = false;
+
+            if ("id".equals(columnName) && (fieldType == Long.class || fieldType == long.class)) {
+                ddl.append("    id BIGINT AUTO_INCREMENT PRIMARY KEY");
+            } else {
+                String sqlType = toSqlType(fieldType);
+                ddl.append("    ").append(columnName).append(" ").append(sqlType);
+            }
+        }
+
+        ddl.append("\n);");
+
+        return ddl.toString();
+    }
+
+    private String toSqlType(Class<?> fieldType) {
+        if (fieldType == Long.class || fieldType == long.class) {
+            return "BIGINT";
+        } else if (fieldType == Integer.class || fieldType == int.class) {
+            return "INT";
+        } else if (fieldType == String.class) {
+            return "VARCHAR(255)";
+        } else if (fieldType == Boolean.class || fieldType == boolean.class) {
+            return "BOOLEAN";
+        } else if (fieldType == LocalDateTime.class ||
+                fieldType == java.util.Date.class ||
+                fieldType == java.sql.Timestamp.class) {
+            return "TIMESTAMP";
+        } else if (fieldType == Double.class || fieldType == double.class) {
+            return "DOUBLE";
+        } else if (fieldType == Float.class || fieldType == float.class) {
+            return "FLOAT";
+        }
+        return "VARCHAR(255)";
+    }
+
+    private String toTableName(Class<?> clazz) {
+        String name = clazz.getSimpleName().toLowerCase();
+
+        for(String word : resolvedWord){
+            if(word.equals(name))
+                return "`" + word + "`";
+        }
+        return name;
+    }
+}

src/main/java/database/ConnectionManager.java

@@ -0,0 +1,7 @@
+package database;
+
+import java.sql.Connection;
+
+public interface ConnectionManager {
+    Connection getConnection();
+}