io.sentry.spring.jakarta.webflux.SentryWebExceptionHandler Maven / Gradle / Ivy
package io.sentry.spring.jakarta.webflux;
import static io.sentry.TypeCheckHint.WEBFLUX_EXCEPTION_HANDLER_EXCHANGE;
import static io.sentry.TypeCheckHint.WEBFLUX_EXCEPTION_HANDLER_REQUEST;
import static io.sentry.TypeCheckHint.WEBFLUX_EXCEPTION_HANDLER_RESPONSE;
import io.sentry.Hint;
import io.sentry.IScopes;
import io.sentry.SentryEvent;
import io.sentry.SentryLevel;
import io.sentry.exception.ExceptionMechanismException;
import io.sentry.protocol.Mechanism;
import io.sentry.util.Objects;
import org.jetbrains.annotations.ApiStatus;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import org.springframework.core.annotation.Order;
import org.springframework.web.server.ResponseStatusException;
import org.springframework.web.server.ServerWebExchange;
import org.springframework.web.server.WebExceptionHandler;
import reactor.core.publisher.Mono;
/** Handles unhandled exceptions in Spring WebFlux integration. */
@Order(
-2) // the DefaultErrorWebExceptionHandler provided by Spring Boot for error handling is ordered
// at -1
@ApiStatus.Experimental
public final class SentryWebExceptionHandler implements WebExceptionHandler {
public static final String MECHANISM_TYPE = "Spring6WebFluxExceptionResolver";
private final @NotNull IScopes scopes;
public SentryWebExceptionHandler(final @NotNull IScopes scopes) {
this.scopes = Objects.requireNonNull(scopes, "scopes are required");
}
@Override
public @NotNull Mono handle(
final @NotNull ServerWebExchange serverWebExchange, final @NotNull Throwable ex) {
final @Nullable IScopes requestScopes =
serverWebExchange.getAttributeOrDefault(SentryWebFilter.SENTRY_SCOPES_KEY, null);
final @NotNull IScopes scopesToUse = requestScopes != null ? requestScopes : scopes;
return ReactorUtils.withSentryScopes(
Mono.just(ex)
.map(
it -> {
if (!(ex instanceof ResponseStatusException)) {
final Mechanism mechanism = new Mechanism();
mechanism.setType(MECHANISM_TYPE);
mechanism.setHandled(false);
final Throwable throwable =
new ExceptionMechanismException(mechanism, ex, Thread.currentThread());
final SentryEvent event = new SentryEvent(throwable);
event.setLevel(SentryLevel.FATAL);
event.setTransaction(
TransactionNameProvider.provideTransactionName(serverWebExchange));
final Hint hint = new Hint();
hint.set(WEBFLUX_EXCEPTION_HANDLER_REQUEST, serverWebExchange.getRequest());
hint.set(
WEBFLUX_EXCEPTION_HANDLER_RESPONSE, serverWebExchange.getResponse());
hint.set(WEBFLUX_EXCEPTION_HANDLER_EXCHANGE, serverWebExchange);
scopes.captureEvent(event, hint);
}
return it;
}),
scopesToUse)
.flatMap(it -> Mono.error(ex));
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy