07_RDBMS_락을활용한_동시성제어보고서 - loveAlakazam/hh-08-concert GitHub Wiki

public ConcertSeat extends BaseEntity {
    public void reserve() {
        // 이미 예약된 좌석
        // if( !isAvailable ) throw new BusinessException(ALREADY_RESERVED_SEAT);
        this.isAvailable = false;
 }
}

 @Test
 void 병렬요청_시간_응답을_기록한다() throws Exception {
  int threadCount = 20;
  ExecutorService executorService= Executors.newFixedThreadPool(threadCount);
  CountDownLatch latch = new CountDownLatch(threadCount);

  List<Future<Long>> results = new ArrayList<>();
  for (int i = 0; i < threadCount; i++) {
   final int index = i;
   results.add(executorService.submit(() -> {
    latch.countDown();
    latch.await();
    long start = System.currentTimeMillis();
    try {
     User user = userRepository.save(User.of("유저 " + index));
     reservationService.temporaryReserve(
      ReservationCommand.TemporaryReserve.of(user, sampleConcertSeat)
     );
    } catch (Exception ignored) {}
    return System.currentTimeMillis() - start;
   }));
  }
  executorService.shutdown();
  executorService.awaitTermination(10, TimeUnit.SECONDS);

  long avg = (long) results.stream().mapToLong(r -> {
   try { return r.get(); } catch (Exception e) { return 0; }
  }).average().orElse(0);

  log.info("평균 응답시간: {}ms", avg);
  assertTrue(avg > 0 , "체크완료");
}

 @Test
 @Order(1)
 void 포인트_충전과_사용은_동시에_실행되면_안된다() throws Exception {
  // given
  long userId = sampleUser.getId();

  // 먼저 10,000원 충전
  userService.chargePoint(UserPointCommand.ChargePoint.of(userId, 10_000L));

  // 두 작업이 동시에 실행되도록 조율하는 CyclicBarrier
  CyclicBarrier barrier = new CyclicBarrier(2);

  ExecutorService executor = Executors.newFixedThreadPool(2);
  List<Future<Void>> results = new ArrayList<>();

  // 충전 쓰레드
  results.add(executor.submit(() -> {
   barrier.await(); // 🔥 다른 스레드가 도달할 때까지 대기
   userService.chargePoint(UserPointCommand.ChargePoint.of(userId, 5_000L));
   return null;
  }));

  // 사용 쓰레드
  results.add(executor.submit(() -> {
   barrier.await(); // 🔥 두 스레드가 동시에 실행되도록 조율
   userService.usePoint(UserPointCommand.UsePoint.of(userId, 5_000L));
   return null;
  }));

  // when: 두 작업이 완료될 때까지 기다림
  for (Future<Void> result : results) {
   result.get(); // 예외 발생 시 여기서 잡힘
  }

  // then: 최종 포인트는 10,000 + 5,000 - 5,000 = 10,000
  UserInfo.GetCurrentPoint info = userService.getCurrentPoint(UserPointCommand.GetCurrentPoint.of(userId));
  assertEquals(10_000L, info.point());
 }

@Entity
@Table(name = "user_points")
public class UserPoint{
 /**
  * 유저 포인트 ID
  */
 @Id
 @GeneratedValue(strategy = GenerationType.IDENTITY)
 @Column(name ="id", nullable = false, updatable = false)
 private long id;
 @Column(name ="point", nullable = false)
 private long point = 0; // 잔액

 @Version
 private int version; // 👈 버젼 필드 추가

    ...
}

@Test
void 포인트_충전과_사용이_동시에_진행할때_둘중하나는_OptimisticLockException예외를_발생시킨다() throws Exception {
  // given
  long userId = sampleUser.getId();

  // 먼저 10,000원 충전
  userService.chargePoint(UserPointCommand.ChargePoint.of(userId, 10_000L));

  // 두 작업이 동시에 실행되도록 조율하는 CyclicBarrier
  CyclicBarrier barrier = new CyclicBarrier(2);

  ExecutorService executor = Executors.newFixedThreadPool(2);
  List<Future<String>> results = new ArrayList<>();

  // 충전 쓰레드
  results.add(executor.submit(() -> {
   try{
    log.info("::: 포인트 충전 스레드 실행");
    barrier.await(); // 🔥 다른 스레드가 도달할 때까지 대기
    userService.chargePoint(UserPointCommand.ChargePoint.of(userId, 5_000L));
    return "충전 성공";
   } catch(ObjectOptimisticLockingFailureException e){
    return "충전 충돌";
   }
  }));

  // 사용 쓰레드
  results.add(executor.submit(() -> {
   try {
    log.info("::: 포인트 사용 스레드 실행");
    barrier.await(); // 🔥 두 스레드가 동시에 실행되도록 조율
    userService.usePoint(UserPointCommand.UsePoint.of(userId, 5_000L));
    return "사용 성공";

   } catch (ObjectOptimisticLockingFailureException e) {
    return "사용 충돌";
   }
  }));

  // when: 두 작업이 완료될 때까지 기다림
  List<String> messages = new ArrayList<>();
  for (Future<String> result : results) {
   messages.add(result.get()); // 예외 발생 시 여기서 잡힘
  }

  // 충전/사용 중 하나는 반드시 성공하고 하나는 충돌이어야 한다
  log.info("결과메시지: {}", messages);
  assertTrue(messages.contains("충전 성공") || messages.contains("사용 성공"));
  assertTrue(messages.contains("충전 충돌") || messages.contains("사용 충돌"));

  // then
  UserInfo.GetCurrentPoint info = userService.getCurrentPoint(UserPointCommand.GetCurrentPoint.of(userId));
  log.info("보유포인트 : {}", info.point());
  assertTrue(info.point() == 15000 || info.point() == 5000);
}

@Service
@RequiredArgsConstructor
public class UserService {
    ...

    @Transactional
    public UserInfo.ChargePoint chargePoint(UserPointCommand.ChargePoint command) {
        // 유저 포인트정보 조회 (공유락 사용)
        UserPoint userPoint = userPointRepository.findUserPointWithSharedLock(command.userId());
        if(userPoint == null)
            throw new BusinessException(UserErrorCode.NOT_EXIST_USER);
            ...
    }
    @Transactional
    public UserInfo.UsePoint usePoint(UserPointCommand.UsePoint command) {
        // 유저 정보 조회 (공유락 사용)
        UserPoint userPoint = userPointRepository.findUserPointWithSharedLock(command.userId());
        if(userPoint == null) throw new BusinessException(UserErrorCode.NOT_EXIST_USER);
        ....
    }
}