de.codecentric.spring.boot.chaos.monkey.component.ChaosMonkeyScheduler Maven / Gradle / Ivy

Go to download
package de.codecentric.spring.boot.chaos.monkey.component;

import de.codecentric.spring.boot.chaos.monkey.configuration.AssaultProperties;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.lang.Nullable;
import org.springframework.scheduling.config.CronTask;
import org.springframework.scheduling.config.ScheduledTask;
import org.springframework.scheduling.config.ScheduledTaskRegistrar;

public class ChaosMonkeyScheduler {

  private static final Logger Logger = LoggerFactory.getLogger(ChaosMonkeyScheduler.class);

  @Nullable private final ScheduledTaskRegistrar scheduler;

  private final AssaultProperties config;

  private final ChaosMonkeyRuntimeScope runtimeScope;

  @Nullable private ScheduledTask currentTask;

  public ChaosMonkeyScheduler(
      ScheduledTaskRegistrar scheduler,
      AssaultProperties config,
      ChaosMonkeyRuntimeScope runtimeScope) {
    this.scheduler = scheduler;
    this.config = config;
    this.runtimeScope = runtimeScope;

    reloadConfig();
  }

  public void reloadConfig() {
    String cronExpression = config.getRuntimeAssaultCronExpression();
    boolean active = !"OFF".equals(cronExpression);

    if (currentTask != null) {
      Logger.info("Cancelling previous task");
      currentTask.cancel();
      currentTask = null;
    }

    if (active) {
      CronTask task = new CronTask(runtimeScope::callChaosMonkey, cronExpression);
      currentTask = scheduler.scheduleCronTask(task);
    }
  }
}