oqtane.framework/Oqtane.Server/Repository/UserRoleRepository.cs
2019-08-25 14:52:25 -04:00

113 lines
2.6 KiB
C#

using Microsoft.EntityFrameworkCore;
using System.Collections.Generic;
using System.Linq;
using Oqtane.Models;
namespace Oqtane.Repository
{
public class UserRoleRepository : IUserRoleRepository
{
private TenantDBContext db;
public UserRoleRepository(TenantDBContext context)
{
db = context;
}
public IEnumerable<UserRole> GetUserRoles()
{
try
{
return db.UserRole.ToList();
}
catch
{
throw;
}
}
public IEnumerable<UserRole> GetUserRoles(int UserId)
{
try
{
return db.UserRole.Where(item => item.UserId == UserId)
.Include(item => item.Role) // eager load roles
.ToList();
}
catch
{
throw;
}
}
public IEnumerable<UserRole> GetUserRoles(int UserId, int SiteId)
{
try
{
return db.UserRole.Where(item => item.UserId == UserId)
.Include(item => item.Role) // eager load roles
.Where(item => item.Role.SiteId == SiteId)
.ToList();
}
catch
{
throw;
}
}
public UserRole AddUserRole(UserRole UserRole)
{
try
{
db.UserRole.Add(UserRole);
db.SaveChanges();
return UserRole;
}
catch
{
throw;
}
}
public UserRole UpdateUserRole(UserRole UserRole)
{
try
{
db.Entry(UserRole).State = EntityState.Modified;
db.SaveChanges();
return UserRole;
}
catch
{
throw;
}
}
public UserRole GetUserRole(int UserRoleId)
{
try
{
return db.UserRole.Include(item => item.Role) // eager load roles
.SingleOrDefault(item => item.UserRoleId == UserRoleId);
}
catch
{
throw;
}
}
public void DeleteUserRole(int UserRoleId)
{
try
{
UserRole UserRole = db.UserRole.Find(UserRoleId);
db.UserRole.Remove(UserRole);
db.SaveChanges();
}
catch
{
throw;
}
}
}
}