78 lines
1.4 KiB
Java
78 lines
1.4 KiB
Java
package it.cavallium.dbengine;
|
|
|
|
import it.cavallium.dbengine.lucene.PriorityQueue;
|
|
import java.io.IOException;
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
import org.apache.lucene.search.HitQueue;
|
|
import reactor.core.publisher.Flux;
|
|
|
|
public class PriorityQueueAdaptor<T> implements PriorityQueue<T> {
|
|
|
|
private final org.apache.lucene.util.PriorityQueue<T> hitQueue;
|
|
|
|
public PriorityQueueAdaptor(org.apache.lucene.util.PriorityQueue<T> hitQueue) {
|
|
this.hitQueue = hitQueue;
|
|
}
|
|
|
|
@Override
|
|
public void add(T element) {
|
|
hitQueue.add(element);
|
|
hitQueue.updateTop();
|
|
}
|
|
|
|
@Override
|
|
public T top() {
|
|
hitQueue.updateTop();
|
|
return hitQueue.top();
|
|
}
|
|
|
|
@Override
|
|
public T pop() {
|
|
var popped = hitQueue.pop();
|
|
hitQueue.updateTop();
|
|
return popped;
|
|
}
|
|
|
|
@Override
|
|
public void replaceTop(T newTop) {
|
|
hitQueue.updateTop(newTop);
|
|
}
|
|
|
|
@Override
|
|
public long size() {
|
|
return hitQueue.size();
|
|
}
|
|
|
|
@Override
|
|
public void clear() {
|
|
hitQueue.clear();
|
|
}
|
|
|
|
@Override
|
|
public boolean remove(T element) {
|
|
var removed = hitQueue.remove(element);
|
|
hitQueue.updateTop();
|
|
return removed;
|
|
}
|
|
|
|
@Override
|
|
public Flux<T> iterate() {
|
|
List<T> items = new ArrayList<>(hitQueue.size());
|
|
T item;
|
|
while ((item = hitQueue.pop()) != null) {
|
|
items.add(item);
|
|
}
|
|
for (T t : items) {
|
|
hitQueue.insertWithOverflow(t);
|
|
}
|
|
return Flux.fromIterable(items);
|
|
}
|
|
|
|
@Override
|
|
public void close() throws IOException {
|
|
hitQueue.clear();
|
|
hitQueue.updateTop();
|
|
}
|
|
}
|