OGR: Add SQL query logger

qgis · Dec 28, 2022 · e166ae4 · e166ae4
1 parent 2712aed
commit e166ae4
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/src/core/providers/ogr/qgsogrfeatureiterator.cpp b/src/core/providers/ogr/qgsogrfeatureiterator.cpp
@@ -30,6 +30,7 @@
 #include "qgsogrtransaction.h"
 #include "qgssymbol.h"
 #include "qgsgeometryengine.h"
+#include "qgsdbquerylog.h"
 
 #include <QTextCodec>
 #include <QFile>
@@ -279,6 +280,23 @@ QgsOgrFeatureIterator::QgsOgrFeatureIterator( QgsOgrFeatureSource *source, bool
     std::sort( mRequestAttributes.begin(), mRequestAttributes.end() );
   }
 
+  // Install query logger
+#if GDAL_VERSION_NUM >= GDAL_COMPUTE_VERSION(3,7,0)
+  GDALDatasetSetQueryLoggerFunc( mConn->ds, [ ]( const char *pszSQL, const char *pszError, int64_t lNumRecords, int64_t lExecutionTimeMilliseconds, void *pQueryLoggerArg )
+  {
+    QgsDatabaseQueryLogEntry entry;
+    entry.initiatorClass = QStringLiteral( "QgsOgrFeatureIterator" );
+    entry.origin = QGS_QUERY_LOG_ORIGIN;
+    entry.provider = QStringLiteral( "ogr" );
+    entry.uri = *reinterpret_cast<QString *>( pQueryLoggerArg );
+    entry.query = QString( pszSQL );
+    entry.error = QString( pszError );
+    entry.startedTime = QDateTime::currentMSecsSinceEpoch() - lExecutionTimeMilliseconds;
+    entry.fetchedRows = lNumRecords;
+    QgsApplication::databaseQueryLog()->log( entry );
+    QgsApplication::databaseQueryLog()->finished( entry );
+  }, reinterpret_cast<void *>( &mConn->path ) );
+#endif
   //start with first feature
   rewind();
 

diff --git a/src/core/providers/ogr/qgsogrfeatureiterator.h b/src/core/providers/ogr/qgsogrfeatureiterator.h
@@ -126,6 +126,7 @@ class QgsOgrFeatureIterator final: public QgsAbstractFeatureIteratorFromSource<Q
     bool fetchFeatureWithId( QgsFeatureId id, QgsFeature &feature ) const;
 
     void resetReading();
+
 };
 
 ///@endcond