app/src/main/java/net/lacolaco/smileessence/view/adapter/OrderedCustomListAdapter.java


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

package net.lacolaco.smileessence.view.adapter;

import net.lacolaco.smileessence.entity.IdObject;

import java.util.*;

public abstract class OrderedCustomListAdapter<T extends IdObject> extends CustomListAdapter<T> {
    private final Map<Long, T> treeMap;

    public OrderedCustomListAdapter() {
        this(Long::compare);
    }

    public OrderedCustomListAdapter(Comparator<Long> comparator) {
        super();
        this.treeMap = new TreeMap<>(Collections.reverseOrder(comparator));
    }

    @Override
    protected synchronized List<T> getList() {
        return new ArrayList<>(treeMap.values());
    }

    public synchronized void add(T item) {
        treeMap.put(item.getId(), item);
    }

    public synchronized void addAll(Collection<T> items) {
        for (T item : items) {
            treeMap.put(item.getId(), item);
        }
    }

    public synchronized void clear() {
        treeMap.clear();
    }

    public synchronized T remove(T item) {
        return treeMap.remove(item.getId());
    }
}