[6/27 TIL] SPRING(JDBC)

🍃프로그래머스 백엔드 데브코스 4기 교육과정을 듣고 정리한 글입니다.🍃

JDBC(Java Database Connectivity)

그림 출처

• 자바 애플리케이션과 데이터베이스 간의 연결과 데이터베이스 작업을 수행하기 위한 자바 API

JDBC Flow

그림 출처
1. DriverManager를 통해서 컨넥션 객체를 받아옴
2. Connection을 통해서 Statement를 가져옴
3. Statement를 통해서 쿼리를 실행해서 ResultSet을 가져오거나 update를 실행
4. 데이터베이스 반드시 컨넥션을 종료

예시 코드

public class JDBCCrudExample {

    // JDBC 연결 정보
    private static final String JDBC_URL = "jdbc:mysql://localhost:3306/sampledb";
    private static final String JDBC_USERNAME = "username";
    private static final String JDBC_PASSWORD = "password";

    public static void main(String[] args) {
        // JDBC 드라이버 로드
        try {
            Class.forName("com.mysql.cj.jdbc.Driver");
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
            return;
        }

        // 데이터베이스 연결
        try (Connection connection = DriverManager.getConnection(JDBC_URL, JDBC_USERNAME, JDBC_PASSWORD)) {
            // 데이터 삽입
            UUID id = UUID.randomUUID();
            String name = "유명한";
            int age = 25;
            insertData(connection, id, name, age);

            // 데이터 검색
            System.out.println("All Data:");
            retrieveData(connection);

            // 단일 행 검색
            System.out.println("Single Row Data:");
            UUID searchId = UUID.fromString("your-id");
            UserData userData = retrieveDataById(connection, searchId);
            if (userData != null) {
                System.out.println("Data Found:");
                System.out.println("ID: " + userData.getId() + ", Name: " + userData.getName() + ", Age: " + userData.getAge());
            } else {
                System.out.println("Data Not Found");
            }

            // 데이터 갱신
            String newName = "김명한";
            int newAge = 30;
            updateData(connection, id, newName, newAge);

            // 데이터 삭제
            deleteData(connection, id);

            // 최종 데이터 검색
            System.out.println("Updated Data:");
            retrieveData(connection);
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }

    // 데이터 삽입
    private static void insertData(Connection connection, UUID id, String name, int age) throws SQLException {
        String sql = "INSERT INTO users (id, name, age) VALUES (?, ?, ?)";
        try (PreparedStatement statement = connection.prepareStatement(sql)) {
            statement.setBytes(1, toBytes(id));
            statement.setString(2, name);
            statement.setInt(3, age);
            statement.executeUpdate();
            System.out.println("Data inserted successfully.");
        }
    }

    // 데이터 검색
    private static void retrieveData(Connection connection) throws SQLException {
        String sql = "SELECT * FROM users";
        try (Statement statement = connection.createStatement();
             ResultSet resultSet = statement.executeQuery(sql)) {
            while (resultSet.next()) {
                UUID id = toUUID(resultSet.getBytes("id"));
                String name = resultSet.getString("name");
                int age = resultSet.getInt("age");
                System.out.println("ID: " + id + ", Name: " + name + ", Age: " + age);
            }
        }
    }
    
    // 단일 행 검색
    private static UserData retrieveDataById(Connection connection, UUID id) throws SQLException {
        String sql = "SELECT * FROM users WHERE id = ?";
        try (PreparedStatement statement = connection.prepareStatement(sql)) {
            statement.setBytes(1, toBytes(id));
            try (ResultSet resultSet = statement.executeQuery()) {
                if (resultSet.next()) {
                    UUID userId = toUUID(resultSet.getBytes("id"));
                    String name = resultSet.getString("name");
                    int age = resultSet.getInt("age");
                    return new UserData(userId, name, age);
                }
            }
        }
        return null;
    }

    // 데이터 갱신
    private static void updateData(Connection connection, UUID id, String name, int age) throws SQLException {
        String sql = "UPDATE users SET name=?, age=? WHERE id=?";
        try (PreparedStatement statement = connection.prepareStatement(sql)) {
            statement.setString(1, name);
            statement.setInt(2, age);
            statement.setBytes(3, toBytes(id));
            statement.executeUpdate();
            System.out.println("Data updated successfully.");
        }
    }

    // 데이터 삭제
    private static void deleteData(Connection connection, UUID id) throws SQLException {
        String sql = "DELETE FROM users WHERE id=?";
        try (PreparedStatement statement = connection.prepareStatement(sql)) {
            statement.setBytes(1, toBytes(id));
            statement.executeUpdate();
            System.out.println("Data deleted successfully.");
        }
    }

    // byte[]을 UUID로 변환
    private static UUID toUUID(byte[] bytes) {
        ByteBuffer byteBuffer = ByteBuffer.wrap(bytes);
        return new UUID(byteBuffer.getLong(), byteBuffer.getLong());
    }

    // UUID를 byte[]로 변환
    private static byte[] toBytes(UUID uuid) {
        ByteBuffer byteBuffer = ByteBuffer.allocate(16);
        byteBuffer.putLong(uuid.getMostSignificantBits());
        byteBuffer.putLong(uuid.getLeastSignificantBits());
        return byteBuffer.array();
    }
}

• sql 인젝션 방지 및 성능 향상(캐시 사용)을 위해 Statement 대신, PreparedStatement 사용
• UUID를 긴 문자열로 저장 하기보단, byte 배열로 저장함으로써 데이터베이스의 성능과 공간 효율성을 향상시킴