파일 count 기능 추가

This commit is contained in:
2026-02-13 10:44:49 +09:00
parent 11d3afe295
commit 12994aab60
2 changed files with 15 additions and 1 deletions

View File

@@ -206,7 +206,7 @@ public class TrainApiController {
})
@PostMapping("/counts/{uuid}")
public ApiResponseDto<Void> 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);

View File

@@ -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 {