コード例 #1
0
        /// <summary>
        /// Runs the middleware described by <see cref="TaskMiddlewareDescriptor"/>.
        /// </summary>
        /// <param name="descriptor">The task middleware to find and run.</param>
        /// <param name="context">The dispatch middleware context.</param>
        /// <param name="next">The next middleware callback.</param>
        /// <returns>A task that completes when the middleware has finished executing.</returns>
        public static Task RunAsync(
            TaskMiddlewareDescriptor descriptor, DispatchMiddlewareContext context, Func <Task> next)
        {
            Check.NotNull(descriptor, nameof(descriptor));
            Check.NotNull(context, nameof(context));
            Check.NotNull(next, nameof(next));

            IServiceProvider serviceProvider = context.GetProperty <IServiceProvider>();
            ITaskMiddleware  middleware      = GetMiddleware(descriptor, serviceProvider);

            if (middleware is null)
            {
                // This _should not_ be possible, as TaskMiddlewareDescriptor is designed to only be constructable with
                // valid values. But a good sanity check here.
                ILogger logger  = serviceProvider.CreateLogger(typeof(TaskMiddlewareRunner));
                string  message = Strings.MiddlewareCreationFailed(descriptor);
                logger.LogError(message);
                throw new InvalidOperationException(message); // TODO: maybe a better exception type.
            }

            return(middleware.InvokeAsync(context, next));
        }
コード例 #2
0
 /// <summary>
 /// Add the middleware to the processing pipeline
 /// </summary>
 /// <param name="session">The current session</param>
 /// <param name="middleware">The middleware to add</param>
 /// <returns></returns>
 public static ProfilerSession AddMiddleware(this ProfilerSession session, ITaskMiddleware middleware)
 {
     session.ProcessingPipeline.SetNext(middleware);
     return(session);
 }