aboutsummaryrefslogtreecommitdiffstats
path: root/uitest/src/main/java/com/vaadin/tests/util/LoggingItemDataProvider.java
blob: 0cc12c1ed62b0ef963e46ae152c2f59a983ad2c8 (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
69
70
71
package com.vaadin.tests.util;

import java.util.Locale;
import java.util.stream.IntStream;
import java.util.stream.Stream;

import com.vaadin.data.provider.CallbackDataProvider;
import com.vaadin.data.provider.Query;
import com.vaadin.ui.Label;
import com.vaadin.ui.VerticalLayout;

/**
 * CallbackDataProvider that logs info in UI on Queries it receives.
 */
public class LoggingItemDataProvider
        extends CallbackDataProvider<String, String> {

    private int counter;

    public LoggingItemDataProvider(int size, VerticalLayout logContainer) {
        super(q -> fetch(logContainer, q, size),
                q -> size(logContainer, q, size));
    }

    private static Stream<String> fetch(VerticalLayout logContainer,
            Query<String, String> query, int size) {
        log("FETCH", query, logContainer);
        return itemStream(query, size).skip(query.getOffset())
                .limit(query.getLimit());
    }

    private static int size(VerticalLayout logContainer,
            Query<String, String> query, int size) {
        log("SIZE", query, logContainer);
        return size(query, size);
    }

    private static Stream<String> itemStream(Query<String, String> q,
            int size) {
        Stream<String> stream = IntStream.range(0, size)
                .mapToObj(i -> "Item " + i);
        String filterText = q.getFilter().orElse("").toLowerCase(Locale.US);

        if (filterText.isEmpty()) {
            return stream;
        }

        return stream.filter(
                text -> text.toLowerCase(Locale.US).contains(filterText));
    }

    private static int size(Query<String, String> q, int size) {
        if (!q.getFilter().orElse("").isEmpty()) {
            return (int) itemStream(q, size).count();
        }
        return size;
    }

    private static Label log(String queryType, Query<String, String> query,
            VerticalLayout logContainer) {
        int componentCount = logContainer.getComponentCount();
        Label label = new Label(componentCount + ": " + queryType + " "
                + query.getOffset() + " " + query.getLimit() + " "
                + query.getFilter().orElse(""));
        logContainer.addComponentAsFirst(label);
        label.setId(queryType + "-" + componentCount);
        label.setStyleName("log");
        return label;
    }

}