InitializingCronTriggerFactoryBean.java

/***************************************************************************
 * Copyright (c) 2012-2015 VMware, Inc. All Rights Reserved.
 * Licensed 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 com.vmware.bdd.manager.collection.quartz;

import org.apache.log4j.Logger;
import org.quartz.CronExpression;
import org.springframework.scheduling.quartz.CronTriggerFactoryBean;

import com.vmware.bdd.manager.collection.CollectionDriver;
import com.vmware.bdd.manager.collection.CollectionDriverManager;
import com.vmware.bdd.utils.CommonUtil;
import com.vmware.bdd.utils.PropertiesUtil;

public class InitializingCronTriggerFactoryBean extends CronTriggerFactoryBean {

   static final Logger logger = Logger
         .getLogger(InitializingCronTriggerFactoryBean.class);
   private CollectionDriverManager collectionDriverManager;

   private void initCronExpression() {
      String cronExpression = getCronExpressionFromConfiguration();
      if (!CommonUtil.isBlank(cronExpression)
            && CronExpression.isValidExpression(cronExpression)) {
         setCronExpression(cronExpression);
      } else {
         CollectionDriver driver = collectionDriverManager.getDriver();
         if (driver != null) {
            setCronExpression(driver.getDefaultCronExpression());
         } else {
            logger.warn("com.vmware.bdd.manager.collection.CollectionDriver is not configured.");
         }
      }
   }

   private String getCronExpressionFromConfiguration() {
      String cronExpression = "";
      CollectionDriver driver = collectionDriverManager.getDriver();
      if (driver != null) {
         cronExpression =
               new PropertiesUtil(
                     CollectionDriverManager.getConfigurationFile())
                     .getProperty(driver.getCronExpressionName());
      }
      return cronExpression;
   }

   public void setCollectionDriverManager(CollectionDriverManager collectionDriverManager) {
      this.collectionDriverManager = collectionDriverManager;
      initCronExpression();
   }

}