import java.util.Iterator;
import java.util.Map.Entry;
+import java.util.Optional;
-final class LNode<K, V> extends MainNode<K, V> implements Iterable<Entry<K, V>> {
+final class LNode<K, V> extends MainNode<K, V> {
private final ListMap<K, V> listmap;
private LNode(final ListMap<K, V> listmap) {
return new LNode<>(listmap.add(k, v));
}
+ // FIXME: can we also get the hashcode ?
MainNode<K, V> removed(final K k, final TrieMap<K, V> ct) {
- final ListMap<K, V> updmap = listmap.remove(k);
- if (updmap.size() > 1) {
- return new LNode<>(updmap);
+ // We only ever create ListMaps with two or more entries, and remove them as soon as they reach one element
+ // (below), so we cannot observe a null return here.
+ final ListMap<K, V> map = listmap.remove(k);
+ final Optional<Entry<K, V>> maybeKv = map.maybeSingleton();
+ if (maybeKv.isPresent()) {
+ final Entry<K, V> kv = maybeKv.get();
+ // create it tombed so that it gets compressed on subsequent accesses
+ return new TNode<>(kv.getKey(), kv.getValue(), ct.computeHash(kv.getKey()));
}
- final Entry<K, V> kv = updmap.iterator().next();
- // create it tombed so that it gets compressed on subsequent accesses
- return new TNode<>(kv.getKey(), kv.getValue(), ct.computeHash(kv.getKey()));
+ return new LNode<>(map);
}
- Option<V> get(final K k) {
+ Optional<V> get(final K k) {
return listmap.get(k);
}
return "LNode";
}
- @Override
- public Iterator<Entry<K, V>> iterator() {
+ Iterator<Entry<K, V>> iterator() {
return listmap.iterator();
}
}
\ No newline at end of file