diff --git a/src/main/java/umc/codeplay/dto/SQSMessageDTO.java b/src/main/java/umc/codeplay/dto/SQSMessageDTO.java index 6fc358d..7581629 100644 --- a/src/main/java/umc/codeplay/dto/SQSMessageDTO.java +++ b/src/main/java/umc/codeplay/dto/SQSMessageDTO.java @@ -14,6 +14,7 @@ public class SQSMessageDTO { @AllArgsConstructor public static class HarmonyMessageDTO { String key; + Long musicId; Long taskId; String jobType; } @@ -24,6 +25,7 @@ public static class HarmonyMessageDTO { @AllArgsConstructor public static class TrackMessageDTO { String key; + Long musicId; Long taskId; String jobType; @@ -34,6 +36,7 @@ public static class TrackMessageDTO { @Setter public static class RemixMessageDTO { String key; + Long musicId; Long taskId; String jobType; diff --git a/src/main/java/umc/codeplay/service/ModelService.java b/src/main/java/umc/codeplay/service/ModelService.java index 751fabc..a422f3b 100644 --- a/src/main/java/umc/codeplay/service/ModelService.java +++ b/src/main/java/umc/codeplay/service/ModelService.java @@ -75,7 +75,7 @@ public Task sendTrackMessage(Music music, String config) { } switch (config) { - case "vocals", "bass", "drums", "none": // TODO: 6-stems guitar, piano 테스트 후 추가 + case "vocals", "bass", "drums", "none", "guitar", "piano": break; default: throw new GeneralException(ErrorStatus.INVALID_CONFIG); @@ -86,6 +86,7 @@ public Task sendTrackMessage(Music music, String config) { queueName, SQSMessageDTO.TrackMessageDTO.builder() .key(music.getTitle()) + .musicId(music.getId()) .taskId(task.getId()) .jobType(JobType.TRACK.toString()) .twoStemConfig(config) @@ -106,6 +107,7 @@ public Task sendHarmonyMessage(Music music) { queueName, SQSMessageDTO.HarmonyMessageDTO.builder() .key(music.getTitle()) + .musicId(music.getId()) .taskId(task.getId()) .jobType(JobType.HARMONY.toString()) .build()); @@ -151,6 +153,7 @@ public Task sendRemixMessage(Music music, MemberRequestDTO.RemixTaskDTO request) Task task = taskService.addTask(newRemix); remixPayLoad.setKey(music.getTitle()); + remixPayLoad.setMusicId(music.getId()); remixPayLoad.setTaskId(task.getId()); remixPayLoad.setJobType(JobType.REMIX.toString());