1 /****************************************************************************
2 **
3 ** Copyright (C) 2016 The Qt Company Ltd.
4 ** Contact: https://www.qt.io/licensing/
5 **
6 ** This file is part of the QtSql module of the Qt Toolkit.
7 **
8 ** $QT_BEGIN_LICENSE:LGPL$
9 ** Commercial License Usage
10 ** Licensees holding valid commercial Qt licenses may use this file in
11 ** accordance with the commercial license agreement provided with the
12 ** Software or, alternatively, in accordance with the terms contained in
13 ** a written agreement between you and The Qt Company. For licensing terms
14 ** and conditions see https://www.qt.io/terms-conditions. For further
15 ** information use the contact form at https://www.qt.io/contact-us.
16 **
17 ** GNU Lesser General Public License Usage
18 ** Alternatively, this file may be used under the terms of the GNU Lesser
19 ** General Public License version 3 as published by the Free Software
20 ** Foundation and appearing in the file LICENSE.LGPL3 included in the
21 ** packaging of this file. Please review the following information to
22 ** ensure the GNU Lesser General Public License version 3 requirements
23 ** will be met: https://www.gnu.org/licenses/lgpl-3.0.html.
24 **
25 ** GNU General Public License Usage
26 ** Alternatively, this file may be used under the terms of the GNU
27 ** General Public License version 2.0 or (at your option) the GNU General
28 ** Public license version 3 or any later version approved by the KDE Free
29 ** Qt Foundation. The licenses are as published by the Free Software
30 ** Foundation and appearing in the file LICENSE.GPL2 and LICENSE.GPL3
31 ** included in the packaging of this file. Please review the following
32 ** information to ensure the GNU General Public License requirements will
33 ** be met: https://www.gnu.org/licenses/gpl-2.0.html and
34 ** https://www.gnu.org/licenses/gpl-3.0.html.
35 **
36 ** $QT_END_LICENSE$
37 **
38 ****************************************************************************/
39 
40 #ifndef QSQLQUERYMODEL_P_H
41 #define QSQLQUERYMODEL_P_H
42 
43 //
44 //  W A R N I N G
45 //  -------------
46 //
47 // This file is not part of the Qt API.  It exists for the convenience
48 // of qsql*model.h .  This header file may change from version to version
49 // without notice, or even be removed.
50 //
51 // We mean it.
52 //
53 
54 #include <QtSql/private/qtsqlglobal_p.h>
55 #include "private/qabstractitemmodel_p.h"
56 #include "QtSql/qsqlerror.h"
57 #include "QtSql/qsqlquery.h"
58 #include "QtSql/qsqlrecord.h"
59 #include "QtCore/qhash.h"
60 #include "QtCore/qvarlengtharray.h"
61 #include "QtCore/qvector.h"
62 
63 QT_REQUIRE_CONFIG(sqlmodel);
64 
65 QT_BEGIN_NAMESPACE
66 
67 class QSqlQueryModelPrivate: public QAbstractItemModelPrivate
68 {
Q_DECLARE_PUBLIC(QSqlQueryModel)69     Q_DECLARE_PUBLIC(QSqlQueryModel)
70 public:
71     QSqlQueryModelPrivate() : atEnd(false), nestedResetLevel(0) {}
72     ~QSqlQueryModelPrivate();
73 
74     void prefetch(int);
75     void initColOffsets(int size);
76     int columnInQuery(int modelColumn) const;
77 
78     mutable QSqlQuery query = { QSqlQuery(nullptr) };
79     mutable QSqlError error;
80     QModelIndex bottom;
81     QSqlRecord rec;
82     uint atEnd : 1;
83     QVector<QHash<int, QVariant> > headers;
84     QVarLengthArray<int, 56> colOffsets; // used to calculate indexInQuery of columns
85     int nestedResetLevel;
86 };
87 
88 // helpers for building SQL expressions
89 class QSqlQueryModelSql
90 {
91 public:
92     // SQL keywords
as()93     inline const static QLatin1String as() { return QLatin1String("AS"); }
asc()94     inline const static QLatin1String asc() { return QLatin1String("ASC"); }
comma()95     inline const static QLatin1String comma() { return QLatin1String(","); }
desc()96     inline const static QLatin1String desc() { return QLatin1String("DESC"); }
eq()97     inline const static QLatin1String eq() { return QLatin1String("="); }
98     // "and" is a C++ keyword
et()99     inline const static QLatin1String et() { return QLatin1String("AND"); }
from()100     inline const static QLatin1String from() { return QLatin1String("FROM"); }
leftJoin()101     inline const static QLatin1String leftJoin() { return QLatin1String("LEFT JOIN"); }
on()102     inline const static QLatin1String on() { return QLatin1String("ON"); }
orderBy()103     inline const static QLatin1String orderBy() { return QLatin1String("ORDER BY"); }
parenClose()104     inline const static QLatin1String parenClose() { return QLatin1String(")"); }
parenOpen()105     inline const static QLatin1String parenOpen() { return QLatin1String("("); }
select()106     inline const static QLatin1String select() { return QLatin1String("SELECT"); }
sp()107     inline const static QLatin1String sp() { return QLatin1String(" "); }
where()108     inline const static QLatin1String where() { return QLatin1String("WHERE"); }
109 
110     // Build expressions based on key words
as(const QString & a,const QString & b)111     inline const static QString as(const QString &a, const QString &b) { return b.isEmpty() ? a : concat(concat(a, as()), b); }
asc(const QString & s)112     inline const static QString asc(const QString &s) { return concat(s, asc()); }
comma(const QString & a,const QString & b)113     inline const static QString comma(const QString &a, const QString &b) { return a.isEmpty() ? b : b.isEmpty() ? a : QString(a).append(comma()).append(b); }
concat(const QString & a,const QString & b)114     inline const static QString concat(const QString &a, const QString &b) { return a.isEmpty() ? b : b.isEmpty() ? a : QString(a).append(sp()).append(b); }
desc(const QString & s)115     inline const static QString desc(const QString &s) { return concat(s, desc()); }
eq(const QString & a,const QString & b)116     inline const static QString eq(const QString &a, const QString &b) { return QString(a).append(eq()).append(b); }
et(const QString & a,const QString & b)117     inline const static QString et(const QString &a, const QString &b) { return a.isEmpty() ? b : b.isEmpty() ? a : concat(concat(a, et()), b); }
from(const QString & s)118     inline const static QString from(const QString &s) { return concat(from(), s); }
leftJoin(const QString & s)119     inline const static QString leftJoin(const QString &s) { return concat(leftJoin(), s); }
on(const QString & s)120     inline const static QString on(const QString &s) { return concat(on(), s); }
orderBy(const QString & s)121     inline const static QString orderBy(const QString &s) { return s.isEmpty() ? s : concat(orderBy(), s); }
paren(const QString & s)122     inline const static QString paren(const QString &s) { return s.isEmpty() ? s : parenOpen() + s + parenClose(); }
select(const QString & s)123     inline const static QString select(const QString &s) { return concat(select(), s); }
where(const QString & s)124     inline const static QString where(const QString &s) { return s.isEmpty() ? s : concat(where(), s); }
125 };
126 
127 QT_END_NAMESPACE
128 
129 #endif // QSQLQUERYMODEL_P_H
130