6. SpringBoot 입문 - Spring DB 접근 기술 - 순수 JDBC

2021. 5. 26. 15:31WEB Dev./Spring Boot 입문

  1. H2 데이터베이스 설치
  2. 순수 JDBC
  3. Spring 통합 테스트
  4. Spring JdbcTemplate
  5. JPA
  6. Spring Data JPA (JPA를 더 편리하게)

2. 순수 JDBC

애플리케이션에서 데이터베이스와 연동하여 CRUD 하는 내용을 다뤄보겠습니다.

순수 JDBC는 오래전에 개발자들이 사용했던 방법입니다. 

 

build.gradle 파일에 jdbc, h2 데이터베이스 관련 라이브러리를 추가해줍니다.

implementation 'org.springframework.boot:spring-boot-starter-jdbc'
runtimeOnly 'com.h2database:h2'

Java는 기본적으로 DB와 연동하려면 JDBC Driver가 꼭 있어야 합니다. 라이브러리를 추가하면 오른쪽 상단에 gradle 아이콘 버튼이 활성화되는데 클릭하셔서 import 과정을 진행해주시면 됩니다.

 

Spring Boot 데이터베이스 연결 설정 추가 (resources/application.properties)

spring.datasource.url=jdbc:h2:tcp://localhost/~/DB파일명
spring.datasource.driver-class-name=org.h2.Driver
spring.datasource.username=sa

 

이제 JDBC API로 개발해봅시다.

기존에는 MemoryMemberRepository를 사용했습니다. 이제 데이터베이스를 추가했기 때문에 데이터베이스에 필요한 구현체가 필요합니다. 

repository/JdbcMemberRepository

package hello.hellospring.repository;

import hello.hellospring.domain.Member;
import org.springframework.jdbc.datasource.DataSourceUtils;

import javax.sql.DataSource;
import java.sql.*;
import java.util.ArrayList;
import java.util.List;
import java.util.Optional;

public class JdbcMemberRepository implements MemberRepository {

    private final DataSource dataSource;
    
    public JdbcMemberRepository(DataSource dataSource) {
        this.dataSource = dataSource;
    }

    @Override
    public Member save(Member member) {
        String sql = "insert into member(name) values(?)";
       
        Connection conn = null;
        PreparedStatement pstmt = null;
        ResultSet rs = null;

        try {
            conn = getConnection();
            pstmt = conn.prepareStatement(sql, Statement.RETURN_GENERATED_KEYS);
            
            pstmt.setString(1, member.getName());
            
            pstmt.executeUpdate();
            rs = pstmt.getGeneratedKeys();

            if (rs.next()) {
                member.setId(rs.getLong(1));
            } 
            else {
                throw new SQLException("id 조회 실패");
            }
            return member;
        } catch (Exception e) {
            throw new IllegalStateException(e);
        } finally {
            close(conn, pstmt, rs);
        }
    }
    
    @Override
    public Optional<Member> findById(Long id) {
        String sql = "select * from member where id = ?";

        Connection conn = null;
        PreparedStatement pstmt = null;
        ResultSet rs = null;

        try {
            conn = getConnection();
            pstmt = conn.prepareStatement(sql); pstmt.setLong(1, id);
            rs = pstmt.executeQuery();

            if(rs.next()) {
                Member member = new Member(); 
                member.setId(rs.getLong("id")); 
                member.setName(rs.getString("name")); 
                return Optional.of(member);
            } else {
                return Optional.empty();
            }
        } catch (Exception e) {
            throw new IllegalStateException(e);
        } finally {
            close(conn, pstmt, rs);
        } 
   }

    @Override
    public List<Member> findAll() {
        String sql = "select * from member";

        Connection conn = null;
        PreparedStatement pstmt = null;
        ResultSet rs = null;

        try {
            conn = getConnection();
            pstmt = conn.prepareStatement(sql);
            rs = pstmt.executeQuery();

            List<Member> members = new ArrayList<>(); 
            
            while(rs.next()) {
                Member member = new Member();
                member.setId(rs.getLong("id"));
                member.setName(rs.getString("name"));
                members.add(member);
            }
            return members;
        } catch (Exception e) {
            throw new IllegalStateException(e);
        } finally {
            close(conn, pstmt, rs);
        }
    }

    @Override
    public Optional<Member> findByName(String name) {
        String sql = "select * from member where name = ?";
        Connection conn = null;
        PreparedStatement pstmt = null;
        ResultSet rs = null;

        try {
            conn = getConnection();
            pstmt = conn.prepareStatement(sql); 
            pstmt.setString(1, name);
            rs = pstmt.executeQuery();

            if(rs.next()) {
                Member member = new Member();
                member.setId(rs.getLong("id"));
                member.setName(rs.getString("name"));
                return Optional.of(member);
            }

            return Optional.empty();
        } catch (Exception e) {
            throw new IllegalStateException(e);
        } finally {
            close(conn, pstmt, rs);
        }
}

    private Connection getConnection() {
        return DataSourceUtils.getConnection(dataSource);
    }
    
    private void close(Connection conn, PreparedStatement pstmt, ResultSet rs) {
        try {
            if (rs != null) {
                rs.close();
            }
        } catch (SQLException e) { 
            e.printStackTrace();
        } 
        
        try {
            if (pstmt != null) { 
                pstmt.close();
            }
        } catch (SQLException e) {
            e.printStackTrace();
        }
        
        try {
            if (conn != null) {
                close(conn);
            }
        } catch (SQLException e) { 
            e.printStackTrace();
        }
    }
    
    private void close(Connection conn) throws SQLException { 
        DataSourceUtils.releaseConnection(conn, dataSource);
    } 
}

DataSource = DB에 연동하려면 DataSource가 필요합니다. 그리고 Spring한테 주입받아야 하는데, Spring Boot는 application.properties를 보고 DataSource를 만들어 놓습니다. 이 DataSource를 JdbcMemberRepository에 주입합니다. (dataSource.getConnection();)

RETURN_GENERATED_KEYS = DB에 id를 자동으로 업데이트할 때 사용됩니다.

pstmt.excuteUpdate(); = 작성된 쿼리가 DB로 전달됩니다.

rs = pstmt.getGeneratedKeys(); = 방금 생성된 id 값을 반환해줍니다.

 

이제 이것을 configuration 해줘야 합니다. SpringConfig 파일로 이동하셔서 확인하시면 MemoryMemberRepository가 Bean으로 등록되어 있는 것을 확인하실 수 있습니다.

이것을 JdbcMemberRepository로 교체해줍니다. DataSource가 파라미터로 필요하기 때문에 @Autowired DataSource dataSource로 호출합니다.

JdbcMemberRepository만 추가해서 구현체만 Config에서 바꿔줬을 뿐 어떤 코드도 수정하지 않았습니다.

 

객체지향 설계가 좋은 이유는 결국은 다형성을 활용할 수 있기 때문입니다. Spring은 이런 작업이 편리할 수 있도록 Spring Container가 지원해줍니다. 

이러한 방식을 개방-폐쇄 원칙(OCP, Open-Closed Principle)이라고 합니다. 즉, 확장에는 열려있고, 수정에는 닫혀있음을 의미합니다.