package com.atlassian.sal.core.executor;
   
   import com.atlassian.sal.api.executor.ThreadLocalContextManager;
   
   import java.util.concurrent.Callable;
   import java.util.concurrent.ScheduledExecutorService;
   import java.util.concurrent.ScheduledFuture;
   import java.util.concurrent.TimeUnit;
   
  import static com.google.common.base.Preconditions.checkNotNull;
  
  /**
   * Scheduled executor service that wraps executing callables and runnables in a wrapper that transfers the thread local
   * state of the caller to the thread of the executing task.
   *
   * @since 2.0
   */
  public class ThreadLocalDelegateScheduledExecutorService<C> extends ThreadLocalDelegateExecutorService<C> implements ScheduledExecutorService {
      private final ScheduledExecutorService delegate;
  
      public ThreadLocalDelegateScheduledExecutorService(ThreadLocalContextManager<C> manager, ScheduledExecutorService delegate) {
          super(checkNotNull(manager), checkNotNull(delegate));
          this.delegate = delegate;
      }
  
      public ScheduledFuture<?> schedule(Runnable command, long delay, TimeUnit unit) {
          return delegate.schedule(new ThreadLocalDelegateRunnable<C>(manager, command), delay, unit);
      }
  
      public <V> ScheduledFuture<V> schedule(Callable<V> callable, long delay, TimeUnit unit) {
          return delegate.schedule(new ThreadLocalDelegateCallable<C, V>(manager, callable), delay, unit);
      }
  
      public ScheduledFuture<?> scheduleAtFixedRate(Runnable command, long initialDelay, long period, TimeUnit unit) {
          return delegate.scheduleAtFixedRate(new ThreadLocalDelegateRunnable<C>(manager, command), initialDelay, period, unit);
      }
  
      public ScheduledFuture<?> scheduleWithFixedDelay(Runnable command, long initialDelay, long delay, TimeUnit unit) {
          return delegate.scheduleWithFixedDelay(new ThreadLocalDelegateRunnable<C>(manager, command), initialDelay, delay, unit);
      }
  }