Update Audit

Author: JonathanMagnan

src/shared/Z.EF.Plus.Audit.Shared/AuditManager.cs

@@ -5,6 +5,8 @@
 // More projects: http://www.zzzprojects.com/
 // Copyright © ZZZ Projects Inc. 2014 - 2016. All rights reserved.
 
+using Z.EntityFramework.Extensions;
+
 namespace Z.EntityFramework.Plus
 {
     /// <summary>Manager for audits.</summary>
@@ -13,6 +15,8 @@ public static class AuditManager
         /// <summary>Static constructor.</summary>
         static AuditManager()
         {
+            EntityFrameworkManager.IsEntityFrameworkPlus = true;
+
             DefaultConfiguration = new AuditConfiguration();
         }
 

src/shared/Z.EF.Plus.Audit.Shared/Extensions/DbSet`AuditEntry/DbSet`AuditEntry.cs

@@ -23,6 +23,11 @@ namespace Z.EntityFramework.Plus
     public static partial class AuditExtensions
     {
         public static IQueryable<AuditEntry> Where<T>(this DbSet<AuditEntry> set, T entry) where T : class
+        {
+            return set.Where<AuditEntry, T>(entry);
+        }
+
+        public static IQueryable<TAuditEntry> Where<TAuditEntry, T>(this DbSet<TAuditEntry> set, T entry) where TAuditEntry : AuditEntry where T : class
         {
             var context = set.GetDbContext();
             var keyNames = context.GetKeyNames<T>();
@@ -49,6 +54,11 @@ public static IQueryable<AuditEntry> Where<T>(this DbSet<AuditEntry> set, T entr
         }
 
         public static IQueryable<AuditEntry> Where<T>(this DbSet<AuditEntry> set, params object[] keyValues) where T : class
+        {
+            return set.Where<AuditEntry, T>(keyValues);
+        }
+
+        public static IQueryable<TAuditEntry> Where<TAuditEntry, T>(this DbSet<TAuditEntry> set, params object[] keyValues) where TAuditEntry : AuditEntry where T : class
         {
             var context = set.GetDbContext();
             var keyNames = context.GetKeyNames<T>();
@@ -59,6 +69,7 @@ public static IQueryable<AuditEntry> Where<T>(this DbSet<AuditEntry> set, params
             {
                 throw new Exception(ExceptionMessage.Audit_Key_Null);
             }
+
             if (keyValues.Length != keyNames.Length)
             {
                 throw new Exception(ExceptionMessage.Audit_Key_OutOfBound);