Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
23 changes: 22 additions & 1 deletion src/System.Data.Entity.Hooks/DbHookContext.cs
Original file line number Diff line number Diff line change
Expand Up @@ -128,7 +128,13 @@ public override int SaveChanges()

try
{
return base.SaveChanges();
BeforeSaveChanges();

var rowsAffected = base.SaveChanges();

AfterSaveChanges(rowsAffected);

return rowsAffected;
}
finally
{
Expand Down Expand Up @@ -169,6 +175,21 @@ protected void RegisterPostSaveHook(IDbHook dbHook)
_postSaveHooks.Add(dbHook);
}

/// <summary>
/// Called right before SaveChanges method of the DbContext is called, and after pre save hooks are executed.
/// </summary>
protected virtual void BeforeSaveChanges()
{
}

/// <summary>
/// Called right after SaveChanges method of the DbContext is called, and before post save hooks are executed.
/// </summary>
/// <param name="rowsAffected">The number of objects written to the underlying database.</param>
protected virtual void AfterSaveChanges(int rowsAffected)
{
}

private void ObjectMaterialized(object sender, ObjectMaterializedEventArgs e)
{
ObjectStateEntry objectStateEntry;
Expand Down