1 | |
|
2 | |
|
3 | |
|
4 | |
|
5 | |
|
6 | |
|
7 | |
|
8 | |
|
9 | |
|
10 | |
|
11 | |
package org.mule.transport.rmi; |
12 | |
|
13 | |
import org.mule.api.MuleEvent; |
14 | |
import org.mule.api.MuleException; |
15 | |
import org.mule.api.MuleMessage; |
16 | |
import org.mule.api.construct.FlowConstruct; |
17 | |
import org.mule.api.endpoint.EndpointURI; |
18 | |
import org.mule.api.endpoint.InboundEndpoint; |
19 | |
import org.mule.api.lifecycle.CreateException; |
20 | |
import org.mule.api.transport.Connector; |
21 | |
import org.mule.transport.AbstractMessageReceiver; |
22 | |
import org.mule.transport.ConnectException; |
23 | |
import org.mule.transport.rmi.i18n.RmiMessages; |
24 | |
import org.mule.util.ClassUtils; |
25 | |
|
26 | |
import java.lang.reflect.Method; |
27 | |
import java.net.InetAddress; |
28 | |
|
29 | |
import javax.naming.Context; |
30 | |
|
31 | |
|
32 | |
|
33 | |
public class RmiCallbackMessageReceiver extends AbstractMessageReceiver |
34 | |
{ |
35 | |
|
36 | |
|
37 | |
|
38 | |
|
39 | |
public static final String PROPERTY_SERVICE_CLASS_NAME = "serviceClassName"; |
40 | |
|
41 | |
protected RmiConnector connector; |
42 | |
|
43 | 0 | protected RmiAble remoteObject = null; |
44 | |
|
45 | 0 | private Context jndi = null; |
46 | |
|
47 | 0 | private String bindName = null; |
48 | |
|
49 | |
private int port; |
50 | |
|
51 | |
|
52 | |
public RmiCallbackMessageReceiver(Connector connector, FlowConstruct flowConstruct, InboundEndpoint endpoint) |
53 | |
throws CreateException |
54 | |
{ |
55 | 0 | super(connector, flowConstruct, endpoint); |
56 | 0 | this.connector = (RmiConnector) connector; |
57 | |
|
58 | 0 | logger.debug("Initializing with endpoint " + endpoint); |
59 | |
|
60 | 0 | String rmiPolicyPath = this.connector.getSecurityPolicy(); |
61 | |
|
62 | 0 | System.setProperty("java.security.policy", rmiPolicyPath); |
63 | |
|
64 | 0 | EndpointURI endpointUri = endpoint.getEndpointURI(); |
65 | |
|
66 | 0 | port = endpointUri.getPort(); |
67 | |
|
68 | 0 | if (port < 1) |
69 | |
{ |
70 | 0 | port = RmiConnector.DEFAULT_RMI_muleRegistry_PORT; |
71 | |
} |
72 | |
|
73 | |
|
74 | 0 | logger.debug("Initialized successfully"); |
75 | 0 | } |
76 | |
|
77 | |
@Override |
78 | |
protected void doDispose() |
79 | |
{ |
80 | |
|
81 | 0 | } |
82 | |
|
83 | |
|
84 | |
|
85 | |
|
86 | |
|
87 | |
|
88 | |
|
89 | |
|
90 | |
@Override |
91 | |
protected void doConnect() throws ConnectException |
92 | |
{ |
93 | |
try |
94 | |
{ |
95 | |
|
96 | |
|
97 | |
|
98 | |
|
99 | |
|
100 | |
|
101 | 0 | if (null == remoteObject) |
102 | |
{ |
103 | |
try |
104 | |
{ |
105 | 0 | InetAddress inetAddress = InetAddress.getByName(endpoint.getEndpointURI().getHost()); |
106 | |
|
107 | 0 | bindName = endpoint.getEndpointURI().getPath(); |
108 | |
|
109 | 0 | remoteObject = getRmiObject(); |
110 | |
|
111 | 0 | Method theMethod = remoteObject.getClass().getMethod("setReceiver", |
112 | |
new Class[]{RmiCallbackMessageReceiver.class}); |
113 | 0 | theMethod.invoke(remoteObject, new Object[]{this}); |
114 | |
|
115 | 0 | jndi = connector.getJndiContext(inetAddress.getHostAddress() + ":" + port); |
116 | |
|
117 | 0 | jndi.rebind(bindName, remoteObject); |
118 | |
} |
119 | 0 | catch (Exception e) |
120 | |
{ |
121 | 0 | throw new ConnectException(e, this); |
122 | 0 | } |
123 | |
|
124 | |
} |
125 | |
} |
126 | 0 | catch (Exception e) |
127 | |
{ |
128 | 0 | throw new ConnectException(e, this); |
129 | 0 | } |
130 | 0 | } |
131 | |
|
132 | |
|
133 | |
@Override |
134 | |
protected void doDisconnect() |
135 | |
{ |
136 | 0 | logger.debug("Disconnecting..."); |
137 | |
|
138 | |
try |
139 | |
{ |
140 | 0 | jndi.unbind(bindName); |
141 | |
} |
142 | 0 | catch (Exception e) |
143 | |
{ |
144 | 0 | logger.error(e); |
145 | 0 | } |
146 | |
|
147 | 0 | logger.debug("Disconnected successfully."); |
148 | 0 | } |
149 | |
|
150 | |
@Override |
151 | |
protected void doStart() throws MuleException |
152 | |
{ |
153 | |
|
154 | 0 | } |
155 | |
|
156 | |
@Override |
157 | |
protected void doStop() throws MuleException |
158 | |
{ |
159 | |
|
160 | 0 | } |
161 | |
|
162 | |
|
163 | |
|
164 | |
|
165 | |
|
166 | |
|
167 | |
|
168 | |
private RmiAble getRmiObject() throws ConnectException |
169 | |
{ |
170 | 0 | String className = (String) endpoint.getProperty(PROPERTY_SERVICE_CLASS_NAME); |
171 | |
|
172 | 0 | if (null == className) |
173 | |
{ |
174 | 0 | throw new ConnectException(RmiMessages.messageReceiverNeedsRmiAble(), this); |
175 | |
} |
176 | |
|
177 | |
RmiAble remote; |
178 | |
|
179 | |
try |
180 | |
{ |
181 | 0 | remote = (RmiAble) ClassUtils.instanciateClass(className, new Object[]{}, this.getClass()); |
182 | |
} |
183 | 0 | catch (Exception e) |
184 | |
{ |
185 | 0 | throw new ConnectException(RmiMessages.serviceClassInvocationFailed(), e, this); |
186 | 0 | } |
187 | |
|
188 | 0 | return (remote); |
189 | |
} |
190 | |
|
191 | |
public MuleEvent routeMessage(Object payload) throws MuleException |
192 | |
{ |
193 | 0 | MuleMessage messageToRoute = createMuleMessage(payload, endpoint.getEncoding()); |
194 | 0 | return routeMessage(messageToRoute); |
195 | |
} |
196 | |
} |