CAMEL-12464 Add support for transform with context

components/camel-jolt/src/main/java/org/apache/camel/component/jolt/JoltEndpoint.java

@@ -17,21 +17,17 @@
 package org.apache.camel.component.jolt;
 
 import java.io.InputStream;
+import java.util.Map;
 
-import com.bazaarvoice.jolt.Chainr;
-import com.bazaarvoice.jolt.Defaultr;
-import com.bazaarvoice.jolt.JsonUtils;
-import com.bazaarvoice.jolt.Removr;
-import com.bazaarvoice.jolt.Shiftr;
-import com.bazaarvoice.jolt.Sortr;
-import com.bazaarvoice.jolt.Transform;
+import com.bazaarvoice.jolt.*;
 
 import org.apache.camel.Exchange;
 import org.apache.camel.ExchangePattern;
 import org.apache.camel.Message;
 import org.apache.camel.component.ResourceEndpoint;
 import org.apache.camel.spi.UriEndpoint;
 import org.apache.camel.spi.UriParam;
+import org.apache.camel.util.ExchangeHelper;
 import org.apache.camel.util.ObjectHelper;
 
 /**
@@ -40,8 +36,8 @@
 @UriEndpoint(firstVersion = "2.16.0", scheme = "jolt", title = "JOLT", syntax = "jolt:resourceUri", producerOnly = true, label = "transformation")
 public class JoltEndpoint extends ResourceEndpoint {
 
-    private Transform transform;
-    
+    private JoltTransform transform;
+
     @UriParam(defaultValue = "Hydrated")
     private JoltInputOutputType outputType;
 
@@ -73,7 +69,7 @@ protected String createEndpointUri() {
         return "jolt:" + getResourceUri();
     }
 
-    private synchronized Transform getTransform() throws Exception {
+    private synchronized JoltTransform getTransform() throws Exception {
         if (transform == null) {
             if (log.isDebugEnabled()) {
                 String path = getResourceUri();
@@ -110,7 +106,7 @@ private synchronized Transform getTransform() throws Exception {
     /**
      * Sets the Transform to use. If not set a Transform specified by the transformDsl will be created
      */
-    public void setTransform(Transform transform) {
+    public void setTransform(JoltTransform transform) {
         this.transform = transform;
     }
 
@@ -167,14 +163,24 @@ protected void onExchange(Exchange exchange) throws Exception {
             newEndpoint.onExchange(exchange);
             return;
         }
+
         Object input;
         if (getInputType() == JoltInputOutputType.JsonString) {
             input = JsonUtils.jsonToObject(exchange.getIn().getBody(InputStream.class));
         } else {
             input = exchange.getIn().getBody();
         }
-        Object output = getTransform().transform(input);
-
+
+        Object output;
+
+        @SuppressWarnings("unchecked")
+        Map<String, Object> inputContextMap = exchange.getIn().getHeader(JoltConstants.JOLT_CONTEXT, Map.class);
+        if (inputContextMap != null) {
+            output = ((ContextualTransform)getTransform()).transform(input, inputContextMap);
+        } else {
+            output = ((Transform)getTransform()).transform(input);
+        }
+
         // now lets output the results to the exchange
         Message out = exchange.getOut();
         if (getOutputType() == JoltInputOutputType.JsonString) {

components/camel-jolt/src/test/java/org/apache/camel/component/jolt/JoltFirstSampleTest.java

@@ -16,12 +16,17 @@
  */
 package org.apache.camel.component.jolt;
 
+import org.apache.camel.Exchange;
+import org.apache.camel.Processor;
 import org.apache.camel.builder.RouteBuilder;
 import org.apache.camel.test.junit4.CamelTestSupport;
 import org.apache.camel.util.IOHelper;
 import org.apache.camel.util.ResourceHelper;
 import org.junit.Test;
 
+import java.util.HashMap;
+import java.util.Map;
+
 /**
  * Unit test based on the first sample test from the JOLT project.
  */
@@ -46,9 +51,19 @@ public void testFirstSampleJolt() throws Exception {
 
     @Override
     protected RouteBuilder createRouteBuilder() throws Exception {
+        final Processor processor = new Processor() {
+            public void process(Exchange exchange) {
+                Map<String, String> contextMap = new HashMap<>();
+                contextMap.put("contextB", "bb");
+
+                exchange.getIn().setHeader(JoltConstants.JOLT_CONTEXT, contextMap);
+            }
+        };
+
         return new RouteBuilder() {
             public void configure() {
                 from("direct://start")
+                        .process(processor)
                     .to("jolt:org/apache/camel/component/jolt/firstSample/spec.json?inputType=JsonString&outputType=JsonString")
                     .to("mock:result");
             }

components/camel-jolt/src/test/resources/org/apache/camel/component/jolt/firstSample/output.json

@@ -1 +1 @@
-{"~a":"aa","~id":"id","~z":"zz","Rating":3,"RatingRange":5,"SecondaryRatings":{"quality":{"Id":"quality","Range":7,"Value":3},"sharpness":{"Id":"sharpness","Range":5,"Value":5}},"zz":"zz"}
+{"~a":"aa","~id":"id","~z":"zz","Rating":3,"RatingRange":5,"SecondaryRatings":{"quality":{"Id":"quality","Range":7,"Value":3},"sharpness":{"Id":"sharpness","Range":5,"Value":5}},"b":"bb","zz":"zz"}

components/camel-jolt/src/test/resources/org/apache/camel/component/jolt/firstSample/spec.json

@@ -34,6 +34,12 @@
                 "~deleteme": ""
             }
         },
+        {
+          "operation": "modify-overwrite-beta",
+          "spec": {
+            "b": "^contextB"
+          }
+        },
         {
             // last operation is to sort the JSON
             "operation": "sort"