스프링 DB 접근 기술

(인프런) 김영한 님의 스프링입문 - 코드로 배우는 스프링부트,웹 MVC, DB접근기술  강의를 보고 리뷰한 글입니다. 

 

<이번 섹션 목차> <이번 포스팅 목차>

1. H2 데이터베이스 설치 
2. 순수 Jdbc
3. 스프링 통합 테스트 
4. 스프링 JdbcTemplate
5. JPA
6. 스프링 데이터 JPA 

H2 데이터베이스 설치 

http://h2database.com/html/main.html

H2 Database Engine

이 사이트에서 H2 설치가 가능하다. 설치를 한 후 db를 실행시켜준다.

//h2 db 실행시키는 명령어들 
$ cd h2
$ cd bin
$ chmod 755 h2.sh
$ ./h2.sh

파일로 접근하게 되면, 웹 애플리케이션과  웹 콘솔이 같이 접근될 수 없는 문제점이 발생할 수 있음. 이를 방지하기 위해, 

JDBC URL에 jdbc:h2:tcp://localhost/~/test을 적어, 소켓을 통해 접속하는 방식을 활용한다.

drop table if exists member CASCADE;
create table member
(
//id 값을 세팅하지 않고 insert하면 db가 값을 자동으로 세팅해줌. 
 id bigint generated by default as identity,
 name varchar(255),
 primary key(id)
);

이 코드를 치고 실행시키면 MEMBER라는 db가 생성된다.

member 만드는 코드

 

 

$SELECT *FROM MEMBER 을 치면 table 형태의 member list를 볼 수 있다. 

디렉토리를 생성해서 코드를 적어놓고 관리한다. 

 

순수 Jdbc

//build.gradle의 dependencies 에 코드 추가

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

//application.properties에 추가 

spring.datasource.url = jdbc:h2:tcp://localhost/~/test
spring.datasource.driver-class-name=org.h2.Driver

spring이 필요한 db랑 연결을 다 해줌. 

 

// JdbcMemberRepository.java

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

 

//SpringConfig.java

package hello.hellospring.service;

import hello.hellospring.repository.JdbcMemberRepository;
import hello.hellospring.repository.MemberRepository;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.sql.DataSource;

@Configuration
public class SpringConfig {
    private DataSource dataSource;

    @Autowired
    public SpringConfig(DataSource dataSource){
        this.dataSource = dataSource;
    }
    @Bean
    public MemberService memberService(){
        return new MemberService(memberRepository());
    }
    @Bean
    public MemberRepository memberRepository(){
        //spring 역할을 하는
        return new JdbcMemberRepository();
    }
}

localhost:8080에서 "spring2"의 이름으로 회원가입을 하고나면 db에 저장되는 것을 확인할 수 있다. 

원래는 메모리를 이용한 MemoryMemberRepository를 데이터 저장소로 사용했었지만, 

 MemoryMemberRepository 스프링빈을 내리고 JdbcMemberRepository를 사용하면 나머지는 수정 안해도 사용이 가능하다. 

 

➜OCP(= Open-Closed Principle) 개방 폐쇄 원칙

: 확장에는 열려있고, 변경 및 수정에는 닫혀있다. 

 

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

 

서버를 닫은 후 재실행해도 db에 데이터가 담겨있으므로 데이터가 그대로 저장되어 있는 것을 알 수 있다.

 

스프링 통합 테스트 

//MemberServiceIntegrationTest
package hello.hellospring.service;

import hello.hellospring.domain.Member;
import hello.hellospring.repository.MemberRepository;
import org.assertj.core.api.Assertions;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;

import static org.junit.jupiter.api.Assertions.assertThrows;

//스프링 컨테이너와 테스트를 함께 실행한다는 의미 
@SpringBootTest
@Transactional
class MemberServiceIntegrationTest {

    @Autowired MemberService memberService ;
    @Autowired MemberRepository memberRepository ;

    @Test
    void 회원가입() {
        //given
        Member member = new Member();
        member.setName("hello");
        //when
        Long saveId = memberService.join(member);
        //then
        Member findMember = memberService.findOne(saveId).get();
        Assertions.assertThat(member.getName()).isEqualTo(findMember.getName());
    }

    @Test
    public void 중복_회원_예외() {
        //given
        Member member1 = new Member();
        member1.setName("spring");

        Member member2 = new Member();
        member2.setName("spring");
        //when
        memberService.join(member1);
        IllegalStateException e = assertThrows(IllegalStateException.class, () ->memberService.join(member2));
        Assertions.assertThat(e.getMessage()).isEqualTo("이미 존재하는 회원입니다.");

        //then

    }

}

db초기화 후 test를 하면 회원가입된 정보가 db에 잘 저장되는 것을 알 수 있다. 

이전 코드에서 보였던 @AfterEach를 통해 데이터를 삭제하는 과정은

@Transactional 어노테이션을 이용해서 동일한 기능을 구현할 수 있다.

@Transactional 어노테이션 
Test에 추가하면 테스트를 실행할 때 transactional을 먼저 실행하고, 그 다음 db에 데이터를 insert query하고(= 데이터를 다 저장함) 테스트가 끝나면 rollback을 해줌. 즉, db에 넣었던 데이터가 깔끔하게 지워져서 다음 테스트에 영향을 주지 않는다. 
@Transactional 어노테이션을 쓰지 않는 경우, db에 넣었던 데이터를 계속 직접 삭제 해줘야 한다는 불편함이 있다.