diff options
Diffstat (limited to 'src/DevHive.Data')
| -rw-r--r-- | src/DevHive.Data/Repositories/FriendsRepository.cs | 38 | ||||
| -rw-r--r-- | src/DevHive.Data/Repositories/UserRepository.cs | 8 |
2 files changed, 38 insertions, 8 deletions
diff --git a/src/DevHive.Data/Repositories/FriendsRepository.cs b/src/DevHive.Data/Repositories/FriendsRepository.cs new file mode 100644 index 0000000..c65ba5e --- /dev/null +++ b/src/DevHive.Data/Repositories/FriendsRepository.cs @@ -0,0 +1,38 @@ +using System; +using System.Linq; +using System.Threading.Tasks; +using DevHive.Common.Models.Data; +using DevHive.Data.Models; +using Microsoft.EntityFrameworkCore; + +namespace DevHive.Data.Repositories +{ + public class FriendsRepository : UserRepository + { + private readonly DbContext _context; + + public FriendsRepository(DbContext context) + : base(context) + { + this._context = context; + } + + //Create + public async Task<bool> AddFriendAsync(User user, User friend) + { + this._context.Update(user); + user.Friends.Add(friend); + + return await RepositoryMethods.SaveChangesAsync(this._context); + } + + //Delete + public async Task<bool> RemoveFriendAsync(User user, User friend) + { + this._context.Update(user); + user.Friends.Remove(friend); + + return await RepositoryMethods.SaveChangesAsync(this._context); + } + } +}
\ No newline at end of file diff --git a/src/DevHive.Data/Repositories/UserRepository.cs b/src/DevHive.Data/Repositories/UserRepository.cs index 577f02f..6d14a65 100644 --- a/src/DevHive.Data/Repositories/UserRepository.cs +++ b/src/DevHive.Data/Repositories/UserRepository.cs @@ -116,13 +116,5 @@ namespace DevHive.Data.Repositories .AsNoTracking() .AnyAsync(u => u.Email == email); } - - public async Task<bool> AddFriend(User user, User friend) - { - this._context.Update(user); - user.Friends.Add(friend); - - return await RepositoryMethods.SaveChangesAsync(this._context); - } } } |
