From 12994aab6095410bd4b6777358a9cf363338fb95 Mon Sep 17 00:00:00 2001 From: teddy Date: Fri, 13 Feb 2026 10:44:49 +0900 Subject: [PATCH] =?UTF-8?q?=ED=8C=8C=EC=9D=BC=20count=20=EA=B8=B0=EB=8A=A5?= =?UTF-8?q?=20=EC=B6=94=EA=B0=80?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../cd/training/train/TrainApiController.java | 2 +- .../train/service/DataSetCountersService.java | 14 ++++++++++++++ 2 files changed, 15 insertions(+), 1 deletion(-) diff --git a/src/main/java/com/kamco/cd/training/train/TrainApiController.java b/src/main/java/com/kamco/cd/training/train/TrainApiController.java index 88783d1..f6e8396 100644 --- a/src/main/java/com/kamco/cd/training/train/TrainApiController.java +++ b/src/main/java/com/kamco/cd/training/train/TrainApiController.java @@ -206,7 +206,7 @@ public class TrainApiController { }) @PostMapping("/counts/{uuid}") public ApiResponseDto getCount( - @Parameter(description = "uuid", example = "80a0e544-36ed-4999-b705-97427f23337d") + @Parameter(description = "uuid", example = "e22181eb-2ac4-4100-9941-d06efce25c49") @PathVariable UUID uuid) { Long modelId = trainJobService.getModelIdByUuid(uuid); diff --git a/src/main/java/com/kamco/cd/training/train/service/DataSetCountersService.java b/src/main/java/com/kamco/cd/training/train/service/DataSetCountersService.java index 0c90ee8..7454b1a 100644 --- a/src/main/java/com/kamco/cd/training/train/service/DataSetCountersService.java +++ b/src/main/java/com/kamco/cd/training/train/service/DataSetCountersService.java @@ -54,6 +54,20 @@ public class DataSetCountersService { return (int) stream.filter(Files::isRegularFile).filter(p -> p.toString().endsWith(".tif")).count(); } + + /* + 대소문자 및 geojson 필요시 + * try (var stream = Files.walk(dir)) { + return (int) + stream + .filter(Files::isRegularFile) + .filter(p -> { + String name = p.getFileName().toString().toLowerCase(); + return name.endsWith(".tif") || name.endsWith(".geojson"); + }) + .count(); + } + * */ } public DatasetCounters countTmpAfterBuild(Path path) throws IOException {