diff --git a/src/main/java/com/kamco/cd/training/common/download/RangeDownloadResponder.java b/src/main/java/com/kamco/cd/training/common/download/RangeDownloadResponder.java index 686741a..6abe707 100644 --- a/src/main/java/com/kamco/cd/training/common/download/RangeDownloadResponder.java +++ b/src/main/java/com/kamco/cd/training/common/download/RangeDownloadResponder.java @@ -36,6 +36,7 @@ public class RangeDownloadResponder { .contentType(MediaType.APPLICATION_OCTET_STREAM) .header(HttpHeaders.CONTENT_DISPOSITION, disposition) .header(HttpHeaders.ACCEPT_RANGES, "bytes") + .header("Access-Control-Expose-Headers", "Content-Disposition") .header("X-Accel-Buffering", "no"); if (rangeHeader == null || rangeHeader.isBlank()) { @@ -71,6 +72,7 @@ public class RangeDownloadResponder { .contentType(MediaType.APPLICATION_OCTET_STREAM) .header(HttpHeaders.CONTENT_DISPOSITION, disposition) .header(HttpHeaders.ACCEPT_RANGES, "bytes") + .header("Access-Control-Expose-Headers", "Content-Disposition") .header("X-Accel-Buffering", "no") .header(HttpHeaders.CONTENT_RANGE, "bytes " + start + "-" + end + "/" + totalSize) .contentLength(regionLength) 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 {