serverscanning/capnproto-java-rpc
16
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

simplify iteration of (weak) question table

Browse source
This commit is contained in:
Vaci Koblizek 2020-11-12 20:40:18 +00:00
parent c8c8c8a085
commit af47f1a825
1 changed files with 7 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
runtime-rpc/src/main/java/org/capnproto/RpcState.java
View file

@ -117,7 +117,7 @@ final class RpcState<VatId> {
}
}
class QuestionExportTable implements Iterable<Question> {
class QuestionExportTable {
private final HashMap<Integer, WeakReference<Question>> slots = new HashMap<>();
private final Queue<Integer> freeIds = new PriorityQueue<>();
private int max = 0;
@ -146,20 +146,12 @@ final class RpcState<VatId> {
return value;
}
@Override
public Iterator<Question> iterator() {
return this.slots.values()
.stream()
.map(Reference::get)
.filter(Objects::nonNull)
.iterator();
}
@Override
public void forEach(Consumer<? super Question> action) {
var iter = this.iterator();
while (iter.hasNext()) {
action.accept(iter.next());
for (var entry: this.slots.values()) {
var question = entry.get();
if (question != null) {
action.accept(question);
}
}
}
}
@ -297,9 +289,7 @@ final class RpcState<VatId> {
var networkExc = RpcException.disconnected(exc.getMessage());
// All current questions complete with exceptions.
for (var question: questions) {
question.reject(networkExc);
}
questions.forEach(question -> question.reject(networkExc));
List<PipelineHook> pipelinesToRelease = new ArrayList<>();
List<ClientHook> clientsToRelease = new ArrayList<>();