qgsqueryresultwidget.h

Go to the documentation of this file.

/***************************************************************************
  qgsqueryresultwidget.h - QgsQueryResultWidget
 
 ---------------------
 begin                : 14.1.2021
 copyright            : (C) 2021 by elpaso
 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.                                   *
 *                                                                         *
 ***************************************************************************/
#ifndef QGSQUERYRESULTWIDGET_H
#define QGSQUERYRESULTWIDGET_H
 
#include "qgis_gui.h"
#include "qgis_sip.h"
#include "ui_qgsqueryresultwidgetbase.h"
#include "qgsabstractdatabaseproviderconnection.h"
#include "qgsqueryresultmodel.h"
#include "qgsfeedback.h"
 
#include <QWidget>
#include <QThread>
#include <QtConcurrent>
#include <QStyledItemDelegate>
 
 
#ifndef SIP_RUN
 
class GUI_EXPORT QgsQueryResultItemDelegate: public QStyledItemDelegate
{
    Q_OBJECT
 
    // QStyledItemDelegate interface
  public:
 
    explicit QgsQueryResultItemDelegate( QObject *parent = nullptr );
 
    QString displayText( const QVariant &value, const QLocale &locale ) const override;
};
 
class GUI_EXPORT QgsConnectionsApiFetcher: public QObject
{
    Q_OBJECT
 
  public:
 
    QgsConnectionsApiFetcher( const QgsAbstractDatabaseProviderConnection *connection )
      : mConnection( connection )
    {}
 
    void fetchTokens();
 
    void stopFetching();
 
  signals:
 
    void tokensReady( const QStringList &newTokens );
 
    void fetchingFinished();
 
  private:
 
    const QgsAbstractDatabaseProviderConnection *mConnection = nullptr;
    QAtomicInt mStopFetching = 0;
 
};
 
#endif
 
 
class GUI_EXPORT QgsQueryResultWidget: public QWidget, private Ui::QgsQueryResultWidgetBase
{
    Q_OBJECT
 
  public:
 
 
    enum class QueryWidgetMode : int
    {
      SqlQueryMode = 1 << 0, 
      QueryLayerUpdateMode = 1 << 1, 
    };
    Q_ENUM( QueryWidgetMode )
 
    
    QgsQueryResultWidget( QWidget *parent = nullptr, QgsAbstractDatabaseProviderConnection *connection SIP_TRANSFER = nullptr );
 
    virtual ~QgsQueryResultWidget();
 
    void setSqlVectorLayerOptions( const QgsAbstractDatabaseProviderConnection::SqlVectorLayerOptions &options );
 
    void setWidgetMode( QueryWidgetMode widgetMode );
 
    void setConnection( QgsAbstractDatabaseProviderConnection *connection SIP_TRANSFER );
 
    void setQuery( const QString &sql );
 
 
  public slots:
 
    void notify( const QString &title, const QString &text, Qgis::MessageLevel level = Qgis::MessageLevel::Info );
 
    void executeQuery();
 
    void showError( const QString &title, const QString &message, bool isSqlError = false );
 
    void tokensReady( const QStringList &tokens );
 
  signals:
 
    void createSqlVectorLayer( const QString &providerKey, const QString &connectionUri, const QgsAbstractDatabaseProviderConnection::SqlVectorLayerOptions &options );
 
    void firstResultBatchFetched();
 
  private slots:
 
    void updateButtons();
 
  private:
 
    std::unique_ptr<QgsAbstractDatabaseProviderConnection> mConnection;
    std::unique_ptr<QgsQueryResultModel> mModel;
    std::unique_ptr<QgsFeedback> mFeedback;
    std::unique_ptr<QgsConnectionsApiFetcher> mApiFetcher;
    QThread mApiFetcherWorkerThread;
    bool mWasCanceled = false;
    mutable QgsAbstractDatabaseProviderConnection::SqlVectorLayerOptions mSqlVectorLayerOptions;
    bool mFirstRowFetched = false;
    QFutureWatcher<QgsAbstractDatabaseProviderConnection::QueryResult> mQueryResultWatcher;
    QString mSqlErrorMessage;
    long long mActualRowCount = -1;
    long long mFetchedRowsBatchCount = 0;
    QueryWidgetMode mQueryWidgetMode = QueryWidgetMode::SqlQueryMode;
 
    void updateSqlLayerColumns();
 
    void cancelRunningQuery();
 
    void cancelApiFetcher();
 
    void startFetching();
 
    QgsAbstractDatabaseProviderConnection::SqlVectorLayerOptions sqlVectorLayerOptions() const;
 
 
    friend class TestQgsQueryResultWidget;
 
};
 
#endif // QGSQUERYRESULTWIDGET_H