server/HuanMengProject
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 53 Wiki Activity

修复问题

Browse Source
This commit is contained in:
zpc 2024-09-21 15:52:35 +08:00
parent 87c5315d03
commit a3b0b24db6
1 changed files with 21 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/0-core/HuanMeng.DotNetCore/MultiTenant/MultiTenantDbContext.cs
View File

@ -75,6 +75,27 @@ namespace HuanMeng.DotNetCore.MultiTenant
}
return base.SaveChanges();
}
public override Task<int> SaveChangesAsync(CancellationToken cancellationToken = default)
{
if (TenantInfo?.TenantId != null)
{
var entries = ChangeTracker.Entries()
.Where(e => e.Entity is IMultiTenantEntity &&
(e.State == EntityState.Added || e.State == EntityState.Modified))
.Select(e => e.Entity as IMultiTenantEntity)
.ToList();
foreach (var entity in entries)
{
if (entity?.TenantId == null || entity?.TenantId == Guid.Empty)
{
entity.TenantId = TenantInfo.TenantId;
}
}
}
return base.SaveChangesAsync(cancellationToken);
}
}
}