jdbc : apply transaction
This commit is contained in:
haerong22
@@ -0,0 +1,175 @@
|
||||
package com.example.jdbc.repository;
|
||||
|
||||
import com.example.jdbc.domain.Member;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.jdbc.support.JdbcUtils;
|
||||
|
||||
import javax.sql.DataSource;
|
||||
import java.sql.*;
|
||||
import java.util.NoSuchElementException;
|
||||
|
||||
/**
|
||||
* JDBC - ConnectionParam
|
||||
*/
|
||||
@Slf4j
|
||||
public class MemberRepositoryV2 {
|
||||
|
||||
private final DataSource dataSource;
|
||||
|
||||
public MemberRepositoryV2(DataSource dataSource) {
|
||||
this.dataSource = dataSource;
|
||||
}
|
||||
|
||||
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());
|
||||
pstmt.executeUpdate();
|
||||
return member;
|
||||
} catch (SQLException e) {
|
||||
log.error("db error", e);
|
||||
throw e;
|
||||
} finally {
|
||||
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 {
|
||||
throw new NoSuchElementException("member not found memberId=" + memberId);
|
||||
}
|
||||
} catch (SQLException e) {
|
||||
log.error("db error", e);
|
||||
throw e;
|
||||
} finally {
|
||||
close(con, pstmt, rs);
|
||||
}
|
||||
}
|
||||
|
||||
public Member findById(Connection con, String memberId) throws SQLException {
|
||||
String sql = "select * from member where member_id = ?";
|
||||
|
||||
PreparedStatement pstmt = null;
|
||||
ResultSet rs = null;
|
||||
|
||||
try {
|
||||
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 {
|
||||
throw new NoSuchElementException("member not found memberId=" + memberId);
|
||||
}
|
||||
} catch (SQLException e) {
|
||||
log.error("db error", e);
|
||||
throw e;
|
||||
} finally {
|
||||
// connection 은 서비스계층에서 닫는다.
|
||||
JdbcUtils.closeResultSet(rs);
|
||||
JdbcUtils.closeStatement(pstmt);
|
||||
}
|
||||
}
|
||||
|
||||
public void update(String memberId, int money) throws SQLException {
|
||||
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.setString(2, memberId);
|
||||
|
||||
int result = pstmt.executeUpdate();
|
||||
log.info("resultSize={}", result);
|
||||
} catch (SQLException e) {
|
||||
log.error("db error", e);
|
||||
throw e;
|
||||
} finally {
|
||||
close(con, pstmt, null);
|
||||
}
|
||||
}
|
||||
|
||||
public void update(Connection con, String memberId, int money) throws SQLException {
|
||||
String sql = "update member set money=? where member_id=?";
|
||||
|
||||
PreparedStatement pstmt = null;
|
||||
|
||||
try {
|
||||
pstmt = con.prepareStatement(sql);
|
||||
pstmt.setInt(1, money);
|
||||
pstmt.setString(2, memberId);
|
||||
|
||||
int result = pstmt.executeUpdate();
|
||||
log.info("resultSize={}", result);
|
||||
} catch (SQLException e) {
|
||||
log.error("db error", e);
|
||||
throw e;
|
||||
} finally {
|
||||
// connection 은 서비스계층에서 닫는다.
|
||||
JdbcUtils.closeStatement(pstmt);
|
||||
}
|
||||
}
|
||||
|
||||
public void delete(String memberId) throws SQLException {
|
||||
String sql = "delete from member where member_id=?";
|
||||
|
||||
Connection con = null;
|
||||
PreparedStatement pstmt = null;
|
||||
|
||||
try {
|
||||
con = getConnection();
|
||||
pstmt = con.prepareStatement(sql);
|
||||
pstmt.setString(1, memberId);
|
||||
pstmt.executeUpdate();
|
||||
} catch (SQLException e) {
|
||||
log.error("db error", e);
|
||||
throw e;
|
||||
} finally {
|
||||
close(con, pstmt, null);
|
||||
}
|
||||
}
|
||||
|
||||
private void close(Connection con, Statement stmt, ResultSet rs) {
|
||||
JdbcUtils.closeResultSet(rs);
|
||||
JdbcUtils.closeStatement(stmt);
|
||||
JdbcUtils.closeConnection(con);
|
||||
}
|
||||
|
||||
private Connection getConnection() throws SQLException {
|
||||
Connection con = dataSource.getConnection();
|
||||
log.info("get connection={}, class={}", con, con.getClass());
|
||||
return con;
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,30 @@
|
||||
package com.example.jdbc.service;
|
||||
|
||||
import com.example.jdbc.domain.Member;
|
||||
import com.example.jdbc.repository.MemberRepositoryV1;
|
||||
import lombok.RequiredArgsConstructor;
|
||||
|
||||
import java.sql.SQLException;
|
||||
|
||||
@RequiredArgsConstructor
|
||||
public class MemberServiceV1 {
|
||||
|
||||
private final MemberRepositoryV1 memberRepository;
|
||||
|
||||
public void accountTransfer(String fromId, String toId, int money) throws SQLException {
|
||||
Member fromMember = memberRepository.findById(fromId);
|
||||
Member toMember = memberRepository.findById(toId);
|
||||
|
||||
memberRepository.update(fromId, fromMember.getMoney() - money);
|
||||
|
||||
validation(toMember);
|
||||
|
||||
memberRepository.update(toId, toMember.getMoney() + money);
|
||||
}
|
||||
|
||||
private void validation(Member toMember) {
|
||||
if (toMember.getMemberId().equals("ex")) {
|
||||
throw new IllegalStateException("이체중 예외 발생");
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,66 @@
|
||||
package com.example.jdbc.service;
|
||||
|
||||
import com.example.jdbc.domain.Member;
|
||||
import com.example.jdbc.repository.MemberRepositoryV2;
|
||||
import lombok.RequiredArgsConstructor;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
|
||||
import javax.sql.DataSource;
|
||||
import java.sql.Connection;
|
||||
import java.sql.SQLException;
|
||||
|
||||
/**
|
||||
* 트랜잭션 - 파라미터 연동, 풀을 고려한 종료
|
||||
*/
|
||||
@Slf4j
|
||||
@RequiredArgsConstructor
|
||||
public class MemberServiceV2 {
|
||||
|
||||
private final DataSource dataSource;
|
||||
private final MemberRepositoryV2 memberRepository;
|
||||
|
||||
public void accountTransfer(String fromId, String toId, int money) throws SQLException {
|
||||
Connection con = dataSource.getConnection();
|
||||
try {
|
||||
con.setAutoCommit(false); // 트랜잭션 시작
|
||||
|
||||
// 비지니스 로직
|
||||
bizLogic(con, fromId, toId, money);
|
||||
|
||||
con.commit(); // 성공시 커밋
|
||||
} catch (Exception e) {
|
||||
con.rollback(); // 실패시 롤백
|
||||
throw new IllegalStateException(e);
|
||||
} finally {
|
||||
release(con);
|
||||
}
|
||||
}
|
||||
|
||||
private void bizLogic(Connection con, String fromId, String toId, int money) throws SQLException {
|
||||
Member fromMember = memberRepository.findById(con, fromId);
|
||||
Member toMember = memberRepository.findById(con, toId);
|
||||
|
||||
memberRepository.update(con, fromId, fromMember.getMoney() - money);
|
||||
|
||||
validation(toMember);
|
||||
|
||||
memberRepository.update(con, toId, toMember.getMoney() + money);
|
||||
}
|
||||
|
||||
private void validation(Member toMember) {
|
||||
if (toMember.getMemberId().equals("ex")) {
|
||||
throw new IllegalStateException("이체중 예외 발생");
|
||||
}
|
||||
}
|
||||
|
||||
private void release(Connection con) {
|
||||
if (con != null) {
|
||||
try {
|
||||
con.setAutoCommit(true); // 커넥션 풀 고려
|
||||
con.close();
|
||||
} catch (Exception e) {
|
||||
log.info("error", e);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,83 @@
|
||||
package com.example.jdbc.service;
|
||||
|
||||
import com.example.jdbc.domain.Member;
|
||||
import com.example.jdbc.repository.MemberRepositoryV1;
|
||||
import org.junit.jupiter.api.AfterEach;
|
||||
import org.junit.jupiter.api.BeforeEach;
|
||||
import org.junit.jupiter.api.DisplayName;
|
||||
import org.junit.jupiter.api.Test;
|
||||
import org.springframework.jdbc.datasource.DriverManagerDataSource;
|
||||
|
||||
import java.sql.SQLException;
|
||||
|
||||
import static com.example.jdbc.connection.ConnectionConst.*;
|
||||
import static org.assertj.core.api.Assertions.assertThat;
|
||||
import static org.assertj.core.api.Assertions.assertThatThrownBy;
|
||||
|
||||
/**
|
||||
* 기본 동작, 트랜잭션이 없어서 문제 발생
|
||||
*/
|
||||
class MemberServiceV1Test {
|
||||
|
||||
public static final String MEMBER_A = "memberA";
|
||||
public static final String MEMBER_B = "memberB";
|
||||
public static final String MEMBER_EX = "ex";
|
||||
|
||||
private MemberRepositoryV1 memberRepository;
|
||||
private MemberServiceV1 memberService;
|
||||
|
||||
@BeforeEach
|
||||
void before() {
|
||||
DriverManagerDataSource dataSource = new DriverManagerDataSource(URL, USERNAME, PASSWORD);
|
||||
memberRepository = new MemberRepositoryV1(dataSource);
|
||||
memberService = new MemberServiceV1(memberRepository);
|
||||
}
|
||||
|
||||
@AfterEach
|
||||
void after() throws SQLException {
|
||||
memberRepository.delete(MEMBER_A);
|
||||
memberRepository.delete(MEMBER_B);
|
||||
memberRepository.delete(MEMBER_EX);
|
||||
}
|
||||
|
||||
@Test
|
||||
@DisplayName("정상 이체")
|
||||
void accountTransfer() throws SQLException {
|
||||
// given
|
||||
Member memberA = new Member(MEMBER_A, 10000);
|
||||
Member memberB = new Member(MEMBER_B, 10000);
|
||||
memberRepository.save(memberA);
|
||||
memberRepository.save(memberB);
|
||||
|
||||
// when
|
||||
memberService.accountTransfer(memberA.getMemberId(), memberB.getMemberId(), 2000);
|
||||
|
||||
// then
|
||||
Member findMemberA = memberRepository.findById(memberA.getMemberId());
|
||||
Member findMemberB = memberRepository.findById(memberB.getMemberId());
|
||||
|
||||
assertThat(findMemberA.getMoney()).isEqualTo(8000);
|
||||
assertThat(findMemberB.getMoney()).isEqualTo(12000);
|
||||
}
|
||||
|
||||
@Test
|
||||
@DisplayName("이체중 예외 발생")
|
||||
void accountTransferEx() throws SQLException {
|
||||
// given
|
||||
Member memberA = new Member(MEMBER_A, 10000);
|
||||
Member memberEx = new Member(MEMBER_EX, 10000);
|
||||
memberRepository.save(memberA);
|
||||
memberRepository.save(memberEx);
|
||||
|
||||
// when
|
||||
assertThatThrownBy(() -> memberService.accountTransfer(memberA.getMemberId(), memberEx.getMemberId(), 2000))
|
||||
.isInstanceOf(IllegalStateException.class);
|
||||
|
||||
// then
|
||||
Member findMemberA = memberRepository.findById(memberA.getMemberId());
|
||||
Member findMemberB = memberRepository.findById(memberEx.getMemberId());
|
||||
|
||||
assertThat(findMemberA.getMoney()).isEqualTo(8000);
|
||||
assertThat(findMemberB.getMoney()).isEqualTo(10000);
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,83 @@
|
||||
package com.example.jdbc.service;
|
||||
|
||||
import com.example.jdbc.domain.Member;
|
||||
import com.example.jdbc.repository.MemberRepositoryV2;
|
||||
import org.junit.jupiter.api.AfterEach;
|
||||
import org.junit.jupiter.api.BeforeEach;
|
||||
import org.junit.jupiter.api.DisplayName;
|
||||
import org.junit.jupiter.api.Test;
|
||||
import org.springframework.jdbc.datasource.DriverManagerDataSource;
|
||||
|
||||
import java.sql.SQLException;
|
||||
|
||||
import static com.example.jdbc.connection.ConnectionConst.*;
|
||||
import static org.assertj.core.api.Assertions.assertThat;
|
||||
import static org.assertj.core.api.Assertions.assertThatThrownBy;
|
||||
|
||||
/**
|
||||
* 트랜잭션 - 커넥션 파라미터 전달 방식 동기화
|
||||
*/
|
||||
class MemberServiceV2Test {
|
||||
|
||||
public static final String MEMBER_A = "memberA";
|
||||
public static final String MEMBER_B = "memberB";
|
||||
public static final String MEMBER_EX = "ex";
|
||||
|
||||
private MemberRepositoryV2 memberRepository;
|
||||
private MemberServiceV2 memberService;
|
||||
|
||||
@BeforeEach
|
||||
void before() {
|
||||
DriverManagerDataSource dataSource = new DriverManagerDataSource(URL, USERNAME, PASSWORD);
|
||||
memberRepository = new MemberRepositoryV2(dataSource);
|
||||
memberService = new MemberServiceV2(dataSource, memberRepository);
|
||||
}
|
||||
|
||||
@AfterEach
|
||||
void after() throws SQLException {
|
||||
memberRepository.delete(MEMBER_A);
|
||||
memberRepository.delete(MEMBER_B);
|
||||
memberRepository.delete(MEMBER_EX);
|
||||
}
|
||||
|
||||
@Test
|
||||
@DisplayName("정상 이체")
|
||||
void accountTransfer() throws SQLException {
|
||||
// given
|
||||
Member memberA = new Member(MEMBER_A, 10000);
|
||||
Member memberB = new Member(MEMBER_B, 10000);
|
||||
memberRepository.save(memberA);
|
||||
memberRepository.save(memberB);
|
||||
|
||||
// when
|
||||
memberService.accountTransfer(memberA.getMemberId(), memberB.getMemberId(), 2000);
|
||||
|
||||
// then
|
||||
Member findMemberA = memberRepository.findById(memberA.getMemberId());
|
||||
Member findMemberB = memberRepository.findById(memberB.getMemberId());
|
||||
|
||||
assertThat(findMemberA.getMoney()).isEqualTo(8000);
|
||||
assertThat(findMemberB.getMoney()).isEqualTo(12000);
|
||||
}
|
||||
|
||||
@Test
|
||||
@DisplayName("이체중 예외 발생")
|
||||
void accountTransferEx() throws SQLException {
|
||||
// given
|
||||
Member memberA = new Member(MEMBER_A, 10000);
|
||||
Member memberEx = new Member(MEMBER_EX, 10000);
|
||||
memberRepository.save(memberA);
|
||||
memberRepository.save(memberEx);
|
||||
|
||||
// when
|
||||
assertThatThrownBy(() -> memberService.accountTransfer(memberA.getMemberId(), memberEx.getMemberId(), 2000))
|
||||
.isInstanceOf(IllegalStateException.class);
|
||||
|
||||
// then
|
||||
Member findMemberA = memberRepository.findById(memberA.getMemberId());
|
||||
Member findMemberB = memberRepository.findById(memberEx.getMemberId());
|
||||
|
||||
assertThat(findMemberA.getMoney()).isEqualTo(10000);
|
||||
assertThat(findMemberB.getMoney()).isEqualTo(10000);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user