Coverage Report - org.mule.routing.inbound.IdempotentReceiver
 
Classes in this File Line Coverage Branch Coverage Complexity
IdempotentReceiver
63%
27/43
64%
9/14
2.6
 
 1  
 /*
 2  
  * $Id: IdempotentReceiver.java 12296 2008-07-11 16:56:21Z 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.routing.inbound;
 12  
 
 13  
 import org.mule.api.MessagingException;
 14  
 import org.mule.api.MuleEvent;
 15  
 import org.mule.api.lifecycle.InitialisationException;
 16  
 import org.mule.api.routing.RoutingException;
 17  
 import org.mule.api.store.ObjectStore;
 18  
 import org.mule.config.i18n.CoreMessages;
 19  
 import org.mule.util.expression.ExpressionEvaluatorManager;
 20  
 import org.mule.util.store.InMemoryObjectStore;
 21  
 
 22  
 /**
 23  
  * <code>IdempotentReceiver</code> ensures that only unique messages are received by a
 24  
  * service. It does this by checking the unique ID of the incoming message. Note that
 25  
  * the underlying endpoint must support unique message IDs for this to work, otherwise a
 26  
  * <code>UniqueIdNotSupportedException</code> is thrown.<br>
 27  
  * By default this implementation uses an instance of
 28  
  * {@link IdempotentInMemoryMessageIdStore}.
 29  
  */
 30  
 public class IdempotentReceiver extends SelectiveConsumer
 31  
 {
 32  
     protected volatile ObjectStore store;
 33  
     protected volatile String assignedComponentName;
 34  
 
 35  2
     protected String idExpression = "${message:id}";
 36  
 
 37  
     public IdempotentReceiver()
 38  
     {
 39  2
         super();
 40  2
     }
 41  
 
 42  
     protected void initialize(MuleEvent event) throws RoutingException
 43  
     {
 44  2
         if (assignedComponentName == null && store == null)
 45  
         {
 46  2
             this.assignedComponentName = event.getService().getName();
 47  
             try
 48  
             {
 49  2
                 this.store = this.createMessageIdStore();
 50  
             }
 51  0
             catch (InitialisationException e)
 52  
             {
 53  0
                 throw new RoutingException(event.getMessage(), event.getEndpoint(), e);
 54  2
             }
 55  
         }
 56  2
     }
 57  
 
 58  
     protected ObjectStore createMessageIdStore() throws InitialisationException
 59  
     {
 60  2
         InMemoryObjectStore s = new InMemoryObjectStore();
 61  2
         s.setName(assignedComponentName);
 62  2
         s.setMaxEntries(-1);
 63  2
         s.setEntryTTL(60 * 5);
 64  2
         s.setExpirationInterval(6000);
 65  2
         s.initialise();
 66  2
         return s;
 67  
     }
 68  
 
 69  
     // @Override
 70  
     public boolean isMatch(MuleEvent event) throws MessagingException
 71  
     {
 72  10
         if (!super.isMatch(event))
 73  
         {
 74  0
             return false;
 75  
         }
 76  
         else
 77  
         {
 78  10
             if (store == null)
 79  
             {
 80  
                 // we need to load this on the first request as we need the service
 81  
                 // name
 82  2
                 synchronized (this)
 83  
                 {
 84  2
                     this.initialize(event);
 85  2
                 }
 86  
             }
 87  
 
 88  
             try
 89  
             {
 90  10
                 return !store.containsObject(this.getIdForEvent(event));
 91  
             }
 92  0
             catch (Exception ex)
 93  
             {
 94  0
                 throw new RoutingException(event.getMessage(), event.getEndpoint(), ex);
 95  
             }
 96  
         }
 97  
     }
 98  
 
 99  
     // @Override
 100  
     public MuleEvent[] process(MuleEvent event) throws MessagingException
 101  
     {
 102  4
         String eventComponentName = event.getService().getName();
 103  4
         if (!assignedComponentName.equals(eventComponentName))
 104  
         {
 105  0
             IllegalArgumentException iex = new IllegalArgumentException(
 106  
                 "This receiver is assigned to service: " + assignedComponentName
 107  
                                 + " but has received an event for service: " + eventComponentName
 108  
                                 + ". Please check your config to make sure each service"
 109  
                                 + "has its own instance of IdempotentReceiver.");
 110  0
             throw new RoutingException(event.getMessage(), event.getEndpoint(), iex);
 111  
         }
 112  
 
 113  4
         String id = this.getIdForEvent(event);
 114  
 
 115  
         try
 116  
         {
 117  4
             if (store.storeObject(id, id))
 118  
             {
 119  4
                 return new MuleEvent[]{event};
 120  
             }
 121  
             else
 122  
             {
 123  0
                 return null;
 124  
             }
 125  
         }
 126  0
         catch (Exception e)
 127  
         {
 128  0
             throw new RoutingException(CoreMessages.failedToWriteMessageToStore(id, assignedComponentName),
 129  
                 event.getMessage(), event.getEndpoint(), e);
 130  
         }
 131  
     }
 132  
 
 133  
     protected String getIdForEvent(MuleEvent event) throws MessagingException
 134  
     {
 135  14
         return ExpressionEvaluatorManager.parse(idExpression, event.getMessage(), true);
 136  
     }
 137  
 
 138  
     public String getIdExpression()
 139  
     {
 140  0
         return idExpression;
 141  
     }
 142  
 
 143  
     public void setIdExpression(String idExpression)
 144  
     {
 145  0
         this.idExpression = idExpression;
 146  0
     }
 147  
 
 148  
     public ObjectStore getStore()
 149  
     {
 150  0
         return store;
 151  
     }
 152  
 
 153  
     public void setStore(ObjectStore store)
 154  
     {
 155  0
         this.store = store;
 156  0
     }
 157  
 }