[4단계 - DB 적용] 이유빈 미션 제출합니다

src/main/java/chess/dao/ChessGameDao.java

@@ -0,0 +1,116 @@
+package chess.dao;
+
+import chess.board.Board;
+import chess.utils.DatabaseUtil;
+import chess.piece.Team;
+import java.sql.Connection;
+import java.sql.PreparedStatement;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+
+public class ChessGameDao {
+    public void createNewRoom() {
+        String query = "INSERT INTO game (boardState, turn, status) VALUES (?, ?, ?)";
+        Board board = new Board();
+
+        try (Connection conn = DatabaseUtil.getConnection(); PreparedStatement pstmt = conn.prepareStatement(query)) {
+            pstmt.setString(1, board.convertBoardStateToString());
+            pstmt.setBoolean(2, true);
+            pstmt.setString(3, "ready");
+            pstmt.executeUpdate();
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+    }
+
+    public int selectRoom(int gameId) {
+        String query = "SELECT status FROM game WHERE gameId = ?";
+
+        try (Connection conn = DatabaseUtil.getConnection();
+             PreparedStatement pstmt = conn.prepareStatement(query)) {
+            pstmt.setInt(1, gameId); // gameId를 쿼리에 설정
+            try (ResultSet rs = pstmt.executeQuery()) {
+                if (rs.next()) {
+                    return gameId;
+                }
+            }
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+        throw new IllegalStateException("존재하지 않는 방 번호입니다.");
+    }
+
+    public void saveGame(int gameId, String boardState, boolean turn, String status) {
+        String query = "INSERT INTO game (gameId, boardState, turn, status) VALUES (?, ?, ?, ?) " + "ON DUPLICATE KEY UPDATE boardState = VALUES(boardState), turn = VALUES(turn), status = VALUES(status)";
+
+        try (Connection conn = DatabaseUtil.getConnection(); PreparedStatement pstmt = conn.prepareStatement(query)) {
+            pstmt.setInt(1, gameId);
+            pstmt.setString(2, boardState);
+            pstmt.setBoolean(3, turn);
+            pstmt.setString(4, status);
+            pstmt.executeUpdate();
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+    }
+
+    public void updateCurrentBoard(int gameId, Board board) {
+        String query = "SELECT boardState, status FROM game WHERE gameId = ?";
+        try (Connection conn = DatabaseUtil.getConnection(); PreparedStatement pstmt = conn.prepareStatement(query)) {
+            pstmt.setInt(1, gameId);
+            try (ResultSet rs = pstmt.executeQuery()) {
+                if (rs.next()) {
+                    String boardState = rs.getString("boardState");
+                    board.loadBoardState(boardState);
+                }
+            }
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+    }
+
+    public Team getCurrentTurn(int gameId, Team turn) {
+        String query = "SELECT turn FROM game WHERE gameID = ?";
+        try (Connection conn = DatabaseUtil.getConnection();
+             PreparedStatement pstmt = conn.prepareStatement(query)) {
+            pstmt.setInt(1, gameId);
+            try (ResultSet rs = pstmt.executeQuery()) {
+                if (rs.next()) {
+                    return turn.getTurnByBinary(rs.getInt("turn"));
+                }
+            }
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+        return turn;
+    }
+
+    public String getCurrentStatus(int gameId) {
+        String query = "SELECT status FROM game WHERE gameId = ?";
+
+        try (Connection conn = DatabaseUtil.getConnection();
+             PreparedStatement pstmt = conn.prepareStatement(query)) {
+            pstmt.setInt(1, gameId);
+            try (ResultSet rs = pstmt.executeQuery()) {
+                if (rs.next()) {
+                    return rs.getString("status");
+                }
+            }
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+        throw new IllegalStateException("존재하지 않는 방입니다.");
+    }
+
+    public void changeStatus(int gameId) {
+        String query = "UPDATE game SET status = ? WHERE gameId = ?";
+        try (Connection conn = DatabaseUtil.getConnection();
+             PreparedStatement pstmt = conn.prepareStatement(query)) {
+            pstmt.setString(1, "end");
+            pstmt.setInt(2, gameId);
+            pstmt.executeUpdate();
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+    }
+}

src/main/java/chess/ChessGameManager.java → src/main/java/chess/game/ChessGameManager.java

@@ -1,6 +1,7 @@
-package chess;
+package chess.game;
 
 import chess.board.Board;
+import chess.dao.ChessGameDao;
 import chess.piece.Piece;
 import chess.ui.InputView;
 import chess.ui.OutputView;
@@ -10,29 +11,56 @@
 public class ChessGameManager {
     private static final String START_COMMAND = "start";
     private static final String STATUS_COMMAND = "status";
+    private static final String END_COMMAND = "end";
+    private static final String CREATE_NEW_ROOM_COMMAND = "create";
     private Team turn = Team.WHITE;
     private boolean isGameOver = false;
+    private ChessGameDao chessGameDao;
+    private int gameId;
+
+    public ChessGameManager() {
+        this.chessGameDao = new ChessGameDao();
+    }
 
     public void startNewGame() {
         OutputView.printStartMessage();
-        if (InputView.inputCommand().equals(START_COMMAND)) {
+        String command = InputView.inputCommand();
+        if (command.equals(START_COMMAND)) {
             Board board = new Board();
+            selectRoom();
+            loadGame(board);
             OutputView.printBoard(board);
             proceedGame(board);
         }
+        if (command.equals(CREATE_NEW_ROOM_COMMAND)) {
+            createNewRoom();
+            startNewGame();
+        }
+    }
+
+    public void createNewRoom() {
+        OutputView.printCreateRoomMessage();
+        chessGameDao.createNewRoom();
     }
 
     public void proceedGame(Board board) {
         if (isGameOver) {
+            changeStatus(gameId);
             return;
         }
         String command = InputView.inputCommand();
+        if (command.equals(END_COMMAND)) {
+            saveGame(gameId, board);
+            return;
+        }
         if (!command.equals(STATUS_COMMAND)) {
             moveProcess(board, command);
+            saveGame(gameId, board);
             proceedGame(board);
         }
         if (command.equals(STATUS_COMMAND)) {
             OutputView.printEndGameByStatusMessage(board);
+            changeStatus(gameId);
         }
     }
 
@@ -57,6 +85,30 @@ public void moveProcess(Board board, String command) {
         turn = turn.changeTurn();
     }
 
+    public void selectRoom() {
+        OutputView.printSelectRoomMessage();
+        int roomNumber = Integer.parseInt(InputView.inputCommand());
+        if (chessGameDao.getCurrentStatus(roomNumber).equals(END_COMMAND)) {
+            throw new IllegalStateException("종료된 게임입니다.");
+        }
+        OutputView.printEnterRoomMessage(chessGameDao.selectRoom(roomNumber));
+        gameId = roomNumber;
+    }
+
+    public void saveGame(int gameId, Board board) {
+        String status = "playing";
+        chessGameDao.saveGame(gameId, board.convertBoardStateToString(), getTurnState(), status);
+    }
+
+    public void loadGame(Board board) {
+        chessGameDao.updateCurrentBoard(gameId, board);
+        turn = chessGameDao.getCurrentTurn(gameId, turn);
+    }
+
+    public void changeStatus(int gameId) {
+        chessGameDao.changeStatus(gameId);
+    }
+
     public Position searchPosition(String location) {
         int file = location.charAt(0) - 'a' + 1;
         int rank = location.charAt(1) - '0';
@@ -69,4 +121,8 @@ public void verifyTurn(Piece piece) {
             throw new IllegalStateException("상대방 기물을 이동시킬 수 없습니다.");
         }
     }
+
+    private boolean getTurnState() {
+        return turn.isSameTeam(Team.WHITE);
+    }
 }

src/main/java/chess/piece/Team.java

@@ -15,4 +15,9 @@ public Team changeTurn() {
         }
         return BLACK;
     }
+
+    public Team getTurnByBinary(int n) {
+        if (n == 0) return Team.BLACK;
+        return Team.WHITE;
+    }
 }