首頁 > 軟體

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

2022-02-21 13:01:48

注意:我使用的是 Entity Framework Core 2.0 (2.0.0-preview2-final)。正式版釋出時,功能可能存在變動。

當您設計資料庫時,有時需要新增列以跟蹤記錄何時更改,以及誰進行了更改。例如,您新增以下列:

  • CreatedAt
  • CreatedBy
  • LastUpdatedAt
  • LastUpdatedBy

您可以輕鬆地使用預設值和觸發器來處理CreatedAtLastUpdatedAt列。老實說,建立觸發器是件無聊的事情,你也不想自己做。此外,很難設定使用者名稱,因為它是應用程式的資訊;實際上,在Web程式的上下文中,它們只是連線到資料庫的一個使用者,因此,您無法使用資料庫中CURRENT_USER函數設定跟蹤列的值。

當然,您不想存取這個屬性。相反,您希望 Entity Framework 為您自動執行,所以,解決的辦法是呼叫SaveChangesSaveChangesAsync方法之前自動設定這些屬性的值。在模型中有多種方法處理這些屬性:您可以在模型中新增讀寫(R/W)屬性;您也可以使用唯讀(R/O)屬性;還有一種方案,您可以使用陰影(Shadow)屬性。在最後一種方案中,是將列不對映到模型中的屬性,只有 Entity Framework 才知道這些屬性。是的,EF Core 非常靈活:)

我們用 Entity Framework 實現這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; }
}

現在,我們將更改SaveChanges方法的預設行為來設定跟蹤屬性的值。要小心,有兩個方法要重寫:SaveChanges和非同步方法SaveChangesAsync。在程式碼中遍歷上下文中跟蹤的實體,並根據狀態設定跟蹤屬性的值。程式碼非常簡單:

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

這個方案有個不容忽視的問題,屬性是可寫的,您不希望在程式碼中設定屬性的值。讓我們看看如何使用唯讀屬性。

唯讀屬性

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

上面的程式碼有點麻煩,但屬性現在是唯讀的。您需要指引 Entity Framework 使用這些欄位來設定列的值。

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

您不能像以前那樣實現OnBeforeSaving方法,因為屬性沒有setter。不過,您可以使用 ChangeTracker來指引 Entity Framework 為屬性設定值。這種方式無需存取屬性或欄位。

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

我認為應是使用的方式,因為這種方式很清晰。

不定義屬性

在某些情況下,您可能需要設定列的值而不在模型中定義屬性。例如,您可能會公開最後更新的日期,但不會顯示誰做了更改。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;
                }
            }
        }
    }
}

總結

使用 Entity Framework Core,您可以根據需求將資料庫表對映到物件模型。這允許建立乾淨的類,不需要公開您不想要的屬性。

到此這篇關於Entity Framework Core生成列並跟蹤列記錄的文章就介紹到這了。希望對大家的學習有所幫助,也希望大家多多支援it145.com。


IT145.com E-mail:sddin#qq.com