애플리케이션과 DB를 연동해서 데이터 입출력이 가능하도록 설정을 해보겠다.
단, 지금 하는 방식은 20년 전의 방식으로 요즘에는 잘 안쓰이는 방식이니 가볍게 읽어보자.
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
여기까지 하면 DB에 접근하기 위한 기초 작업은 다한 것이다.
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{
// DB에 붙으려면 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 rs = null; // 결과를 받는다
try {
// DB 커넥션을 가져온다
conn = getConnection();
// 1. 쿼리(sql)을 넣어준다
// 2. RETURN_GENERATED_KEYS는 id값 얻을 때 사용한다
pstmt = conn.prepareStatement(sql, Statement.RETURN_GENERATED_KEYS);
pstmt.setString(1, member.getName());
pstmt.executeUpdate(); // 이때 DB에 실제 쿼리가 날아간다
rs = pstmt.getGeneratedKeys(); // 1번이면 1번 반환, 2번이면 2번 반환...
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() {
// 스프링에서는 DataSourceUtils에서 커넥션을 얻어야 기존 DB 커넥션이 가능하다
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 통해서 닫아야한다
DataSourceUtils.releaseConnection(conn, dataSource);
}
}
@Configuration
public class SpringConfig {
private final DataSource dataSource;
@Autowired
public SpringConfig(DataSource dataSource) {
this.dataSource = dataSource;
}
@Bean
public MemberService memberService() {
return new MemberService(memberRepository());
}
@Bean
public MemberRepository memberRepository() {
//return new MemoryMemberRepository();
return new JdbcMemberRepository(dataSource); // JdbcMemberRepository를 스프링 빈으로 등록
}
}
여기까지 하고 톰캣 서버를 돌려보면 회원조회가 정상적으로 되며, 회원가입도 할 수 있다.
구현 클래스에 대한 지금 현재 상황을 이미지로 확인하면 다음과 같다.
스프링 설정에 대한 지금 현재 상황을 이미지로 확인하면 다음과 같다.