blob: 606e197c5dd43ae2c61557323d7e71f75ddb18be (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
/*
* SPDX-FileCopyrightText: 2023 The Refinery Authors <https://refinery.tools/>
*
* SPDX-License-Identifier: EPL-2.0
*/
package tools.refinery.store.query.resultset;
import tools.refinery.store.map.Cursor;
import tools.refinery.store.query.ModelQueryAdapter;
import tools.refinery.logic.dnf.Query;
import tools.refinery.store.query.utils.OrderStatisticTree;
import tools.refinery.store.tuple.Tuple;
import java.util.Objects;
public class OrderedResultSet<T> implements AutoCloseable, ResultSet<T> {
private final ResultSet<T> resultSet;
private final OrderStatisticTree<Tuple> tree = new OrderStatisticTree<>();
private final ResultSetListener<T> listener = (key, fromValue, toValue) -> {
var defaultValue = getCanonicalQuery().defaultValue();
if (Objects.equals(defaultValue, toValue)) {
tree.remove(key);
} else {
tree.add(key);
}
};
public OrderedResultSet(ResultSet<T> resultSet) {
this.resultSet = resultSet;
resultSet.addListener(listener);
var cursor = resultSet.getAll();
while (cursor.move()) {
tree.add(cursor.getKey());
}
}
@Override
public ModelQueryAdapter getAdapter() {
return resultSet.getAdapter();
}
@Override
public int size() {
return resultSet.size();
}
@Override
public Query<T> getCanonicalQuery() {
return resultSet.getCanonicalQuery();
}
@Override
public T get(Tuple parameters) {
return resultSet.get(parameters);
}
public Tuple getKey(int index) {
return tree.get(index);
}
@Override
public Cursor<Tuple, T> getAll() {
return resultSet.getAll();
}
@Override
public void addListener(ResultSetListener<T> listener) {
resultSet.addListener(listener);
}
@Override
public void removeListener(ResultSetListener<T> listener) {
resultSet.removeListener(listener);
}
@Override
public void close() {
resultSet.removeListener(listener);
}
}
|