Add extension method UseStackTraceFactory

Author: 彭伟

src/Sentry.AspNetCore/ApplicationBuilderExtensions.cs

@@ -38,6 +38,12 @@ public static IApplicationBuilder UseSentry(this IApplicationBuilder app)
                     o.DiagnosticLogger = new MelDiagnosticLogger(logger, o.DiagnosticsLevel);
                 }
 
+                var stackTraceFactory = app.ApplicationServices.GetService<ISentryStackTraceFactory>();
+                if (stackTraceFactory != null)
+                {
+                    o.UseStackTraceFactory(stackTraceFactory);
+                }
+
                 if (app.ApplicationServices.GetService<IEnumerable<ISentryEventProcessor>>().Any())
                 {
                     o.AddEventProcessorProvider(app.ApplicationServices.GetServices<ISentryEventProcessor>);

src/Sentry/SentryOptionsExtensions.cs

@@ -120,6 +120,24 @@ public static IEnumerable<ISentryEventProcessor> GetAllEventProcessors(this Sent
         public static IEnumerable<ISentryEventExceptionProcessor> GetAllExceptionProcessors(this SentryOptions options)
             => options.ExceptionProcessorsProviders.SelectMany(p => p());
 
+        /// <summary>
+        /// Use custom <see cref="ISentryStackTraceFactory" />
+        /// </summary>
+        /// <param name="options">The SentryOptions to hold the processor provider.</param>
+        /// <param name="sentryStackTraceFactory">The stack trace factory.</param>
+        public static SentryOptions UseStackTraceFactory(this SentryOptions options, ISentryStackTraceFactory sentryStackTraceFactory)
+        {
+            if (sentryStackTraceFactory == null) throw new ArgumentNullException(nameof(sentryStackTraceFactory));
+
+            options.EventProcessors = options.EventProcessors.RemoveAt(1)
+                .Insert(1, new MainSentryEventProcessor(options, sentryStackTraceFactory));
+
+            options.ExceptionProcessors = options.ExceptionProcessors.RemoveAt(0)
+                .Insert(0, new MainExceptionProcessor(options, sentryStackTraceFactory));
+
+            return options;
+        }
+
         internal static void SetupLogging(this SentryOptions options)
         {
             if (options.Debug)

test/Sentry.Tests/SentryOptionsExtensionsTests.cs

@@ -1,3 +1,4 @@
+using System;
 using System.Linq;
 using NSubstitute;
 using Sentry.Extensibility;
@@ -187,6 +188,32 @@ public void GetAllEventProcessors_NoAdding_SecondReturned_MainSentryEventProcess
             Assert.IsType<MainSentryEventProcessor>(Sut.GetAllEventProcessors().Skip(1).First());
         }
 
+        [Fact]
+        public void UseStackTraceFactory()
+        {
+            var eventProcessor1 = Sut.GetAllEventProcessors().Skip(1).First();
+            var exceptionProcessor1 = Sut.GetAllExceptionProcessors().First();
+
+            Sut.UseStackTraceFactory(Substitute.For<ISentryStackTraceFactory>());
+
+            var eventProcessor2 = Sut.GetAllEventProcessors().Skip(1).First();
+            var exceptionProcessor2 = Sut.GetAllExceptionProcessors().First();
+
+            Assert.IsType<MainSentryEventProcessor>(eventProcessor1);
+            Assert.IsType<MainExceptionProcessor>(exceptionProcessor1);
+            Assert.IsType<MainSentryEventProcessor>(eventProcessor2);
+            Assert.IsType<MainExceptionProcessor>(exceptionProcessor2);
+
+            Assert.NotEqual(eventProcessor1, eventProcessor2);
+            Assert.NotEqual(exceptionProcessor1, exceptionProcessor2);
+        }
+
+        [Fact]
+        public void UseStackTraceFactory_NotNull()
+        {
+            Assert.Throws<ArgumentNullException>(() => Sut.UseStackTraceFactory(null));
+        }
+
         [Fact]
         public void GetAllEventProcessors_AddingMore_FirstReturned_DuplicateDetectionProcessor()
         {