Coverage Report - org.mule.enricher.MessageEnricher
 
Classes in this File Line Coverage Branch Coverage Complexity
MessageEnricher
0%
0/27
0%
0/10
0
MessageEnricher$EnrichExpressionPair
0%
0/15
N/A
0
 
 1  
 /*
 2  
  * $Id: MessageEnricher.java 20555 2010-12-09 17:54:31Z aperepel $
 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.enricher;
 12  
 
 13  
 import org.mule.RequestContext;
 14  
 import org.mule.api.MuleEvent;
 15  
 import org.mule.api.MuleException;
 16  
 import org.mule.api.MuleMessage;
 17  
 import org.mule.api.expression.ExpressionManager;
 18  
 import org.mule.api.processor.MessageProcessor;
 19  
 import org.mule.processor.AbstractMessageProcessorOwner;
 20  
 import org.mule.util.StringUtils;
 21  
 
 22  
 import java.util.ArrayList;
 23  
 import java.util.Collections;
 24  
 import java.util.List;
 25  
 
 26  0
 public class MessageEnricher extends AbstractMessageProcessorOwner implements MessageProcessor
 27  
 {
 28  
 
 29  0
     private List<EnrichExpressionPair> enrichExpressionPairs = new ArrayList<EnrichExpressionPair>();
 30  
 
 31  
     private MessageProcessor enrichmentProcessor;
 32  
 
 33  
     public MuleEvent process(MuleEvent event) throws MuleException
 34  
     {
 35  0
         ExpressionManager expressionManager = event.getMuleContext().getExpressionManager();
 36  0
         MuleMessage enrichmentMessage = enrichmentProcessor.process(
 37  
             RequestContext.cloneAndUpdateEventEndpoint(event, enrichmentProcessor)).getMessage();
 38  
 
 39  0
         if (enrichmentMessage != null)
 40  
         {
 41  0
             for (EnrichExpressionPair pair : enrichExpressionPairs)
 42  
             {
 43  0
                 enrich(event.getMessage(), enrichmentMessage, pair.getSource(), pair.getTarget(),
 44  
                     expressionManager);
 45  
             }
 46  
         }
 47  0
         return event;
 48  
     }
 49  
 
 50  
     protected void enrich(MuleMessage currentMessage,
 51  
                           MuleMessage enrichmentMessage,
 52  
                           String sourceExpressionArg,
 53  
                           String targetExpressionArg,
 54  
                           ExpressionManager expressionManager)
 55  
     {
 56  0
         if (StringUtils.isEmpty(sourceExpressionArg))
 57  
         {
 58  0
             sourceExpressionArg = "#[payload]";
 59  
         }
 60  
 
 61  0
         Object enrichmentObject = expressionManager.evaluate(sourceExpressionArg, enrichmentMessage);
 62  0
         if (enrichmentObject instanceof MuleMessage)
 63  
         {
 64  0
             enrichmentObject = ((MuleMessage) enrichmentObject).getPayload();
 65  
         }
 66  
 
 67  0
         if (!StringUtils.isEmpty(targetExpressionArg))
 68  
         {
 69  0
             expressionManager.enrich(targetExpressionArg, currentMessage, enrichmentObject);
 70  
         }
 71  
         else
 72  
         {
 73  0
             currentMessage.setPayload(enrichmentObject);
 74  
         }
 75  0
     }
 76  
 
 77  
     public void setEnrichmentMessageProcessor(MessageProcessor enrichmentProcessor)
 78  
     {
 79  0
         this.enrichmentProcessor = enrichmentProcessor;
 80  0
     }
 81  
 
 82  
     /**
 83  
      * For spring
 84  
      */
 85  
     public void setMessageProcessor(MessageProcessor enrichmentProcessor)
 86  
     {
 87  0
         this.enrichmentProcessor = enrichmentProcessor;
 88  0
     }
 89  
 
 90  
     public void setEnrichExpressionPairs(List<EnrichExpressionPair> enrichExpressionPairs)
 91  
     {
 92  0
         this.enrichExpressionPairs = enrichExpressionPairs;
 93  0
     }
 94  
 
 95  
     public void addEnrichExpressionPair(EnrichExpressionPair pair)
 96  
     {
 97  0
         this.enrichExpressionPairs.add(pair);
 98  0
     }
 99  
 
 100  0
     public static class EnrichExpressionPair
 101  
     {
 102  
 
 103  
         private String source;
 104  
         private String target;
 105  
 
 106  
         public EnrichExpressionPair()
 107  0
         {
 108  
             // for spring
 109  0
         }
 110  
 
 111  
         public EnrichExpressionPair(String target)
 112  0
         {
 113  0
             this.target = target;
 114  0
         }
 115  
 
 116  
         public EnrichExpressionPair(String source, String target)
 117  0
         {
 118  0
             this.source = source;
 119  0
             this.target = target;
 120  0
         }
 121  
 
 122  
         public String getSource()
 123  
         {
 124  0
             return source;
 125  
         }
 126  
 
 127  
         public void setSource(String source)
 128  
         {
 129  0
             this.source = source;
 130  0
         }
 131  
 
 132  
         public String getTarget()
 133  
         {
 134  0
             return target;
 135  
         }
 136  
 
 137  
         public void setTarget(String target)
 138  
         {
 139  0
             this.target = target;
 140  0
         }
 141  
     }
 142  
 
 143  
     @Override
 144  
     protected List<MessageProcessor> getOwnedMessageProcessors()
 145  
     {
 146  0
         return Collections.singletonList(enrichmentProcessor);
 147  
     }
 148  
 }