implement import and export tables
This commit is contained in:
parent
76c81a76a2
commit
10f8f5e7d5
3 changed files with 135 additions and 1 deletions
56
runtime/src/main/java/org/capnproto/ExportTable.java
Normal file
56
runtime/src/main/java/org/capnproto/ExportTable.java
Normal file
|
@ -0,0 +1,56 @@
|
||||||
|
package org.capnproto;
|
||||||
|
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.Iterator;
|
||||||
|
import java.util.PriorityQueue;
|
||||||
|
import java.util.Queue;
|
||||||
|
import java.util.function.Consumer;
|
||||||
|
|
||||||
|
abstract class ExportTable<T> implements Iterable<T> {
|
||||||
|
|
||||||
|
final HashMap<Integer, T> slots = new HashMap<>();
|
||||||
|
final Queue<Integer> freeIds = new PriorityQueue<>();
|
||||||
|
int max = 0;
|
||||||
|
|
||||||
|
protected abstract T newExportable();
|
||||||
|
|
||||||
|
public T find(int id) {
|
||||||
|
return slots.get(id);
|
||||||
|
}
|
||||||
|
|
||||||
|
public T erase(int id, T entry) {
|
||||||
|
var value = slots.get(id);
|
||||||
|
if (value == entry) {
|
||||||
|
freeIds.add(id);
|
||||||
|
return slots.remove(id);
|
||||||
|
} else {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public T next() {
|
||||||
|
if (freeIds.isEmpty()) {
|
||||||
|
var id = max;
|
||||||
|
max++;
|
||||||
|
var value = newExportable();
|
||||||
|
slots.put(id, value);
|
||||||
|
return value;
|
||||||
|
} else {
|
||||||
|
var id = freeIds.remove();
|
||||||
|
var value = newExportable();
|
||||||
|
slots.put(id, value);
|
||||||
|
return value;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Iterator<T> iterator() {
|
||||||
|
return slots.values().iterator();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void forEach(Consumer<? super T> action) {
|
||||||
|
slots.values().forEach(action);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
40
runtime/src/main/java/org/capnproto/ImportTable.java
Normal file
40
runtime/src/main/java/org/capnproto/ImportTable.java
Normal file
|
@ -0,0 +1,40 @@
|
||||||
|
package org.capnproto;
|
||||||
|
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.Iterator;
|
||||||
|
import java.util.function.Consumer;
|
||||||
|
|
||||||
|
abstract class ImportTable<T> implements Iterable<T> {
|
||||||
|
|
||||||
|
private final HashMap<Integer, T> slots = new HashMap<>();
|
||||||
|
|
||||||
|
protected abstract T newImportable();
|
||||||
|
|
||||||
|
public T put(int id) {
|
||||||
|
return slots.computeIfAbsent(id, key -> newImportable());
|
||||||
|
}
|
||||||
|
|
||||||
|
public T find(int id) {
|
||||||
|
return slots.get(id);
|
||||||
|
}
|
||||||
|
|
||||||
|
public T erase(int id, T entry) {
|
||||||
|
var removed = slots.remove(id, entry);
|
||||||
|
assert removed;
|
||||||
|
return entry;
|
||||||
|
}
|
||||||
|
|
||||||
|
public T erase(int id) {
|
||||||
|
return slots.remove(id);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Iterator<T> iterator() {
|
||||||
|
return slots.values().iterator();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void forEach(Consumer<? super T> action) {
|
||||||
|
slots.values().forEach(action);
|
||||||
|
}
|
||||||
|
}
|
|
@ -1,7 +1,8 @@
|
||||||
package org.capnproto;
|
package org.capnproto;
|
||||||
|
|
||||||
import java.util.List;
|
import java.util.*;
|
||||||
import java.util.concurrent.CompletableFuture;
|
import java.util.concurrent.CompletableFuture;
|
||||||
|
import java.util.function.Consumer;
|
||||||
|
|
||||||
final class RpcState {
|
final class RpcState {
|
||||||
|
|
||||||
|
@ -46,6 +47,43 @@ final class RpcState {
|
||||||
CompletableFuture<java.lang.Void> fulfiller;
|
CompletableFuture<java.lang.Void> fulfiller;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private final ExportTable<Export> exports = new ExportTable<Export>() {
|
||||||
|
@Override
|
||||||
|
protected Export newExportable() {
|
||||||
|
return new Export();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
private final ExportTable<Question> questions = new ExportTable<Question>() {
|
||||||
|
@Override
|
||||||
|
protected Question newExportable() {
|
||||||
|
return new Question();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
private final ImportTable<Answer> answers = new ImportTable<Answer>() {
|
||||||
|
@Override
|
||||||
|
protected Answer newImportable() {
|
||||||
|
return new Answer();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
private final ImportTable<Import> imports = new ImportTable<Import>() {
|
||||||
|
@Override
|
||||||
|
protected Import newImportable() {
|
||||||
|
return new Import();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
private final ExportTable<Embargo> embargos = new ExportTable<Embargo>() {
|
||||||
|
@Override
|
||||||
|
protected Embargo newExportable() {
|
||||||
|
return new Embargo();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
private final HashMap<ClientHook, Integer> exportsByCap = new HashMap<>();
|
||||||
|
|
||||||
interface RpcResponse extends ResponseHook {
|
interface RpcResponse extends ResponseHook {
|
||||||
AnyPointer.Reader getResults();
|
AnyPointer.Reader getResults();
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue