qgsspatialiteutils.cpp

Go to the documentation of this file.

/***************************************************************************
                          qgsspatialiteutils.cpp
                           -------------------
    begin                : Nov, 2017
    copyright            : (C) 2017 by Matthias Kuhn
    email                : [email protected]
 ***************************************************************************/
 
/***************************************************************************
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/
 
 
#include "qgsspatialiteutils.h"
#include "qgslogger.h"
 
#include <sqlite3.h>
 
#ifdef HAVE_SPATIALITE
#include <spatialite.h>
#endif
 
// Define this variable to print all spatialite SQL statements
#ifdef SPATIALITE_PRINT_ALL_SQL
// Debugging code
#include <QDebug>
#include <QThread>
static int trace_callback( unsigned, void *ctx, void *p, void * )
{
  sqlite3_stmt *stmt = ( sqlite3_stmt * )p;
  char *sql = sqlite3_expanded_sql( stmt );
  qDebug() << "SPATIALITE" << QThread::currentThreadId() << ( sqlite3 * ) ctx << sql;
  sqlite3_free( sql );
  return 0;
}
#endif
 
 
int spatialite_database_unique_ptr::open( const QString &path )
{
#ifdef HAVE_SPATIALITE
  auto &deleter = get_deleter();
  deleter.mSpatialiteContext = spatialite_alloc_connection();
#endif
 
  sqlite3 *database = nullptr;
  int result = sqlite3_open( path.toUtf8(), &database );
  std::unique_ptr< sqlite3, QgsSpatialiteCloser>::reset( database );
 
#ifdef HAVE_SPATIALITE
  if ( result == SQLITE_OK )
    spatialite_init_ex( database, deleter.mSpatialiteContext, 0 );
#endif
 
  return result;
}
 
void spatialite_database_unique_ptr::reset()
{
  std::unique_ptr< sqlite3, QgsSpatialiteCloser>::reset();
}
 
int spatialite_database_unique_ptr::open_v2( const QString &path, int flags, const char *zVfs )
{
#ifdef HAVE_SPATIALITE
  auto &deleter = get_deleter();
  deleter.mSpatialiteContext = spatialite_alloc_connection();
#endif
 
  sqlite3 *database = nullptr;
  int result = sqlite3_open_v2( path.toUtf8(), &database, flags, zVfs );
  std::unique_ptr< sqlite3, QgsSpatialiteCloser>::reset( database );
 
#ifdef HAVE_SPATIALITE
  if ( result == SQLITE_OK )
    spatialite_init_ex( database, deleter.mSpatialiteContext, 0 );
#endif
 
#ifdef SPATIALITE_PRINT_ALL_SQL
  // Log all queries
  sqlite3_trace_v2(
    database,
    SQLITE_TRACE_STMT,
    trace_callback,
    database
  );
#endif
 
  return result;
}
 
QString spatialite_database_unique_ptr::errorMessage() const
{
  return QString( sqlite3_errmsg( get() ) );
}
 
sqlite3_statement_unique_ptr spatialite_database_unique_ptr::prepare( const QString &sql, int &resultCode )
{
  sqlite3_stmt *preparedStatement = nullptr;
  const char *tail = nullptr;
  const QByteArray sqlUtf8 = sql.toUtf8();
  resultCode = sqlite3_prepare( get(), sqlUtf8, sqlUtf8.length(), &preparedStatement, &tail );
  sqlite3_statement_unique_ptr s;
  s.reset( preparedStatement );
  return s;
}
 
void QgsSpatialiteCloser::operator()( sqlite3 *handle )
{
  int res = sqlite3_close_v2( handle );
  if ( res != SQLITE_OK )
  {
    QgsDebugMsg( QStringLiteral( "sqlite3_close_v2() failed: %1" ).arg( res ) );
  }
 
#ifdef HAVE_SPATIALITE
  spatialite_cleanup_ex( mSpatialiteContext );
#endif
  mSpatialiteContext = nullptr;
}