[스프링 DB 1편] #1. JDBC의 이해 (2) - 등록/조회/수정/삭제

[Pracrobo]

JDBC 개발 - 등록

JDBC를 사용해서 회원 데이터를 데이터베이스에 관리하는 기능을 개발해본다.

주의 ~

H2데이터베이스 설정 마지막에 있는 테이블과 샘플 데이터 만들기를 통해서 member테이블을 미리 만들어두어야 한다.

DROP TABLE member IF EXIST S CASCADE;

CREATE TABLE member (
	member_id varchar(10),
	money integer not null default 0,
	primary key(member_id)
);

package hello.jdbc.domain;

import lombok.Data;

@Data
public class Member {
	private String memberId;
	private int money;
	
	public Member() {
		
		
	}
	
	public Member(String memberId, int money) {
		this.memberId = memberId;
		this.money = money;
	}
}

package : repository

MemberRepositoryV0

package hello.jdbc.repository;

import lombok.extern.slf4j.Slf4j;

/**
* JDBC = DriverManager 사용
**/
@Slf4j
public class MemberRepositoryV0 {

	public Member save(Member member) throws SQLException {
		String sql = "insert into member(member_id, money) values (?, ?);"
		Connection con = null;
		PreparedStatement pstmt = null;
		
		try {
			con = getConnection();
			pstmt = con.prepareStatement(sql);
			pstmt.setString(1, member.getMemberId());
			pstmt.setInt(2, member.getMoney());
			return member;
		}catch (SQLExceeption e){
			//checked Exception
			log.error("db error", e);
			//e.printStackTrace();
			throw e; 
		}finally {
			//시작과 역순으로 닫는다.
//			pstmt.close(); //Exception 되면 con이 닫히지않는 문제가 발생할 수 있다.
//			con.close();
				close(con, pstmt, null);	
		}
	}
	private void close(Connection con, Statement stmt, ResultSet rs) {
		if(rs!= null) {
			try {
				rs.close(); 
			catch (SQLException e) {
				log.info("error", e);
			}
		}

		if(stmt != null) {
			try {
				stmt.close(); //pstmt 는 파라미터를 넘길 수 있는 더 많은 기능을 가지고 있고 stmt를 상속받는다
			catch (SQLException e) {
				log.info("error", e);
			}
		}
		// 이렇게 하면 각자 에러가 터져도 다른 것에 영향을 주지 않기에 닫을 수 있다.
		if(con != null) {
			try {
				con.close();		
			}catch (SQLException e) {
				log.info("error", e);
			}
	}
	
	private Connection getConnection() {
		return DBConnectionUtil.getConnection();	
	}
}

커넥션 획득

• getConnection() : 이전에 만들어둔 DBConnectionUtil 을 통해서 데이터 베이스 커넥션을 획득한다.

save() : SQL 전달

• sql : 데이터 베이스에 전달할 SQL 을 정의한다. 여기서는 데이터를 등록해야 하므로 insert sql을 했다.
• con.prepareStatement(sql) : 데이터 베이스에 전달할 SQL과 파라미터로 전달할 데이터들을 준비한다.
• pstmt.executeUpdate() : Statement 를 통해 준비된 SQL을 커넥션을 통해 실제 데이터 베이스에 전달한다. 참고로 executeUpdate() 는 int를 반환하는데 영향받은 DB row 수를 반환한다. 여기서는 하나의 row를 등록했으므로 1을 반환한다.

executeUpdate()

int executeUpdate() throws SQLException;

리소스 정리

쿼리를 실행하고 나면 리소스를 정리해야 한다. 여기서는 Connection, PreparedStatement 를 사용했다. 리소스를 정리할 때는 항상 역순으로 해야한다. Connection을 먼저 획득하고 Connection을 통해 PreparedStatement 를 만들었기 때문에 리소스를 반환할때는 PreparedStatement를 먼저 종료하고 그 다음, Connection을 종료하면 된다. 여기서는 사용하지 않은 ResultSet은 결과를 조회할때 사용한다.

!! 리소스 정리는 꼭해야한다. 안하면 리소스 누수로 인해 커넥션 부족으로 장애가 발생할 수 있다.

참고
PreparedStatement는 Statement의 자식타입인데, ? 를 통한 파라미터 바인딩을 가능하게 해준다. 참고로 SQL Injection 공격을 예방하려면 PrepareStatement를 통한 파라미터 바인딩 방식을 사용해야 한다.

package hello.jdbc.repository;

import

class MemberRepositoryV0Test {
	MemberRepositoryV0 repository = new MemberRepositoryV0();
	
	@Test
	void crud() {
		Member member = new Member("memberV0", 10000);
		repository.save(member);
	}
}

JDBC 개발 - 조회

package hello.jdbc.repository;

import lombok.extern.slf4j.Slf4j;
// 추가하기
import java.util.NoSuchElementException;

/**
* JDBC = DriverManager 사용
**/
@Slf4j
public class MemberRepositoryV0 {

	public Member save(Member member) throws SQLException {
		String sql = "insert into member(member_id, money) values (?, ?);"
		Connection con = null;
		PreparedStatement pstmt = null;
		
		try {
			con = getConnection();
			pstmt = con.prepareStatement(sql);
			pstmt.setString(1, member.getMemberId());
			pstmt.setInt(2, member.getMoney());
			return member;
		}catch (SQLExceeption e){
			//checked Exception
			log.error("db error", e);
			//e.printStackTrace();
			throw e; 
		}finally {
			//시작과 역순으로 닫는다.
//			pstmt.close(); //Exception 되면 con이 닫히지않는 문제가 발생할 수 있다.
//			con.close();
				close(con, pstmt, null);	
		}
	}
	
	
	// 조회
	public Member findById(String memberId) throws SQLException {
		String sql = "select * from member where member_id = ?";
		
		Connection con = null;
		PreparedStatement pstmt = null;
		ResultSet rs = null;
		
		try {
			con  = getConnection();
			pstmt = con.prepareStatement(sql);
			pstmt.setString(1, memberId);
		
			rs = pstmt.executeQuery();
			if(rs.next()) {
				Member member = new Member();
				member.setMemberId(rs.getString("member_id"));
				member.setMoney(rs.getInt("money"));
				return member;
			}else {
			// rs.next() 값이 없다면
				throw new NoSuchElementException("member not found memberId = " + memberId);
			}		
		}catch (SQLException e) {
			log.error("db error", e);
			throw e;
		}	finally {
			close(con, pstmt, rs);
		}	
	}
	
	private void close(Connection con, Statement stmt, ResultSet rs) {
		if(rs!= null) {
			try {
				rs.close(); 
			catch (SQLException e) {
				log.info("error", e);
			}
		}

		if(stmt != null) {
			try {
				stmt.close(); //pstmt 는 파라미터를 넘길 수 있는 더 많은 기능을 가지고 있고 stmt를 상속받는다
			catch (SQLException e) {
				log.info("error", e);
			}
		}
		// 이렇게 하면 각자 에러가 터져도 다른 것에 영향을 주지 않기에 닫을 수 있다.
		if(con != null) {
			try {
				con.close();		
			}catch (SQLException e) {
				log.info("error", e);
			}
	}
	
	private Connection getConnection() {
		return DBConnectionUtil.getConnection();	
	}
}

• rs = pstmt.executeQuery() : 데이터를 변경할 때는 executeUpdate() 를 사용하지만 데이터를 조회할 때는 executeQuery 를 사용한다. executeQuery() 는 결과를 ResultSet에 담아서 반환한다.

exeuteQuery()

ResultSet executeQuery() throws SQLException;

ResultSet

• ResultSet 은 다음과 같이 생긴 데이터 구조이다. 보통 select 쿼리의 결과가 순서대로 들어간다.
  • 예를 들어서 select member_id, money 라고 저장하면 member_id, money라는 이름으로 데이터가 저장된다.
  • 참고로 select* 를 사용하면 테이블의 모든 컬럼을 다 지정한다.
• ResultSet 내부에 있는 커서(cursor)를 이동해서 다음 데이터를 조회할 수 잇따.
• rs.next() : 이것을 호출하면 커서가 다음으로 이동한다. 참고로 최초의 커서는 데이터를 가리키고 있지 않기 때문에 rs.next()룰 최초 한번은 호출해야 데이터를 조회할 수 있다.
  • rs.next() 의 결과가 true면 커서의 이동 결과 데이터가 있다는 뜻이다.
  • rs.next()의 결과가 false면 더이상 커서가 가리키는 데이터가 없다는 뜻이다.
• rs.getString(”member_id”) : 현재 커서가 가리키고 있는 위치의 member_id 데이터를 String 타입으로 반환한다.
• rs.getInt(”money”) : 현재 커서가 가리키고 있는 위치의 money 데이터를 int 타입으로 반환한다.

ResultSet 결과 예시

• 회원 2명이 조회되는 경우이다.
• 1-1 에서 rs.next()로 호출한다.
• 1-2 의 결과로 cursor 가 다음으로 이동한다. 이 경우 cursor 가 가리키는 데이터가 있으므로 true를 반환한다.
• 2-1에서 rs.next() 를 호출한다.
• 2-2 의 결과로 cursor가 다음으로 이동한다. 이 경우 cursor가 가리키는 데이터가 있으므로 true를 반환한다.
• 3-1 에서 rs.next()를 호출한다.
• 3-2의 결과로 cursor가 다음으로 이동한다. 이 경우 cursor가 가리키는 데이터가 없으므로 false를 반환한다.
• 그래서 데이터가 여러개면 While 문 을 써준다.

Test

package hello.jdbc.repository;

import

@Slf4j
class MemberRepositoryV0Test {
	MemberRepositoryV0 repository = new MemberRepositoryV0();
	
	@Test
	void crud() {
	//save
		Member member = new Member("memberV0", 10000);
		repository.save(member);
	
	//findById
		repository.findById(member.getMemberId());
		
		log.info("findMember={}  ", findMember);
	// 검증
	// static import Assertions
	assertThat(findMember).isEqaulTo(member);
	}
}

• 실행 결과에 member객체의 참조값이 아니라 실제 데이터가 보이는 이유는 @DaTa가 toString() 을 적절히 오버라이딩해서 보여주기 때문이다.

• @lombok을 쓰면 equals&hashCode 를 잡아준다.

log.info("member = findMember {} ", member = findMember); //false
log.info("member equals findMember {}", member.equals(findMember); //true

• isEqualTo() : findMember.equals(member) 를 비교한다. 결과가 참인 이유는 롬복의 @Data 는 해당 객체의 모든 필드를 사용하도록 equals() 를 오버라이딩하기 때문이다.

JDBC 개발 - 수정, 삭제

package hello.jdbc.repository;

import lombok.extern.slf4j.Slf4j;

/**
* JDBC = DriverManager 사용
**/
@Slf4j
public class MemberRepositoryV0 {

	public Member save(Member member) throws SQLException {
		String sql = "insert into member(member_id, money) values (?, ?);"
		Connection con = null;
		PreparedStatement pstmt = null;
		
		try {
			con = getConnection();
			pstmt = con.prepareStatement(sql);
			pstmt.setString(1, member.getMemberId());
			pstmt.setInt(2, member.getMoney());
			return member;
		}catch (SQLExceeption e){
			//checked Exception
			log.error("db error", e);
			//e.printStackTrace();
			throw e; 
		}finally {
			//시작과 역순으로 닫는다.
//			pstmt.close(); //Exception 되면 con이 닫히지않는 문제가 발생할 수 있다.
//			con.close();
				close(con, pstmt, null);	
		}
	}
	
	
	// 조회
	public Member findById(String memberId) throws SQLException {
		String sql = "select * from member where member_id = ?";
		
		Connection con = null;
		PreparedStatement pstmt = null;
		ResultSet rs = null;
		
		try {
			con  = getConnection();
			pstmt = con.prepareStatement(sql);
			pstmt.setString(1, memberId);
		
			rs = pstmt.executeQuery();
			if(rs.next()) {
				Member member = new Member();
				member.setMemberId(rs.getString("member_id"));
				member.setMoney(rs.getInt("money"));
				return member;
			}else {
			// rs.next() 값이 없다면
				throw new NoSuchElementException("member not found memberId = " + memberId);
			}		
		}catch (SQLException e) {
			log.error("db error", e);
			throw e;
		}	finally {
			close(con, pstmt, rs);
		}	
	}
	// 수정
	public void update(String memberId, int money) {
		String sql = update member set money = ? where member_id = ?";
		Connection con = null;
		PreparedStatement pstmt = null;
		
		try {
			con = getConnection();
			pstmt = con.prepareStatement(sql);
			pstmt.setInt(1, money);
			pstmt.setStrint(2, memberId);
			int resultSize = pstmt.executeUpdate();
			log.info("resultSize={}. resultSize); // 1 또는 0 이 나온다.
		}catch (SQLExceeption e){
			//checked Exception
			log.error("db error", e);
			//e.printStackTrace();
			throw e; 
		}finally {
			close(con, pstmt, null);	
		}
		
	}	
	
	// 삭제 
	public void delete(String memberId) {
		String sql = "delete from member where member_id =?";
		
		Connection con = null;
		PreparedStatement pstmt = null;
		
		try {
			con = getConnection();
			pstmt = con.prepareStatement(sql);
			pstmt.setInt(1, memberId)
			pstmt.executeUpdate();

		}catch (SQLExceeption e){
			//checked Exception
			log.error("db error", e);
			throw e; 
		}finally{
				close(con, pstmt, null);	
		}

	}
	
		
	private void close(Connection con, Statement stmt, ResultSet rs) {
		if(rs!= null) {
			try {
				rs.close(); 
			catch (SQLException e) {
				log.info("error", e);
			}
		}

		if(stmt != null) {
			try {
				stmt.close(); //pstmt 는 파라미터를 넘길 수 있는 더 많은 기능을 가지고 있고 stmt를 상속받는다
			catch (SQLException e) {
				log.info("error", e);
			}
		}
		// 이렇게 하면 각자 에러가 터져도 다른 것에 영향을 주지 않기에 닫을 수 있다.
		if(con != null) {
			try {
				con.close();		
			}catch (SQLException e) {
				log.info("error", e);
			}
	}
	
	private Connection getConnection() {
		return DBConnectionUtil.getConnection();	
	}
}

Test

package hello.jdbc.repository;

import

@Slf4j
class MemberRepositoryV0Test {
	MemberRepositoryV0 repository = new MemberRepositoryV0();
	
	@Test
	void crud() {
	//save
		Member member = new Member("memberV100", 10000);
		repository.save(member);
	
	//findById
		repository.findById(member.getMemberId());
		
		log.info("findMember={}  ", findMember);
	// 검증
	// static import Assertions
		assertThat(findMember).isEqaulTo(member);
	
	
		//update : money : 10000 -> 20000
		repository.update(member.getMemberId(), 20000);
		Member updateMember = repository.findById(member.getMemberId());
		assertThat(updatedMember.getMoney()).isEqualTo(20000);
		
		//delete
		repository.delete(member.getMemberId());
		assertThatThrownBy(() -> repository.findById(member.getMemberId())
			.isInstanceOf(NoSuchElementException.class);
		Member deleteMember = repository.findById(member.getMemberId());
	}
}

실행로그(pstmt.executeUpdate()의 결과)

MemberRepositoryV0 - resultSize = 1

• 회원을 삭제한 다음 findById()를 통해서 조회한다. 회원이 없기 때문에 NoSuchElementException 이 발생한다. assertThatThrownBy 는 해당 예외가 발생해야 검증에 성공한다.
• 마지막에 삭제를 하기 때문에 전체 사이클을 돌리며 계속 테스트 해볼 수 있다.
• 그러나 중간에 에러가 나면 삭제 로직이 수행이 안되기 때문에 반복실행할 수 없다.
• 좋은 로직은 아니다.
• 해결방법은 트랜잭션을 활용하는 것이다.