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.ajax;
8   
9   import org.mule.api.MuleEvent;
10  import org.mule.api.MuleException;
11  import org.mule.api.MuleMessage;
12  import org.mule.api.transformer.Transformer;
13  import org.mule.api.transport.Connector;
14  import org.mule.transport.DefaultReplyToHandler;
15  
16  import java.util.List;
17  
18  import org.cometd.Channel;
19  import org.cometd.Client;
20  import org.mortbay.cometd.AbstractBayeux;
21  
22  /**
23   * Handles the sending of sending result messages back to the client when the a replyTo 
24   * channel is specified in the client request.
25   */
26  public class AjaxReplyToHandler extends DefaultReplyToHandler
27  {
28      private Connector connector;
29      
30      public AjaxReplyToHandler(List<Transformer> transformers, Connector connector)
31      {
32          super(transformers, connector.getMuleContext());
33          this.connector = connector;
34      }
35  
36      @Override
37      public void processReplyTo(MuleEvent event, MuleMessage returnMessage, Object replyTo) throws MuleException
38      {
39          AbstractBayeux bayeux = ((BayeuxAware)connector).getBayeux();
40          Channel channel = bayeux.getChannel(replyTo.toString(), false);
41          if(channel==null)
42          {
43              logger.warn("No ajax Channel: " + replyTo + ". Maybe the client unregistered interest.");
44              return;
45          }
46          
47          Object ret;
48          if(returnMessage.getExceptionPayload()!=null)
49          {
50              //If we are using RPC make sure we still send something back to the client so that the subscription is cancelled
51              ret = returnMessage.getExceptionPayload().getMessage();
52          }
53          else
54          {
55              returnMessage.applyTransformers(event, getTransformers());
56              ret = returnMessage.getPayload();
57          }
58          //Publish to interested clients
59          for (Client client : channel.getSubscribers())
60          {
61              channel.publish(client, ret, null);
62          }
63      }
64  }