blob: 256a5de3528e570ec6a54b9a873d935223132852 (
plain)
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
|
package com.vaadin.data.provider;
import java.util.Comparator;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
import java.util.stream.Stream;
import com.vaadin.server.SerializablePredicate;
import com.vaadin.shared.data.sort.SortDirection;
public class BackendDataProviderTest extends
DataProviderTestBase<BackEndDataProvider<StrBean, SerializablePredicate<StrBean>>> {
private static Map<String, Comparator<StrBean>> propertyToComparatorMap = new HashMap<>();
static {
propertyToComparatorMap.put("value",
Comparator.comparing(StrBean::getValue));
propertyToComparatorMap.put("id", Comparator.comparing(StrBean::getId));
propertyToComparatorMap.put("randomNumber",
Comparator.comparing(StrBean::getRandomNumber));
}
private static Comparator<StrBean> getComparator(SortOrder<String> so) {
Comparator<StrBean> comparator = propertyToComparatorMap
.get(so.getSorted());
if (so.getDirection() == SortDirection.DESCENDING) {
comparator = comparator.reversed();
}
return comparator;
}
public static class StrBeanBackEndDataProvider extends
CallbackDataProvider<StrBean, SerializablePredicate<StrBean>> {
public StrBeanBackEndDataProvider(List<StrBean> data) {
super(query -> {
Stream<StrBean> stream = data.stream().filter(
t -> query.getFilter().orElse(s -> true).test(t));
if (!query.getSortOrders().isEmpty()) {
Comparator<StrBean> sorting = query.getSortOrders().stream()
.map(BackendDataProviderTest::getComparator)
.reduce((c1, c2) -> c1.thenComparing(c2)).get();
stream = stream.sorted(sorting);
}
List<StrBean> list = stream.skip(query.getOffset())
.limit(query.getLimit()).collect(Collectors.toList());
return list.stream();
}, query -> (int) data.stream()
.filter(t -> query.getFilter().orElse(s -> true).test(t))
.count());
}
}
@Override
protected BackEndDataProvider<StrBean, SerializablePredicate<StrBean>> createDataProvider() {
return dataProvider = new StrBeanBackEndDataProvider(data);
}
@Override
protected void setSortOrder(List<QuerySortOrder> sortOrder,
Comparator<StrBean> comp) {
getDataProvider().setSortOrders(sortOrder);
}
}
|
