[BE] refactor: 하이라이트 매핑 로직 간소화

backend/src/main/java/reviewme/highlight/domain/HighlightedLines.java

@@ -2,6 +2,8 @@
 
 import java.util.Arrays;
 import java.util.List;
+import java.util.function.Function;
+import java.util.stream.IntStream;
 import lombok.Getter;
 import reviewme.highlight.domain.exception.InvalidHighlightLineIndexException;
 import reviewme.highlight.domain.exception.NegativeHighlightLineIndexException;
@@ -37,4 +39,12 @@ private void validateLineIndexRange(int lineIndex) {
             throw new InvalidHighlightLineIndexException(lineIndex, lines.size());
         }
     }
+
+    public List<Highlight> toHighlights(long answerId) {
+        return IntStream.range(0, lines.size())
+                .mapToObj(lineIndex -> lines.get(lineIndex).getRanges().stream()
+                        .map(range -> new Highlight(answerId, lineIndex, range)))
+                .flatMap(Function.identity())
+                .toList();
+    }
 }

backend/src/main/java/reviewme/highlight/service/HighlightService.java

@@ -30,7 +30,6 @@ public void editHighlight(HighlightsRequest highlightsRequest, ReviewGroup revie
 
         Set<Long> answerIds = answerRepository.findIdsByQuestionId(highlightsRequest.questionId());
         highlightRepository.deleteAllByAnswerIds(answerIds);
-
         highlightRepository.saveAll(highlights);
     }
 }

backend/src/main/java/reviewme/highlight/service/dto/HighlightRequest.java

@@ -4,6 +4,8 @@
 import jakarta.validation.constraints.NotEmpty;
 import jakarta.validation.constraints.NotNull;
 import java.util.List;
+import java.util.stream.Stream;
+import reviewme.highlight.service.mapper.HighlightFragment;
 
 public record HighlightRequest(
 
@@ -13,4 +15,15 @@ public record HighlightRequest(
         @Valid @NotEmpty(message = "하이라이트 된 라인을 입력해주세요.")
         List<HighlightedLineRequest> lines
 ) {
+    public List<HighlightFragment> toFragments() {
+        return lines.stream()
+                .flatMap(this::mapRangesToFragment)
+                .toList();
+    }
+
+    private Stream<HighlightFragment> mapRangesToFragment(HighlightedLineRequest line) {
+        return line.ranges()
+                .stream()
+                .map(range -> new HighlightFragment(answerId, line.index(), range.startIndex(), range.endIndex()));
+    }
 }

backend/src/main/java/reviewme/highlight/service/dto/HighlightsRequest.java

@@ -3,6 +3,7 @@
 import jakarta.validation.Valid;
 import jakarta.validation.constraints.NotNull;
 import java.util.List;
+import reviewme.highlight.service.mapper.HighlightFragment;
 
 public record HighlightsRequest(
 
@@ -20,4 +21,10 @@ public List<Long> getUniqueAnswerIds() {
                 .distinct()
                 .toList();
     }
+
+    public List<HighlightFragment> toFragments() {
+        return highlights.stream()
+                .flatMap(request -> request.toFragments().stream())
+                .toList();
+    }
 }

backend/src/main/java/reviewme/highlight/service/mapper/HighlightFragment.java

@@ -0,0 +1,4 @@
+package reviewme.highlight.service.mapper;
+
+public record HighlightFragment(long answerId, int lineIndex, int startIndex, int endIndex) {
+}

backend/src/main/java/reviewme/highlight/service/mapper/HighlightMapper.java

@@ -1,21 +1,14 @@
 package reviewme.highlight.service.mapper;
 
-import java.util.ArrayList;
 import java.util.List;
 import java.util.Map;
-import java.util.Map.Entry;
 import java.util.stream.Collectors;
 import lombok.RequiredArgsConstructor;
 import org.springframework.stereotype.Component;
-import reviewme.highlight.domain.HighlightedLines;
-import reviewme.highlight.domain.HighlightedLine;
 import reviewme.highlight.domain.Highlight;
-import reviewme.highlight.domain.HighlightRange;
-import reviewme.highlight.service.dto.HighlightIndexRangeRequest;
-import reviewme.highlight.service.dto.HighlightRequest;
-import reviewme.highlight.service.dto.HighlightedLineRequest;
+import reviewme.highlight.domain.HighlightedLines;
 import reviewme.highlight.service.dto.HighlightsRequest;
-import reviewme.review.domain.Answer;
+import reviewme.review.domain.TextAnswer;
 import reviewme.review.repository.TextAnswerRepository;
 
 @Component
@@ -25,53 +18,19 @@ public class HighlightMapper {
     private final TextAnswerRepository textAnswerRepository;
 
     public List<Highlight> mapToHighlights(HighlightsRequest highlightsRequest) {
-        Map<Long, HighlightedLines> answerHighlightLines = textAnswerRepository
+        Map<Long, HighlightedLines> answerIdHighlightedLines = textAnswerRepository
                 .findAllById(highlightsRequest.getUniqueAnswerIds())
                 .stream()
-                .collect(Collectors.toMap(Answer::getId, answer -> new HighlightedLines(answer.getContent())));
-        addIndexRanges(highlightsRequest, answerHighlightLines);
-        return mapLinesToHighlights(answerHighlightLines);
-    }
-
-    private void addIndexRanges(HighlightsRequest highlightsRequest, Map<Long, HighlightedLines> answerHighlightLines) {
-        for (HighlightRequest highlightRequest : highlightsRequest.highlights()) {
-            HighlightedLines highlightedLines = answerHighlightLines.get(highlightRequest.answerId());
-            addIndexRangesForAnswer(highlightRequest, highlightedLines);
-        }
-    }
-
-    private void addIndexRangesForAnswer(HighlightRequest highlightRequest, HighlightedLines highlightedLines) {
-        for (HighlightedLineRequest lineRequest : highlightRequest.lines()) {
-            int lineIndex = lineRequest.index();
-            for (HighlightIndexRangeRequest rangeRequest : lineRequest.ranges()) {
-                highlightedLines.addRange(lineIndex, rangeRequest.startIndex(), rangeRequest.endIndex());
-            }
-        }
-    }
-
-    private List<Highlight> mapLinesToHighlights(Map<Long, HighlightedLines> answerHighlightLines) {
-        List<Highlight> highlights = new ArrayList<>();
-        for (Entry<Long, HighlightedLines> answerHighlightLine : answerHighlightLines.entrySet()) {
-            createHighlightsForAnswer(answerHighlightLine, highlights);
-        }
-        return highlights;
-    }
+                .collect(Collectors.toMap(TextAnswer::getId, answer -> new HighlightedLines(answer.getContent())));
 
-    private void createHighlightsForAnswer(Entry<Long, HighlightedLines> answerHighlightLine,
-                                           List<Highlight> highlights) {
-        long answerId = answerHighlightLine.getKey();
-        List<HighlightedLine> highlightedLines = answerHighlightLine.getValue().getLines();
-
-        for (int lineIndex = 0; lineIndex < highlightedLines.size(); lineIndex++) {
-            createHighlightForLine(highlightedLines, lineIndex, answerId, highlights);
+        for (HighlightFragment fragment : highlightsRequest.toFragments()) {
+            HighlightedLines highlightedLines = answerIdHighlightedLines.get(fragment.answerId());
+            highlightedLines.addRange(fragment.lineIndex(), fragment.startIndex(), fragment.endIndex());
         }
-    }
 
-    private void createHighlightForLine(List<HighlightedLine> highlightedLines, int lineIndex, long answerId,
-                                        List<Highlight> highlights) {
-        for (HighlightRange range : highlightedLines.get(lineIndex).getRanges()) {
-            Highlight highlight = new Highlight(answerId, lineIndex, range);
-            highlights.add(highlight);
-        }
+        return answerIdHighlightedLines.entrySet()
+                .stream()
+                .flatMap(entry -> entry.getValue().toHighlights(entry.getKey()).stream())
+                .toList();
     }
 }

backend/src/test/java/reviewme/highlight/domain/HighlightedLinesTest.java

@@ -84,4 +84,21 @@ class HighlightedLinesTest {
         assertThatCode(() -> highlightedLines.addRange(invalidLineIndex, 0, 1))
                 .isInstanceOf(InvalidHighlightLineIndexException.class);
     }
+
+    @Test
+    void 하이라이트가_존재하는_부분만_엔티티로_변환한다() {
+        // given
+        HighlightedLines lines = new HighlightedLines("0\n11\n222");
+        lines.addRange(0, 0, 0);
+        lines.addRange(2, 2, 2);
+
+        // when
+        List<Highlight> highlights = lines.toHighlights(1L);
+
+        // then
+        assertThat(highlights).containsExactly(
+                new Highlight(1L, 0, new HighlightRange(0, 0)),
+                new Highlight(1L, 2, new HighlightRange(2, 2))
+        );
+    }
 }