-
Notifications
You must be signed in to change notification settings - Fork 1.9k
/
ShortBranchIssuesLoader.java
98 lines (88 loc) · 3.95 KB
/
ShortBranchIssuesLoader.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
/*
* SonarQube
* Copyright (C) 2009-2017 SonarSource SA
* mailto:info AT sonarsource DOT com
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 3 of the License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program; if not, write to the Free Software Foundation,
* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
*/
package org.sonar.server.computation.task.projectanalysis.issue;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;
import org.sonar.core.issue.DefaultIssue;
import org.sonar.core.issue.ShortBranchIssue;
import org.sonar.db.DbClient;
import org.sonar.db.DbSession;
import org.sonar.db.component.ComponentDto;
import org.sonar.db.issue.IssueChangeDto;
import org.sonar.db.issue.IssueDto;
import org.sonar.db.issue.ShortBranchIssueDto;
import org.sonar.server.computation.task.projectanalysis.component.Component;
import org.sonar.server.computation.task.projectanalysis.component.ShortBranchComponentsWithIssues;
import static java.util.stream.Collectors.groupingBy;
import static java.util.stream.Collectors.toMap;
public class ShortBranchIssuesLoader {
private final ShortBranchComponentsWithIssues shortBranchComponentsWithIssues;
private final DbClient dbClient;
public ShortBranchIssuesLoader(ShortBranchComponentsWithIssues shortBranchComponentsWithIssues, DbClient dbClient) {
this.shortBranchComponentsWithIssues = shortBranchComponentsWithIssues;
this.dbClient = dbClient;
}
public Collection<ShortBranchIssue> loadCandidateIssuesForMergingInTargetBranch(Component component) {
String componentKey = ComponentDto.removeBranchFromKey(component.getKey());
Set<String> uuids = shortBranchComponentsWithIssues.getUuids(componentKey);
if (uuids.isEmpty()) {
return Collections.emptyList();
}
try (DbSession session = dbClient.openSession(false)) {
return dbClient.issueDao().selectResolvedOrConfirmedByComponentUuids(session, uuids)
.stream()
.map(ShortBranchIssueDto::toShortBranchIssue)
.collect(Collectors.toList());
}
}
public Map<ShortBranchIssue, DefaultIssue> loadDefaultIssuesWithChanges(Collection<ShortBranchIssue> lightIssues) {
if (lightIssues.isEmpty()) {
return Collections.emptyMap();
}
Map<String, ShortBranchIssue> issuesByKey = lightIssues.stream().collect(Collectors.toMap(ShortBranchIssue::getKey, i -> i));
try (DbSession session = dbClient.openSession(false)) {
Map<String, List<IssueChangeDto>> changeDtoByIssueKey = dbClient.issueChangeDao()
.selectByIssueKeys(session, issuesByKey.keySet()).stream().collect(groupingBy(IssueChangeDto::getIssueKey));
return dbClient.issueDao().selectByKeys(session, issuesByKey.keySet())
.stream()
.map(IssueDto::toDefaultIssue)
.peek(i -> setChanges(changeDtoByIssueKey, i))
.collect(toMap(i -> issuesByKey.get(i.key()), i -> i));
}
}
private static void setChanges(Map<String, List<IssueChangeDto>> changeDtoByIssueKey, DefaultIssue i) {
changeDtoByIssueKey.get(i.key()).forEach(c -> {
switch (c.getChangeType()) {
case IssueChangeDto.TYPE_FIELD_CHANGE:
i.addChange(c.toFieldDiffs());
break;
case IssueChangeDto.TYPE_COMMENT:
i.addComment(c.toComment());
break;
default:
throw new IllegalStateException("Unknow change type: " + c.getChangeType());
}
});
}
}