dev 契约锁

P15GEN2\59518

2024-05-29 d4210c7c4b04abde20037ea8aa0f54ef8a2649aa

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 foundation.route;
 
import java.util.HashMap;
import java.util.Map;
import java.util.Set;
 
 
public class URIRouteMap {
 
    private Map<String, FilterNode<Route>> nodes;
    private Map<String, Route> routeMap;
    
    
    public URIRouteMap() {
        nodes = new HashMap<String, FilterNode<Route>>();
        routeMap = new HashMap<String, Route>();
    }
    
    public void clear() {
        nodes.clear();
    }
 
    public Route get(String[] segments) {
        int idx = segments.length - 1;
        
        if (idx < 0) {
            return null;
        }
        
        FilterNode<Route> node = nodes.get(segments[idx]);
        
        if (node == null) {
            return null;
        }
        
        return node.match(segments, idx - 1);
    }
 
    public Route get(String key) {
        return routeMap.get(key);
    }
    
    public void add(String uri, Route route) {
        if (uri == null) {
            return;
        }
        
        //1. add to map
        routeMap.put(uri, route);
        
        //2. add add to nodes
        String[] segments = uri.split("/");
        
        int idx = segments.length - 1;
        String segment = segments[idx];
        
        FilterNode<Route> node = nodes.get(segment);
        
        if (node == null) {
            node = new FilterNode<Route>(segment, route);
            
            nodes.put(segment, node);
        }
        
        node.add(segments, idx - 1, route);
    }
 
    public Set<String> keySet() {
        return routeMap.keySet();
    }
}