1// Copyright (C) 2016 The Qt Company Ltd.
2// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR LGPL-3.0-only OR GPL-2.0-only OR GPL-3.0-only
3
4#ifndef QSQLTABLEMODEL_P_H
5#define QSQLTABLEMODEL_P_H
6
7//
8// W A R N I N G
9// -------------
10//
11// This file is not part of the Qt API. It exists for the convenience
12// of qsql*model.h . This header file may change from version to version
13// without notice, or even be removed.
14//
15// We mean it.
16//
17
18#include <QtSql/private/qtsqlglobal_p.h>
19#include "private/qsqlquerymodel_p.h"
20#include "QtSql/qsqlindex.h"
21#include "QtCore/qmap.h"
22
23QT_REQUIRE_CONFIG(sqlmodel);
24
25QT_BEGIN_NAMESPACE
26
27class Q_AUTOTEST_EXPORT QSqlTableModelPrivate: public QSqlQueryModelPrivate
28{
29 Q_DECLARE_PUBLIC(QSqlTableModel)
30
31public:
32 QSqlTableModelPrivate()
33 : sortColumn(-1),
34 sortOrder(Qt::AscendingOrder),
35 strategy(QSqlTableModel::OnRowChange),
36 busyInsertingRows(false)
37 {}
38
39 void clear();
40 virtual void clearCache();
41 QSqlRecord record(const QList<QVariant> &values) const;
42
43 bool exec(const QString &stmt, bool prepStatement,
44 const QSqlRecord &rec, const QSqlRecord &whereValues);
45 virtual void revertCachedRow(int row);
46 virtual int nameToIndex(const QString &name) const;
47 QString strippedFieldName(const QString &name) const;
48 int insertCount(int maxRow = -1) const;
49 void initRecordAndPrimaryIndex();
50
51 QSqlDatabase db;
52
53 int sortColumn;
54 Qt::SortOrder sortOrder;
55
56 QSqlTableModel::EditStrategy strategy;
57 bool busyInsertingRows;
58
59 QSqlQuery editQuery = { QSqlQuery(nullptr) };
60 QSqlIndex primaryIndex;
61 QString tableName;
62 QString filter;
63 QString autoColumn;
64
65 enum Op { None, Insert, Update, Delete };
66
67 class ModifiedRow
68 {
69 public:
70 inline ModifiedRow(Op o = None, const QSqlRecord &r = QSqlRecord())
71 : m_op(None), m_db_values(r), m_insert(o == Insert)
72 { setOp(o); }
73 inline Op op() const { return m_op; }
74 inline void setOp(Op o)
75 {
76 if (o == None)
77 m_submitted = true;
78 if (o == m_op)
79 return;
80 m_submitted = (o != Insert && o != Delete);
81 m_op = o;
82 m_rec = m_db_values;
83 setGenerated(r&: m_rec, g: m_op == Delete);
84 }
85 inline const QSqlRecord &rec() const { return m_rec; }
86 inline QSqlRecord& recRef() { return m_rec; }
87 inline void setValue(int c, const QVariant &v)
88 {
89 m_submitted = false;
90 m_rec.setValue(i: c, val: v);
91 m_rec.setGenerated(i: c, generated: true);
92 }
93 inline bool submitted() const { return m_submitted; }
94 inline void setSubmitted()
95 {
96 m_submitted = true;
97 setGenerated(r&: m_rec, g: false);
98 if (m_op == Delete) {
99 m_rec.clearValues();
100 }
101 else {
102 m_op = Update;
103 m_db_values = m_rec;
104 setGenerated(r&: m_db_values, g: true);
105 }
106 }
107 inline void refresh(bool exists, const QSqlRecord& newvals)
108 {
109 m_submitted = true;
110 if (exists) {
111 m_op = Update;
112 m_db_values = newvals;
113 m_rec = newvals;
114 setGenerated(r&: m_rec, g: false);
115 } else {
116 m_op = Delete;
117 m_rec.clear();
118 m_db_values.clear();
119 }
120 }
121 inline bool insert() const { return m_insert; }
122 inline void revert()
123 {
124 if (m_submitted)
125 return;
126 if (m_op == Delete)
127 m_op = Update;
128 m_rec = m_db_values;
129 setGenerated(r&: m_rec, g: false);
130 m_submitted = true;
131 }
132 inline QSqlRecord primaryValues(const QSqlRecord& pi) const
133 {
134 if (m_op == None || m_op == Insert)
135 return QSqlRecord();
136
137 return m_db_values.keyValues(keyFields: pi);
138 }
139 private:
140 inline static void setGenerated(QSqlRecord& r, bool g)
141 {
142 for (int i = r.count() - 1; i >= 0; --i)
143 r.setGenerated(i, generated: g);
144 }
145 Op m_op;
146 QSqlRecord m_rec;
147 QSqlRecord m_db_values;
148 bool m_submitted;
149 bool m_insert;
150 };
151
152 typedef QMap<int, ModifiedRow> CacheMap;
153 CacheMap cache;
154};
155
156QT_END_NAMESPACE
157
158#endif // QSQLTABLEMODEL_P_H
159

source code of qtbase/src/sql/models/qsqltablemodel_p.h