qgsabstractdatabaseproviderconnection.h

Go to the documentation of this file.

/***************************************************************************
  qgsabstractdatabaseproviderconnection.h - QgsAbstractDatabaseProviderConnection
 
 ---------------------
 begin                : 2.8.2019
 copyright            : (C) 2019 by Alessandro Pasotti
 email                : elpaso at itopen dot it
 ***************************************************************************
 *                                                                         *
 *   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.                                   *
 *                                                                         *
 ***************************************************************************/
#ifndef QGSABSTRACTDATABASEPROVIDERCONNECTION_H
#define QGSABSTRACTDATABASEPROVIDERCONNECTION_H
 
#include "qgsabstractproviderconnection.h"
#include "qgscoordinatereferencesystem.h"
#include "qgis_core.h"
#include "qgsfields.h"
#include "qgsexception.h"
#include "qgsvectordataprovider.h"
 
#include <QObject>
 
class QgsFeedback;
 
class CORE_EXPORT QgsAbstractDatabaseProviderConnection : public QgsAbstractProviderConnection
{
 
    Q_GADGET
 
  public:
 
    enum TableFlag
    {
      Aspatial = 1 << 1,          
      Vector = 1 << 2,            
      Raster = 1 << 3,            
      View = 1 << 4,              
      MaterializedView = 1 << 5,  
      Foreign = 1 << 6,           
    };
 
    Q_ENUM( TableFlag )
    Q_DECLARE_FLAGS( TableFlags, TableFlag )
    Q_FLAG( TableFlags )
 
    
    struct CORE_EXPORT QueryResult
    {
 
        QStringList columns() const;
 
        QList<QList<QVariant> > rows( QgsFeedback *feedback = nullptr );
 
        bool hasNextRow() const;
 
        QList<QVariant> nextRow() const;
 
        long long fetchedRowCount( ) const;
 
        long long rowCount( ) const;
 
 
#ifdef SIP_RUN
        // Python iterator
        QueryResult *__iter__();
        % MethodCode
        sipRes = sipCpp;
        % End
 
        SIP_PYOBJECT __next__();
        % MethodCode
        QList<QVariant> result;
        Py_BEGIN_ALLOW_THREADS
        result = sipCpp->nextRow( );
        Py_END_ALLOW_THREADS
        if ( ! result.isEmpty() )
        {
          const sipTypeDef *qvariantlist_type = sipFindType( "QList<QVariant>" );
          sipRes = sipConvertFromNewType( new QList<QVariant>( result ), qvariantlist_type, Py_None );
        }
        else
        {
          PyErr_SetString( PyExc_StopIteration, "" );
        }
        % End
#endif
 
 
        struct CORE_EXPORT QueryResultIterator SIP_SKIP
        {
            QVariantList nextRow();
 
            bool hasNextRow() const;
 
            long long fetchedRowCount();
 
            long long rowCount();
 
            virtual ~QueryResultIterator() = default;
 
          private:
 
            virtual QVariantList nextRowPrivate() = 0;
            virtual bool hasNextRowPrivate() const = 0;
            virtual long long rowCountPrivate() const = 0;
 
            mutable qlonglong mFetchedRowCount = 0;
            mutable QMutex mMutex;
 
        };
 
        void appendColumn( const QString &columnName ) SIP_SKIP;
 
        QueryResult( std::shared_ptr<QueryResultIterator> iterator ) SIP_SKIP;
 
        QueryResult( ) = default SIP_SKIP;
 
        double queryExecutionTime( );
 
        void setQueryExecutionTime( double queryExecutionTime );
 
 
      private:
 
        mutable std::shared_ptr<QueryResultIterator> mResultIterator;
        QStringList mColumns;
        double mQueryExecutionTime = 0;
 
    };
 
    struct CORE_EXPORT SqlVectorLayerOptions
    {
      QString sql;
      QString filter;
      QString layerName;
      QStringList primaryKeyColumns;
      QString geometryColumn;
      bool disableSelectAtId = false;
 
    };
 
    struct CORE_EXPORT TableProperty
    {
 
#ifdef SIP_RUN
        SIP_PYOBJECT __repr__();
        % MethodCode
        QString str = QStringLiteral( "<QgsAbstractDatabaseProviderConnection.TableProperty: '%1'>" ).arg( sipCpp->tableName() );
        sipRes = PyUnicode_FromString( str.toUtf8().constData() );
        % End
#endif
 
        struct CORE_EXPORT GeometryColumnType
        {
#ifdef SIP_RUN
          SIP_PYOBJECT __repr__();
          % MethodCode
          QString str = QStringLiteral( "<QgsAbstractDatabaseProviderConnection.TableProperty.GeometryColumnType: '%1, %2'>" ).arg( QgsWkbTypes::displayString( sipCpp->wkbType ), sipCpp->crs.authid() );
          sipRes = PyUnicode_FromString( str.toUtf8().constData() );
          % End
#endif
          QgsWkbTypes::Type wkbType;
          QgsCoordinateReferenceSystem crs;
 
          // TODO c++20 - replace with = default
          inline bool operator==( const GeometryColumnType &other ) const
          {
            return this->crs == other.crs && this->wkbType == other.wkbType;
          }
        };
 
      public:
 
        QString tableName() const;
 
        void setTableName( const QString &name );
 
        void addGeometryColumnType( const QgsWkbTypes::Type &type, const QgsCoordinateReferenceSystem &crs );
 
        QList<QgsAbstractDatabaseProviderConnection::TableProperty::GeometryColumnType> geometryColumnTypes() const;
 
        void setGeometryColumnTypes( const QList<QgsAbstractDatabaseProviderConnection::TableProperty::GeometryColumnType> &geometryColumnTypes );
 
        QString defaultName() const;
 
        TableProperty at( int index ) const;
 
        QString schema() const;
 
        void setSchema( const QString &schema );
 
        QString geometryColumn() const;
 
        void setGeometryColumn( const QString &geometryColumn );
 
        QStringList primaryKeyColumns() const;
 
        void setPrimaryKeyColumns( const QStringList &primaryKeyColumns );
 
        QList<QgsCoordinateReferenceSystem> crsList() const;
 
        TableFlags flags() const;
 
        void setFlags( const TableFlags &flags );
 
        QString comment() const;
 
        void setComment( const QString &comment );
 
        QVariantMap info() const;
 
        void setInfo( const QVariantMap &info );
 
        int geometryColumnCount() const;
 
        void setGeometryColumnCount( int geometryColumnCount );
 
        void setFlag( const TableFlag &flag );
 
        int maxCoordinateDimensions() const;
 
        bool operator==( const QgsAbstractDatabaseProviderConnection::TableProperty &other ) const;
 
      private:
 
        QList<GeometryColumnType>     mGeometryColumnTypes;
        QString                       mSchema;
        QString                       mTableName;
        QString                       mGeometryColumn;
        int                           mGeometryColumnCount;
        QStringList                   mPkColumns;
        TableFlags                    mFlags;
        QString                       mComment;
        QVariantMap                   mInfo;
    };
 
    struct CORE_EXPORT SpatialIndexOptions
    {
      QString geometryColumnName;
    };
 
    enum Capability
    {
      CreateVectorTable = 1 << 1,                     
      DropRasterTable = 1 << 2,                       
      DropVectorTable = 1 << 3,                       
      RenameVectorTable = 1 << 4,                     
      RenameRasterTable = 1 << 5,                     
      CreateSchema = 1 << 6,                          
      DropSchema = 1 << 7,                            
      RenameSchema = 1 << 8,                          
      ExecuteSql = 1 << 9,                            
      Vacuum = 1 << 10,                               
      Tables = 1 << 11,                               
      Schemas = 1 << 12,                              
      SqlLayers = 1 << 13,                            
      TableExists = 1 << 14,                          
      Spatial = 1 << 15,                              
      CreateSpatialIndex = 1 << 16,                   
      SpatialIndexExists = 1 << 17,                   
      DeleteSpatialIndex = 1 << 18,                   
      DeleteField = 1 << 19,                          
      DeleteFieldCascade = 1 << 20,                   
      AddField = 1 << 21,                             
    };
    Q_ENUM( Capability )
    Q_DECLARE_FLAGS( Capabilities, Capability )
    Q_FLAG( Capabilities )
 
    
    enum GeometryColumnCapability
    {
      Z = 1 << 1,                    
      M = 1 << 2,                    
      SinglePart = 1 << 3,           
      Curves = 1 << 4                
    };
 
    Q_ENUM( GeometryColumnCapability )
    Q_DECLARE_FLAGS( GeometryColumnCapabilities, GeometryColumnCapability )
    Q_FLAG( GeometryColumnCapabilities )
 
    
    QgsAbstractDatabaseProviderConnection( const QString &name );
 
    QgsAbstractDatabaseProviderConnection( const QString &uri, const QVariantMap &configuration );
 
 
    // Public interface
 
    Capabilities capabilities() const;
 
    virtual GeometryColumnCapabilities geometryColumnCapabilities();
 
    virtual Qgis::SqlLayerDefinitionCapabilities sqlLayerDefinitionCapabilities();
 
    // Operations interface
 
    virtual QString tableUri( const QString &schema, const QString &name ) const SIP_THROW( QgsProviderConnectionException );
 
    virtual void createVectorTable( const QString &schema, const QString &name, const QgsFields &fields, QgsWkbTypes::Type wkbType, const QgsCoordinateReferenceSystem &srs, bool overwrite, const QMap<QString, QVariant> *options ) const SIP_THROW( QgsProviderConnectionException );
 
    virtual bool tableExists( const QString &schema, const QString &name ) const SIP_THROW( QgsProviderConnectionException );
 
    virtual void dropVectorTable( const QString &schema, const QString &name ) const SIP_THROW( QgsProviderConnectionException );
 
    virtual void dropRasterTable( const QString &schema, const QString &name ) const SIP_THROW( QgsProviderConnectionException );
 
    virtual void renameVectorTable( const QString &schema, const QString &name, const QString &newName ) const SIP_THROW( QgsProviderConnectionException );
 
    virtual void renameRasterTable( const QString &schema, const QString &name, const QString &newName ) const SIP_THROW( QgsProviderConnectionException );
 
    virtual void createSchema( const QString &name ) const SIP_THROW( QgsProviderConnectionException );
 
    virtual void dropSchema( const QString &name, bool force = false ) const SIP_THROW( QgsProviderConnectionException );
 
    virtual void deleteField( const QString &fieldName, const QString &schema, const QString &tableName, bool force = false ) const SIP_THROW( QgsProviderConnectionException );
 
    virtual void addField( const QgsField &field, const QString &schema, const QString &tableName ) const SIP_THROW( QgsProviderConnectionException );
 
    virtual void renameSchema( const QString &name, const QString &newName ) const SIP_THROW( QgsProviderConnectionException );
 
    virtual QList<QList<QVariant>> executeSql( const QString &sql, QgsFeedback *feedback = nullptr ) const SIP_THROW( QgsProviderConnectionException );
 
    virtual QgsVectorLayer *createSqlVectorLayer( const SqlVectorLayerOptions &options ) const SIP_THROW( QgsProviderConnectionException ) SIP_FACTORY;
 
    virtual SqlVectorLayerOptions sqlOptions( const QString &layerSource ) SIP_THROW( QgsProviderConnectionException );
 
    virtual QueryResult execSql( const QString &sql, QgsFeedback *feedback = nullptr ) const SIP_THROW( QgsProviderConnectionException );
 
    virtual void vacuum( const QString &schema, const QString &name ) const SIP_THROW( QgsProviderConnectionException );
 
    virtual void createSpatialIndex( const QString &schema, const QString &name, const QgsAbstractDatabaseProviderConnection::SpatialIndexOptions &options = QgsAbstractDatabaseProviderConnection::SpatialIndexOptions() ) const SIP_THROW( QgsProviderConnectionException );
 
    virtual bool spatialIndexExists( const QString &schema, const QString &name, const QString &geometryColumn ) const SIP_THROW( QgsProviderConnectionException );
 
    virtual void deleteSpatialIndex( const QString &schema, const QString &name, const QString &geometryColumn ) const SIP_THROW( QgsProviderConnectionException );
 
    virtual QList<QgsAbstractDatabaseProviderConnection::TableProperty> tables( const QString &schema = QString(), const QgsAbstractDatabaseProviderConnection::TableFlags &flags = QgsAbstractDatabaseProviderConnection::TableFlags() ) const SIP_SKIP;
 
    virtual QgsAbstractDatabaseProviderConnection::TableProperty table( const QString &schema, const QString &table ) const SIP_THROW( QgsProviderConnectionException );
 
    QList<QgsAbstractDatabaseProviderConnection::TableProperty> tablesInt( const QString &schema = QString(), const int flags = 0 ) const SIP_THROW( QgsProviderConnectionException ) SIP_PYNAME( tables );
 
 
    // TODO: return more schema information and not just the name
 
    virtual QStringList schemas() const SIP_THROW( QgsProviderConnectionException );
 
    virtual QgsFields fields( const QString &schema, const QString &table ) const SIP_THROW( QgsProviderConnectionException );
 
    virtual QList< QgsVectorDataProvider::NativeType > nativeTypes() const SIP_THROW( QgsProviderConnectionException ) = 0;
 
    QString providerKey() const;
 
    virtual QMultiMap<Qgis::SqlKeywordCategory, QStringList> sqlDictionary();
 
  protected:
 
 
    void checkCapability( Capability capability ) const;
 
    Capabilities mCapabilities = Capabilities() SIP_SKIP;
    GeometryColumnCapabilities mGeometryColumnCapabilities = GeometryColumnCapabilities() SIP_SKIP;
    Qgis::SqlLayerDefinitionCapabilities mSqlLayerDefinitionCapabilities = Qgis::SqlLayerDefinitionCapabilities() SIP_SKIP;
    QString mProviderKey;
 
};
 
Q_DECLARE_OPERATORS_FOR_FLAGS( QgsAbstractDatabaseProviderConnection::Capabilities )
 
#endif // QGSABSTRACTDATABASEPROVIDERCONNECTION_H