모델학습 1단계 실행중인 것이 있는지 count API

src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java

@@ -502,4 +502,8 @@ public class ModelTrainMngCoreService {
   public List<String> findDatasetUid(List<Long> datasetIds) {
     return datasetRepository.findDatasetUid(datasetIds);
   }
+
+  public Long findModelStep1InProgressCnt() {
+    return modelMngRepository.findModelStep1InProgressCnt();
+  }
 }

src/main/java/com/kamco/cd/training/postgres/repository/model/ModelMngRepositoryCustom.java

@@ -22,4 +22,6 @@ public interface ModelMngRepositoryCustom {
   Optional<ModelMasterEntity> findFirstByStatusCdAndDelYn(String statusCd, Boolean delYn);
 
   TrainRunRequest findTrainRunRequest(Long modelId);
+
+  Long findModelStep1InProgressCnt();
 }

src/main/java/com/kamco/cd/training/postgres/repository/model/ModelMngRepositoryImpl.java

@@ -4,6 +4,7 @@ import static com.kamco.cd.training.postgres.entity.QModelConfigEntity.modelConf
 import static com.kamco.cd.training.postgres.entity.QModelHyperParamEntity.modelHyperParamEntity;
 import static com.kamco.cd.training.postgres.entity.QModelMasterEntity.modelMasterEntity;
 
+import com.kamco.cd.training.common.enums.TrainStatusType;
 import com.kamco.cd.training.model.dto.ModelTrainMngDto;
 import com.kamco.cd.training.postgres.entity.ModelMasterEntity;
 import com.kamco.cd.training.train.dto.TrainRunRequest;
@@ -147,4 +148,13 @@ public class ModelMngRepositoryImpl implements ModelMngRepositoryCustom {
         .where(modelMasterEntity.id.eq(modelId))
         .fetchOne();
   }
+
+  @Override
+  public Long findModelStep1InProgressCnt() {
+    return queryFactory
+        .select(modelMasterEntity.id.count())
+        .from(modelMasterEntity)
+        .where(modelMasterEntity.step1State.eq(TrainStatusType.IN_PROGRESS.getId()))
+        .fetchOne();
+  }
 }