修复问题
This commit is contained in:
parent
87c5315d03
commit
a3b0b24db6
|
|
@ -76,5 +76,26 @@ namespace HuanMeng.DotNetCore.MultiTenant
|
||||||
return base.SaveChanges();
|
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);
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue
Block a user