1 | |
|
2 | |
|
3 | |
|
4 | |
|
5 | |
|
6 | |
|
7 | |
|
8 | |
|
9 | |
|
10 | |
|
11 | |
package org.mule.providers.vm; |
12 | |
|
13 | |
import org.mule.MuleException; |
14 | |
import org.mule.config.i18n.CoreMessages; |
15 | |
import org.mule.impl.MuleMessage; |
16 | |
import org.mule.providers.TransactedPollingMessageReceiver; |
17 | |
import org.mule.umo.UMOComponent; |
18 | |
import org.mule.umo.UMOEvent; |
19 | |
import org.mule.umo.UMOException; |
20 | |
import org.mule.umo.UMOMessage; |
21 | |
import org.mule.umo.endpoint.UMOEndpoint; |
22 | |
import org.mule.umo.lifecycle.InitialisationException; |
23 | |
import org.mule.umo.provider.UMOConnector; |
24 | |
import org.mule.util.queue.Queue; |
25 | |
import org.mule.util.queue.QueueSession; |
26 | |
|
27 | |
import java.util.List; |
28 | |
|
29 | |
import edu.emory.mathcs.backport.java.util.concurrent.TimeUnit; |
30 | |
|
31 | |
|
32 | |
|
33 | |
|
34 | |
|
35 | |
public class VMMessageReceiver extends TransactedPollingMessageReceiver |
36 | |
{ |
37 | |
public static final long DEFAULT_VM_POLL_FREQUENCY = 10; |
38 | 0 | public static final TimeUnit DEFAULT_VM_POLL_TIMEUNIT = TimeUnit.MILLISECONDS; |
39 | |
|
40 | |
private VMConnector connector; |
41 | 0 | private final Object lock = new Object(); |
42 | |
|
43 | |
public VMMessageReceiver(UMOConnector connector, UMOComponent component, UMOEndpoint endpoint) |
44 | |
throws InitialisationException |
45 | |
{ |
46 | 0 | super(connector, component, endpoint); |
47 | 0 | this.setReceiveMessagesInTransaction(endpoint.getTransactionConfig().isTransacted()); |
48 | 0 | this.setFrequency(DEFAULT_VM_POLL_FREQUENCY); |
49 | 0 | this.setTimeUnit(DEFAULT_VM_POLL_TIMEUNIT); |
50 | 0 | this.connector = (VMConnector) connector; |
51 | 0 | } |
52 | |
|
53 | |
protected void doDispose() |
54 | |
{ |
55 | |
|
56 | 0 | } |
57 | |
|
58 | |
protected void doConnect() throws Exception |
59 | |
{ |
60 | 0 | if (connector.isQueueEvents()) |
61 | |
{ |
62 | |
|
63 | 0 | QueueSession queueSession = connector.getQueueSession(); |
64 | 0 | Queue q = queueSession.getQueue(endpoint.getEndpointURI().getAddress()); |
65 | 0 | if (logger.isDebugEnabled()) |
66 | |
{ |
67 | 0 | logger.debug("Current queue depth for queue: " + endpoint.getEndpointURI().getAddress() |
68 | |
+ " is: " + q.size()); |
69 | |
} |
70 | |
} |
71 | 0 | } |
72 | |
|
73 | |
protected void doDisconnect() throws Exception |
74 | |
{ |
75 | |
|
76 | 0 | } |
77 | |
|
78 | |
|
79 | |
|
80 | |
|
81 | |
|
82 | |
|
83 | |
public void onEvent(UMOEvent event) throws UMOException |
84 | |
{ |
85 | 0 | if (connector.isQueueEvents()) |
86 | |
{ |
87 | 0 | QueueSession queueSession = connector.getQueueSession(); |
88 | 0 | Queue queue = queueSession.getQueue(endpoint.getEndpointURI().getAddress()); |
89 | |
try |
90 | |
{ |
91 | 0 | queue.put(event); |
92 | |
} |
93 | 0 | catch (InterruptedException e) |
94 | |
{ |
95 | 0 | throw new MuleException(CoreMessages.interruptedQueuingEventFor(this.endpoint |
96 | |
.getEndpointURI()), e); |
97 | 0 | } |
98 | |
} |
99 | |
else |
100 | |
{ |
101 | 0 | UMOMessage msg = new MuleMessage(event.getTransformedMessage(), event.getMessage()); |
102 | 0 | synchronized (lock) |
103 | |
{ |
104 | 0 | routeMessage(msg); |
105 | 0 | } |
106 | |
} |
107 | 0 | } |
108 | |
|
109 | |
public Object onCall(UMOEvent event) throws UMOException |
110 | |
{ |
111 | 0 | return routeMessage(new MuleMessage(event.getTransformedMessage(), event.getMessage()), event |
112 | |
.isSynchronous()); |
113 | |
} |
114 | |
|
115 | |
protected List getMessages() throws Exception |
116 | |
{ |
117 | 0 | QueueSession qs = connector.getQueueSession(); |
118 | 0 | Queue queue = qs.getQueue(endpoint.getEndpointURI().getAddress()); |
119 | 0 | UMOEvent event = (UMOEvent) queue.poll(connector.getQueueTimeout()); |
120 | 0 | if (event != null) |
121 | |
{ |
122 | 0 | routeMessage(new MuleMessage(event.getTransformedMessage(), event.getMessage())); |
123 | |
} |
124 | 0 | return null; |
125 | |
} |
126 | |
|
127 | |
|
128 | |
|
129 | |
|
130 | |
|
131 | |
|
132 | |
protected void processMessage(Object msg) throws Exception |
133 | |
{ |
134 | |
|
135 | 0 | } |
136 | |
|
137 | |
} |