io.sentry.spring.SentryTaskDecorator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of sentry-spring Show documentation

SDK for sentry.io

There is a newer version: 8.0.0-rc.3

package io.sentry.spring;

import io.sentry.IHub;
import io.sentry.Sentry;
import java.util.concurrent.Callable;
import org.jetbrains.annotations.NotNull;
import org.springframework.core.task.TaskDecorator;
import org.springframework.scheduling.annotation.Async;

/**
 * Sets a current hub on a thread running a {@link Runnable} given by parameter. Used to propagate
 * the current {@link IHub} on the thread executing async task - like MVC controller methods
 * returning a {@link Callable} or Spring beans methods annotated with {@link Async}.
 */
public final class SentryTaskDecorator implements TaskDecorator {
  @Override
  public @NotNull Runnable decorate(final @NotNull Runnable runnable) {
    final IHub newHub = Sentry.getCurrentHub().clone();

    return () -> {
      final IHub oldState = Sentry.getCurrentHub();
      Sentry.setCurrentHub(newHub);
      try {
        runnable.run();
      } finally {
        Sentry.setCurrentHub(oldState);
      }
    };
  }
}