라벨 자동할당 update 로직 -> 추후 라벨링 할당 테이블로 변경할 예정

src/main/java/com/kamco/cd/kamcoback/postgres/core/LabelAllocateCoreService.java

@@ -0,0 +1,21 @@
+package com.kamco.cd.kamcoback.postgres.core;
+
+import com.kamco.cd.kamcoback.postgres.repository.label.LabelAllocateRepository;
+import java.util.List;
+import lombok.RequiredArgsConstructor;
+import org.springframework.stereotype.Service;
+
+@Service
+@RequiredArgsConstructor
+public class LabelAllocateCoreService {
+
+  private final LabelAllocateRepository labelAllocateRepository;
+
+  public List<Long> fetchNextIds(Long lastId, int batchSize) {
+    return labelAllocateRepository.fetchNextIds(lastId, batchSize);
+  }
+
+  public Long assignOwner(List<Long> ids, Long userId) {
+    return labelAllocateRepository.assignOwner(ids, userId);
+  }
+}

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

@@ -0,0 +1,8 @@
+package com.kamco.cd.kamcoback.postgres.repository.label;
+
+import com.kamco.cd.kamcoback.postgres.entity.MapSheetAnalDataInferenceGeomEntity;
+import org.springframework.data.jpa.repository.JpaRepository;
+
+public interface LabelAllocateRepository
+    extends JpaRepository<MapSheetAnalDataInferenceGeomEntity, Long>,
+        LabelAllocateRepositoryCustom {}

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

@@ -0,0 +1,10 @@
+package com.kamco.cd.kamcoback.postgres.repository.label;
+
+import java.util.List;
+
+public interface LabelAllocateRepositoryCustom {
+
+  List<Long> fetchNextIds(Long lastId, int batchSize);
+
+  Long assignOwner(List<Long> ids, Long userId);
+}

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

@@ -0,0 +1,51 @@
+package com.kamco.cd.kamcoback.postgres.repository.label;
+
+import static com.kamco.cd.kamcoback.postgres.entity.QMapSheetAnalDataInferenceGeomEntity.mapSheetAnalDataInferenceGeomEntity;
+
+import com.kamco.cd.kamcoback.postgres.entity.MapSheetAnalDataGeomEntity;
+import com.querydsl.core.types.dsl.Expressions;
+import com.querydsl.core.types.dsl.StringExpression;
+import com.querydsl.jpa.impl.JPAQueryFactory;
+import java.util.List;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.data.jpa.repository.support.QuerydslRepositorySupport;
+import org.springframework.stereotype.Repository;
+
+@Slf4j
+@Repository
+public class LabelAllocateRepositoryImpl extends QuerydslRepositorySupport
+    implements LabelAllocateRepositoryCustom {
+
+  private final JPAQueryFactory queryFactory;
+  private final StringExpression NULL_STRING = Expressions.stringTemplate("cast(null as text)");
+
+  public LabelAllocateRepositoryImpl(JPAQueryFactory queryFactory) {
+    super(MapSheetAnalDataGeomEntity.class);
+    this.queryFactory = queryFactory;
+  }
+
+  @Override
+  public List<Long> fetchNextIds(Long lastId, int batchSize) {
+    return queryFactory
+        .select(mapSheetAnalDataInferenceGeomEntity.geoUid)
+        .from(mapSheetAnalDataInferenceGeomEntity)
+        .where(
+            //        mapSheetAnalDataGeomEntity.pnu.isNotNull(),   //TODO: Mockup 진행 이후 확인하기
+            lastId == null ? null : mapSheetAnalDataInferenceGeomEntity.geoUid.gt(lastId),
+            mapSheetAnalDataInferenceGeomEntity.compareYyyy.eq(2022),
+            mapSheetAnalDataInferenceGeomEntity.targetYyyy.eq(2024),
+            mapSheetAnalDataInferenceGeomEntity.labelerUid.isNull())
+        .orderBy(mapSheetAnalDataInferenceGeomEntity.geoUid.asc())
+        .limit(batchSize)
+        .fetch();
+  }
+
+  @Override
+  public Long assignOwner(List<Long> ids, Long userId) {
+    return queryFactory
+        .update(mapSheetAnalDataInferenceGeomEntity)
+        .set(mapSheetAnalDataInferenceGeomEntity.labelerUid, userId)
+        .where(mapSheetAnalDataInferenceGeomEntity.geoUid.in(ids))
+        .execute();
+  }
+}