快捷導(dǎo)航

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

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

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

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

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

CreatedAt
CreatedBy
LastUpdatedAt
LastUpdatedBy

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

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

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

讀寫屬性

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

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方法的默認(rèn)行為來(lái)設(shè)置跟蹤屬性的值。要小心，有兩個(gè)方法要重寫：SaveChanges和異步方法SaveChangesAsync。在代碼中遍歷上下文中跟蹤的實(shí)體，并根據(jù)狀態(tài)設(shè)置跟蹤屬性的值。代碼非常簡(jiǎn)單：

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
    }
}

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

只讀屬性

Entity Framework Core 允許將列映射到字段。這樣，您可以使用只讀屬性，我們來(lái)修改 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;
}

上面的代碼有點(diǎn)麻煩，但屬性現(xiàn)在是只讀的。您需要指引 Entity Framework 使用這些字段來(lái)設(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");
    }
}

您不能像以前那樣實(shí)現(xiàn)OnBeforeSaving方法，因?yàn)閷傩詻]有setter。不過，您可以使用 ChangeTracker來(lái)指引 Entity Framework 為屬性設(shè)置值。這種方式無(wú)需訪問屬性或字段。

    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;
                }
            }
        }
    }
}

我認(rèn)為應(yīng)是使用的方式，因?yàn)檫@種方式很清晰。

不定義屬性

在某些情況下，您可能需要設(shè)置列的值而不在模型中定義屬性。例如，您可能會(huì)公開最后更新的日期，但不會(huì)顯示誰(shuí)做了更改。Entity Framework Core 可以通過使用陰影（Shadow ）屬性來(lái)處理這個(gè)。陰影屬性在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;
                }
            }
        }
    }
}