리팩토링

Daemawiki · Feb 13, 2024 · 890b55b · 890b55b
1 parent 57a13bc
commit 890b55b
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 10 deletions.
diff --git a/src/main/java/com/example/daemawiki/domain/document/service/GetDocument.java b/src/main/java/com/example/daemawiki/domain/document/service/GetDocument.java
@@ -2,28 +2,23 @@
 
 import com.example.daemawiki.domain.document.dto.response.GetDocumentResponse;
 import com.example.daemawiki.domain.document.service.facade.DocumentFacade;
+import com.example.daemawiki.domain.document.service.mapper.DocumentMapper;
 import org.springframework.stereotype.Service;
 import reactor.core.publisher.Mono;
 
 @Service
 public class GetDocument {
     private final DocumentFacade documentFacade;
+    private final DocumentMapper documentMapper;
 
-    public GetDocument(DocumentFacade documentFacade) {
+    public GetDocument(DocumentFacade documentFacade, DocumentMapper documentMapper) {
         this.documentFacade = documentFacade;
+        this.documentMapper = documentMapper;
     }
 
     public Mono<GetDocumentResponse> execute(String id) {
         return documentFacade.findDocumentById(id)
-                        .map(document -> GetDocumentResponse.builder()
-                                .title(document.getTitle())
-                                .type(document.getType())
-                                .dateTime(document.getDateTime())
-                                .groups(document.getGroups())
-                                .editor(document.getEditor())
-                                .content(document.getContent())
-                                .build());
+                        .flatMap(documentMapper::defaultDocumentToGetResponse);
     }
 
-
 }
diff --git a/src/main/java/com/example/daemawiki/domain/document/service/mapper/DocumentMapper.java b/src/main/java/com/example/daemawiki/domain/document/service/mapper/DocumentMapper.java
@@ -0,0 +1,22 @@
+package com.example.daemawiki.domain.document.service.mapper;
+
+import com.example.daemawiki.domain.document.dto.response.GetDocumentResponse;
+import com.example.daemawiki.domain.document.model.DefaultDocument;
+import org.springframework.stereotype.Component;
+import reactor.core.publisher.Mono;
+
+@Component
+public class DocumentMapper {
+
+    public Mono<GetDocumentResponse> defaultDocumentToGetResponse(DefaultDocument document) {
+        return Mono.justOrEmpty(GetDocumentResponse.builder()
+                .title(document.getTitle())
+                .type(document.getType())
+                .dateTime(document.getDateTime())
+                .groups(document.getGroups())
+                .editor(document.getEditor())
+                .content(document.getContent())
+                .build());
+    }
+
+}