41 lines
864 B
Java
41 lines
864 B
Java
|
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);
|
||
|
|
}
|
||
|
|
}
|