Implement Disable/EnableForeignKeyCheck for MySql + SQL Server

Author: mythz

src/ServiceStack.OrmLite.MySql/MySqlDialectProviderBase.cs

@@ -554,6 +554,13 @@ public override string SqlCast(object fieldOrValue, string castAs) =>
 
         public override string SqlBool(bool value) => value ? "1" : "0";
 
+        public override void EnableForeignKeysCheck(IDbCommand cmd) => cmd.ExecNonQuery("SET FOREIGN_KEY_CHECKS=1;");
+        public override Task EnableForeignKeysCheckAsync(IDbCommand cmd, CancellationToken token = default) => 
+	        cmd.ExecNonQueryAsync("SET FOREIGN_KEY_CHECKS=1;", null, token);
+        public override void DisableForeignKeysCheck(IDbCommand cmd) => cmd.ExecNonQuery("SET FOREIGN_KEY_CHECKS=0;");
+        public override Task DisableForeignKeysCheckAsync(IDbCommand cmd, CancellationToken token = default) => 
+	        cmd.ExecNonQueryAsync("SET FOREIGN_KEY_CHECKS=0;", null, token);
+
         protected DbConnection Unwrap(IDbConnection db)
         {
             return (DbConnection)db.ToDbConnection();

src/ServiceStack.OrmLite.SqlServer/SqlServerOrmLiteDialectProvider.cs

@@ -685,6 +685,13 @@ public override string SqlCast(object fieldOrValue, string castAs) =>
                 : $"CAST({fieldOrValue} AS {castAs})";
 
         public override string SqlRandom => "NEWID()";
+
+        public override void EnableForeignKeysCheck(IDbCommand cmd) => cmd.ExecNonQuery("EXEC sp_msforeachtable \"ALTER TABLE ? WITH CHECK CHECK CONSTRAINT all\"");
+        public override Task EnableForeignKeysCheckAsync(IDbCommand cmd, CancellationToken token = default) => 
+            cmd.ExecNonQueryAsync("EXEC sp_msforeachtable \"ALTER TABLE ? WITH CHECK CHECK CONSTRAINT all\"", null, token);
+        public override void DisableForeignKeysCheck(IDbCommand cmd) => cmd.ExecNonQuery("EXEC sp_msforeachtable \"ALTER TABLE ? NOCHECK CONSTRAINT all\"");
+        public override Task DisableForeignKeysCheckAsync(IDbCommand cmd, CancellationToken token = default) => 
+            cmd.ExecNonQueryAsync("EXEC sp_msforeachtable \"ALTER TABLE ? NOCHECK CONSTRAINT all\"", null, token);
 
         protected SqlConnection Unwrap(IDbConnection db) => (SqlConnection)db.ToDbConnection();
 

src/ServiceStack.OrmLite.Sqlite/SqliteOrmLiteDialectProviderBase.cs

@@ -248,13 +248,13 @@ public override string SqlConflict(string sql, string conflictResolution)
 
         public override string SqlRandom => "random()";
 
-        public override void EnableForeignKeysCheck(IDbCommand cmd) => cmd.ExecNonQuery($"PRAGMA foreign_keys = ON;");
+        public override void EnableForeignKeysCheck(IDbCommand cmd) => cmd.ExecNonQuery("PRAGMA foreign_keys = ON;");
         public override Task EnableForeignKeysCheckAsync(IDbCommand cmd, CancellationToken token = default) => 
-            cmd.ExecNonQueryAsync($"PRAGMA foreign_keys = ON;", null, token);
+            cmd.ExecNonQueryAsync("PRAGMA foreign_keys = ON;", null, token);
 
-        public override void DisableForeignKeysCheck(IDbCommand cmd) => cmd.ExecNonQuery($"PRAGMA foreign_keys = OFF;");
+        public override void DisableForeignKeysCheck(IDbCommand cmd) => cmd.ExecNonQuery("PRAGMA foreign_keys = OFF;");
         public override Task DisableForeignKeysCheckAsync(IDbCommand cmd, CancellationToken token = default) => 
-            cmd.ExecNonQueryAsync($"PRAGMA foreign_keys = OFF;", null, token);
+            cmd.ExecNonQueryAsync("PRAGMA foreign_keys = OFF;", null, token);
     }
 
     public static class SqliteExtensions

tests/ServiceStack.OrmLite.Tests/Issues/LoadReferenceIssueWithCyclicalForeignKeys.cs

@@ -6,7 +6,7 @@
 
 namespace ServiceStack.OrmLite.Tests.Issues
 {
-    [TestFixtureOrmLiteDialects(Dialect.Sqlite)]
+    [TestFixtureOrmLiteDialects(Dialect.Sqlite | Dialect.MySql)]
     public class LoadReferenceIssueWithCyclicalForeignKeys : OrmLiteProvidersTestBase
     {
         public LoadReferenceIssueWithCyclicalForeignKeys(DialectContext context) : base(context) { }
@@ -44,8 +44,10 @@ public class NameEntity : BaseEntity
         private void RecreateTables(IDbConnection db)
         {
             db.DisableForeignKeysCheck();
-            db.DropAndCreateTable<NameEntity>();
-            db.DropAndCreateTable<ResearchEntity>();
+            db.DropTable<NameEntity>();
+            db.DropTable<ResearchEntity>();
+            db.CreateTable<NameEntity>();
+            db.CreateTable<ResearchEntity>();
             db.EnableForeignKeysCheck();
         }