快捷導(dǎo)航

Entity Framework Core生成列并跟蹤列記錄

更新時間：2022年02月21日 11:56:44 作者：Sweet-Tang

這篇文章介紹了Entity Framework Core生成列并跟蹤列記錄的方法，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

注意：我使用的是 Entity Framework Core 2.0 (2.0.0-preview2-final)。正式版發(fā)布時，功能可能存在變動。

當您設(shè)計數(shù)據(jù)庫時，有時需要添加列以跟蹤記錄何時更改，以及誰進行了更改。例如，您添加以下列：

CreatedAt
CreatedBy
LastUpdatedAt
LastUpdatedBy

您可以輕松地使用默認值和觸發(fā)器來處理CreatedAt和LastUpdatedAt列。老實說，創(chuàng)建觸發(fā)器是件無聊的事情，你也不想自己做。此外，很難設(shè)置用戶名，因為它是應(yīng)用程序的信息；實際上，在Web程序的上下文中，它們只是連接到數(shù)據(jù)庫的一個用戶，因此，您無法使用數(shù)據(jù)庫中CURRENT_USER函數(shù)設(shè)置跟蹤列的值。

當然，您不想訪問這個屬性。相反，您希望 Entity Framework 為您自動執(zhí)行，所以，解決的辦法是調(diào)用SaveChanges或SaveChangesAsync方法之前自動設(shè)置這些屬性的值。在模型中有多種方法處理這些屬性：您可以在模型中添加讀寫（R/W）屬性；您也可以使用只讀（R/O）屬性；還有一種方案，您可以使用陰影（Shadow）屬性。在最后一種方案中，是將列不映射到模型中的屬性，只有 Entity Framework 才知道這些屬性。是的，EF Core 非常靈活:)

我們用 Entity Framework 實現(xiàn)這3種方案！

讀寫屬性

在這種情況下，實體將具有讀/寫屬性。這意味著，您可以在代碼中更改屬性的值。

public interface ITrackable
{
    DateTime CreatedAt { get; set; }
    string CreatedBy { get; set; }
    DateTime LastUpdatedAt { get; set; }
    string LastUpdatedBy { get; set; }
}

public class Post : ITrackable
{
    public int Id { get; set; }
    public string Title { get; set; }
    public string Content { get; set; }

    public DateTime CreatedAt { get; set; }
    public string CreatedBy { get; set; }
    public DateTime LastUpdatedAt { get; set; }
    public string LastUpdatedBy { get; set; }
}

現(xiàn)在，我們將更改SaveChanges方法的默認行為來設(shè)置跟蹤屬性的值。要小心，有兩個方法要重寫：SaveChanges和異步方法SaveChangesAsync。在代碼中遍歷上下文中跟蹤的實體，并根據(jù)狀態(tài)設(shè)置跟蹤屬性的值。代碼非常簡單：

public class BloggingContext : DbContext
{
    public DbSet<Post> Posts { get; set; }

    public override int SaveChanges(bool acceptAllChangesOnSuccess)
    {
        OnBeforeSaving();
        return base.SaveChanges(acceptAllChangesOnSuccess);
    }

    public override Task<int> SaveChangesAsync(bool acceptAllChangesOnSuccess, CancellationToken cancellationToken = default(CancellationToken))
    {
        OnBeforeSaving();
        return base.SaveChangesAsync(acceptAllChangesOnSuccess, cancellationToken);
    }

    private void OnBeforeSaving()
    {
        var entries = ChangeTracker.Entries();
        foreach (var entry in entries)
        {
            if (entry.Entity is ITrackable trackable)
            {
                var now = DateTime.UtcNow;
                var user = GetCurrentUser();
                switch (entry.State)
                {
                    case EntityState.Modified:
                        trackable.LastUpdatedAt = now;
                        trackable.LastUpdatedBy = user;
                        break;

                    case EntityState.Added:
                        trackable.CreatedAt = now;
                        trackable.CreatedAt = user;
                        trackable.LastUpdatedAt = now;
                        trackable.LastUpdatedBy = user;
                        break;
                }
            }
        }
    }

    private string GetCurrentUser()
    {
        return "UserName"; // TODO implement your own logic
    }
}

這個方案有個不容忽視的問題，屬性是可寫的，您不希望在代碼中設(shè)置屬性的值。讓我們看看如何使用只讀屬性。

只讀屬性

Entity Framework Core 允許將列映射到字段。這樣，您可以使用只讀屬性，我們來修改 Post類：

public class Post
{
    private DateTime _createdAt;
    private string _createdBy;
    private DateTime _lastUpdatedAt;
    private string _lastUpdatedBy;

    public int Id { get; set; }
    public string Title { get; set; }
    public string Content { get; set; }

    public DateTime CreatedAt => _createdAt;
    public string CreatedBy => _createdBy;
    public DateTime LastUpdatedAt => _lastUpdatedAt;
    public string LastUpdatedBy => _lastUpdatedBy;
}

上面的代碼有點麻煩，但屬性現(xiàn)在是只讀的。您需要指引 Entity Framework 使用這些字段來設(shè)置列的值。

public class BloggingContext : DbContext
{
    public DbSet<Post> Posts { get; set; }

    protected override void OnModelCreating(ModelBuilder modelBuilder)
    {
        base.OnModelCreating(modelBuilder);

        modelBuilder.Entity<Post>()
            .Property(post => post.CreatedAt)
            .HasField("_createdAt");

        modelBuilder.Entity<Post>()
            .Property(post => post.CreatedBy)
            .HasField("_createdBy");

        modelBuilder.Entity<Post>()
            .Property(post => post.LastUpdatedAt)
            .HasField("_lastUpdatedAt");

        modelBuilder.Entity<Post>()
            .Property(post => post.LastUpdatedBy)
            .HasField("_lastUpdatedBy");
    }
}

您不能像以前那樣實現(xiàn)OnBeforeSaving方法，因為屬性沒有setter。不過，您可以使用 ChangeTracker來指引 Entity Framework 為屬性設(shè)置值。這種方式無需訪問屬性或字段。

    private void OnBeforeSaving()
    {
        var entries = ChangeTracker.Entries();
        foreach (var entry in entries)
        {
            if (entry.Entity is Post post)
            {
                var now = DateTime.UtcNow;
                var user = GetCurrentUser();
                switch (entry.State)
                {
                    case EntityState.Modified:
                        entry.CurrentValues["LastUpdatedAt"] = now;
                        entry.CurrentValues["LastUpdatedBy"] = user;
                        break;

                    case EntityState.Added:
                        entry.CurrentValues["CreatedAt"] = now;
                        entry.CurrentValues["CreatedBy"] = user;
                        entry.CurrentValues["LastUpdatedAt"] = now;
                        entry.CurrentValues["LastUpdatedBy"] = user;
                        break;
                }
            }
        }
    }
}

我認為應(yīng)是使用的方式，因為這種方式很清晰。

不定義屬性

在某些情況下，您可能需要設(shè)置列的值而不在模型中定義屬性。例如，您可能會公開最后更新的日期，但不會顯示誰做了更改。Entity Framework Core 可以通過使用陰影（Shadow ）屬性來處理這個。陰影屬性在OnModelCreating方法中聲明，但在類中不存在。您可以使用ChangeTracker讀取和寫入陰影屬性的值。

public class Post
{
    public int Id { get; set; }
    public string Title { get; set; }
    public string Content { get; set; }
}

public class BloggingContext : DbContext
{
    protected override void OnModelCreating(ModelBuilder modelBuilder)
    {
        base.OnModelCreating(modelBuilder);

        // Declare properties
        modelBuilder.Entity<Post>().Property<DateTime>("CreatedAt");
        modelBuilder.Entity<Post>().Property<string>("CreatedBy");
        modelBuilder.Entity<Post>().Property<DateTime>("LastUpdatedAt");
        modelBuilder.Entity<Post>().Property<string>("LastUpdatedBy");
    }

    private void OnBeforeSaving()
    {
        var entries = ChangeTracker.Entries();
        foreach (var entry in entries)
        {
            if (entry.Entity is Post post)
            {
                var now = DateTime.UtcNow;
                var user = GetCurrentUser();
                switch (entry.State)
                {
                    case EntityState.Modified:
                        entry.CurrentValues["LastUpdatedAt"] = now;
                        entry.CurrentValues["LastUpdatedBy"] = user;
                        break;

                    case EntityState.Added:
                        entry.CurrentValues["CreatedAt"] = now;
                        entry.CurrentValues["CreatedBy"] = user;
                        entry.CurrentValues["LastUpdatedAt"] = now;
                        entry.CurrentValues["LastUpdatedBy"] = user;
                        break;
                }
            }
        }
    }
}