1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20 package org.apache.mina.filter.executor;
21
22 import java.lang.reflect.Field;
23 import java.lang.reflect.Modifier;
24 import java.util.HashSet;
25 import java.util.Set;
26 import java.util.concurrent.ConcurrentHashMap;
27 import java.util.concurrent.ConcurrentMap;
28
29 import org.apache.mina.core.buffer.IoBuffer;
30 import org.apache.mina.core.session.IoEvent;
31 import org.apache.mina.core.write.WriteRequest;
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46 public class DefaultIoEventSizeEstimator implements IoEventSizeEstimator {
47
48 private final ConcurrentMap<Class<?>, Integer> class2size = new ConcurrentHashMap<>();
49
50
51
52
53
54 public DefaultIoEventSizeEstimator() {
55 class2size.put(boolean.class, 4);
56 class2size.put(byte.class, 1);
57 class2size.put(char.class, 2);
58 class2size.put(int.class, 4);
59 class2size.put(short.class, 2);
60 class2size.put(long.class, 8);
61 class2size.put(float.class, 4);
62 class2size.put(double.class, 8);
63 class2size.put(void.class, 0);
64 }
65
66
67
68
69 @Override
70 public int estimateSize(IoEvent event) {
71 return estimateSize((Object) event) + estimateSize(event.getParameter());
72 }
73
74
75
76
77
78
79 public int estimateSize(Object message) {
80 if (message == null) {
81 return 8;
82 }
83
84 int answer = 8 + estimateSize(message.getClass(), null);
85
86 if (message instanceof IoBuffer) {
87 answer += ((IoBuffer) message).remaining();
88 } else if (message instanceof WriteRequest) {
89 answer += estimateSize(((WriteRequest) message).getMessage());
90 } else if (message instanceof CharSequence) {
91 answer += ((CharSequence) message).length() << 1;
92 } else if (message instanceof Iterable) {
93 for (Object m : (Iterable<?>) message) {
94 answer += estimateSize(m);
95 }
96 }
97
98 return align(answer);
99 }
100
101 private int estimateSize(Class<?> clazz, Set<Class<?>> visitedClasses) {
102 Integer objectSize = class2size.get(clazz);
103
104 if (objectSize != null) {
105 return objectSize;
106 }
107
108 if (visitedClasses != null) {
109 if (visitedClasses.contains(clazz)) {
110 return 0;
111 }
112 } else {
113 visitedClasses = new HashSet<>();
114 }
115
116 visitedClasses.add(clazz);
117
118 int answer = 8;
119
120 for (Class<?> c = clazz; c != null; c = c.getSuperclass()) {
121 Field[] fields = c.getDeclaredFields();
122
123 for (Field f : fields) {
124 if ((f.getModifiers() & Modifier.STATIC) != 0) {
125
126 continue;
127 }
128
129 answer += estimateSize(f.getType(), visitedClasses);
130 }
131 }
132
133 visitedClasses.remove(clazz);
134
135
136 answer = align(answer);
137
138
139 Integer tmpAnswer = class2size.putIfAbsent(clazz, answer);
140
141 if (tmpAnswer != null) {
142 answer = tmpAnswer;
143 }
144
145 return answer;
146 }
147
148 private static int align(int size) {
149 if (size % 8 != 0) {
150 size /= 8;
151 size++;
152 size *= 8;
153 }
154
155 return size;
156 }
157 }