Coverage Report - org.mule.transport.rmi.RmiMessageDispatcher
 
Classes in this File Line Coverage Branch Coverage Complexity
RmiMessageDispatcher
73%
22/30
50%
6/12
2.143
 
 1  
 /*
 2  
  * $Id: RmiMessageDispatcher.java 10961 2008-02-22 19:01:02Z dfeist $
 3  
  * --------------------------------------------------------------------------------------
 4  
  * Copyright (c) MuleSource, Inc.  All rights reserved.  http://www.mulesource.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  
 
 11  
 package org.mule.transport.rmi;
 12  
 
 13  
 import org.mule.DefaultMuleMessage;
 14  
 import org.mule.api.MuleEvent;
 15  
 import org.mule.api.MuleMessage;
 16  
 import org.mule.api.endpoint.OutboundEndpoint;
 17  
 import org.mule.api.transformer.TransformerException;
 18  
 import org.mule.transport.AbstractMessageDispatcher;
 19  
 
 20  
 import java.lang.reflect.Method;
 21  
 import java.rmi.RMISecurityManager;
 22  
 import java.rmi.Remote;
 23  
 import java.util.Collections;
 24  
 
 25  
 /**
 26  
  * <code>RmiMessageDispatcher</code> will send transformed mule events over
 27  
  * RMI-JRMP.
 28  
  */
 29  
 public class RmiMessageDispatcher extends AbstractMessageDispatcher
 30  
 {
 31  
     private final RmiConnector connector;
 32  
     protected volatile Remote remoteObject;
 33  
     protected volatile Method invokedMethod;
 34  
 
 35  
     public RmiMessageDispatcher(OutboundEndpoint endpoint)
 36  
     {
 37  48
         super(endpoint);
 38  48
         this.connector = (RmiConnector)endpoint.getConnector();
 39  48
     }
 40  
 
 41  
     protected void doConnect() throws Exception
 42  
     {
 43  48
         if (remoteObject == null)
 44  
         {
 45  
             // Shouldn't all this be in the connector?
 46  48
             String rmiPolicyPath = connector.getSecurityPolicy();
 47  48
             System.setProperty("java.security.policy", rmiPolicyPath);
 48  
 
 49  
             // Set security manager
 50  48
             if (System.getSecurityManager() == null)
 51  
             {
 52  0
                 System.setSecurityManager(new RMISecurityManager());
 53  
             }
 54  
 
 55  48
             remoteObject = connector.getRemoteObject(endpoint);
 56  
         }
 57  48
     }
 58  
 
 59  
     protected void doDisconnect() throws Exception
 60  
     {
 61  48
         remoteObject = null;
 62  48
         invokedMethod = null;
 63  48
     }
 64  
 
 65  
     private Object[] getArgs(MuleEvent event) throws TransformerException
 66  
     {
 67  42
         Object payload = event.transformMessage();
 68  42
         if (payload instanceof Object[])
 69  
         {
 70  0
             return (Object[])payload;
 71  
         }
 72  
         else
 73  
         {
 74  42
             return new Object[]{payload};
 75  
         }
 76  
     }
 77  
 
 78  
     /*
 79  
      * (non-Javadoc)
 80  
      * 
 81  
      * @see org.mule.api.transport.UMOConnectorSession#dispatch(org.mule.api.MuleEvent)
 82  
      */
 83  
     protected void doDispatch(MuleEvent event) throws Exception
 84  
     {
 85  0
         Object[] arguments = getArgs(event);
 86  0
         if (invokedMethod == null)
 87  
         {
 88  0
             invokedMethod = connector.getMethodObject(remoteObject, event);
 89  
         }
 90  0
         invokedMethod.invoke(remoteObject, arguments);
 91  0
     }
 92  
 
 93  
     /*
 94  
      * (non-Javadoc)
 95  
      * 
 96  
      * @see org.mule.api.transport.UMOConnectorSession#send(org.mule.api.MuleEvent)
 97  
      */
 98  
     public MuleMessage doSend(MuleEvent event) throws Exception
 99  
     {
 100  60
         if (invokedMethod == null)
 101  
         {
 102  48
             invokedMethod = connector.getMethodObject(remoteObject, event);
 103  
         }
 104  
 
 105  42
         Object[] arguments = getArgs(event);
 106  42
         Object result = invokedMethod.invoke(remoteObject, arguments);
 107  
 
 108  42
         if (result == null)
 109  
         {
 110  0
             return null;
 111  
         }
 112  
         else
 113  
         {
 114  42
             return new DefaultMuleMessage(connector.getMessageAdapter(result).getPayload(), Collections.EMPTY_MAP);
 115  
         }
 116  
     }
 117  
 
 118  
     protected void doDispose()
 119  
     {
 120  
         // template method
 121  48
     }
 122  
 }