[Feature] 퀘스트 생성 시 AI서버에 라벨링 요청을 한다 (#100)

feature: 퀘스트 생성 시 AI서버에 라벨링 요청을 한다

Author: vectorch9

build.gradle

@@ -29,6 +29,7 @@ dependencies {
     implementation 'org.springframework.boot:spring-boot-starter-validation:3.1.2'
     implementation 'org.springframework.boot:spring-boot-starter-web:3.1.2'
     implementation 'org.springframework.boot:spring-boot-starter-actuator:3.1.2'
+    implementation 'org.springframework.boot:spring-boot-starter-webflux:3.1.2'
 
     implementation 'org.springdoc:springdoc-openapi-starter-webmvc-ui:2.2.0'
     asciidoctorExt 'org.springframework.restdocs:spring-restdocs-asciidoctor'

src/main/java/daybyquest/global/config/PropertiesConfig.java

@@ -1,11 +1,12 @@
 package daybyquest.global.config;
 
+import daybyquest.global.infra.AiServerProperties;
 import daybyquest.image.domain.BaseImageProperties;
 import org.springframework.boot.context.properties.EnableConfigurationProperties;
 import org.springframework.context.annotation.Configuration;
 
 @Configuration
-@EnableConfigurationProperties(BaseImageProperties.class)
+@EnableConfigurationProperties({BaseImageProperties.class, AiServerProperties.class})
 public class PropertiesConfig {
 
 }

src/main/java/daybyquest/global/config/WebClientConfig.java

@@ -0,0 +1,20 @@
+package daybyquest.global.config;
+
+import daybyquest.global.infra.AiServerProperties;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.MediaType;
+import org.springframework.web.reactive.function.client.WebClient;
+
+@Configuration
+public class WebClientConfig {
+
+    @Bean
+    public WebClient webClient(final AiServerProperties properties) {
+        return WebClient.builder()
+                .baseUrl(properties.getDomain())
+                .defaultHeader(HttpHeaders.CONTENT_TYPE, MediaType.APPLICATION_JSON_VALUE)
+                .build();
+    }
+}

src/main/java/daybyquest/global/infra/AiServerProperties.java

@@ -0,0 +1,18 @@
+package daybyquest.global.infra;
+
+
+import lombok.Getter;
+import org.springframework.boot.context.properties.ConfigurationProperties;
+import org.springframework.boot.context.properties.bind.ConstructorBinding;
+
+@Getter
+@ConfigurationProperties(prefix = "ai")
+public class AiServerProperties {
+
+    private final String domain;
+
+    @ConstructorBinding
+    public AiServerProperties(final String domain) {
+        this.domain = domain;
+    }
+}

src/main/java/daybyquest/quest/application/QuestClient.java

@@ -0,0 +1,8 @@
+package daybyquest.quest.application;
+
+import java.util.List;
+
+public interface QuestClient {
+
+    void requestLabels(final Long questId, final List<String> identifiers);
+}

src/main/java/daybyquest/quest/application/SaveQuestService.java

@@ -27,18 +27,23 @@ public class SaveQuestService {
 
     private final ImageIdentifierGenerator generator;
 
+    private final QuestClient questClient;
+
     public SaveQuestService(final Quests quests, final Badges badges, final Images images,
-            final ImageIdentifierGenerator generator) {
+            final ImageIdentifierGenerator generator, final QuestClient questClient) {
         this.quests = quests;
         this.badges = badges;
         this.images = images;
         this.generator = generator;
+        this.questClient = questClient;
     }
 
     @Transactional
     public Long invoke(final SaveQuestRequest request, final List<MultipartFile> files) {
         final Quest quest = toEntity(request, toImageList(files));
-        return quests.save(quest);
+        final Long questId = quests.save(quest);
+        questClient.requestLabels(questId, quest.getImages().stream().map(Image::getIdentifier).toList());
+        return questId;
     }
 
     private List<Image> toImageList(final List<MultipartFile> files) {

src/main/java/daybyquest/quest/dto/request/LabelQuestRequest.java

@@ -0,0 +1,7 @@
+package daybyquest.quest.dto.request;
+
+import java.util.List;
+
+public record LabelQuestRequest(List<String> imageIdentifiers) {
+
+}

src/main/java/daybyquest/quest/infra/QuestStubClient.java

@@ -0,0 +1,23 @@
+package daybyquest.quest.infra;
+
+import daybyquest.quest.application.QuestClient;
+import java.util.List;
+import org.springframework.context.annotation.Profile;
+import org.springframework.stereotype.Component;
+import org.springframework.web.reactive.function.client.WebClient;
+
+@Component
+@Profile("!prod")
+public class QuestStubClient implements QuestClient {
+
+    private final WebClient webClient;
+
+    public QuestStubClient(final WebClient webClient) {
+        this.webClient = webClient;
+    }
+
+    @Override
+    public void requestLabels(final Long questId, final List<String> identifiers) {
+        return;
+    }
+}

src/main/java/daybyquest/quest/infra/QuestWebClient.java

@@ -0,0 +1,38 @@
+package daybyquest.quest.infra;
+
+import daybyquest.global.error.exception.BadRequestException;
+import daybyquest.quest.application.QuestClient;
+import daybyquest.quest.dto.request.LabelQuestRequest;
+import java.util.List;
+import org.springframework.context.annotation.Profile;
+import org.springframework.http.HttpStatusCode;
+import org.springframework.stereotype.Component;
+import org.springframework.web.reactive.function.client.WebClient;
+
+@Component
+@Profile("prod")
+public class QuestWebClient implements QuestClient {
+
+    private final WebClient webClient;
+
+    public QuestWebClient(final WebClient webClient) {
+        this.webClient = webClient;
+    }
+
+    @Override
+    public void requestLabels(final Long questId, final List<String> identifiers) {
+        final LabelQuestRequest request = new LabelQuestRequest(identifiers);
+        webClient.post()
+                .uri(uriBuilder -> uriBuilder.pathSegment("quest", questId.toString(), "shot").build())
+                .bodyValue(request)
+                .retrieve()
+                .onStatus(HttpStatusCode::is4xxClientError, response -> {
+                    throw new BadRequestException();
+                })
+                .onStatus(HttpStatusCode::is5xxServerError, response -> {
+                    throw new BadRequestException();
+                })
+                .toBodilessEntity()
+                .block();
+    }
+}

src/main/resources/application-web.yml

@@ -12,6 +12,9 @@ image:
 tracker:
   days: 60
 
+ai:
+  domain: http://localhost:8080
+
 spring:
   servlet:
     multipart: