IM服务端

hefeixia

2021-02-18 5b8c95c760840f09910730943b21391e47187315

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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
package frame.util;
 
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.TreeSet;
 
import org.apache.log4j.Logger;
 
public class MapTree<T> implements Iterable<T> {
 
    protected static Logger logger;
    protected TreeSet<T> itemList;
    protected Map<String, T> itemMap;
    
    static {
        logger = Logger.getLogger(MapTree.class);
    }
    
    public MapTree() {
        itemList = new TreeSet<T>();
        itemMap = new HashMap<String, T>();
    }
    
    public void add(String key, T item) {
        if (key == null) {
            key = "empty";
        }
        key = key.toLowerCase();
        
        if (itemMap.containsKey(key)) {
            logger.error("duplicate " + this.getClass().getName() +  " key: " + key);
            return;
        }
        
        itemMap.put(key, item);
        itemList.add(item);
    }
    
    public T get(String key) {
        if (key == null) {
            key = "empty";
        }
        key = key.toLowerCase();
        
        return itemMap.get(key);
    }
    
    public T remove(String key) {
        if (key == null) {
            return null;
        }
        
        T obj = itemMap.get(key.toLowerCase());
        
        if (obj != null) {
            itemMap.remove(key);
            itemList.remove(obj);
        }
        
        return obj;
    }
    
    public TreeSet<T> getItemList() {
        return itemList;
    }
 
    public boolean isEmpty() {
        return itemList.isEmpty();
    }
    
    public int size() {
        return itemList.size();
    }
    
    public boolean containsKey(String key) {
        if (key == null) {
            return false;
        }
        
        key = key.toLowerCase();
        return itemMap.containsKey(key);
    }
 
    @Override
    public Iterator<T> iterator() {
        return itemList.iterator();
    }
 
    public void clear() {
        itemList.clear();
        itemMap.clear();
    }
    
    public String toString(String separator) {
        ContentBuilder builder = new ContentBuilder(separator);
        
        for (String key: itemMap.keySet()) {
            builder.append(key);
        }
        
        return builder.toString();
    }
    
}