Improvements to the CrudModel

The new data of modified rows is now stored directly within the row_mapping
also changed how new rows are handled so the new empty row for inserting
is not a special case but is part of the list.
This commit is contained in:
eelke 2022-01-22 16:22:29 +01:00
parent a7f247bdee
commit 06504ecc1f
9 changed files with 213 additions and 317 deletions

View file

@ -1,6 +1,6 @@
#include "DatabaseWindow.h" #include "DatabaseWindow.h"
#include "util.h" #include "util.h"
#include "CrudTab.h" #include "crud/CrudTab.h"
#include "widgets/CatalogTablesPage.h" #include "widgets/CatalogTablesPage.h"
#include "OpenDatabase.h" #include "OpenDatabase.h"
#include "catalog/PgDatabaseCatalog.h" #include "catalog/PgDatabaseCatalog.h"

View file

@ -9,6 +9,7 @@
#include <QtConcurrent> #include <QtConcurrent>
#include <QFuture> #include <QFuture>
#include <QFutureWatcher> #include <QFutureWatcher>
#include <QMessageBox>
#include "Pgsql_oids.h" #include "Pgsql_oids.h"
#include "Pgsql_PgException.h" #include "Pgsql_PgException.h"
#include "Pgsql_Params.h" #include "Pgsql_Params.h"
@ -62,16 +63,9 @@ QVariant CrudModel::headerData(int section, Qt::Orientation orientation, int rol
return r; return r;
} }
// Basic functionality:
int CrudModel::rowCount(const QModelIndex &/*parent*/) const int CrudModel::rowCount(const QModelIndex &/*parent*/) const
{ {
// int row_count = m_roData ? m_roData->rows() : 0; return (int)m_rowMapping.size();
// todo there will be rownumbers that are not in m_roData
// ++row_count; // one empty new row at the end
return m_rowCount;
} }
int CrudModel::columnCount(const QModelIndex &/*parent*/) const int CrudModel::columnCount(const QModelIndex &/*parent*/) const
@ -85,52 +79,56 @@ Oid CrudModel::getType(int column) const
return m_roData ? m_roData->type(column) : InvalidOid; return m_roData ? m_roData->type(column) : InvalidOid;
} }
CrudModel::Value CrudModel::getData(const QModelIndex &index) const CrudModel::Value CrudModel::getLatestData(int columnIndex, int rowIndex) const
{ {
Value value;
if (m_roData) { if (m_roData) {
int grid_row = index.row(); auto row_mapping = m_rowMapping[rowIndex];
int col = index.column();
auto row_mapping = m_rowMapping[grid_row];
const int last_row = rowCount() - 1;
//Oid o = m_roData->type(col);
// First see if we have buffered editted values that still need saving
std::optional<Value> val; std::optional<Value> val;
if (row_mapping.pending) { if (row_mapping.pending) {
val = m_pendingRowList.getValue(col, row_mapping.rowKey); val = m_pendingRowList.getValue(columnIndex, row_mapping.rowKey);
} }
if (row_mapping.modified && !val && grid_row < last_row) { // last_row should never be in modified list, when it is put in modified list a new last row should be created if (!val.has_value())
// No pending save have a look if we have modified saved data in the modified list {
auto find_res = m_modifiedRowList.find(row_mapping.rowKey); val = getSavedData(row_mapping, columnIndex);
if (find_res != m_modifiedRowList.end()) {
val = find_res->second.data()[col];
} }
return val.value_or(Value());
}
return {};
}
CrudModel::Value CrudModel::getSavedData(int columnIndex, int rowIndex) const
{
if (m_roData) {
auto row_mapping = m_rowMapping[rowIndex];
return getSavedData(row_mapping, columnIndex);
}
return {};
}
CrudModel::Value CrudModel::getSavedData(const RowMapping &row_mapping, int columnIndex) const
{
// First see if we have buffered editted values that still need saving
std::optional<Value> val;
if (!val.has_value() && row_mapping.isModified()) {
val = row_mapping.modifiedValue(columnIndex);
} }
//Value value;
// If we did not have pending or modified data // If we did not have pending or modified data
if (!val && row_mapping.rowKey < m_roData->rows()) { if (!val.has_value() && row_mapping.rowKey < m_roData->rows()) {
// Then we are going to read the original data. // Then we are going to read the original data.
if (!m_roData->null(col, row_mapping.rowKey)) { if (!m_roData->null(columnIndex, row_mapping.rowKey)) {
value = std::string(m_roData->val(col, row_mapping.rowKey)); val = std::string(m_roData->val(columnIndex, row_mapping.rowKey));
} }
} }
else { return val.value_or(Value());
if (val) {
value = *val;
}
}
}
return value;
} }
QVariant CrudModel::data(const QModelIndex &index, int role) const QVariant CrudModel::data(const QModelIndex &index, int role) const
{ {
QVariant v; QVariant v;
if (role == Qt::EditRole) { if (role == Qt::EditRole) {
auto value = getData(index); auto value = getLatestData(index);
if (value) { if (value) {
QString s = QString::fromUtf8(value->c_str()); QString s = QString::fromUtf8(value->c_str());
v = s; v = s;
@ -138,7 +136,7 @@ QVariant CrudModel::data(const QModelIndex &index, int role) const
} }
else if (role == Qt::DisplayRole) { else if (role == Qt::DisplayRole) {
auto value = getData(index); auto value = getLatestData(index);
if (value) { if (value) {
Oid o = m_roData->type(index.column()); Oid o = m_roData->type(index.column());
if (o == Pgsql::bool_oid) { if (o == Pgsql::bool_oid) {
@ -159,7 +157,6 @@ QVariant CrudModel::data(const QModelIndex &index, int role) const
return v; return v;
} }
void CrudModel::loadData() void CrudModel::loadData()
{ {
QString table_name = m_table->fullyQualifiedQuotedObjectName(); // genFQTableName(*m_database->catalog(), *m_table); QString table_name = m_table->fullyQualifiedQuotedObjectName(); // genFQTableName(*m_database->catalog(), *m_table);
@ -180,10 +177,8 @@ void CrudModel::loadIntoModel(std::shared_ptr<Pgsql::Result> data)
{ {
beginResetModel(); beginResetModel();
m_pendingRowList.clear(); m_pendingRowList.clear();
m_modifiedRowList.clear();
m_roData = data; m_roData = data;
lastRowKey = data->rows(); lastRowKey = data->rows() - 1;
m_rowCount = data->rows();
initRowMapping(); initRowMapping();
appendNewRow(); appendNewRow();
endResetModel(); endResetModel();
@ -191,9 +186,10 @@ void CrudModel::loadIntoModel(std::shared_ptr<Pgsql::Result> data)
void CrudModel::initRowMapping() void CrudModel::initRowMapping()
{ {
m_rowMapping.resize(static_cast<size_t>(m_rowCount)); size_t cnt = m_roData->rows();
for (int i = 0; i < m_rowCount; ++i) m_rowMapping.reserve(cnt + 1);
m_rowMapping[i] = RowMapping{ i }; for (int i = 0; i < cnt; ++i)
m_rowMapping.emplace_back(i);
} }
void CrudModel::connectionStateChanged(ASyncDBConnection::State state) void CrudModel::connectionStateChanged(ASyncDBConnection::State state)
@ -252,63 +248,68 @@ bool CrudModel::setData(const QModelIndex &index, const QVariant &value, int rol
return false; return false;
} }
const CrudModel::ModifiedRow* CrudModel::getModifiedRow(int row) const std::tuple<bool, std::vector<CrudModel::Value>> CrudModel::saveRow(const PendingRow &pending_row)
{ {
auto iter = m_modifiedRowList.find(row); auto data = pending_row.data();
if (iter == m_modifiedRowList.end()) RowMapping& rowmapping = m_rowMapping[pending_row.row()];
return nullptr;
if (!data.empty()) {
QString buffer;
Pgsql::Params params;
if (rowmapping.isNew()){
std::tie(buffer, params) = createInsertQuery(pending_row);
}
else {
Pgsql::Params pkey_params = getPKeyParamsForRow(pending_row.row());
std::tie(buffer, params) = createUpdateQuery(pkey_params, pending_row);
}
Pgsql::Connection db_update_conn;
auto dbconfig = m_database->config();
db_update_conn.connect(dbconfig.connectionString());
try {
auto result = db_update_conn.queryParam(buffer, params);
if (result && result.rows() == 1) {
std::vector<Value> values;
auto row = *result.begin();
for (auto v : row) {
if (v.null())
values.push_back(Value());
else else
return &iter->second; values.push_back(std::string(v.c_str()));
} }
CrudModel::PKeyValues CrudModel::getPKeyForRow(int row) const return { true, values };
{
PKeyValues values;
values.reserve(m_primaryKey->key.size());
auto mod_row = getModifiedRow(row);
if (mod_row){
for (auto attnum : m_primaryKey->key) {
const int col = attNumToCol(attnum);
values.push_back(*(mod_row->data()[col]));
} }
} }
else if (row < m_roData->rows()){ catch (const Pgsql::PgResultError &ex) {
for (auto attnum : m_primaryKey->key) { QMessageBox msgBox;
int col = attNumToCol(attnum); msgBox.setText(ex.what());
values.push_back(m_roData->get(col, row).c_str()); msgBox.exec();
} }
} }
return { false, {} };
return values;
} }
Pgsql::Params CrudModel::getPKeyParamsForRow(int row) const Pgsql::Params CrudModel::getPKeyParamsForRow(int row) const
{ {
Pgsql::Params params; Pgsql::Params params;
auto mod_row = getModifiedRow(row);
for (auto attnum : m_primaryKey->key) { for (auto attnum : m_primaryKey->key) {
const int col = attNumToCol(attnum); const int col = attNumToCol(attnum);
Oid t = getType(col); Oid t = getType(col);
std::string s; auto s = getSavedData(col, row);
if (mod_row){
s = *(mod_row->data()[col]);
}
else if (row < m_roData->rows()){
s = m_roData->get(col, row).c_str();
}
params.add(s, t); params.add(s, t);
} }
return params; return params;
} }
QString CrudModel::columnName(int col) const QString CrudModel::columnName(int col) const
{ {
return m_roData->getColName(col); return m_roData->getColName(col);
} }
std::tuple<QString, Pgsql::Params> CrudModel::createUpdateQuery(const PKeyValues &pkey_values, const PendingRow &pending_row) std::tuple<QString, Pgsql::Params> CrudModel::createUpdateQuery(const Pgsql::Params &pkey_params, const PendingRow &pending_row)
{ {
Pgsql::Params params; Pgsql::Params params;
auto data = pending_row.data(); auto data = pending_row.data();
@ -317,7 +318,7 @@ std::tuple<QString, Pgsql::Params> CrudModel::createUpdateQuery(const PKeyValues
QTextStream q(&buffer); QTextStream q(&buffer);
q << "UPDATE " << table_name << " AS d\n SET "; q << "UPDATE " << table_name << " AS d\n SET ";
int param = 0; int param = 0;
for (auto e : data) { for (auto& e : data) {
if (param > 0) if (param > 0)
q << ","; q << ",";
q << quoteIdent(columnName(e.first)) << "=$" << ++param; q << quoteIdent(columnName(e.first)) << "=$" << ++param;
@ -333,10 +334,9 @@ std::tuple<QString, Pgsql::Params> CrudModel::createUpdateQuery(const PKeyValues
if (i > 0) if (i > 0)
q << " AND "; q << " AND ";
q << quoteIdent(columnName(col)) << "=$" << ++param; q << quoteIdent(columnName(col)) << "=$" << ++param;
params.add(pkey_values[i].c_str(), getType(col));
++i; ++i;
} }
params.addParams(pkey_params);
q << "\nRETURNING *"; q << "\nRETURNING *";
q.flush(); q.flush();
return { buffer, params }; return { buffer, params };
@ -353,7 +353,7 @@ std::tuple<QString, Pgsql::Params> CrudModel::createInsertQuery(const PendingRow
auto columns = m_database->catalog()->attributes()->getColumnsForRelation(m_table->oid()); auto columns = m_database->catalog()->attributes()->getColumnsForRelation(m_table->oid());
bool first = true; bool first = true;
for (auto e : data) { for (const auto& e : data) {
int num = e.first + 1; int num = e.first + 1;
auto find_res = std::find_if(columns.begin(), columns.end(), auto find_res = std::find_if(columns.begin(), columns.end(),
[num] (const auto &elem) -> bool { return num == elem.num; }); [num] (const auto &elem) -> bool { return num == elem.num; });
@ -368,8 +368,6 @@ std::tuple<QString, Pgsql::Params> CrudModel::createInsertQuery(const PendingRow
q << ",$" << p; q << ",$" << p;
q << ") RETURNING *"; q << ") RETURNING *";
for (auto e : data) { for (auto e : data) {
// Add value to paramlist
params.add(e.second, getType(e.first)); params.add(e.second, getType(e.first));
} }
q.flush(); q.flush();
@ -410,74 +408,25 @@ QString CrudModel::createDeleteStatement() const
} }
std::tuple<bool, CrudModel::ModifiedRow> CrudModel::updateRow(const PendingRow &pending_row)
{
auto data = pending_row.data();
if (!data.empty()) {
auto pkey_values = getPKeyForRow(pending_row.row());
QString buffer;
Pgsql::Params params;
if (pkey_values.empty()){
std::tie(buffer, params) = createInsertQuery(pending_row);
}
else {
std::tie(buffer, params) = createUpdateQuery(pkey_values, pending_row);
}
int row_number = pending_row.row();
Pgsql::Connection db_update_conn;
auto dbconfig = m_database->config();
db_update_conn.connect(dbconfig.connectionString());
try {
auto result = db_update_conn.queryParam(buffer, params);
if (result && result.rows() == 1) {
std::vector<Value> values;
auto row = *result.begin();
for (auto v : row) {
if (v.null())
values.push_back(Value());
else
values.push_back(std::string(v.c_str()));
}
ModifiedRow modified_row(row_number, values);
return { true, modified_row };
}
}
catch (const Pgsql::PgResultError &ex) {
//Mes
}
}
return { false, {} };
}
bool CrudModel::savePendingChanges() bool CrudModel::savePendingChanges()
{ {
while (!m_pendingRowList.m_rows.empty()) { while (!m_pendingRowList.m_rows.empty()) {
auto iter = m_pendingRowList.m_rows.begin(); auto iter = m_pendingRowList.m_rows.begin();
auto [ok, modified_row] = updateRow(iter->second); auto [ok, modified_row] = saveRow(iter->second);
if (ok) { if (ok) {
int rowKey = iter->first; int rowKey = iter->first;
m_modifiedRowList.insert_or_assign(rowKey, modified_row);
m_pendingRowList.m_rows.erase(iter); m_pendingRowList.m_rows.erase(iter);
auto iter = std::find_if(m_rowMapping.begin(), m_rowMapping.end(), auto iter = std::find_if(m_rowMapping.begin(), m_rowMapping.end(),
[rowKey](const RowMapping &rhs) -> auto { return rhs.rowKey == rowKey; }); [rowKey](const RowMapping &rhs) -> bool { return rhs.rowKey == rowKey; });
if (iter != m_rowMapping.end()) { if (iter != m_rowMapping.end()) {
iter->modified = true; iter->setModifiedRowData(modified_row);
iter->pending = false; iter->pending = false;
int row = iter - m_rowMapping.begin(); if (IsLastRow(iter))
if (row == m_rowCount - 1) {
appendNewRow(); appendNewRow();
} }
}
} }
else { else {
return false; return false;
@ -487,6 +436,11 @@ bool CrudModel::savePendingChanges()
return true; return true;
} }
bool CrudModel::IsLastRow(RowMappingVector::iterator mapping_iter) const
{
return mapping_iter == --m_rowMapping.end();
}
bool CrudModel::submit() bool CrudModel::submit()
{ {
return savePendingChanges(); return savePendingChanges();
@ -499,15 +453,12 @@ void CrudModel::revert()
void CrudModel::appendNewRow() void CrudModel::appendNewRow()
{ {
int row = m_rowCount; int row = m_rowMapping.size();
beginInsertRows(QModelIndex(), row, row); beginInsertRows(QModelIndex(), row, row);
m_rowMapping.emplace_back(allocNewRowKey()); m_rowMapping.emplace_back(allocNewRowKey(), std::vector<Value>(m_roData->cols()));
++m_rowCount;
endInsertRows(); endInsertRows();
} }
std::tuple<bool, QString> CrudModel::removeRows(const std::set<IntegerRange<int>> &row_ranges) std::tuple<bool, QString> CrudModel::removeRows(const std::set<IntegerRange<int>> &row_ranges)
{ {
if (row_ranges.empty()) return { true, "" }; if (row_ranges.empty()) return { true, "" };
@ -545,29 +496,7 @@ std::tuple<bool, QString> CrudModel::removeRows(const std::set<IntegerRange<int>
// If something goes wrong after this commit we should reload contents of model // If something goes wrong after this commit we should reload contents of model
} }
// Then from model // Then from model
{ RemoveRangesOfRowsFromModel(row_ranges);
int rows_deleted = 0;
for (auto range : row_ranges) {
range.setStart(range.start() - rows_deleted);
beginRemoveRows(QModelIndex(), range.start(), range.end() - 1);
SCOPE_EXIT { endRemoveRows(); };
for (int current_row = range.start(); current_row < range.end(); ++current_row) {
auto&& mapping = m_rowMapping[static_cast<size_t>(current_row)];
// if the row is in modified it can be removed from modified
if (mapping.modified) {
m_modifiedRowList.erase(mapping.rowKey);
}
/// \todo can it be pending? should be removed if it is.
}
// remove the rows from m_rowMapping
auto first = m_rowMapping.begin() + range.start();
m_rowMapping.erase(first, first + range.length());
rows_deleted += range.length();
}
}
return { true, "" }; return { true, "" };
} catch (const Pgsql::PgResultError &error) { } catch (const Pgsql::PgResultError &error) {
return { false, QString::fromUtf8(error.details().messageDetail.c_str()) }; return { false, QString::fromUtf8(error.details().messageDetail.c_str()) };
@ -583,4 +512,21 @@ bool CrudModel::removeRows(int row, int count, const QModelIndex &)
return res; return res;
} }
void CrudModel::RemoveRangesOfRowsFromModel(const std::set<IntegerRange<int>> &row_ranges)
{
int rows_deleted = 0;
for (auto range : row_ranges) {
range.setStart(range.start() - rows_deleted);
RemoveRangeOfRowsFromModel(range);
rows_deleted += range.length();
}
}
void CrudModel::RemoveRangeOfRowsFromModel(IntegerRange<int> range)
{
beginRemoveRows(QModelIndex(), range.start(), range.end() - 1);
SCOPE_EXIT { endRemoveRows(); };
auto first = m_rowMapping.begin() + range.start();
m_rowMapping.erase(first, first + range.length());
}

View file

@ -1,14 +1,13 @@
#ifndef CRUDMODEL_H #ifndef CRUDMODEL_H
#define CRUDMODEL_H #define CRUDMODEL_H
#include <QAbstractTableModel>
#include "ASyncDBConnection.h" #include "ASyncDBConnection.h"
#include "Pgsql_Connection.h" #include "Pgsql_Connection.h"
#include "IntegerRange.h" #include "IntegerRange.h"
#include "catalog/PgClass.h" #include "catalog/PgClass.h"
#include "catalog/PgConstraint.h" #include "catalog/PgConstraint.h"
#include "Pgsql_Connection.h" #include <QAbstractTableModel>
#include <map> #include <map>
#include <memory> #include <memory>
#include <set> #include <set>
@ -19,38 +18,6 @@
class PgConstraint; class PgConstraint;
class OpenDatabase; class OpenDatabase;
/**
* @brief The CrudModel class
*
* Features
* - order by one or more user selectable columns
* - user filter condition
* - user limit and offset
* - hide columns (will not be retrieved can greatly speed up things when you disable wide columns)
* - can use foreign keys to display dropdown
*
*
* Need to keep track of changes from the original. Things to track
* - new data being entered for existing row
* - new data for new row
* - current data of modified row
* - current data of new row
* - which rows are deleted
* We need to be able to:
* - determine row count easily
* - find specific row from table easily
* - support resorting (without requerying)
* //- support custom filtering (without requerying)
*
* Keep data as much in original result
* - use redirect list to go from index to actual row in the result
* - use an additional map index by original row number containing changed data
* - might be beneficial to have vector<bool> in sync with original list to signal changes from the result
* this could prevent doing a lot of slower searches in the changed row set by only doing this when the direty markes
* has been set vector<bool> takes 1 bit per value making it very efficient and fast. We also could put a bit in the mapping
* vector but this probably would double the size of that list taking 32 times the size of vector<bool>.
*
*/
class CrudModel: public QAbstractTableModel { class CrudModel: public QAbstractTableModel {
Q_OBJECT Q_OBJECT
public: public:
@ -110,41 +77,8 @@ private:
}; };
// using RowData = std::vector<std::string>;
// using RowDataPtr = std::unique_ptr<RowData>;
// /** Choosen for a vector of pointers while vector is relatively slow with insertion and removal in the middle
// * I hope the small size of the elements will make this not much of an issue while the simple linear and sequential
// * nature of a vector will keep memory allocation fairly efficient.
// */
// using RowList = std::vector<RowDataPtr>;
// std::shared_ptr<RowList> resultToRowList(std::shared_ptr<Pgsql::Result> result);
using Value = std::optional<std::string>; using Value = std::optional<std::string>;
/** Used to remember the changes that have been made to rows.
*
* For simplicity it also holds the values for columns that have not been changed.
*/
class ModifiedRow {
public:
ModifiedRow() = default;
ModifiedRow(int row, const std::vector<Value> &values)
: m_row(row), m_values(values)
{}
ModifiedRow(int row, const std::vector<Value> &&values)
: m_row(row), m_values(values)
{}
const auto& data() const { return m_values; }
int row() const { return m_row; }
private:
int m_row = -1;
std::vector<Value> m_values;
};
using ModifiedRowList = std::map<int, ModifiedRow>;
/** Similar to a modified row but it only stores values for columns that actually have been edited. /** Similar to a modified row but it only stores values for columns that actually have been edited.
* *
*/ */
@ -218,16 +152,52 @@ private:
class RowMapping { class RowMapping {
public: public:
bool modified = false; ///< Row has been modified, look there for actual data
bool pending = false; ///< Row has pending changes, look first in pending for current data bool pending = false; ///< Row has pending changes, look first in pending for current data
int rowKey = -1; ///< value to use as key/index into the data lists int rowKey = -1; ///< value to use as key/index into the data lists
RowMapping() = default; RowMapping() = default;
explicit RowMapping(int row_key) explicit RowMapping(int row_key, std::vector<Value> data)
: rowKey(row_key)
, modifiedRow(std::move(data))
{}
RowMapping(int row_key)
: rowKey(row_key) : rowKey(row_key)
{} {}
bool isModified() const
{
return !modifiedRow.empty();
}
Value modifiedValue(int columnIndex) const
{
return modifiedRow[columnIndex];
}
void setModifiedRowData(std::vector<Value> data)
{
modifiedRow = std::move(data);
}
bool isNew() const
{
// row in m_roData, thus it is not new
if (modifiedRow.empty())
return false;
// if all elements are empty then this is a new row
for (auto &c : modifiedRow)
if (c.has_value())
return false;
return true;
}
private:
std::vector<Value> modifiedRow;
}; };
using RowMappingVector = std::vector<RowMapping>; using RowMappingVector = std::vector<RowMapping>;
std::shared_ptr<OpenDatabase> m_database; std::shared_ptr<OpenDatabase> m_database;
@ -238,76 +208,52 @@ private:
bool callLoadData = false; bool callLoadData = false;
std::shared_ptr<Pgsql::Result> m_roData; std::shared_ptr<Pgsql::Result> m_roData;
/// \brief Total number of rows to show.
///
/// The first n rows are the rows in m_roData.
/// While a row is being edited the modifications are in the PendingRowList
/// When the edits have been submitted the new values are put in the ModifiedRowList
///
/// The last row is for creating new rows. There is no data for this row or it is
/// in the PendingRowList. As soon as the row has been inserted a new last empty row
/// will be created by increasing this variable. And the newly stored row will be in
/// the modified list.
///
int m_rowCount = 0;
PendingRowList m_pendingRowList; PendingRowList m_pendingRowList;
/// \brief Maintains a list of all modified rows.
///
/// The key values are the indexes of the row before any rows were deleted.
///
ModifiedRowList m_modifiedRowList;
/// Keeps track of mapping grid rows to data rows /// Keeps track of mapping grid rows to data rows
RowMappingVector m_rowMapping; RowMappingVector m_rowMapping;
/// call on initial load to fill in the mappings /// call on initial load to fill in the mappings
void initRowMapping(); void initRowMapping();
Value getData(const QModelIndex &index) const; Value getLatestData(int column, int row) const;
Value getLatestData(const QModelIndex &index) const
{
return getLatestData(index.column(), index.row());
}
Value getSavedData(int columnIndex, int rowIndex) const;
Value getSavedData(const RowMapping &row_mapping, int columnIndex) const;
Oid getType(int column) const; Oid getType(int column) const;
///
/// \brief columnName
/// \param col
/// \return
///
QString columnName(int col) const; QString columnName(int col) const;
/// /// Get the PKey values from when the row was last restored or retrieved.
/// \brief getModifiedRow searches for the specified row in the modified set
/// \param row
/// \return Pointer to the modified element or nullptr if there is no modification
///
const ModifiedRow* getModifiedRow(int row) const;
///
/// \brief getPKeyForRow retrieve the primary key of the specified row
///
/// This function should not be called when there is no primarykey. (Editing should be disabled anyway in that case)
///
/// \param row Actual result row not intermeddiat model row
/// \return The values of the primary key column. If this is a new row it will return an empty list
///
PKeyValues getPKeyForRow(int row) const;
Pgsql::Params getPKeyParamsForRow(int row) const; Pgsql::Params getPKeyParamsForRow(int row) const;
bool savePendingChanges(); bool savePendingChanges();
std::tuple<QString, Pgsql::Params> createUpdateQuery(const PKeyValues &pkey_values, const PendingRow &pending_row); std::tuple<QString, Pgsql::Params> createUpdateQuery(const Pgsql::Params &pkey_params, const PendingRow &pending_row);
std::tuple<QString, Pgsql::Params> createInsertQuery(const PendingRow &pending_row); std::tuple<QString, Pgsql::Params> createInsertQuery(const PendingRow &pending_row);
std::tuple<QString, Pgsql::Params> createDeleteStatement(const PKeyValues &pkey_values); std::tuple<QString, Pgsql::Params> createDeleteStatement(const PKeyValues &pkey_values);
QString createDeleteStatement() const; QString createDeleteStatement() const;
std::tuple<bool, ModifiedRow> updateRow(const PendingRow &pending_row); std::tuple<bool, std::vector<Value>> saveRow(const PendingRow &pending_row);
void appendNewRow(); void appendNewRow();
int lastRowKey = 0; int lastRowKey = -1;
int allocNewRowKey() { return ++lastRowKey; } int allocNewRowKey() { return ++lastRowKey; }
/// Convert an attnum from the database catalog to the corresponding column in the model /// Convert an attnum from the database catalog to the corresponding column in the model
/// ///
/// \todo still assumes columns are in order, all being shown and no special column like oid shown. /// \todo still assumes columns are in order, all being shown and no special column like oid shown.
int attNumToCol(int attnum) const { return attnum - 1; } int attNumToCol(int attnum) const { return attnum - 1; }
void RemoveRangesOfRowsFromModel(const std::set<IntegerRange<int>> &row_ranges);
void RemoveRangeOfRowsFromModel(IntegerRange<int> row_range);
bool IsLastRow(RowMappingVector::iterator mapping_iter) const;
private slots: private slots:
void loadIntoModel(std::shared_ptr<Pgsql::Result> data); void loadIntoModel(std::shared_ptr<Pgsql::Result> data);

View file

@ -16,8 +16,8 @@
CrudTab::CrudTab(IDatabaseWindow *context, QWidget *parent) CrudTab::CrudTab(IDatabaseWindow *context, QWidget *parent)
: QWidget(parent) : QWidget(parent)
, m_context(context)
, ui(new Ui::CrudTab) , ui(new Ui::CrudTab)
, m_context(context)
{ {
ui->setupUi(this); ui->setupUi(this);

View file

@ -59,8 +59,8 @@ SOURCES += main.cpp\
ConstraintModel.cpp \ ConstraintModel.cpp \
IconColumnDelegate.cpp \ IconColumnDelegate.cpp \
IndexModel.cpp \ IndexModel.cpp \
CrudTab.cpp \ crud/CrudTab.cpp \
CrudModel.cpp \ crud/CrudModel.cpp \
PgLabItemDelegate.cpp \ PgLabItemDelegate.cpp \
Module.cpp \ Module.cpp \
EditorGutter.cpp \ EditorGutter.cpp \
@ -133,8 +133,8 @@ HEADERS += \
ConstraintModel.h \ ConstraintModel.h \
IconColumnDelegate.h \ IconColumnDelegate.h \
IndexModel.h \ IndexModel.h \
CrudTab.h \ crud/CrudTab.h \
CrudModel.h \ crud/CrudModel.h \
PgLabItemDelegate.h \ PgLabItemDelegate.h \
Module.h \ Module.h \
EditorGutter.h \ EditorGutter.h \
@ -176,7 +176,7 @@ FORMS += \
QueryTab.ui \ QueryTab.ui \
ProcessStdioWidget.ui \ ProcessStdioWidget.ui \
NamespaceFilterWidget.ui \ NamespaceFilterWidget.ui \
CrudTab.ui \ crud/CrudTab.ui \
CodeGenerator.ui CodeGenerator.ui
RESOURCES += \ RESOURCES += \

View file

@ -12,7 +12,7 @@ Params::Params(const Params& rhs)
, m_paramFormats(rhs.m_paramFormats) , m_paramFormats(rhs.m_paramFormats)
{ {
//std::vector<const char *> m_paramValues; //std::vector<const char *> m_paramValues;
copyValues(rhs.m_paramValues); appendValues(rhs.m_paramValues);
} }
Params& Params::operator=(const Params& rhs) Params& Params::operator=(const Params& rhs)
@ -21,7 +21,7 @@ Params& Params::operator=(const Params& rhs)
m_paramTypes = rhs.m_paramTypes; m_paramTypes = rhs.m_paramTypes;
m_paramLengths = rhs.m_paramLengths; m_paramLengths = rhs.m_paramLengths;
m_paramFormats = rhs.m_paramFormats; m_paramFormats = rhs.m_paramFormats;
copyValues(rhs.m_paramValues); appendValues(rhs.m_paramValues);
} }
return *this; return *this;
} }
@ -82,20 +82,21 @@ Param Params::add(const char *data, Oid oid)
p[len] = 0; p[len] = 0;
} }
return addText(p, oid); return addText(p, oid);
// m_paramTypes.push_back(oid);
// m_paramValues.push_back(p);
// m_paramLengths.push_back(len);
// m_paramFormats.push_back(0);
} }
//void Params::addBinary(const char *data, int length, Oid oid) template <typename T, typename U>
//{ void concatContainers(T &d, const U& s)
// m_paramTypes.push_back(oid); {
// m_paramValues.push_back(data); d.insert(d.end(), s.begin(), s.end());
// m_paramLengths.push_back(length); }
// m_paramFormats.push_back(1);
//} void Params::addParams(const Params &params)
{
concatContainers(m_paramTypes, params.m_paramTypes);
concatContainers(m_paramLengths, params.m_paramLengths);
concatContainers(m_paramFormats, params.m_paramFormats);
appendValues(params.m_paramValues);
}
void Params::clear() void Params::clear()
{ {
@ -106,13 +107,14 @@ void Params::clear()
m_paramFormats.clear(); m_paramFormats.clear();
} }
void Params::copyValues(const t_paramValues &r) void Params::appendValues(const t_paramValues &r)
{ {
const int n = static_cast<int>(m_paramTypes.size()); const int n = static_cast<int>(r.size());
m_paramValues.reserve(n); const int ofs = m_paramValues.size();
m_paramValues.reserve(ofs + n);
for (int i = 0; i < n; ++i) { for (int i = 0; i < n; ++i) {
if (r[i]) { if (r[i]) {
const int len = m_paramLengths[i]; const int len = m_paramLengths[ofs + i];
char * p = new char[len+1]; char * p = new char[len+1];
std::memcpy(p, r[i], len+1); std::memcpy(p, r[i], len+1);
m_paramValues.push_back(p); m_paramValues.push_back(p);

View file

@ -41,6 +41,8 @@ namespace Pgsql {
{ {
return add(s ? s->c_str() : nullptr, oid); return add(s ? s->c_str() : nullptr, oid);
} }
void addParams(const Params &params);
//void addBinary(const char *data, int length, Oid oid); //void addBinary(const char *data, int length, Oid oid);
void clear(); void clear();
@ -62,7 +64,7 @@ namespace Pgsql {
} }
/* Assumes other lists already have been copied */ /* Assumes other lists already have been copied */
void copyValues(const t_paramValues &r); void appendValues(const t_paramValues &r);
std::vector<Oid> m_paramTypes; std::vector<Oid> m_paramTypes;
t_paramValues m_paramValues; t_paramValues m_paramValues;