Resolves issue #2081 : provides an option to include statements to disable and re-enable foreign key checks in a MySQL database backup output statement.

3 files changed, 28 insertions, 1 deletions

diff --git a/system/database/DB_utility.php b/system/database/DB_utility.php
index c4140aef3..3ed75967e 100644
--- a/system/database/DB_utility.php
+++ b/system/database/DB_utility.php
@@ -326,7 +326,8 @@ abstract class CI_DB_utility {
 				'format'		=> 'gzip', // gzip, zip, txt
 				'add_drop'		=> TRUE,
 				'add_insert'		=> TRUE,
-				'newline'		=> "\n"
+				'newline'		=> "\n",
+				'fk_checks'		=> TRUE
 			);
 
 		// Did the user submit any preferences? If so set them....
diff --git a/system/database/drivers/mysql/mysql_utility.php b/system/database/drivers/mysql/mysql_utility.php
index 8aa051755..e2de82d0b 100644
--- a/system/database/drivers/mysql/mysql_utility.php
+++ b/system/database/drivers/mysql/mysql_utility.php
@@ -76,6 +76,13 @@ class CI_DB_mysql_utility extends CI_DB_utility {
 
 		// Build the output
 		$output = '';
+
+		// Do we need to include a statement to disable FK checks?
+		if ($fk_checks === FALSE)
+		{
+			$output .= "SET foreign_key_checks = 0;".$newline;
+		}
+
 		foreach ( (array) $tables as $table)
 		{
 			// Is the table in the "ignore" list?
@@ -181,6 +188,12 @@ class CI_DB_mysql_utility extends CI_DB_utility {
 			$output .= $newline.$newline;
 		}
 
+		// Do we need to include a statement to re-enable FK checks?
+		if ($fk_checks === FALSE)
+		{
+			$output .= "SET foreign_key_checks = 1;".$newline;
+		}
+
 		return $output;
 	}
 
diff --git a/system/database/drivers/mysqli/mysqli_utility.php b/system/database/drivers/mysqli/mysqli_utility.php
index 345691e84..ff4949046 100644
--- a/system/database/drivers/mysqli/mysqli_utility.php
+++ b/system/database/drivers/mysqli/mysqli_utility.php
@@ -76,6 +76,13 @@ class CI_DB_mysqli_utility extends CI_DB_utility {
 
 		// Build the output
 		$output = '';
+
+		// Do we need to include a statement to disable FK checks?
+		if ($fk_checks === FALSE)
+		{
+			$output .= "SET foreign_key_checks = 0;".$newline;
+		}
+
 		foreach ( (array) $tables as $table)
 		{
 			// Is the table in the "ignore" list?
@@ -181,6 +188,12 @@ class CI_DB_mysqli_utility extends CI_DB_utility {
 			$output .= $newline.$newline;
 		}
 
+		// Do we need to include a statement to re-enable FK checks?
+		if ($fk_checks === FALSE)
+		{
+			$output .= "SET foreign_key_checks = 1;".$newline;
+		}
+
 		return $output;
 	}