OGR/SQLite/GPKG pragma-defer_foreign_keys=ON

When in transaction forces defer FK evaluation.
qgis · Jun 30, 2021 · 3df8e58 · 3df8e58
1 parent 2745732
commit 3df8e58
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/src/core/providers/ogr/qgsogrproviderutils.h b/src/core/providers/ogr/qgsogrproviderutils.h
@@ -270,6 +270,7 @@ class CORE_EXPORT QgsOgrProviderUtils
 class QgsOgrDataset
 {
     friend class QgsOgrProviderUtils;
+    friend class QgsOgrTransaction;
     QgsOgrProviderUtils::DatasetIdentification mIdent;
     QgsOgrProviderUtils::DatasetWithLayers *mDs;
 

diff --git a/src/core/providers/ogr/qgsogrtransaction.cpp b/src/core/providers/ogr/qgsogrtransaction.cpp
@@ -31,6 +31,16 @@ QgsOgrTransaction::QgsOgrTransaction( const QString &connString, QgsOgrDatasetSh
 
 bool QgsOgrTransaction::beginTransaction( QString &error, int /* statementTimeout */ )
 {
+  GDALDriverH hDriver = GDALGetDatasetDriver( mSharedDS.get()->mDs->hDS );
+  const QString driverName = GDALGetDriverShortName( hDriver );
+  if ( driverName == QLatin1String( "GPKG" ) || driverName == QLatin1String( "SQLite" ) )
+  {
+    QString fkDeferError;
+    if ( ! executeSql( QStringLiteral( "PRAGMA defer_foreign_keys = ON" ), fkDeferError ) )
+    {
+      QgsDebugMsg( QStringLiteral( "Error setting PRAGMA defer_foreign_keys = ON: %1" ).arg( fkDeferError ) );
+    }
+  }
   return executeSql( QStringLiteral( "BEGIN" ), error );
 }