Feat: bulk create batch (#73)

Author: yjm07

com.bkmarriott.charge/src/main/java/com/bkmarriott/charge/application/outputport/RoomChargeOutputPort.java

@@ -18,6 +18,8 @@ public interface RoomChargeOutputPort {
 
     void bulkCreate(List<RoomChargeForCreate> roomChargeForCreateList);
 
+    void bulkCreateBatch(int batchSize, List<RoomChargeForCreate> roomChargeForCreateList);
+
     RoomCharge updateCharge(RoomCharge roomCharge, Integer charge);
 
     List<DefaultRoomCharge> findAllDefault();

com.bkmarriott.charge/src/main/java/com/bkmarriott/charge/application/service/RoomChargeService.java

@@ -76,6 +76,6 @@ public void createAllNextDefaultCharge() {
                 .map(hotelType -> RoomChargeForCreate.of(hotelType.getHotelId(), hotelType.getRoomType(), date, roomChargeMap.get(hotelType.getRoomType())))
                 .toList();
 
-        roomChargeOutputPort.bulkCreate(roomChargeForCreateList);
+        roomChargeOutputPort.bulkCreateBatch(1000, roomChargeForCreateList);
     }
 }

com.bkmarriott.charge/src/main/java/com/bkmarriott/charge/infrastructure/persistence/adapter/RoomChargeAdapter.java

@@ -11,7 +11,9 @@
 import com.bkmarriott.charge.infrastructure.persistence.repository.DefaultRoomChargeRepository;
 import com.bkmarriott.charge.infrastructure.persistence.repository.RoomChargeRepository;
 import lombok.RequiredArgsConstructor;
+import org.springframework.jdbc.core.JdbcTemplate;
 import org.springframework.stereotype.Component;
+import org.springframework.transaction.annotation.Transactional;
 
 import java.util.List;
 import java.util.Optional;
@@ -22,6 +24,7 @@ public class RoomChargeAdapter implements RoomChargeOutputPort {
 
     private final RoomChargeRepository roomChargeRepository;
     private final DefaultRoomChargeRepository defaultRoomChargeRepository;
+    private final JdbcTemplate jdbcTemplate;
 
     public Optional<RoomCharge> findById(RoomChargeId roomChargeId) {
         return roomChargeRepository.findByIdAndIsDeletedFalse(RoomChargeEntityId.fromDomain(roomChargeId))
@@ -47,6 +50,23 @@ public void bulkCreate(List<RoomChargeForCreate> roomChargeForCreateList) {
         roomChargeRepository.saveAll(roomChargeEntities);
     }
 
+    @Transactional
+    public void bulkCreateBatch(int batchSize, List<RoomChargeForCreate> roomChargeForCreateList) {
+        batchUpdate(batchSize, roomChargeForCreateList);
+    }
+
+    private void batchUpdate(int batchSize, List<RoomChargeForCreate> roomChargeForCreateList) {
+        String sql = "INSERT INTO m_room_charge (hotel_id, room_type, date, charge) VALUES (?, ?, ?, ?)";
+
+        jdbcTemplate.batchUpdate(sql, roomChargeForCreateList, batchSize, (ps, argument) -> {
+            RoomChargeEntity entity = RoomChargeEntity.from(argument);
+            ps.setLong(1, entity.getId().getHotelId());
+            ps.setString(2, entity.getId().getRoomType().toString());
+            ps.setString(3, entity.getId().getDate().toString());
+            ps.setInt(4, entity.getCharge());
+        });
+    }
+
     public RoomCharge updateCharge(RoomCharge roomCharge, Integer charge) {
         roomCharge.updateCharge(charge);
         RoomChargeEntity roomChargeEntity = RoomChargeEntity.fromDomain(roomCharge);

com.bkmarriott.charge/src/test/java/com/bkmarriott/charge/application/service/RoomChargeServiceTest.java

@@ -227,7 +227,7 @@ void createAllNextDefaultCharge_successTest() {
 
         // Then
         ArgumentCaptor<List<RoomChargeForCreate>> captor = ArgumentCaptor.forClass(List.class);
-        Mockito.verify(roomChargeOutputPort).bulkCreate(captor.capture());
+        Mockito.verify(roomChargeOutputPort).bulkCreateBatch(ArgumentMatchers.anyInt(), captor.capture());
 
         List<RoomChargeForCreate> capturedCharges = captor.getValue();
         LocalDate expectedDate = LocalDate.now().plusMonths(3);

com.bkmarriott.charge/src/test/java/com/bkmarriott/charge/infrastructure/persistence/adapter/RoomChargeAdapterTest.java

@@ -47,6 +47,15 @@ void bulkCreate_successTest() {
         Assertions.assertDoesNotThrow(() -> roomChargeAdapter.bulkCreate(roomChargeForCreateList));
     }
 
+    @Test
+    @DisplayName("[객실 요금 batch 벌크 등록 성공 테스트] 객실 요금을 batch 벌크 생성한다.")
+    void bulkCreateBatch_successTest() {
+        // Given
+        List<RoomChargeForCreate> roomChargeForCreateList = genRoomChargeForCreateList();
+        // When & Then
+        Assertions.assertDoesNotThrow(() -> roomChargeAdapter.bulkCreateBatch(5, roomChargeForCreateList));
+    }
+
     @Test
     @DisplayName("[객실 요금 조회 성공 테스트] 객실 요금 ID로 객실 요금을 조회한 뒤 Optional 객체에 담아 반환한다.")
     void findById_successTest() {
@@ -67,7 +76,7 @@ void findById_successTest() {
     void findAll_successTest() {
         // Given
         List<RoomChargeForCreate> roomChargeForCreateList = genRoomChargeForCreateList();
-        roomChargeAdapter.bulkCreate(roomChargeForCreateList);
+        roomChargeAdapter.bulkCreateBatch(5, roomChargeForCreateList);
         // When
         List<RoomCharge> roomChargeList = roomChargeAdapter.findAll(
                 roomChargeForCreateList.stream().map(RoomChargeForCreate::id).toList()

com.bkmarriott.charge/src/test/java/com/bkmarriott/charge/infrastructure/persistence/config/PersistenceTestConfig.java

@@ -8,16 +8,18 @@
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
+import org.springframework.jdbc.core.JdbcTemplate;
 
 @Configuration
 public class PersistenceTestConfig {
 
     @Bean
     public RoomChargeAdapter roomChargeAdapter(
             @Autowired RoomChargeRepository roomChargeRepository,
-            @Autowired DefaultRoomChargeRepository defaultRoomChargeRepository
+            @Autowired DefaultRoomChargeRepository defaultRoomChargeRepository,
+            @Autowired JdbcTemplate jdbcTemplate
     ) {
-        return new RoomChargeAdapter(roomChargeRepository, defaultRoomChargeRepository);
+        return new RoomChargeAdapter(roomChargeRepository, defaultRoomChargeRepository, jdbcTemplate);
     }
 
     @Bean