Author: hadrian
Date: Tue Jan 27 16:04:02 2009
New Revision: 738122

URL: http://svn.apache.org/viewvc?rev=738122&view=rev
Log:
CAMEL-1002.  Patch applied with thanks to Martin.

Added:
    
camel/branches/camel-1.x/components/camel-quartz/src/main/java/org/apache/camel/component/quartz/StatefulCamelJob.java
    
camel/branches/camel-1.x/components/camel-quartz/src/test/java/org/apache/camel/component/quartz/StatefulQuartzRouteTest.java
Modified:
    
camel/branches/camel-1.x/components/camel-quartz/src/main/java/org/apache/camel/component/quartz/QuartzComponent.java
    
camel/branches/camel-1.x/components/camel-quartz/src/main/java/org/apache/camel/component/quartz/QuartzEndpoint.java

Modified: 
camel/branches/camel-1.x/components/camel-quartz/src/main/java/org/apache/camel/component/quartz/QuartzComponent.java
URL: 
http://svn.apache.org/viewvc/camel/branches/camel-1.x/components/camel-quartz/src/main/java/org/apache/camel/component/quartz/QuartzComponent.java?rev=738122&r1=738121&r2=738122&view=diff
==============================================================================
--- 
camel/branches/camel-1.x/components/camel-quartz/src/main/java/org/apache/camel/component/quartz/QuartzComponent.java
 (original)
+++ 
camel/branches/camel-1.x/components/camel-quartz/src/main/java/org/apache/camel/component/quartz/QuartzComponent.java
 Tue Jan 27 16:04:02 2009
@@ -46,12 +46,12 @@
     public QuartzComponent() {
     }
 
-    public QuartzComponent(CamelContext context) {
+    public QuartzComponent(final CamelContext context) {
         super(context);
     }
 
     @Override
-    protected QuartzEndpoint createEndpoint(String uri, String remaining, Map 
parameters) throws Exception {
+    protected QuartzEndpoint createEndpoint(final String uri, final String 
remaining, final Map parameters) throws Exception {
         QuartzEndpoint answer = new QuartzEndpoint(uri, this, getScheduler());
 
         // lets split the remaining into a group/name
@@ -126,7 +126,7 @@
         return factory;
     }
 
-    public void setFactory(SchedulerFactory factory) {
+    public void setFactory(final SchedulerFactory factory) {
         this.factory = factory;
     }
 
@@ -137,7 +137,7 @@
         return scheduler;
     }
 
-    public void setScheduler(Scheduler scheduler) {
+    public void setScheduler(final Scheduler scheduler) {
         this.scheduler = scheduler;
     }
 
@@ -145,7 +145,7 @@
         return triggers;
     }
 
-    public void setTriggers(Map triggers) {
+    public void setTriggers(final Map triggers) {
         this.triggers = triggers;
     }
 
@@ -156,6 +156,8 @@
     }
 
     protected Scheduler createScheduler() throws SchedulerException {
-        return getFactory().getScheduler();
+        Scheduler scheduler = getFactory().getScheduler();
+        scheduler.getContext().put(QuartzEndpoint.CONTEXT_KEY, 
getCamelContext());
+        return scheduler;
     }
 }

Modified: 
camel/branches/camel-1.x/components/camel-quartz/src/main/java/org/apache/camel/component/quartz/QuartzEndpoint.java
URL: 
http://svn.apache.org/viewvc/camel/branches/camel-1.x/components/camel-quartz/src/main/java/org/apache/camel/component/quartz/QuartzEndpoint.java?rev=738122&r1=738121&r2=738122&view=diff
==============================================================================
--- 
camel/branches/camel-1.x/components/camel-quartz/src/main/java/org/apache/camel/component/quartz/QuartzEndpoint.java
 (original)
+++ 
camel/branches/camel-1.x/components/camel-quartz/src/main/java/org/apache/camel/component/quartz/QuartzEndpoint.java
 Tue Jan 27 16:04:02 2009
@@ -44,24 +44,27 @@
  */
 public class QuartzEndpoint extends DefaultEndpoint<QuartzExchange> {
     public static final String ENDPOINT_KEY = "org.apache.camel.quartz";
+    public static final String CONTEXT_KEY = "org.apache.camel.CamelContext";
+
     private static final transient Log LOG = 
LogFactory.getLog(QuartzEndpoint.class);
     private Scheduler scheduler;
     private LoadBalancer loadBalancer;
     private Trigger trigger;
     private JobDetail jobDetail;
     private boolean started;
+    private boolean stateful;
 
-    public QuartzEndpoint(String endpointUri, QuartzComponent component, 
Scheduler scheduler) {
+    public QuartzEndpoint(final String endpointUri, final QuartzComponent 
component, final Scheduler scheduler) {
         super(endpointUri, component);
         this.scheduler = scheduler;
     }
 
-    public QuartzEndpoint(String endpointUri, Scheduler scheduler) {
+    public QuartzEndpoint(final String endpointUri, final Scheduler scheduler) 
{
         super(endpointUri);
         this.scheduler = scheduler;
     }
 
-    public void addTriggers(Map<Trigger, JobDetail> triggerMap) throws 
SchedulerException {
+    public void addTriggers(final Map<Trigger, JobDetail> triggerMap) throws 
SchedulerException {
         if (triggerMap != null) {
             Set<Map.Entry<Trigger, JobDetail>> entries = triggerMap.entrySet();
             for (Map.Entry<Trigger, JobDetail> entry : entries) {
@@ -75,7 +78,7 @@
         }
     }
 
-    public void addTrigger(Trigger trigger, JobDetail detail) throws 
SchedulerException {
+    public void addTrigger(final Trigger trigger, final JobDetail detail) 
throws SchedulerException {
         // lets default the trigger name to the job name
         if (trigger.getName() == null) {
             trigger.setName(detail.getName());
@@ -88,10 +91,17 @@
         if (trigger.getStartTime() == null) {
             trigger.setStartTime(new Date());
         }
-        detail.getJobDataMap().put(ENDPOINT_KEY, this);
-        Class jobClass = detail.getJobClass();
-        if (jobClass == null) {
-            detail.setJobClass(CamelJob.class);
+        if (isStateful()) {
+            detail.getJobDataMap().put(ENDPOINT_KEY, getEndpointUri());
+        } else {
+            detail.getJobDataMap().put(ENDPOINT_KEY, this);
+        }
+        if (null == detail.getJobClass()) {
+            if (isStateful()) {
+                detail.setJobClass(StatefulCamelJob.class);
+            } else {
+                detail.setJobClass(CamelJob.class);
+            }
         }
         if (detail.getName() == null) {
             detail.setName(getEndpointUri());
@@ -99,7 +109,7 @@
         getScheduler().scheduleJob(detail, trigger);
     }
 
-    public void removeTrigger(Trigger trigger, JobDetail jobDetail) throws 
SchedulerException {
+    public void removeTrigger(final Trigger trigger, final JobDetail 
jobDetail) throws SchedulerException {
         getScheduler().unscheduleJob(trigger.getName(), trigger.getGroup());
     }
 
@@ -108,7 +118,7 @@
      *
      * @param jobExecutionContext the Quartz Job context
      */
-    public void onJobExecute(JobExecutionContext jobExecutionContext) throws 
JobExecutionException {
+    public void onJobExecute(final JobExecutionContext jobExecutionContext) 
throws JobExecutionException {
         if (LOG.isDebugEnabled()) {
             LOG.debug("Firing Quartz Job with context: " + 
jobExecutionContext);
         }
@@ -123,11 +133,11 @@
     }
 
     @Override
-    public QuartzExchange createExchange(ExchangePattern pattern) {
+    public QuartzExchange createExchange(final ExchangePattern pattern) {
         return new QuartzExchange(getCamelContext(), pattern, null);
     }
 
-    public QuartzExchange createExchange(JobExecutionContext 
jobExecutionContext) {
+    public QuartzExchange createExchange(final JobExecutionContext 
jobExecutionContext) {
         return new QuartzExchange(getCamelContext(), getExchangePattern(), 
jobExecutionContext);
     }
 
@@ -135,7 +145,7 @@
         throw new UnsupportedOperationException("You cannot send messages to 
this endpoint");
     }
 
-    public QuartzConsumer createConsumer(Processor processor) throws Exception 
{
+    public QuartzConsumer createConsumer(final Processor processor) throws 
Exception {
         return new QuartzConsumer(this, processor);
     }
 
@@ -162,7 +172,7 @@
         return loadBalancer;
     }
 
-    public void setLoadBalancer(LoadBalancer loadBalancer) {
+    public void setLoadBalancer(final LoadBalancer loadBalancer) {
         this.loadBalancer = loadBalancer;
     }
 
@@ -173,7 +183,7 @@
         return jobDetail;
     }
 
-    public void setJobDetail(JobDetail jobDetail) {
+    public void setJobDetail(final JobDetail jobDetail) {
         this.jobDetail = jobDetail;
     }
 
@@ -184,13 +194,27 @@
         return trigger;
     }
 
-    public void setTrigger(Trigger trigger) {
+    public void setTrigger(final Trigger trigger) {
         this.trigger = trigger;
     }
 
+    /**
+     * @return the stateful
+     */
+    public boolean isStateful() {
+        return this.stateful;
+    }
+
+    /**
+     * @param stateful the stateful to set
+     */
+    public void setStateful(final boolean stateful) {
+        this.stateful = stateful;
+    }
+
     // Implementation methods
     // 
-------------------------------------------------------------------------
-    public synchronized void consumerStarted(QuartzConsumer consumer) throws 
SchedulerException {
+    public synchronized void consumerStarted(final QuartzConsumer consumer) 
throws SchedulerException {
         getLoadBalancer().addProcessor(consumer.getProcessor());
 
         // if we have not yet added our default trigger, then lets do it
@@ -200,7 +224,7 @@
         }
     }
 
-    public synchronized void consumerStopped(QuartzConsumer consumer) throws 
SchedulerException {
+    public synchronized void consumerStopped(final QuartzConsumer consumer) 
throws SchedulerException {
         getLoadBalancer().removeProcessor(consumer.getProcessor());
         if (getLoadBalancer().getProcessors().isEmpty() && started) {
             removeTrigger(getTrigger(), getJobDetail());

Added: 
camel/branches/camel-1.x/components/camel-quartz/src/main/java/org/apache/camel/component/quartz/StatefulCamelJob.java
URL: 
http://svn.apache.org/viewvc/camel/branches/camel-1.x/components/camel-quartz/src/main/java/org/apache/camel/component/quartz/StatefulCamelJob.java?rev=738122&view=auto
==============================================================================
--- 
camel/branches/camel-1.x/components/camel-quartz/src/main/java/org/apache/camel/component/quartz/StatefulCamelJob.java
 (added)
+++ 
camel/branches/camel-1.x/components/camel-quartz/src/main/java/org/apache/camel/component/quartz/StatefulCamelJob.java
 Tue Jan 27 16:04:02 2009
@@ -0,0 +1,51 @@
+/**
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.camel.component.quartz;
+
+import org.apache.camel.CamelContext;
+import org.quartz.JobExecutionContext;
+import org.quartz.JobExecutionException;
+import org.quartz.SchedulerContext;
+import org.quartz.SchedulerException;
+import org.quartz.StatefulJob;
+
+/**
+ * @author martin.gilday
+ *
+ */
+public class StatefulCamelJob implements StatefulJob {
+
+    /* (non-Javadoc)
+     * @see org.quartz.Job#execute(org.quartz.JobExecutionContext)
+     */
+    public void execute(final JobExecutionContext context) throws 
JobExecutionException {
+
+        SchedulerContext schedulerContext;
+        try {
+            schedulerContext = context.getScheduler().getContext();
+        } catch (SchedulerException e) {
+            throw new JobExecutionException("Failed to obtain scheduler 
context for job " + context.getJobDetail().getName());
+        }
+
+        CamelContext camelContext = (CamelContext) 
schedulerContext.get(QuartzEndpoint.CONTEXT_KEY);
+        String endpointUri = (String) 
context.getJobDetail().getJobDataMap().get(QuartzEndpoint.ENDPOINT_KEY);
+        QuartzEndpoint quartzEndpoint =    (QuartzEndpoint) 
camelContext.getEndpoint(endpointUri);
+        quartzEndpoint.onJobExecute(context);
+    }
+
+}

Added: 
camel/branches/camel-1.x/components/camel-quartz/src/test/java/org/apache/camel/component/quartz/StatefulQuartzRouteTest.java
URL: 
http://svn.apache.org/viewvc/camel/branches/camel-1.x/components/camel-quartz/src/test/java/org/apache/camel/component/quartz/StatefulQuartzRouteTest.java?rev=738122&view=auto
==============================================================================
--- 
camel/branches/camel-1.x/components/camel-quartz/src/test/java/org/apache/camel/component/quartz/StatefulQuartzRouteTest.java
 (added)
+++ 
camel/branches/camel-1.x/components/camel-quartz/src/test/java/org/apache/camel/component/quartz/StatefulQuartzRouteTest.java
 Tue Jan 27 16:04:02 2009
@@ -0,0 +1,67 @@
+/**
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.camel.component.quartz;
+
+import java.util.List;
+
+import org.apache.camel.ContextTestSupport;
+import org.apache.camel.Exchange;
+import org.apache.camel.Message;
+import org.apache.camel.builder.RouteBuilder;
+import org.apache.camel.component.mock.MockEndpoint;
+
+/**
+ * @author martin.gilday
+ *
+ */
+public class StatefulQuartzRouteTest extends ContextTestSupport {
+    protected MockEndpoint resultEndpoint;
+
+    public void testSendAndReceiveMails() throws Exception {
+        resultEndpoint = getMockEndpoint("mock:result");
+        resultEndpoint.expectedMessageCount(2);
+        
resultEndpoint.message(0).header("triggerName").isEqualTo("myTimerName");
+        resultEndpoint.message(0).header("triggerGroup").isEqualTo("myGroup");
+
+        // lets test the receive worked
+        resultEndpoint.assertIsSatisfied();
+
+        List<Exchange> list = resultEndpoint.getReceivedExchanges();
+        for (Exchange exchange : list) {
+            Message in = exchange.getIn();
+            log.debug("Received: " + in + " with headers: " + in.getHeaders());
+        }
+    }
+
+
+    /* (non-Javadoc)
+     * @see org.apache.camel.ContextTestSupport#createRouteBuilder()
+     */
+    @Override
+    protected RouteBuilder createRouteBuilder() throws Exception {
+        return new RouteBuilder() {
+            @Override
+            public void configure() {
+                // START SNIPPET: example
+                
from("quartz://myGroup/myTimerName?trigger.repeatInterval=2&trigger.repeatCount=1&stateful=true").to("mock:result");
+                // END SNIPPET: example
+            }
+        };
+    }
+
+}


Reply via email to