1 |
| |
2 |
| |
3 |
| |
4 |
| |
5 |
| |
6 |
| |
7 |
| |
8 |
| |
9 |
| |
10 |
| |
11 |
| |
12 |
| |
13 |
| |
14 |
| |
15 |
| package org.apache.hivemind.impl; |
16 |
| |
17 |
| import java.lang.reflect.Constructor; |
18 |
| import java.lang.reflect.Modifier; |
19 |
| |
20 |
| import org.apache.hivemind.ApplicationRuntimeException; |
21 |
| import org.apache.hivemind.events.RegistryShutdownListener; |
22 |
| import org.apache.hivemind.internal.ServiceModel; |
23 |
| import org.apache.hivemind.internal.ServicePoint; |
24 |
| import org.apache.hivemind.service.BodyBuilder; |
25 |
| import org.apache.hivemind.service.ClassFab; |
26 |
| import org.apache.hivemind.service.ClassFabUtils; |
27 |
| import org.apache.hivemind.service.MethodSignature; |
28 |
| import org.apache.hivemind.util.ConstructorUtils; |
29 |
| |
30 |
| |
31 |
| |
32 |
| |
33 |
| |
34 |
| |
35 |
| |
36 |
| public final class ProxyUtils |
37 |
| { |
38 |
| public static final String SERVICE_ACCESSOR_METHOD_NAME = "_service"; |
39 |
| |
40 |
| public static final String DELEGATE_ACCESSOR_METHOD_NAME = "_delegate"; |
41 |
| |
42 |
0
| private ProxyUtils()
|
43 |
| { |
44 |
| |
45 |
| } |
46 |
| |
47 |
| |
48 |
| |
49 |
| |
50 |
| |
51 |
| |
52 |
| |
53 |
33
| public static Object createDelegatingProxy(String type, ServiceModel serviceModel,
|
54 |
| String delegationMethodName, ServicePoint servicePoint) |
55 |
| { |
56 |
33
| ProxyBuilder builder = new ProxyBuilder(type, servicePoint);
|
57 |
| |
58 |
33
| ClassFab classFab = builder.getClassFab();
|
59 |
| |
60 |
33
| addConstructor(classFab, serviceModel);
|
61 |
| |
62 |
33
| addServiceAccessor(classFab, delegationMethodName, servicePoint);
|
63 |
| |
64 |
33
| builder.addServiceMethods(SERVICE_ACCESSOR_METHOD_NAME + "()");
|
65 |
| |
66 |
33
| Class proxyClass = classFab.createClass();
|
67 |
| |
68 |
33
| try
|
69 |
| { |
70 |
33
| Constructor c = proxyClass.getConstructor(new Class[]
|
71 |
| { serviceModel.getClass() }); |
72 |
| |
73 |
33
| return c.newInstance(new Object[]
|
74 |
| { serviceModel }); |
75 |
| } |
76 |
| catch (Exception ex) |
77 |
| { |
78 |
0
| throw new ApplicationRuntimeException(ex);
|
79 |
| } |
80 |
| } |
81 |
| |
82 |
| |
83 |
| |
84 |
| |
85 |
| |
86 |
| |
87 |
| |
88 |
| |
89 |
| |
90 |
| |
91 |
| |
92 |
| |
93 |
| |
94 |
| |
95 |
| |
96 |
| |
97 |
33
| public static RegistryShutdownListener createOuterProxy(Object delegate,
|
98 |
| ServicePoint servicePoint) |
99 |
| { |
100 |
33
| ProxyBuilder builder = new ProxyBuilder("OuterProxy", servicePoint, true);
|
101 |
| |
102 |
33
| ClassFab classFab = builder.getClassFab();
|
103 |
| |
104 |
33
| addDelegateAccessor(classFab, servicePoint, delegate);
|
105 |
| |
106 |
33
| builder.addServiceMethods(DELEGATE_ACCESSOR_METHOD_NAME + "()");
|
107 |
| |
108 |
33
| Class proxyClass = classFab.createClass();
|
109 |
| |
110 |
33
| try
|
111 |
| { |
112 |
33
| return (RegistryShutdownListener) ConstructorUtils.invokeConstructor(
|
113 |
| proxyClass, |
114 |
| new Object[] |
115 |
| { delegate }); |
116 |
| } |
117 |
| catch (Exception ex) |
118 |
| { |
119 |
0
| throw new ApplicationRuntimeException(ex);
|
120 |
| } |
121 |
| } |
122 |
| |
123 |
| |
124 |
| |
125 |
33
| private static void addDelegateAccessor(ClassFab classFab, ServicePoint servicePoint,
|
126 |
| Object delegate) |
127 |
| { |
128 |
33
| classFab.addField("_shutdown", boolean.class);
|
129 |
| |
130 |
33
| Class delegateClass = ClassFabUtils.getInstanceClass(delegate, servicePoint
|
131 |
| .getServiceInterface()); |
132 |
| |
133 |
33
| classFab.addField("_delegate", delegateClass);
|
134 |
| |
135 |
33
| classFab.addConstructor(new Class[]
|
136 |
| { delegateClass }, null, "{ super(); _delegate = $1; }"); |
137 |
| |
138 |
33
| classFab.addInterface(RegistryShutdownListener.class);
|
139 |
33
| if( RegistryShutdownListener.class.isAssignableFrom( delegateClass ) )
|
140 |
| { |
141 |
2
| classFab.addMethod(Modifier.PUBLIC | Modifier.FINAL, new MethodSignature(void.class,
|
142 |
| "registryDidShutdown", null, null), "{ _delegate.registryDidShutdown(); _delegate = null; _shutdown = true; }"); |
143 |
| } |
144 |
| else |
145 |
| { |
146 |
31
| classFab.addMethod(Modifier.PUBLIC | Modifier.FINAL, new MethodSignature(void.class,
|
147 |
| "registryDidShutdown", null, null), "{ _delegate = null; _shutdown = true; }"); |
148 |
| } |
149 |
33
| BodyBuilder builder = new BodyBuilder();
|
150 |
| |
151 |
33
| builder.begin();
|
152 |
| |
153 |
33
| builder.addln("if (_shutdown)");
|
154 |
33
| builder.addln(" throw org.apache.hivemind.HiveMind#createRegistryShutdownException();");
|
155 |
| |
156 |
33
| builder.add("return _delegate;");
|
157 |
| |
158 |
33
| builder.end();
|
159 |
| |
160 |
33
| classFab.addMethod(Modifier.FINAL | Modifier.PRIVATE, new MethodSignature(delegateClass,
|
161 |
| DELEGATE_ACCESSOR_METHOD_NAME, null, null), builder.toString()); |
162 |
| } |
163 |
| |
164 |
| |
165 |
| |
166 |
| |
167 |
| |
168 |
33
| private static void addConstructor(ClassFab classFab, ServiceModel model)
|
169 |
| { |
170 |
33
| Class modelClass = model.getClass();
|
171 |
| |
172 |
33
| classFab.addField("_serviceModel", modelClass);
|
173 |
| |
174 |
33
| classFab.addConstructor(new Class[]
|
175 |
| { modelClass }, null, "{ super(); _serviceModel = $1; }"); |
176 |
| } |
177 |
| |
178 |
| |
179 |
| |
180 |
| |
181 |
| |
182 |
33
| private static void addServiceAccessor(ClassFab classFab, String serviceModelMethodName,
|
183 |
| ServicePoint servicePoint) |
184 |
| { |
185 |
33
| Class serviceInterface = servicePoint.getServiceInterface();
|
186 |
| |
187 |
33
| classFab.addField(SERVICE_ACCESSOR_METHOD_NAME, serviceInterface);
|
188 |
| |
189 |
33
| BodyBuilder builder = new BodyBuilder();
|
190 |
33
| builder.begin();
|
191 |
| |
192 |
33
| builder.add("return (");
|
193 |
33
| builder.add(serviceInterface.getName());
|
194 |
33
| builder.add(") _serviceModel.");
|
195 |
33
| builder.add(serviceModelMethodName);
|
196 |
33
| builder.add("();");
|
197 |
| |
198 |
33
| builder.end();
|
199 |
| |
200 |
33
| classFab.addMethod(Modifier.PRIVATE | Modifier.FINAL, new MethodSignature(serviceInterface,
|
201 |
| SERVICE_ACCESSOR_METHOD_NAME, null, null), builder.toString()); |
202 |
| } |
203 |
| } |