|
6 | 6 | import java.util.Map; |
7 | 7 | import java.util.stream.Collectors; |
8 | 8 |
|
9 | | -import org.springframework.data.domain.PageRequest; |
10 | | -import org.springframework.data.domain.Pageable; |
11 | 9 | import org.springframework.stereotype.Service; |
12 | 10 | import org.springframework.transaction.annotation.Transactional; |
13 | 11 |
|
| 12 | +import in.koreatech.koin.common.event.ArticleKeywordEvent; |
14 | 13 | import in.koreatech.koin.domain.community.article.model.Article; |
15 | 14 | import in.koreatech.koin.domain.community.keyword.model.ArticleKeyword; |
16 | 15 | import in.koreatech.koin.domain.community.keyword.model.ArticleKeywordUserMap; |
17 | | -import in.koreatech.koin.common.event.ArticleKeywordEvent; |
18 | 16 | import in.koreatech.koin.domain.community.keyword.repository.ArticleKeywordRepository; |
19 | 17 | import in.koreatech.koin.domain.community.keyword.repository.ArticleKeywordUserMapRepository; |
20 | 18 | import lombok.RequiredArgsConstructor; |
|
24 | 22 | @Transactional(readOnly = true) |
25 | 23 | public class KeywordExtractor { |
26 | 24 |
|
27 | | - private static final int KEYWORD_BATCH_SIZE = 100; |
28 | | - |
29 | 25 | private final ArticleKeywordRepository articleKeywordRepository; |
30 | 26 | private final ArticleKeywordUserMapRepository articleKeywordUserMapRepository; |
31 | 27 |
|
32 | 28 | public List<ArticleKeywordEvent> matchKeyword(List<Article> articles, Integer authorId) { |
33 | | - Map<Integer, Map<Integer, String>> matchedKeywordByUserIdByArticleId = new LinkedHashMap<>(); |
34 | | - int offset = 0; |
| 29 | + List<ArticleKeyword> keywords = articleKeywordRepository.findAll(); |
35 | 30 |
|
36 | | - while (true) { |
37 | | - Pageable pageable = PageRequest.of(offset / KEYWORD_BATCH_SIZE, KEYWORD_BATCH_SIZE); |
38 | | - List<ArticleKeyword> keywords = articleKeywordRepository.findAll(pageable); |
| 31 | + if (keywords.isEmpty()) { |
| 32 | + return List.of(); |
| 33 | + } |
39 | 34 |
|
40 | | - if (keywords.isEmpty()) { |
41 | | - break; |
42 | | - } |
43 | | - List<Integer> keywordIds = keywords.stream() |
44 | | - .map(ArticleKeyword::getId) |
45 | | - .toList(); |
46 | | - Map<Integer, List<ArticleKeywordUserMap>> userMapsByKeywordId = articleKeywordUserMapRepository |
47 | | - .findAllByArticleKeywordIdIn(keywordIds) |
48 | | - .stream() |
49 | | - .filter(keywordUserMap -> !keywordUserMap.getIsDeleted()) |
50 | | - .collect(Collectors.groupingBy( |
51 | | - keywordUserMap -> keywordUserMap.getArticleKeyword().getId(), |
52 | | - LinkedHashMap::new, |
53 | | - Collectors.toList() |
54 | | - )); |
| 35 | + List<Integer> keywordIds = keywords.stream() |
| 36 | + .map(ArticleKeyword::getId) |
| 37 | + .toList(); |
| 38 | + Map<Integer, List<ArticleKeywordUserMap>> userMapsByKeywordId = articleKeywordUserMapRepository |
| 39 | + .findAllByArticleKeywordIdIn(keywordIds) |
| 40 | + .stream() |
| 41 | + .filter(keywordUserMap -> !keywordUserMap.getIsDeleted()) |
| 42 | + .collect(Collectors.groupingBy( |
| 43 | + keywordUserMap -> keywordUserMap.getArticleKeyword().getId(), |
| 44 | + LinkedHashMap::new, |
| 45 | + Collectors.toList() |
| 46 | + )); |
55 | 47 |
|
56 | | - for (Article article : articles) { |
57 | | - String title = article.getTitle(); |
58 | | - for (ArticleKeyword keyword : keywords) { |
59 | | - if (!title.contains(keyword.getKeyword())) { |
60 | | - continue; |
61 | | - } |
62 | | - Map<Integer, String> matchedKeywordByUserId = matchedKeywordByUserIdByArticleId |
63 | | - .computeIfAbsent(article.getId(), ignored -> new LinkedHashMap<>()); |
| 48 | + Map<Integer, Map<Integer, String>> matchedKeywordByUserIdByArticleId = new LinkedHashMap<>(); |
| 49 | + for (Article article : articles) { |
| 50 | + String title = article.getTitle(); |
| 51 | + for (ArticleKeyword keyword : keywords) { |
| 52 | + if (!title.contains(keyword.getKeyword())) { |
| 53 | + continue; |
| 54 | + } |
| 55 | + Map<Integer, String> matchedKeywordByUserId = matchedKeywordByUserIdByArticleId |
| 56 | + .computeIfAbsent(article.getId(), ignored -> new LinkedHashMap<>()); |
64 | 57 |
|
65 | | - for (ArticleKeywordUserMap keywordUserMap : |
66 | | - userMapsByKeywordId.getOrDefault(keyword.getId(), List.of())) { |
67 | | - Integer userId = keywordUserMap.getUser().getId(); |
68 | | - matchedKeywordByUserId.merge( |
69 | | - userId, |
70 | | - keyword.getKeyword(), |
71 | | - this::pickHigherPriorityKeyword |
72 | | - ); |
73 | | - } |
| 58 | + for (ArticleKeywordUserMap keywordUserMap : |
| 59 | + userMapsByKeywordId.getOrDefault(keyword.getId(), List.of())) { |
| 60 | + Integer userId = keywordUserMap.getUser().getId(); |
| 61 | + matchedKeywordByUserId.merge( |
| 62 | + userId, |
| 63 | + keyword.getKeyword(), |
| 64 | + this::pickHigherPriorityKeyword |
| 65 | + ); |
74 | 66 | } |
75 | 67 | } |
76 | | - offset += KEYWORD_BATCH_SIZE; |
77 | 68 | } |
78 | 69 |
|
79 | 70 | List<ArticleKeywordEvent> keywordEvents = new ArrayList<>(); |
|
0 commit comments