huahaofeng
/
aspnetcore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Updated antiforgery ServiceCollectionExtensions

This commit is contained in:
jacalvar 2016-02-22 11:49:15 -08:00
parent 4629148519
commit aa8fd48c64
1 changed files with 18 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
src/Microsoft.AspNetCore.Antiforgery/ServiceCollectionExtensions.cs
View File

@ -10,9 +10,16 @@ using Microsoft.Extensions.Options;
namespace Microsoft.Extensions.DependencyInjection namespace Microsoft.Extensions.DependencyInjection
{ {
/// <summary>
/// Extension methods for setting up antiforgery services in an <see cref="IServiceCollection" />.
/// </summary>
public static class ServiceCollectionExtensions public static class ServiceCollectionExtensions
{ {
public static IServiceCollection AddAntiforgery(this IServiceCollection services) /// <summary>
/// Adds antiforgery services to the specified <see cref="IServiceCollection" />.
/// </summary>
/// <param name="services">The <see cref="IServiceCollection" /> to add services to.</param>
public static void AddAntiforgery(this IServiceCollection services)
{ {
if (services == null) if (services == null)
{ {
@ -40,27 +47,27 @@ namespace Microsoft.Extensions.DependencyInjection
var policy = new AntiforgerySerializationContextPooledObjectPolicy(); var policy = new AntiforgerySerializationContextPooledObjectPolicy();
return provider.Create(policy); return provider.Create(policy);
}); });
return services;
} }
public static IServiceCollection AddAntiforgery( /// <summary>
this IServiceCollection services, /// Adds antiforgery services to the specified <see cref="IServiceCollection" />.
Action<AntiforgeryOptions> setupAction) /// </summary>
/// <param name="services">The <see cref="IServiceCollection" /> to add services to.</param>
/// <param name="setupAction">An <see cref="Action{AntiforgeryOptions}"/> to configure the provided <see cref="AntiforgeryOptions"/>.</param>
public static void AddAntiforgery(this IServiceCollection services, Action<AntiforgeryOptions> setupAction)
{ {
if (services == null) if (services == null)
{ {
throw new ArgumentNullException(nameof(services)); throw new ArgumentNullException(nameof(services));
} }
services.AddAntiforgery(); if (setupAction == null)
if (setupAction != null)
{ {
services.Configure(setupAction); throw new ArgumentNullException(nameof(setupAction));
} }
return services; services.AddAntiforgery();
services.Configure(setupAction);
} }
} }
} }