[Java Spring 무작정 따라하기] JdbcTemplate

순수 JDBC로 작성했던 코드를 JdbcTemplate으로 변경해 볼 차례다.
먼저 JdbcTemplate 라이브러리를 사용하기 위한 리포지토리를 생성했다.

// repository/JdbcTemplateMemberRepository.java
package hello.hello_spring.repository;

import hello.hello_spring.domain.Member;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.core.RowMapper;
import org.springframework.jdbc.core.namedparam.MapSqlParameterSource;
import org.springframework.jdbc.core.simple.SimpleJdbcInsert;

import javax.sql.DataSource;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Optional;

public class JdbcTemplateMemberRepository implements MemberRepository {

    private final JdbcTemplate jdbcTemplate;

    public JdbcTemplateMemberRepository(DataSource dataSource) {
        this.jdbcTemplate = new JdbcTemplate(dataSource);
    }

    @Override
    public Member save(Member member) {
        SimpleJdbcInsert jdbcInsert = new SimpleJdbcInsert(jdbcTemplate);
        jdbcInsert.withTableName("member").usingGeneratedKeyColumns("id");

        Map<String, Object> parameters = new HashMap<>();
        parameters.put("name", member.getName());

        Number key = jdbcInsert.executeAndReturnKey(new MapSqlParameterSource(parameters));
        member.setId(key.longValue());
        return member;
    }

    @Override
    public Optional<Member> findById(Long id) {
        List<Member> result = jdbcTemplate.query("select * from member where id = ?", memberRowMapper(), id);
        return result.stream().findAny();
    }

    @Override
    public Optional<Member> findByName(String name) {
        List<Member> result = jdbcTemplate.query("select * from member where name = ?", memberRowMapper(), name);
        return result.stream().findAny();
    }

    @Override
    public List<Member> findAll() {
        return jdbcTemplate.query("select * from member", memberRowMapper());
    }

    private RowMapper<Member> memberRowMapper() {
        return (rs, rowNum) -> {
            Member member = new Member();
            member.setId(rs.getLong("id"));
            member.setName(rs.getString("name"));
            return member;
        };
    }
}

기능을 작성할 때 SQL쿼리 결과를 member 객체로 맵핑해 줄 RowMapper 메서드를 생성했다.
대부분의 기능이 쿼리를 날려 받은 결과를 RowMapper로 맵핑하는 과정의 반복이다.

그리고 SpringConfig에서 작성한 JdbcTemplate 리포지토리를 사용하도록 수정했다.

// hello.hello_spring/SpringConfig.java
import hello.hello_spring.repository.JdbcTemplateMemberRepository;

@Configuration
public class SpringConfig {
    private final DataSource dataSource;

    @Bean
    public MemberRepository memberRepository(){
//        return new JdbcMemberRepository(dataSource);
        return new JdbcTemplateMemberRepository(dataSource);
    }
}

테스트는 코드가 잘 작성되었다면 그대로 H2 DB와 연결될테니 이전에 작성했던 테스트 케이스를 그대로 사용했다.

테스트가 문제 없이 끝난 것을 확인할 수 있었다.