Simplify registering traces sample callback

CHANGELOG.md

@@ -4,6 +4,7 @@
 * Enhancement: Pass request to CustomSamplingContext in Spring integration (#1172)
 * Fix: Free Local Refs manually due to Android local ref. count limits
 * Enhancement: Add overload for `transaction/span.finish(SpanStatus)` (#1182)
+* Enhancement: Simplify registering traces sample callback in Spring integration (#1184)
 
 # 4.0.0-alpha.3
 

sentry-spring/src/main/java/io/sentry/spring/SentryInitBeanPostProcessor.java

@@ -3,6 +3,7 @@
 import com.jakewharton.nopen.annotation.Open;
 import io.sentry.Sentry;
 import io.sentry.SentryOptions;
+import io.sentry.SentryOptions.TracesSamplerCallback;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
 import org.springframework.beans.BeansException;
@@ -29,6 +30,9 @@ public Object postProcessAfterInitialization(
                 sentryUserProvider ->
                     options.addEventProcessor(
                         new SentryUserProviderEventProcessor(options, sentryUserProvider)));
+        applicationContext
+            .getBeanProvider(TracesSamplerCallback.class)
+            .ifAvailable(options::setTracesSampler);
       }
       Sentry.init(options);
     }

sentry-spring/src/test/kotlin/io/sentry/spring/EnableSentryTest.kt

@@ -6,6 +6,7 @@ import kotlin.test.Test
 import org.assertj.core.api.Assertions.assertThat
 import org.springframework.boot.context.annotation.UserConfigurations
 import org.springframework.boot.test.context.runner.ApplicationContextRunner
+import org.springframework.context.annotation.Bean
 
 class EnableSentryTest {
     private val contextRunner = ApplicationContextRunner()
@@ -88,6 +89,16 @@ class EnableSentryTest {
             }
     }
 
+    @Test
+    fun `configures custom TracesSamplerCallback`() {
+        ApplicationContextRunner().withConfiguration(UserConfigurations.of(AppConfigWithCustomTracesSamplerCallback::class.java))
+            .run {
+                val options = it.getBean(SentryOptions::class.java)
+                val samplerCallback = it.getBean(SentryOptions.TracesSamplerCallback::class.java)
+                assertThat(options.tracesSampler).isEqualTo(samplerCallback)
+            }
+    }
+
     @EnableSentry(dsn = "http://key@localhost/proj")
     class AppConfig
 
@@ -99,4 +110,15 @@ class EnableSentryTest {
 
     @EnableSentry(dsn = "http://key@localhost/proj", exceptionResolverOrder = Integer.MAX_VALUE)
     class AppConfigWithExceptionResolverOrderIntegerMaxValue
+
+    @EnableSentry(dsn = "http://key@localhost/proj")
+    class AppConfigWithCustomTracesSamplerCallback {
+
+        @Bean
+        fun tracesSampler(): SentryOptions.TracesSamplerCallback {
+            return SentryOptions.TracesSamplerCallback {
+                return@TracesSamplerCallback 1.0
+            }
+        }
+    }
 }