1
2
3
4
5
6
7 package org.mule.transport.tcp;
8
9 import org.mule.DefaultMuleMessage;
10 import org.mule.api.construct.FlowConstruct;
11 import org.mule.api.endpoint.InboundEndpoint;
12 import org.mule.api.lifecycle.CreateException;
13 import org.mule.api.transport.Connector;
14 import org.mule.transport.AbstractPollingMessageReceiver;
15 import org.mule.transport.tcp.i18n.TcpMessages;
16 import org.mule.util.MapUtils;
17
18 import java.net.Socket;
19 import java.net.SocketTimeoutException;
20
21
22
23
24
25
26
27 public class PollingTcpMessageReceiver extends AbstractPollingMessageReceiver
28 {
29 private int timeout;
30
31 private PollingTcpConnector connector;
32
33 public PollingTcpMessageReceiver(Connector connector, FlowConstruct flowConstruct, InboundEndpoint endpoint)
34 throws CreateException
35 {
36 super(connector, flowConstruct, endpoint);
37
38 if (connector instanceof PollingTcpConnector)
39 {
40 this.connector = (PollingTcpConnector) connector;
41 }
42 else
43 {
44 throw new CreateException(TcpMessages.pollingReceiverCannotbeUsed(), this);
45 }
46
47 timeout = MapUtils.getIntValue(endpoint.getProperties(), "clientSoTimeout",
48 this.connector.getClientSoTimeout());
49
50 if (timeout > Integer.MAX_VALUE || timeout < 0)
51 {
52 throw new IllegalArgumentException("Timeout incorrect: " + timeout);
53 }
54
55 long pollingFrequency = MapUtils.getLongValue(endpoint.getProperties(), "pollingFrequency",
56 this.connector.getPollingFrequency());
57 if (pollingFrequency > 0)
58 {
59 this.setFrequency(pollingFrequency);
60 }
61 }
62
63 @Override
64 public void poll() throws Exception
65 {
66 Socket socket = connector.getSocket(endpoint);
67 try
68 {
69 Object result = TcpMessageDispatcher.receiveFromSocket(socket, (int) timeout, endpoint);
70 if (!(result == null))
71 {
72 this.routeMessage(new DefaultMuleMessage(result, connector.getMuleContext()));
73 if (logger.isDebugEnabled())
74 {
75 logger.debug("Routing new message: " + result);
76 }
77 }
78 }
79 catch (SocketTimeoutException e)
80 {
81 if (logger.isDebugEnabled())
82 {
83 logger.debug("Socket timed out normally while doing a synchronous receive on endpointUri: "
84 + endpoint.getEndpointURI());
85 }
86 }
87 finally
88 {
89 connector.releaseSocket(socket, endpoint);
90 }
91 }
92 }