스프링 DB 접근 기술 - 순수 JDBC

2022. 7. 31. 14:002022/Spring

build.gradle 파일에 jdbc, h2 데이터베이스 관련 라이브러리 추가
implementation 'org.springframework.boot:spring-boot-starter-jdbc' - 자바가 DB와 붙기위해서 JDBC 필요
runtimeOnly 'com.h2database:h2' - DB와 붙을 때 DB가 제공하는 클라이언트가 필요
scr>main>resources>application.properties 파일에 DB 접속 정보 추가
spring.datasource.url=jdbc:h2:tcp://localhost/~/test
spring.datasource.driver-class-name=org.h2.Driver
spring.datasource.username=sa 

이렇게 하면 Spring이 DB와 연결하는 작업을 알아서 다 해준다.

 

이제까지 했던것은 구현체를 만들어서 MemoryMemberRepository라는 메모리에 데이터를 저장하는 방식이었다. 

이번에는 JdbcMemberRepository 클래스를 만들어 H2 DB와 자바를 연결하고 db로 데이터를 저장하는 방식으로 사용해본다.

 

package hello.hellospring.repositroy;

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;

import static java.sql.DriverManager.getConnection;

public class JdbcMemberRepository implements MemberRepository{

    //application.properties에서 설정한 것을 Spring이 datasource를 만들어둔다. -> 여기서 datadource를 주입받음
    private final DataSource dataSource;

    public JdbcMemberRepository(DataSource dataSource){
        this.dataSource = dataSource;
//        dataSource.getConnection(); //데이터베이스와 연결 -> 여기에 이제 SQL만 날려서 DB로 전달하면 끝!
    }

    //alt+enter -> Implements methods
    @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()); //insert into member(name) values(?) ? 자리가 1번 위치로 getName() 값이 들어간다.

            pstmt.executeUpdate(); //DB에 실제 쿼리 날라감
            rs = pstmt.getGeneratedKeys(); //DB가 방금 생성한 키 값(ID)을 반환

            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);
        }
    }

    //spring framework를 통해서 데이터 커넥션을 할 때, DataSourceUtils를 통해서 연결
    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);
    }
}

 

해당 클래스를 만들고 난 후에 Configuration 쪽만 변경하면 완료!

JdbcMemberRepository 는 DataSource라는 객체를 받아야 하므로 이를 주입시켜준다. 그리고 나서 repository 반환 구현체를 JdbcMemberRepository로 바꿔주기만 하면된다.

SpringConfig.java

JdbcMemberRepository 클래스 생성, 인터페이스 확장, 스프링이 제공하는 Configration(애플리케이션을 조립하는 인터페이스 코드만 변경)변경 이 세가지만 하면 OK

 

이제 H2에 잘 연결되어 데이터가 저장되는지 확인해보기 위해서는

 

현재 db에 저장된 데이터 확인

이제 웹으로 들어가서 회원가입을 한 이후에 다시 DB에서 조회해본다.

데이터가 새롭게 들어옴

잘 연동되어 데이터가 쌓이는 것을 확인해 볼 수 있다.

 

정리

스프링을 왜 사용하는가? = 객체 지향 설계가 왜 좋은가? = 스프링 컨테이너가 이런 모든 것을 지원해줌!

스프링 DI(의존성 주입)을 사용함으로써 기존코드는 전혀 손대지 않고, 설정만으로 구현 클래스를 변경해보았다. 이처럼 객체지향에서 말하는 다형성을 잘 활용하면 기능을 변경해도, 애플리케이션 전체 코드를 수정하지 않아도 된다.

 

OCP(개발-폐쇄 원칙)
확장에는 열려있고, 수정 및 변경에는 닫혀있다.