Asp.Net Core 项目 EntityFramework Core 根据登录用户名过滤数据

小程序：扫一扫查出行
【扫一扫了解最新限行尾号】
复制小程序

1、创建ASP.NET Core Web Applicatoin (MVC)项目，并且使用 Individual User Account

2、创建数据筛选接口 Models->IDataFilter.cs

public interface IDataFilter
{
string UserName { get; set; }
}

3、创建实体 Models->Book.cs 并继承 IDataFilter接口，并将实体加入到 Data->ApplicatoinDbContext.cs 上下文中.

public class Book : IDataFilter
{
public int Id { get; set; }
[Display(Name = "书名")]
public string Name { get; set; }
public string UserName { get; set; }
}

public class ApplicationDbContext : IdentityDbContext
{
public DbSet<Book> Books { get; set; }
public ApplicationDbContext(DbContextOptions<ApplicationDbContext> options)
: base(options)
{
}
}

4、利用模板创建Book类的 CRUD界面。在 Controllers文件夹上右键选择 Add(新增)->Controller(控制器)

5、自动生成Book的增删改查之后，我们在这里要做一个细微的修改，因为我们IDataFilter字段 UserName是系统生成的，所以我们要修改两个地方

将BooksController.cs 下的 Create Action(由于只做演示，没有去修改Update页面)里面的 Bind UserName去掉。修改后结果如下

并将创建页面 Views->Books->Create.cshtml 中的 UserName 部份备注

在母版页添加Book菜单链接 Views->Shared->_Layout.cshtml

6、打开Nuget管理控制台迁移数据库，并F5运行，点击Book链接。检查一下程序有没问题。

Add-Migration Init //创建迁移文件
Update-Database //更新到数据库

7、由于Book实体实现了IDataFilter，UserName我们会通过重写ApplicationDbContext的SaveChanges的实现，进行自动填充用户名.由于要获取登录信息，在ApplicationDbContext中我们需注入 IHttpContextAccessor

public class ApplicationDbContext : IdentityDbContext
{
private readonly IHttpContextAccessor _httpContextAccessor;
public DbSet<Book> Books { get; set; }
public ApplicationDbContext(DbContextOptions<ApplicationDbContext> options, IHttpContextAccessor httpContextAccessor)
: base(options)
{
_httpContextAccessor = httpContextAccessor;
} public override int SaveChanges()
{
FillDataFilterInfo();
return base.SaveChanges();
}
public override Task<int> SaveChangesAsync(CancellationToken cancellationToken = default)
{
FillDataFilterInfo();
return base.SaveChangesAsync(cancellationToken);
}
protected void FillDataFilterInfo() =>
ChangeTracker
.Entries()
.Where(w => w.Entity is IDataFilter && w.State == EntityState.Added)
.ToList()
.ForEach(entry => ((IDataFilter)entry.Entity).UserName = CurrentUserName); private string CurrentUserName => _httpContextAccessor.HttpContext.User?.Identity?.Name;
}

8、验证结果，运行项目，首先注册一个帐号。然后进行创建一本书。最终结果。

9、最后我们来实现数据过滤部份代码,打开 ApplicatonDbContex.cs

添加私有方法DataFilters

private void DataFilters<T>(ModelBuilder builder)
where T : class
{
builder.Entity<T>().HasQueryFilter(s => ((IDataFilter)s).UserName == CurrentUserName);
}

添加一个静态MethodInfo方法。这里用到返射实现

private static readonly MethodInfo _dataFiltersMethodInfo = typeof(ApplicationDbContext).GetMethod(nameof(DataFilters), BindingFlags.Instance | BindingFlags.NonPublic);

在重写OnModelCreating ，针对所有实现 IDataFilter的实体添加数据过滤

protected override void OnModelCreating(ModelBuilder builder)
{
builder.Model
.GetEntityTypes()
.Where(w => typeof(IDataFilter).IsAssignableFrom(w.ClrType))
.ToList().ForEach(entityType =>
{
_dataFiltersMethodInfo
.MakeGenericMethod(entityType.ClrType)
.Invoke(this, new object[] { builder });
});
base.OnModelCreating(builder);
}

10、运行程序，分别用不同的浏览器。注册两个帐号。然后创建几本书。最终结果