Coverage Report - org.mule.util.queue.TransactionalQueueSession
 
Classes in this File Line Coverage Branch Coverage Complexity
TransactionalQueueSession
0%
0/6
N/A
3.778
TransactionalQueueSession$QueueImpl
0%
0/41
0%
0/7
3.778
 
 1  
 /*
 2  
  * $Id: TransactionalQueueSession.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.util.queue;
 12  
 
 13  
 import org.mule.util.xa.AbstractXAResourceManager;
 14  
 import org.mule.util.xa.DefaultXASession;
 15  
 
 16  
 import java.io.IOException;
 17  
 
 18  
 /**
 19  
  * A Queue session that is used to manage the transaction context of a Queue
 20  
  * 
 21  
  * @author <a href="mailto:ross.mason@symphonysoft.com">Ross Mason</a>
 22  
  * @version $Revision: 7976 $
 23  
  */
 24  0
 class TransactionalQueueSession extends DefaultXASession implements QueueSession
 25  
 {
 26  
 
 27  
     protected TransactionalQueueManager queueManager;
 28  
 
 29  
     public TransactionalQueueSession(AbstractXAResourceManager resourceManager,
 30  
                                      TransactionalQueueManager queueManager)
 31  
     {
 32  0
         super(resourceManager);
 33  0
         this.queueManager = queueManager;
 34  0
     }
 35  
 
 36  
     /*
 37  
      * (non-Javadoc)
 38  
      * 
 39  
      * @see org.mule.transaction.xa.queue.QueueSession#getQueue(java.lang.String)
 40  
      */
 41  
     public Queue getQueue(String name)
 42  
     {
 43  0
         QueueInfo queue = queueManager.getQueue(name);
 44  0
         return new QueueImpl(queue);
 45  
     }
 46  
 
 47  
     protected class QueueImpl implements Queue
 48  
     {
 49  
 
 50  
         protected QueueInfo queue;
 51  
 
 52  
         public QueueImpl(QueueInfo queue)
 53  0
         {
 54  0
             this.queue = queue;
 55  0
         }
 56  
 
 57  
         public void put(Object item) throws InterruptedException
 58  
         {
 59  0
             offer(item, Long.MAX_VALUE);
 60  0
         }
 61  
 
 62  
         public boolean offer(Object item, long timeout) throws InterruptedException
 63  
         {
 64  0
             if (localContext != null)
 65  
             {
 66  0
                 return ((TransactionalQueueManager.QueueTransactionContext) localContext).offer(queue, item,
 67  
                     timeout);
 68  
             }
 69  
             else
 70  
             {
 71  
                 try
 72  
                 {
 73  0
                     Object id = queueManager.doStore(queue, item);
 74  
                     try
 75  
                     {
 76  0
                         if (!queue.offer(id, 0, timeout))
 77  
                         {
 78  0
                             queueManager.doRemove(queue, item);
 79  0
                             return false;
 80  
                         }
 81  
                         else
 82  
                         {
 83  0
                             return true;
 84  
                         }
 85  
                     }
 86  0
                     catch (InterruptedException e)
 87  
                     {
 88  0
                         queueManager.doRemove(queue, item);
 89  0
                         throw e;
 90  
                     }
 91  
                 }
 92  0
                 catch (IOException e)
 93  
                 {
 94  0
                     throw new RuntimeException(e);
 95  
                 }
 96  
             }
 97  
         }
 98  
 
 99  
         public Object take() throws InterruptedException
 100  
         {
 101  0
             return poll(Long.MAX_VALUE);
 102  
         }
 103  
 
 104  
         public Object poll(long timeout) throws InterruptedException
 105  
         {
 106  
             try
 107  
             {
 108  0
                 if (localContext != null)
 109  
                 {
 110  0
                     return ((TransactionalQueueManager.QueueTransactionContext) localContext).poll(queue,
 111  
                         timeout);
 112  
                 }
 113  
                 else
 114  
                 {
 115  0
                     Object id = queue.poll(timeout);
 116  0
                     if (id != null)
 117  
                     {
 118  0
                         Object item = queueManager.doLoad(queue, id);
 119  0
                         queueManager.doRemove(queue, id);
 120  0
                         return item;
 121  
                     }
 122  0
                     return null;
 123  
                 }
 124  
             }
 125  0
             catch (IOException e)
 126  
             {
 127  0
                 throw new RuntimeException(e);
 128  
             }
 129  
         }
 130  
 
 131  
         public Object peek() throws InterruptedException
 132  
         {
 133  
             try
 134  
             {
 135  0
                 if (localContext != null)
 136  
                 {
 137  0
                     return ((TransactionalQueueManager.QueueTransactionContext) localContext).peek(queue);
 138  
                 }
 139  
                 else
 140  
                 {
 141  0
                     Object id = queue.peek();
 142  0
                     if (id != null)
 143  
                     {
 144  0
                         Object item = queueManager.doLoad(queue, id);
 145  0
                         queueManager.doRemove(queue, id);
 146  0
                         return item;
 147  
                     }
 148  0
                     return null;
 149  
                 }
 150  
             }
 151  0
             catch (IOException e)
 152  
             {
 153  0
                 throw new RuntimeException(e);
 154  
             }
 155  
         }
 156  
 
 157  
         public int size()
 158  
         {
 159  0
             if (localContext != null)
 160  
             {
 161  0
                 return ((TransactionalQueueManager.QueueTransactionContext) localContext).size(queue);
 162  
             }
 163  
             else
 164  
             {
 165  0
                 return queue.list.size();
 166  
             }
 167  
         }
 168  
 
 169  
     }
 170  
 }