Application/VoteForComment/VoteForCommentCommandHandler.cs

@@ -31,8 +31,6 @@ namespace HN.Application
           break;
       }
 
-      await _commentRepository.UpdateAsync(comment);
-
       return Unit.Value;
     }
   }

Application/VoteForLink/VoteForLinkCommandHandler.cs

@@ -31,8 +31,6 @@ namespace HN.Application
           break;
       }
 
-      await _linkRepository.UpdateAsync(link);
-
       return Unit.Value;
     }
   }

Apps/Website/Controllers/AccountsController.cs

@@ -1,6 +1,6 @@
 using System.Linq;
 using System.Threading.Tasks;
-using HN.Infrastructure;
+using HN.Infrastructure.Identity;
 using Microsoft.AspNetCore.Authorization;
 using Microsoft.AspNetCore.Identity;
 using Microsoft.AspNetCore.Mvc;

Apps/Website/HttpExecutingUserProvider.cs

@@ -1,6 +1,6 @@
 using System;
 using HN.Application;
-using HN.Infrastructure;
+using HN.Infrastructure.Identity;
 using Microsoft.AspNetCore.Http;
 using Microsoft.AspNetCore.Identity;
 

Apps/Website/Startup.cs

@@ -1,6 +1,9 @@
 using HN.Application;
 using HN.Domain;
 using HN.Infrastructure;
+using HN.Infrastructure.Behaviors;
+using HN.Infrastructure.Identity;
+using HN.Infrastructure.Repositories;
 using MediatR;
 using Microsoft.AspNetCore.Authentication.Cookies;
 using Microsoft.AspNetCore.Builder;
@@ -32,6 +35,7 @@ namespace Website
       services.AddScoped<ILinkRepository, LinkRepository>();
       services.AddScoped<ICommentRepository, CommentRepository>();
       services.AddScoped<IExecutingUserProvider, HttpExecutingUserProvider>();
+      services.AddScoped(typeof(IPipelineBehavior<,>), typeof(UnitOfWorkBehavior<,>));
       services.AddMediatR(typeof(HN.Application.IHNContext));
 
       // Permet d'avoir des routes en lowercase

Domain/ICommentRepository.cs

@@ -6,7 +6,6 @@ namespace HN.Domain
   public interface ICommentRepository
   {
     Task AddAsync(Comment comment);
-    Task UpdateAsync(Comment comment);
     Task<Comment> GetByIdAsync(Guid id);
   }
 }

Domain/ILinkRepository.cs

@@ -6,7 +6,6 @@ namespace HN.Domain
   public interface ILinkRepository
   {
     Task AddAsync(Link link);
-    Task UpdateAsync(Link link);
     Task<Link> GetByIdAsync(Guid id);
   }
 }

Infrastructure/Behaviors/UnitOfWorkBehavior.cs

@@ -0,0 +1,25 @@
+using System.Threading;
+using System.Threading.Tasks;
+using MediatR;
+
+namespace HN.Infrastructure.Behaviors
+{
+  public sealed class UnitOfWorkBehavior<TRequest, TResponse> : IPipelineBehavior<TRequest, TResponse>
+  {
+    private readonly HNDbContext _context;
+
+    public UnitOfWorkBehavior(HNDbContext context)
+    {
+      _context = context;
+    }
+
+    public async Task<TResponse> Handle(TRequest request, CancellationToken cancellationToken, RequestHandlerDelegate<TResponse> next)
+    {
+      var response = await next();
+
+      await _context.SaveChangesAsync();
+
+      return response;
+    }
+  }
+}

Infrastructure/EntityTypes/CommentEntityType.cs

@@ -1,8 +1,9 @@
 using HN.Domain;
+using HN.Infrastructure.Identity;
 using Microsoft.EntityFrameworkCore;
 using Microsoft.EntityFrameworkCore.Metadata.Builders;
 
-namespace HN.Infrastructure
+namespace HN.Infrastructure.EntityTypes
 {
   public sealed class CommentEntityType : IEntityTypeConfiguration<Comment>
   {

Infrastructure/EntityTypes/LinkEntityType.cs

@@ -1,4 +1,5 @@
 using HN.Domain;
+using HN.Infrastructure.Identity;
 using Microsoft.EntityFrameworkCore;
 using Microsoft.EntityFrameworkCore.Metadata.Builders;
 

Infrastructure/HNDbContext.cs

@@ -2,6 +2,7 @@
 using System.Linq;
 using HN.Application;
 using HN.Domain;
+using HN.Infrastructure.Identity;
 using Microsoft.AspNetCore.Identity.EntityFrameworkCore;
 using Microsoft.EntityFrameworkCore;
 using Microsoft.Extensions.Logging;

Infrastructure/Identity/Role.cs

@@ -1,7 +1,7 @@
 using System;
 using Microsoft.AspNetCore.Identity;
 
-namespace HN.Infrastructure
+namespace HN.Infrastructure.Identity
 {
   public sealed class Role : IdentityRole<Guid>
   {

Infrastructure/Identity/User.cs

@@ -2,7 +2,7 @@ using System;
 using HN.Application;
 using Microsoft.AspNetCore.Identity;
 
-namespace HN.Infrastructure
+namespace HN.Infrastructure.Identity
 {
   public sealed class User : IdentityUser<Guid>, IUser
   {

Infrastructure/Repositories/CommentRepository.cs

@@ -3,7 +3,7 @@ using System.Threading.Tasks;
 using HN.Domain;
 using Microsoft.EntityFrameworkCore;
 
-namespace HN.Infrastructure
+namespace HN.Infrastructure.Repositories
 {
   public sealed class CommentRepository : Repository<Comment>, ICommentRepository
   {
@@ -20,10 +20,5 @@ namespace HN.Infrastructure
     {
       return Entries.SingleOrDefaultAsync(o => o.Id == id);
     }
-
-    public Task UpdateAsync(Comment comment)
-    {
-      return base.UpdateAsync(comment);
-    }
   }
 }

Infrastructure/Repositories/LinkRepository.cs

@@ -3,7 +3,7 @@ using System.Threading.Tasks;
 using HN.Domain;
 using Microsoft.EntityFrameworkCore;
 
-namespace HN.Infrastructure
+namespace HN.Infrastructure.Repositories
 {
   public sealed class LinkRepository : Repository<Link>, ILinkRepository
   {
@@ -20,10 +20,5 @@ namespace HN.Infrastructure
     {
       return Entries.SingleOrDefaultAsync(o => o.Id == id);
     }
-
-    public Task UpdateAsync(Link link)
-    {
-      return base.UpdateAsync(link);
-    }
   }
 }

Infrastructure/Repositories/Repository.cs

@@ -1,7 +1,7 @@
 using System.Threading.Tasks;
 using Microsoft.EntityFrameworkCore;
 
-namespace HN.Infrastructure
+namespace HN.Infrastructure.Repositories
 {
   public abstract class Repository<TEntity> where TEntity : class
   {
@@ -18,12 +18,6 @@ namespace HN.Infrastructure
     protected async Task AddAsync(params TEntity[] entities)
     {
       await Entries.AddRangeAsync(entities);
-      await _context.SaveChangesAsync();
-    }
-
-    protected async Task UpdateAsync(params TEntity[] entities)
-    {
-      await _context.SaveChangesAsync();
     }
   }
 }