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
99
100
101
|
/*
* SonarQube
* Copyright (C) 2009-2025 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.process.cluster.hz;
import com.hazelcast.cluster.Member;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.Set;
import java.util.stream.Collectors;
import static org.sonar.process.cluster.hz.HazelcastMember.Attribute.NODE_NAME;
/**
* Answer of {@link DistributedCall}, aggregating the answers from
* all the target members.
*/
public class DistributedAnswer<T> {
private final Map<Member, T> answers = new HashMap<>();
private final Set<Member> timedOutMembers = new HashSet<>();
private final Map<Member, Exception> failedMembers = new HashMap<>();
public Optional<T> getAnswer(Member member) {
return Optional.ofNullable(answers.get(member));
}
public boolean hasTimedOut(Member member) {
return timedOutMembers.contains(member);
}
public Optional<Exception> getFailed(Member member) {
return Optional.ofNullable(failedMembers.get(member));
}
public Collection<Member> getMembers() {
List<Member> members = new ArrayList<>();
members.addAll(answers.keySet());
members.addAll(timedOutMembers);
members.addAll(failedMembers.keySet());
return members;
}
public void setAnswer(Member member, T answer) {
this.answers.put(member, answer);
}
public void setTimedOut(Member member) {
this.timedOutMembers.add(member);
}
public void setFailed(Member member, Exception e) {
failedMembers.put(member, e);
}
public void propagateExceptions() {
if (!failedMembers.isEmpty()) {
String failedMemberNames = failedMembers.keySet().stream()
.map(m -> m.getAttribute(NODE_NAME.getKey()))
.collect(Collectors.joining(", "));
throw new IllegalStateException("Distributed cluster action in cluster nodes " + failedMemberNames + " (other nodes may have timed out)",
failedMembers.values().iterator().next());
}
if (!timedOutMembers.isEmpty()) {
String timedOutMemberNames = timedOutMembers.stream()
.map(m -> m.getAttribute(NODE_NAME.getKey()))
.collect(Collectors.joining(", "));
throw new IllegalStateException("Distributed cluster action timed out in cluster nodes " + timedOutMemberNames);
}
}
/**
* Returns any answer. No guarantees are made on the order. Use this method if you only expect exactly one answer.
* @return the first answer, if any
*/
public Optional<T> getSingleAnswer() {
return answers.values().stream().findFirst();
}
}
|
