com.github.kagkarlsson.examples.boot.config.SchedulerConfiguration Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spring-boot-example Show documentation
Examples on how to use db-scheduler with Spring Boot
The newest version!
/*
 * Copyright (C) Gustav Karlsson
 *
 * 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.github.kagkarlsson.examples.boot.config;

import com.github.kagkarlsson.scheduler.SchedulerName;
import com.github.kagkarlsson.scheduler.boot.config.DbSchedulerCustomizer;
import com.github.kagkarlsson.scheduler.event.AbstractSchedulerListener;
import com.github.kagkarlsson.scheduler.event.ExecutionInterceptor;
import com.github.kagkarlsson.scheduler.event.SchedulerListener;
import com.github.kagkarlsson.scheduler.serializer.JacksonSerializer;
import com.github.kagkarlsson.scheduler.serializer.Serializer;
import com.github.kagkarlsson.scheduler.task.ExecutionComplete;
import java.util.Optional;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.slf4j.MDC;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
public class SchedulerConfiguration {

  public static final String MDC_TASKNAME = "task-name";
  public static final String MDC_TASKINSTANCEID = "task-instance-id";
  private static final Logger LOG = LoggerFactory.getLogger(SchedulerConfiguration.class);

  /** Bean defined when a configuration-property in DbSchedulerCustomizer needs to be overridden. */
  @Bean
  DbSchedulerCustomizer customizer() {
    return new DbSchedulerCustomizer() {
      @Override
      public Optional schedulerName() {
        return Optional.of(new SchedulerName.Fixed("spring-boot-scheduler-1"));
      }

      @Override
      public Optional serializer() {
        return Optional.of(new JacksonSerializer());
      }
    };
  }

  @Bean
  SchedulerListener schedulerListener() {
    return new AbstractSchedulerListener() {

      @Override
      public void onExecutionComplete(ExecutionComplete executionComplete) {
        LOG.info(
            "SchedulerListener.onExecutionComplete. Result={}, took={}ms ",
            executionComplete.getResult(),
            executionComplete.getDuration().toMillis());
      }
    };
  }

  @Bean
  ExecutionInterceptor mdcExecutionInterceptor() {
    return (taskInstance, executionContext, chain) -> {
      LOG.info("Setting MDC before execution");
      MDC.put(MDC_TASKNAME, taskInstance.getTaskName());
      MDC.put(MDC_TASKINSTANCEID, taskInstance.getId());
      try {
        return chain.proceed(taskInstance, executionContext);
      } finally {
        MDC.remove(MDC_TASKNAME);
        MDC.remove(MDC_TASKINSTANCEID);
      }
    };
  }
}