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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
package frame.object.meta;
 
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
 
import org.apache.log4j.Logger;
 
import frame.persist.loader.ResultMetaFieldReader;
import frame.variant.ValueType;
import frame.variant.translator.ITranslator;
import frame.variant.translator.Translator;
 
 
public class EntityMeta implements Iterable<Field> {
 
    protected static Logger logger;
    
    protected String tableName;
    protected List<Field> fieldList;
    protected String[] names;
    protected String[] lowerNames;
    protected Map<String, Integer> nameIndexMap;
    protected ITranslator[] translators;
 
    static {
        logger = Logger.getLogger(EntityMeta.class);
    }
 
    protected EntityMeta() {
        
    }
    
    public EntityMeta(String tableName, IFieldReader fieldReader) throws SQLException {
        this.tableName = tableName;
        nameIndexMap = new HashMap<String, Integer>();
        fieldList = new ArrayList<Field>();
 
        initFields(fieldReader);
    }
 
    public static EntityMeta getInstance(String tableName) throws Exception {
        EntityMetaLoader loader = new EntityMetaLoader();
        return  loader.getEntityMetaByTable(tableName);
    }
 
    public static EntityMeta getInstance(ResultSetMetaData metaData) throws Exception {
        ResultMetaFieldReader fieldReader = new ResultMetaFieldReader(metaData);
        return new EntityMeta("tablemeta", fieldReader);
    }
 
    private void initFields(IFieldReader fieldReader) throws SQLException {
        String fieldName;
        int columnType;
        int fieldLength;
        int nullable;
        Field field;
 
        //1. fields
        fieldReader.first();
 
        while (fieldReader.next()) {
            fieldName = fieldReader.getFieldName();
            columnType = fieldReader.getFieldType();
            fieldLength = fieldReader.getFieldLength();
            nullable = fieldReader.getNullable();
 
            if (fieldName == null) {
                continue;
            }
 
            field = new Field(fieldName);
            fieldList.add(field);
 
            field.setType(columnType);
            field.setLength(fieldLength);
            field.setNullable(nullable);
        }
 
        //2. names
        int fieldCount = fieldList.size();
        names = new String[fieldCount];
        lowerNames = new String[fieldCount];
 
        String name;
        for (int i = 0; i < fieldCount; i++) {
            field = fieldList.get(i);
            name = field.getName();
            names[i] = name;
 
            name = name.toLowerCase();
            lowerNames[i] = name;
            nameIndexMap.put(name, i);
        }
        
        //3. translators
        translators = new ITranslator[fieldCount];
        
        for (int i = 0; i < fieldCount; i++) {
            field = fieldList.get(i);
            ValueType valueType = field.getValueType();
            ITranslator translator = Translator.getInstance(valueType);
            
            translators[i] = translator;
            field.setTranslator(translator);
        }
    }
 
    public int getFieldCount() {
        return fieldList.size();
    }
 
    public Iterator<Field> iterator() {
        return fieldList.iterator();
    }
 
    public String getTableName() {
        return tableName;
    }
 
    public List<Field> getFields() {
        return fieldList;
    }
 
    public Field getField(String name) {
        if (name == null)
            return null;
 
        Integer idx = nameIndexMap.get(name.toLowerCase());
 
        if (idx == null) {
            return null;
        }
 
        return fieldList.get(idx);
    }
 
    public Field getField(int idx) {
        return fieldList.get(idx);
    }
    
    public ValueType getValueType(int idx) {
        Field field = fieldList.get(idx);
        return field.getValueType();
    }
 
    public Integer getIndex(String fieldName) {
        if (fieldName == null) {
            return null;
        }
        
        try {
            return nameIndexMap.get(fieldName.toLowerCase());
        }
        catch (Exception e) {
            return null;
        }
    }
 
    public String[] getLowerNames() {
        return lowerNames;
    }
 
    public String[] getFieldNames() {
        return names;
    }
    
    public boolean contains(String name) {
        if (name == null) {
            return false;
        }
 
        name = name.toLowerCase();
        return nameIndexMap.containsKey(name);
    }
 
    public ITranslator getTranslator(int idx) {
        return translators[idx];
    }
    
    public ITranslator getTranslator(String name) {
        name = name.toLowerCase();
        Integer idx = nameIndexMap.get(name);
 
        if (idx == null) {
            return null;
        }
        
        return translators[idx];
    }
 
}