oqtane.framework/Oqtane.Server/Repository/RoleRepository.cs
2024-03-19 09:19:56 +08:00

80 lines
2.4 KiB
C#

using System.Collections.Generic;
using System.Linq;
using Microsoft.EntityFrameworkCore;
using Oqtane.Models;
namespace Oqtane.Repository
{
public class RoleRepository : IRoleRepository
{
private readonly IDbContextFactory<TenantDBContext> _dbContextFactory;
public RoleRepository(IDbContextFactory<TenantDBContext> dbContextFactory)
{
_dbContextFactory = dbContextFactory;
}
public IEnumerable<Role> GetRoles(int siteId)
{
return GetRoles(siteId, false);
}
public IEnumerable<Role> GetRoles(int siteId, bool includeGlobalRoles)
{
using var db = _dbContextFactory.CreateDbContext();
if (includeGlobalRoles)
{
return db.Role.Where(item => item.SiteId == siteId || item.SiteId == null).ToList();
}
else
{
return db.Role.Where(item => item.SiteId == siteId).ToList();
}
}
public Role AddRole(Role role)
{
using var db = _dbContextFactory.CreateDbContext();
role.Description = role.Description.Substring(0, (role.Description.Length > 256) ? 256 : role.Description.Length);
db.Role.Add(role);
db.SaveChanges();
return role;
}
public Role UpdateRole(Role role)
{
using var db = _dbContextFactory.CreateDbContext();
role.Description = role.Description.Substring(0, (role.Description.Length > 256) ? 256 : role.Description.Length);
db.Entry(role).State = EntityState.Modified;
db.SaveChanges();
return role;
}
public Role GetRole(int roleId)
{
return GetRole(roleId, true);
}
public Role GetRole(int roleId, bool tracking)
{
using var db = _dbContextFactory.CreateDbContext();
if (tracking)
{
return db.Role.Find(roleId);
}
else
{
return db.Role.AsNoTracking().FirstOrDefault(item => item.RoleId == roleId);
}
}
public void DeleteRole(int roleId)
{
using var db = _dbContextFactory.CreateDbContext();
Role role = db.Role.Find(roleId);
db.Role.Remove(role);
db.SaveChanges();
}
}
}