View Javadoc

1   /*
2    * $Id: AjaxReplyToHandler.java 20321 2010-11-24 15:21:24Z dfeist $
3    * --------------------------------------------------------------------------------------
4    * Copyright (c) MuleSoft, Inc.  All rights reserved.  http://www.mulesoft.com
5    *
6    * The software in this package is published under the terms of the CPAL v1.0
7    * license, a copy of which has been included with this distribution in the
8    * LICENSE.txt file.
9    */
10  package org.mule.transport.ajax;
11  
12  import org.mule.api.MuleEvent;
13  import org.mule.api.MuleException;
14  import org.mule.api.MuleMessage;
15  import org.mule.api.transformer.Transformer;
16  import org.mule.api.transport.Connector;
17  import org.mule.transport.DefaultReplyToHandler;
18  
19  import java.util.List;
20  
21  import org.cometd.Channel;
22  import org.cometd.Client;
23  import org.mortbay.cometd.AbstractBayeux;
24  
25  /**
26   * Handles the sending of sending result messages back to the client when the a replyTo 
27   * channel is specified in the client request.
28   */
29  public class AjaxReplyToHandler extends DefaultReplyToHandler
30  {
31      private Connector connector;
32      
33      public AjaxReplyToHandler(List<Transformer> transformers, Connector connector)
34      {
35          super(transformers, connector.getMuleContext());
36          this.connector = connector;
37      }
38  
39      @Override
40      public void processReplyTo(MuleEvent event, MuleMessage returnMessage, Object replyTo) throws MuleException
41      {
42          AbstractBayeux bayeux = ((BayeuxAware)connector).getBayeux();
43          Channel channel = bayeux.getChannel(replyTo.toString(), false);
44          if(channel==null)
45          {
46              logger.warn("No ajax Channel: " + replyTo + ". Maybe the client unregistered interest.");
47              return;
48          }
49          
50          Object ret;
51          if(returnMessage.getExceptionPayload()!=null)
52          {
53              //If we are using RPC make sure we still send something back to the client so that the subscription is cancelled
54              ret = returnMessage.getExceptionPayload().getMessage();
55          }
56          else
57          {
58              returnMessage.applyTransformers(event, getTransformers());
59              ret = returnMessage.getPayload();
60          }
61          //Publish to interested clients
62          for (Client client : channel.getSubscribers())
63          {
64              channel.publish(client, ret, null);
65          }
66      }
67  }