View Javadoc
1   /*
2    * Copyright (c) MuleSoft, Inc.  All rights reserved.  http://www.mulesoft.com
3    * The software in this package is published under the terms of the CPAL v1.0
4    * license, a copy of which has been included with this distribution in the
5    * LICENSE.txt file.
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   * <code>PollingTcpMessageReceiver</code> acts like a TCP client polling for new
23   * messages.
24   * 
25   * @author esteban.robles
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  }