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
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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
package foundation.data.entity;
 
import foundation.data.meta.field.FieldsRuntime;
import foundation.json.IJSONWriter;
import foundation.util.Util;
 
 
public class EntityTree extends EntitySet {
 
    private EntitySet rootSet;
    private boolean selectActive;
    
    
    public EntityTree(FieldsRuntime fieldMetas) {
        super(fieldMetas);
        rootSet = new EntitySet(fieldMetas);
        selectActive = false;
    }
 
    public EntitySet getChildren(String parentId) throws Exception {
        EntityNode parent = getNode(parentId);
        
        if (parent == null) {
            return null;
        }
        
        return parent.children;
    }
 
    public EntityNode getNode(String id) throws Exception {
        if (Util.isEmpty(id)) {
            return null;
        }
 
        Entity result = getEntity("id", id);
        return (EntityNode) result;
    }
 
    public EntitySet getRootSet() {
        return rootSet;
    }
    
    public void initRelation() throws Exception {
        initFamilyRelation();
        initLevelRelation();
    }
    
    public void initFamilyRelation() throws Exception {
        for (Entity entity: this) {
            EntityNode node = (EntityNode) entity;
            String parentId = node.getParentId();
            
            EntityNode parent = getNode(parentId);
            
            if (parent == null) {
                node.level = 0;
                rootSet.append(node);
                continue;
            }
            
            parent.addChild(node);
        }
    }
    
    public void initLevelRelation() throws Exception {
        for (Entity entity: rootSet) {
            EntityNode node = (EntityNode) entity;
            doInitLevelRelation(node);
        }
    }
 
    private void doInitLevelRelation(EntityNode parent) {
        EntitySet children = parent.getChildren();
        
        for (Entity entity: children) {
            EntityNode child = (EntityNode) entity;
            child.level = parent.level + 1;
            
            doInitLevelRelation(child);
        }
    }
    
    public void setSelectActive(boolean selectActive) {
        this.selectActive = selectActive;
    }
 
    @Override
    public void writeJSON(IJSONWriter writer) {
        writer.beginArray();
        if (selectActive) {
            for (Entity entity: rootSet) {
                if (SelectState.Unselect == entity.selectState) {
                    continue;
                }
                
                EntityNode node = (EntityNode)entity;
                node.writeJSON(writer, node.selectState);
            }    
        }
        else {
            for (Entity entity: rootSet) {
                EntityNode node = (EntityNode)entity;
                node.writeJSON(writer, SelectState.CurrentAndChildren);
            }    
        }
        
        writer.endArray();    
    }
 
    @Override
    public void writeJSONBody(IJSONWriter writer) {
        if (selectActive) {
            for (Entity entity: rootSet) {
                if (SelectState.Unselect == entity.selectState) {
                    continue;
                }
                
                EntityNode node = (EntityNode)entity;
                node.writeJSON(writer, node.selectState);
            }
        }
        else {
            for (Entity entity: rootSet) {
                EntityNode node = (EntityNode)entity;
                node.writeJSON(writer, SelectState.CurrentAndChildren);
            }
        }
    }
}