라벨할당 로직 수정

src/main/java/com/kamco/cd/kamcoback/postgres/repository/label/LabelAllocateRepositoryImpl.java

@@ -34,6 +34,7 @@ import jakarta.persistence.EntityManager;
 import jakarta.persistence.EntityNotFoundException;
 import jakarta.persistence.PersistenceContext;
 import jakarta.transaction.Transactional;
+import java.sql.PreparedStatement;
 import java.time.LocalDate;
 import java.time.LocalTime;
 import java.time.ZoneId;
@@ -43,6 +44,7 @@ import java.util.Objects;
 import java.util.UUID;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
+import org.hibernate.Session;
 import org.springframework.data.domain.Page;
 import org.springframework.data.domain.PageImpl;
 import org.springframework.data.domain.Pageable;
@@ -106,6 +108,10 @@ public class LabelAllocateRepositoryImpl implements LabelAllocateRepositoryCusto
         .execute();
 
     // 라벨러 할당 테이블에 insert
+    Session session = em.unwrap(Session.class);
+
+    session.doWork(
+        connection -> {
           String sql =
               """
         insert into tb_labeling_assignment
@@ -114,19 +120,29 @@ public class LabelAllocateRepositoryImpl implements LabelAllocateRepositoryCusto
         values (?, ?, ?, ?, ?, ?)
         """;
 
+          try (PreparedStatement ps = connection.prepareStatement(sql)) {
+            int batchSize = 0;
+
             for (AllocateInfoDto info : ids) {
-      em.createNativeQuery(sql)
+              ps.setObject(1, UUID.randomUUID());
-          .setParameter(1, UUID.randomUUID())
+              ps.setLong(2, info.getGeoUid());
-          .setParameter(2, info.getGeoUid())
+              ps.setString(3, userId);
-          .setParameter(3, userId)
+              ps.setString(4, LabelState.ASSIGNED.getId());
-          .setParameter(4, LabelState.ASSIGNED.getId())
+              ps.setString(5, String.valueOf(info.getMapSheetNum()));
-          .setParameter(5, info.getMapSheetNum())
+              ps.setLong(6, analEntity.getId());
-          .setParameter(6, analEntity.getId())
+
-          .executeUpdate();
+              ps.addBatch();
+              batchSize++;
+
+              if (batchSize % 1000 == 0) {
+                ps.executeBatch();
+                ps.clearBatch();
+              }
             }
 
-    em.flush();
+            ps.executeBatch();
-    em.clear();
+          }
+        });
   }
 
   @Override