Disable ID generation during duplicate database

Should fix #1978
sequelpro · Sep 18, 2014 · a86bed3db5f1f5528b9ce38052373286719c2313 · a86bed3
1 parent d1425fb
commit a86bed3db5f1f5528b9ce38052373286719c2313
Unified Split

Showing with 14 additions and 0 deletions.

+14 −0 Source/SPTableCopy.m
diff --git a/Source/SPTableCopy.m b/Source/SPTableCopy.m
@@ -98,6 +98,13 @@ - (BOOL)copyTables:(NSArray *)tablesArray from:(NSString *)sourceDB to:(NSString
 		success = NO;
 	}
 	
+	// Disable auto-id creation for '0' values
+	[connection queryString:@"/*!40101 SET @OLD_SQL_MODE=@@SQL_MODE, SQL_MODE='NO_AUTO_VALUE_ON_ZERO' */"];
+	
+	if([connection queryErrored]) {
+		success = NO;
+	}
+	
 	for (NSString *tableName in tablesArray) 
 	{
 		if (![self copyTable:tableName from:sourceDB to:targetDB withContent:copyWithContent]) {
@@ -112,6 +119,13 @@ - (BOOL)copyTables:(NSArray *)tablesArray from:(NSString *)sourceDB to:(NSString
 		success = NO;
 	}
 	
+	// re-enable id creation
+	[connection queryString:@"/*!40101 SET SQL_MODE=@OLD_SQL_MODE */"];
+	
+	if ([connection queryErrored]) {
+		success = NO;
+	}
+	
 	return success;
 }