오늘이라도

[VI. 스프링 DB 접근 기술] 19. 순수 JDBC 본문

인프런/스프링 입문 (김영한)

[VI. 스프링 DB 접근 기술] 19. 순수 JDBC

upcake_ 2022. 1. 10. 20:48
반응형

https://github.com/upcake/hello-spring

강의 링크

 

* 오늘의 단축키 & 기능


1. 환경 설정

- build.gradle 파일에 jdbc, h2 데이터베이스 관련 라이브러리 추가

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

- 스프링 부트 데이터베이스 연결 설정 추가

- application.properties

spring.datasource.url=jdbc:h2:tcp://localhost/D:/workspace_ij/db/test
spring.datasource.driver-class-name=org.h2.Driver
spring.datasource.username=sa
spring.datasource.password=

2. JDBC 리포지토리 구현

- 이렇게 JDBC API로 직접 코딩하는 것은 15년 전 이야기라고 한다.

- JdbcMemberRepository.java

더보기
public class JdbcMemberRepository implements MemberRepository {

    //DB에 붙으려면 DataSource가 필요함
    //스프링을 통해 DataSource를 주입받는다.
    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은 결과를 받는 곳
        ResultSet rs = null;

        try {
            conn = getConnection();
            //pstmt 설정
            pstmt = conn.prepareStatement(sql, Statement.RETURN_GENERATED_KEYS);

            pstmt.setString(1, member.getName());

            pstmt.executeUpdate(); //이 단계에서 DB에 실제 쿼리가 날라감
            rs = pstmt.getGeneratedKeys(); //생성된 key를 반환받음

            //rs에서 값이 있으면,
            if (rs.next()) {
                //값을 꺼내서 셋팅
                member.setId(rs.getLong(1));
            } else {
                throw new SQLException("id 조회 실패");
            }
            return member;
        } catch (Exception e) {
            throw new IllegalStateException(e);
        } finally {
            //사용한 자원 release
            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() {
        //스프링을 통해서 dataSource를 받을때는 DataSourceUtils를 통해서 받아야 한다.
        //같은 DataSource를 유지시켜 주는 역할을 한다.
        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);
    }
}

 

3. Spring 설정

- SpringConfig.java

    private DataSource dataSource;

    @Autowired
    public SpringConfig(DataSource dataSource) {
        this.dataSource = dataSource;
    }
    
    //중간 생략
    
    @Bean
    public MemberRepository memberRepository() {
        //interface는 new가 안되므로 구현체를 return 해준다.
        //return new MemoryMemberRepository();
        return new JdbcMemberRepository(dataSource);
    }

- DataSource는 데이터베이스 커넥션을 획득할 때 사용하는 객체다. 스프링 부트는 데이터베이스 커넥션
정보를 바탕으로 DataSource를 생성하고 스프링 빈으로 만들어둔다. 그래서 DI를 받을 수 있다.

4. 테스트

- 회원가입, 조회가 정상적으로 작동하는 것을 확인할 수 있다.

 

 

- 개방-폐쇄 원칙 (OCP, Open-Closed Principle) : 확장에는 열려있고, 수정, 변경에는 닫혀있다.

- 스프링의 DI (Dependencies Injection)을 사용하면 기존 코드를 전혀 손대지 않고, 설정만으로 구현 클래스를 변경할 수 있다.

- 데이터를 DB에 저장하므로 스프링 서버를 다시 실행해도 데이터가 안전하게 저장된다.

반응형