Changed object to Guid in IRepository

2 files changed, 20 insertions, 10 deletions

diff --git a/src/DevHive.Data/Repositories/IRepository.cs b/src/DevHive.Data/Repositories/IRepository.cs
index 81a1a35..f9ad2c5 100644
--- a/src/DevHive.Data/Repositories/IRepository.cs
+++ b/src/DevHive.Data/Repositories/IRepository.cs
@@ -1,3 +1,4 @@
+using System;
 using System.Collections.Generic;
 using System.Threading.Tasks;
 
@@ -13,12 +14,12 @@ namespace Data.Models.Interfaces.Database
 		IEnumerable<TEntity> Query(int count);
 
 		//Find entity by id
-		Task<TEntity> FindByIdAsync(object id);
+		Task<TEntity> FindByIdAsync(Guid id);
 
 		//Modify Entity from database
-		Task EditAsync(object id, TEntity newEntity);
+		Task EditAsync(Guid id, TEntity newEntity);
 
 		//Delete Entity from database
-		Task DeleteAsync(object id);
+		Task DeleteAsync(Guid id);
 	}
 }
\ No newline at end of file
diff --git a/src/DevHive.Data/Repositories/UserRepository.cs b/src/DevHive.Data/Repositories/UserRepository.cs
index b884732..2363200 100644
--- a/src/DevHive.Data/Repositories/UserRepository.cs
+++ b/src/DevHive.Data/Repositories/UserRepository.cs
@@ -1,3 +1,4 @@
+using System;
 using System.Collections.Generic;
 using System.Linq;
 using System.Threading.Tasks;
@@ -9,12 +10,11 @@ namespace DevHive.Data.Repositories
 {
 	public class UserRepository : IRepository<User>
 	{
-		private readonly UserRepository _dbRepository;
+		private readonly DbContext _context;
 
 		public UserRepository(DbContext context)
-			: base (context)
 		{
-			this._dbRepository = new DbRepository<User>(context);
+			this._context = context;
 		}
 		/*
 		public User FindByUsername(string username)
@@ -42,17 +42,26 @@ namespace DevHive.Data.Repositories
 					x.UserName == username);
 		} */
 
-		public Task AddAsync(User entity)
+		public async Task AddAsync(User entity)
 		{
-			throw new System.NotImplementedException();
+			await this._context
+				.Set<User>()
+				.AddAsync(entity);
+
+			await this._context.SaveChangesAsync();
 		}
 
 		public IEnumerable<User> Query(int count)
 		{
-			throw new System.NotImplementedException();
+			return this._context
+				.Set<User>()
+				.AsNoTracking()
+				.Take(count)
+				.AsEnumerable();
+
 		}
 
-		public Task<User> FindByIdAsync(object id)
+		public Task<User> FindByIdAsync(Guid id)
 		{
 			throw new System.NotImplementedException();
 		}