001/**
002 * Licensed to the Apache Software Foundation (ASF) under one or more
003 * contributor license agreements.  See the NOTICE file distributed with
004 * this work for additional information regarding copyright ownership.
005 * The ASF licenses this file to You under the Apache License, Version 2.0
006 * (the "License"); you may not use this file except in compliance with
007 * the License.  You may obtain a copy of the License at
008 *
009 *      http://www.apache.org/licenses/LICENSE-2.0
010 *
011 * Unless required by applicable law or agreed to in writing, software
012 * distributed under the License is distributed on an "AS IS" BASIS,
013 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
014 * See the License for the specific language governing permissions and
015 * limitations under the License.
016 */
017package org.apache.camel.spring.pollingconsumer;
018
019import java.util.TimeZone;
020import java.util.concurrent.ScheduledFuture;
021
022import org.apache.camel.CamelContext;
023import org.apache.camel.Consumer;
024import org.apache.camel.spi.ScheduledPollConsumerScheduler;
025import org.apache.camel.spring.util.CamelThreadPoolTaskScheduler;
026import org.apache.camel.support.ServiceSupport;
027import org.apache.camel.util.ObjectHelper;
028import org.slf4j.Logger;
029import org.slf4j.LoggerFactory;
030import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;
031import org.springframework.scheduling.support.CronTrigger;
032
033/**
034 * A Spring based {@link ScheduledPollConsumerScheduler} which uses a {@link CronTrigger} to define when the
035 * poll should be triggered.
036 */
037public class SpringScheduledPollConsumerScheduler extends ServiceSupport implements ScheduledPollConsumerScheduler {
038
039    private static final Logger LOG = LoggerFactory.getLogger(SpringScheduledPollConsumerScheduler.class);
040    private CamelContext camelContext;
041    private Consumer consumer;
042    private Runnable runnable;
043    private String cron;
044    private TimeZone timeZone = TimeZone.getDefault();
045    private volatile CronTrigger trigger;
046    private volatile ThreadPoolTaskScheduler taskScheduler;
047    private boolean destroyTaskScheduler;
048    private volatile ScheduledFuture<?> future;
049
050    @Override
051    public void onInit(Consumer consumer) {
052        this.consumer = consumer;
053    }
054
055    @Override
056    public void scheduleTask(Runnable runnable) {
057        this.runnable = runnable;
058    }
059
060    @Override
061    public void unscheduleTask() {
062        if (future != null) {
063            future.cancel(false);
064            future = null;
065        }
066    }
067
068    @Override
069    public void startScheduler() {
070        // we start the scheduler in doStart
071    }
072
073    @Override
074    public boolean isSchedulerStarted() {
075        return taskScheduler != null && !taskScheduler.getScheduledExecutor().isShutdown();
076    }
077
078    @Override
079    public void setCamelContext(CamelContext camelContext) {
080        this.camelContext = camelContext;
081    }
082
083    @Override
084    public CamelContext getCamelContext() {
085        return camelContext;
086    }
087
088    public String getCron() {
089        return cron;
090    }
091
092    public void setCron(String cron) {
093        this.cron = cron;
094    }
095
096    public TimeZone getTimeZone() {
097        return timeZone;
098    }
099
100    public void setTimeZone(TimeZone timeZone) {
101        this.timeZone = timeZone;
102    }
103
104    public ThreadPoolTaskScheduler getTaskScheduler() {
105        return taskScheduler;
106    }
107
108    public void setTaskScheduler(ThreadPoolTaskScheduler taskScheduler) {
109        this.taskScheduler = taskScheduler;
110    }
111
112    @Override
113    protected void doStart() throws Exception {
114        ObjectHelper.notEmpty(cron, "cron", this);
115
116        trigger = new CronTrigger(getCron(), getTimeZone());
117
118        if (taskScheduler == null) {
119            taskScheduler = new CamelThreadPoolTaskScheduler(getCamelContext(), consumer, consumer.getEndpoint().getEndpointUri());
120            taskScheduler.afterPropertiesSet();
121            destroyTaskScheduler = true;
122        }
123
124        LOG.debug("Scheduling cron trigger {}", getCron());
125        future = taskScheduler.schedule(runnable, trigger);
126    }
127
128    @Override
129    protected void doStop() throws Exception {
130        if (future != null) {
131            future.cancel(false);
132            future = null;
133        }
134
135        if (destroyTaskScheduler) {
136            taskScheduler.destroy();
137            taskScheduler = null;
138        }
139    }
140}