Coverage Report - org.mule.impl.MuleSessionHandler
 
Classes in this File Line Coverage Branch Coverage Complexity
MuleSessionHandler
0%
0/42
0%
0/9
5
 
 1  
 /*
 2  
  * $Id: MuleSessionHandler.java 7976 2007-08-21 14:26:13Z dirk.olmes $
 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.impl;
 12  
 
 13  
 import org.mule.config.MuleProperties;
 14  
 import org.mule.config.i18n.CoreMessages;
 15  
 import org.mule.transformers.codec.Base64Decoder;
 16  
 import org.mule.transformers.codec.Base64Encoder;
 17  
 import org.mule.umo.UMOException;
 18  
 import org.mule.umo.UMOMessage;
 19  
 import org.mule.umo.UMOSession;
 20  
 import org.mule.umo.provider.UMOSessionHandler;
 21  
 import org.mule.umo.transformer.UMOTransformer;
 22  
 
 23  
 import java.io.UnsupportedEncodingException;
 24  
 import java.util.Iterator;
 25  
 import java.util.StringTokenizer;
 26  
 
 27  
 import org.apache.commons.logging.Log;
 28  
 import org.apache.commons.logging.LogFactory;
 29  
 
 30  
 /**
 31  
  * A default session handler used to store and retrieve session information on an
 32  
  * event. The MuleSession information is stored as a header on the message (does not
 33  
  * support Tcp, Udp, etc. unless the UMOMessage object is serialised across the
 34  
  * wire). The session is stored in the "MULE_SESSION" property as String key/value
 35  
  * pairs that are Base64 encoded, for example:
 36  
  * ID=dfokokdf-3ek3oke-dkfokd;MySessionProp1=Value1;MySessionProp2=Value2
 37  
  */
 38  0
 public class MuleSessionHandler implements UMOSessionHandler
 39  
 {
 40  
 
 41  
     /**
 42  
      * logger used by this class
 43  
      */
 44  0
     protected transient Log logger = LogFactory.getLog(getClass());
 45  
 
 46  0
     private static UMOTransformer encoder = new Base64Encoder();
 47  0
     private static UMOTransformer decoder = new Base64Decoder();
 48  
 
 49  
     public void retrieveSessionInfoFromMessage(UMOMessage message, UMOSession session) throws UMOException
 50  
     {
 51  0
         String sessionId = (String) message.removeProperty(MuleProperties.MULE_SESSION_ID_PROPERTY);
 52  0
         Object sessionHeader = message.removeProperty(MuleProperties.MULE_SESSION_PROPERTY);
 53  
 
 54  0
         if (sessionId != null)
 55  
         {
 56  
             // TODO Mule 2.0 grab session from the context
 57  0
             throw new IllegalStateException(
 58  
                 "This session handler does not know how to look up session information for session id: "
 59  
                                 + sessionId);
 60  
         }
 61  0
         if (sessionHeader != null)
 62  
         {
 63  0
             String sessionString = null;
 64  
             try
 65  
             {
 66  0
                 sessionString = new String((byte[]) decoder.transform(sessionHeader), message.getEncoding());
 67  
             }
 68  0
             catch (UnsupportedEncodingException e)
 69  
             {
 70  0
                 sessionString = new String((byte[]) decoder.transform(sessionHeader));
 71  0
             }
 72  0
             if (logger.isDebugEnabled())
 73  
             {
 74  0
                 logger.debug("Parsing session header: " + sessionString);
 75  
             }
 76  
             String pair;
 77  
             String name;
 78  
             String value;
 79  0
             for (StringTokenizer stringTokenizer = new StringTokenizer(sessionString, ";"); stringTokenizer.hasMoreTokens();)
 80  
             {
 81  0
                 pair = stringTokenizer.nextToken();
 82  0
                 int i = pair.indexOf("=");
 83  0
                 if (i == -1)
 84  
                 {
 85  0
                     throw new IllegalArgumentException(
 86  
                         CoreMessages.sessionValueIsMalformed(pair).toString());
 87  
                 }
 88  0
                 name = pair.substring(0, i).trim();
 89  0
                 value = pair.substring(i + 1).trim();
 90  0
                 session.setProperty(name, value);
 91  0
                 if (logger.isDebugEnabled())
 92  
                 {
 93  0
                     logger.debug("Added Session variable: " + pair);
 94  
                 }
 95  
             }
 96  
 
 97  
         }
 98  0
     }
 99  
 
 100  
     public void storeSessionInfoToMessage(UMOSession session, UMOMessage message) throws UMOException
 101  
     {
 102  0
         StringBuffer buf = new StringBuffer();
 103  0
         buf.append(getSessionIDKey()).append("=").append(session.getId());
 104  0
         for (Iterator iterator = session.getPropertyNames(); iterator.hasNext();)
 105  
         {
 106  0
             Object o = iterator.next();
 107  0
             buf.append(";");
 108  0
             buf.append(o).append("=").append(session.getProperty(o));
 109  0
             if (logger.isDebugEnabled())
 110  
             {
 111  0
                 logger.debug("Adding property to session header: " + o + "=" + session.getProperty(o));
 112  
             }
 113  
         }
 114  0
         String sessionString = buf.toString();
 115  0
         if (logger.isDebugEnabled())
 116  
         {
 117  0
             logger.debug("Adding session header to message: " + sessionString);
 118  
         }
 119  0
         sessionString = (String) encoder.transform(sessionString);
 120  0
         message.setProperty(MuleProperties.MULE_SESSION_PROPERTY, sessionString);
 121  0
     }
 122  
 
 123  
     public String getSessionIDKey()
 124  
     {
 125  0
         return "ID";
 126  
     }
 127  
 }