Entity Framework/SQL2008 — jak automatycznie aktualizować pola LastModified dla jednostek?

Wiem, że jestem trochę spóźniony na imprezę, ale właśnie rozwiązałem to dla projektu, nad którym pracuję i pomyślałem, że podzielę się swoim rozwiązaniem.

Po pierwsze, aby rozwiązanie było bardziej przydatne do ponownego użycia, stworzyłem klasę bazową z właściwościami znacznika czasu:

public class EntityBase
{
    public DateTime? CreatedDate { get; set; }
    public DateTime? LastModifiedDate { get; set; }
}

Następnie nadpisałem metodę SaveChanges w moim DbContext:

public class MyContext : DbContext
{
    public override int SaveChanges()
    {
        ObjectContext context = ((IObjectContextAdapter)this).ObjectContext;

        //Find all Entities that are Added/Modified that inherit from my EntityBase
        IEnumerable<ObjectStateEntry> objectStateEntries =
            from e in context.ObjectStateManager.GetObjectStateEntries(EntityState.Added | EntityState.Modified)
            where
                e.IsRelationship == false &&
                e.Entity != null &&
                typeof(EntityBase).IsAssignableFrom(e.Entity.GetType())
            select e;

        var currentTime = DateTime.Now;

        foreach (var entry in objectStateEntries)
        {
            var entityBase = entry.Entity as EntityBase;

            if (entry.State == EntityState.Added)
            {
                entityBase.CreatedDate = currentTime;
            }

            entityBase.LastModifiedDate = currentTime;
        }

        return base.SaveChanges();
    }
}