c# - 保存前设置值

Question

我想知道是否有任何方法可以为实体 onsave 设置一个值？
因为我正在开发一个 Multi-Tenancy Web 应用程序，所以我想设置当前租户 ID(通过简单的 DI 服务)。

我尝试在 Fluent API 中使用 HasDefaultValue()，但这会尝试转换为 SQL 函数。所以这对我不起作用。

builder.Entity<Order>( )
    .HasQueryFilter(p => p.TenantId == _tenantProvider.GetTenantId())
    .Property(p => p.TenantId)
    .HasDefaultValue(_tenantProvider.GetTenantId());

非常感谢任何建议。

Answer 1

您可以覆盖 DbContext.SaveChanges()方法并迭代 ChangeTracker 条目:

public override int SaveChanges()
{
    foreach (var entityEntry in ChangeTracker.Entries()) // Iterate all made changes
    {
        if (entityEntry.Entity is Order order)
        {
            if (entityEntry.State == EntityState.Added) // If you want to update TenantId when Order is added
            {
                order.TenantId = _tenantProvider.GetTenantId();
            }
            else if (entityEntry.State == EntityState.Modified) // If you want to update TenantId when Order is modified
            {
                order.TenantId = _tenantProvider.GetTenantId();
            }
        }
    }
    return base.SaveChanges();
}

当然，这需要将租户提供者注入(inject)到您的上下文中。