Transform to CMake project.
This commit is contained in:
SET
446
LPQResultSet.cpp
Normal file
446
LPQResultSet.cpp
Normal file
@@ -0,0 +1,446 @@
|
||||
/*
|
||||
* File: LPQResultSet.cpp
|
||||
* Author: SET - nmset@yandex.com
|
||||
* License : LGPL version 2.1
|
||||
* Copyright SET, M. D. - © 2014
|
||||
*
|
||||
* Created on 29 mai 2014, 16:19
|
||||
*/
|
||||
|
||||
#ifdef USE_LIBPQ
|
||||
|
||||
#include "LPQResultSet.h"
|
||||
#include "LLightPQResultSet.h"
|
||||
using namespace PQ;
|
||||
|
||||
LPQResultSet::LPQResultSet() : LResultSet()
|
||||
{
|
||||
}
|
||||
|
||||
LPQResultSet::LPQResultSet(LConnection* newConnection) : LResultSet(newConnection)
|
||||
{
|
||||
}
|
||||
|
||||
LPQResultSet::~LPQResultSet()
|
||||
{
|
||||
if (m_rs)
|
||||
{
|
||||
PGresult * prs = static_cast<PGresult*> (m_rs);
|
||||
PQclear(prs);
|
||||
m_rs = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
bool LPQResultSet::SetSQL(const wxString& newSql)
|
||||
{
|
||||
UpdateSQL(newSql);
|
||||
if (RunSQL()) First();
|
||||
return (m_rs != NULL);
|
||||
}
|
||||
|
||||
bool LPQResultSet::RunSQL()
|
||||
{
|
||||
if (m_curSql.IsEmpty())
|
||||
{
|
||||
return false;
|
||||
}
|
||||
wxASSERT_MSG(m_conn != NULL, _T("m_conn est NULL."));
|
||||
if (m_rs)
|
||||
{
|
||||
PGresult * prs = static_cast<PGresult*> (m_rs);
|
||||
PQclear(prs);
|
||||
m_rs = NULL;
|
||||
}
|
||||
void * vrs = m_conn->ExecuteSQL(m_curSql);
|
||||
m_rs = static_cast<PGresult*> (vrs);
|
||||
if (m_rs != NULL)
|
||||
{
|
||||
m_cursor = (HasData()) ? 0 : -1;
|
||||
Display();
|
||||
m_inserting = false;
|
||||
if (!m_initialised)
|
||||
{
|
||||
m_initialised = true;
|
||||
InformInitialised();
|
||||
}
|
||||
InformDataChanged();
|
||||
return true;
|
||||
}
|
||||
else
|
||||
{
|
||||
m_cursor = -1;
|
||||
Display();
|
||||
m_inserting = false;
|
||||
const LInformation inf(RS000C, RS000M);
|
||||
InformLibMessage(inf);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
bool LPQResultSet::HasData() const
|
||||
{
|
||||
if (m_rs == NULL) return false;
|
||||
PGresult * prs = static_cast<PGresult*> (m_rs);
|
||||
if (PQntuples(prs) == 0) return false;
|
||||
return true;
|
||||
}
|
||||
|
||||
bool LPQResultSet::Absolute(const unsigned int newRowIndex)
|
||||
{
|
||||
if (!m_readOnly)
|
||||
{
|
||||
if (m_inserting)
|
||||
{
|
||||
InformInserting();
|
||||
return false;
|
||||
}
|
||||
if (IsDirty())
|
||||
{
|
||||
InformDirty();
|
||||
return false;
|
||||
}
|
||||
}
|
||||
if (!HasData())
|
||||
{
|
||||
m_cursor = -1;
|
||||
Display();
|
||||
return false;
|
||||
}
|
||||
wxASSERT_MSG(newRowIndex < GetRowCount(), _T("Paramètre newRowIndex invalide."));
|
||||
InformBeforeChangeRow();
|
||||
if (!m_canMove) return false;
|
||||
m_cursor = newRowIndex;
|
||||
Display();
|
||||
m_inserting = false;
|
||||
InformAfterChangeRow();
|
||||
return true;
|
||||
}
|
||||
|
||||
bool LPQResultSet::IsFirst() const
|
||||
{
|
||||
if (m_inserting) return false;
|
||||
if (!(HasData())) return false;
|
||||
return (m_cursor == 0);
|
||||
}
|
||||
|
||||
bool LPQResultSet::IsLast() const
|
||||
{
|
||||
if (m_inserting) return false;
|
||||
if (!(HasData())) return false;
|
||||
PGresult * prs = static_cast<PGresult*> (m_rs);
|
||||
return (m_cursor == (PQntuples(prs) - 1));
|
||||
}
|
||||
|
||||
bool LPQResultSet::First()
|
||||
{
|
||||
if (IsFirst()) return false;
|
||||
return Absolute(0);
|
||||
}
|
||||
|
||||
bool LPQResultSet::Last()
|
||||
{
|
||||
if (IsLast()) return false;
|
||||
PGresult * prs = static_cast<PGresult*> (m_rs);
|
||||
return Absolute(PQntuples(prs) - 1);
|
||||
}
|
||||
|
||||
bool LPQResultSet::Next()
|
||||
{
|
||||
if (IsLast()) return false;
|
||||
return Absolute(m_cursor + 1);
|
||||
}
|
||||
|
||||
bool LPQResultSet::Previous()
|
||||
{
|
||||
if (IsFirst()) return false;
|
||||
return Absolute(m_cursor - 1);
|
||||
}
|
||||
|
||||
const unsigned int LPQResultSet::GetRowCount() const
|
||||
{
|
||||
wxASSERT_MSG(m_rs != NULL, _T("m_rs est NULL."));
|
||||
PGresult * prs = static_cast<PGresult*> (m_rs);
|
||||
return PQntuples(prs);
|
||||
}
|
||||
|
||||
const unsigned int LPQResultSet::GetColumnCount() const
|
||||
{
|
||||
wxASSERT_MSG(m_rs != NULL, _T("m_rs est NULL."));
|
||||
PGresult * prs = static_cast<PGresult*> (m_rs);
|
||||
return PQnfields(prs);
|
||||
}
|
||||
|
||||
const wxString LPQResultSet::GetColumnName(const unsigned int colIndex) const
|
||||
{
|
||||
wxASSERT_MSG(m_rs != NULL, _T("m_rs est NULL."));
|
||||
PGresult * prs = static_cast<PGresult*> (m_rs);
|
||||
const char * name = PQfname(prs, colIndex);
|
||||
if (name) return wxString(name);
|
||||
return wxEmptyString;
|
||||
}
|
||||
|
||||
const int LPQResultSet::GetColumnIndex(const wxString& colName) const
|
||||
{
|
||||
wxASSERT_MSG(m_rs != NULL, _T("m_rs est NULL."));
|
||||
PGresult * prs = static_cast<PGresult*> (m_rs);
|
||||
return PQfnumber(prs, colName.c_str());
|
||||
}
|
||||
|
||||
const wxAny LPQResultSet::GetData(const unsigned int rowIdx, const unsigned int colIdx) const
|
||||
{
|
||||
PGresult * prs = static_cast<PGresult*> (m_rs);
|
||||
const char * data = PQgetvalue(prs, rowIdx, colIdx);
|
||||
return wxString(data, wxConvUTF8);
|
||||
}
|
||||
|
||||
const wxAny LPQResultSet::GetData(const wxString& columnName) const
|
||||
{
|
||||
PGresult * prs = static_cast<PGresult*> (m_rs);
|
||||
if (m_cursor < 0
|
||||
|| IsInserting()
|
||||
|| !HasData()
|
||||
|| m_cursor > PQntuples(prs) - 1)
|
||||
return wxEmptyString;
|
||||
const char* colName = columnName.c_str();
|
||||
int colIdx = PQfnumber(prs, colName);
|
||||
if (colIdx < 0 || colIdx >= PQnfields(prs))
|
||||
{
|
||||
const wxString msg = RS004M + columnName;
|
||||
const LInformation inf(RS004C, msg);
|
||||
InformLibMessage(inf);
|
||||
return wxEmptyString;
|
||||
}
|
||||
return GetData(m_cursor, colIdx);
|
||||
}
|
||||
|
||||
void LPQResultSet::Display()
|
||||
{
|
||||
for (unsigned int i = 0; i < m_BoundControls.GetCount(); i++)
|
||||
{
|
||||
LBoundControl * control = static_cast<LBoundControl*> (m_BoundControls.Item(i));
|
||||
if (!control) continue;
|
||||
wxAny aData = GetData(control->GetColumnName());
|
||||
control->SetData(aData);
|
||||
}
|
||||
}
|
||||
|
||||
bool LPQResultSet::Save()
|
||||
{
|
||||
wxASSERT_MSG(m_conn != NULL, _T("m_conn est NULL."));
|
||||
if (m_readOnly || m_conn->IsReadOnly()) return false;
|
||||
if (!m_inserting && !HasData()) return false;
|
||||
InformBeforeSave();
|
||||
if (!m_canSave) return false;
|
||||
LBoundControl* pkControl = GetControl(m_pkColName);
|
||||
if (!pkControl) return false;
|
||||
if (pkControl->IsNull())
|
||||
{
|
||||
InformBeforeInsert();
|
||||
if (!m_canInsert) return false;
|
||||
}
|
||||
else
|
||||
{
|
||||
InformBeforeUpdate();
|
||||
if (!m_canUpdate) return false;
|
||||
}
|
||||
const wxString sql = MakeUpdateSQL();
|
||||
if (sql.IsEmpty()) return false;
|
||||
bool updating = (sql.Mid(0, 7) == _T("UPDATE "));
|
||||
wxAny pkValue;
|
||||
bool res;
|
||||
if (updating) pkValue = GetData(m_pkColName);
|
||||
if (m_conn->ExecuteUpdateSQL(sql))
|
||||
{
|
||||
const int oldCursor = m_cursor;
|
||||
if (updating)
|
||||
{
|
||||
InformAfterUpdate(pkValue, sql);
|
||||
res = RunSQL();
|
||||
if (res)
|
||||
{
|
||||
m_cursor = GetRowByPKValue(pkValue);
|
||||
Display();
|
||||
if (oldCursor != m_cursor) InformAfterChangeRow();
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
void * vretKeys = m_conn->GetReturnedKeys();
|
||||
PGresult * retKeys = static_cast<PGresult*> (vretKeys);
|
||||
if (retKeys != NULL)
|
||||
{
|
||||
const char * data = PQgetvalue(retKeys, 0, 0); // RETURNING pkColName from MakeUpdateSQL
|
||||
wxAny newPK = wxString(data);
|
||||
InformAfterInsert(newPK, sql); // Before RunSQL so as UpdateSQL can be called by client app
|
||||
res = RunSQL();
|
||||
if (res)
|
||||
{
|
||||
m_cursor = GetRowByPKValue(newPK);
|
||||
Display();
|
||||
m_inserting = false;
|
||||
if (oldCursor != m_cursor) InformAfterChangeRow();
|
||||
}
|
||||
//m_conn->ClearReturnedKeys(); // Not here
|
||||
}
|
||||
else
|
||||
{
|
||||
const LInformation inf(RS002C, RS002M);
|
||||
InformLibMessage(inf);
|
||||
}
|
||||
}
|
||||
InformAfterSave();
|
||||
}
|
||||
else
|
||||
{
|
||||
const LInformation inf(RSPQMC, wxString(m_conn->GetLastLibMessage()));
|
||||
InformLibMessage(inf);
|
||||
}
|
||||
m_conn->ClearReturnedKeys(); // Always clear, even if update.
|
||||
if (res) InformAfterAction(LResultSetEvent::ACTION_SAVE);
|
||||
return res;
|
||||
}
|
||||
|
||||
const wxString LPQResultSet::MakeUpdateSQL() const
|
||||
{
|
||||
if (m_readOnly || !m_canSave || !m_canInsert || !m_canUpdate) return wxEmptyString;
|
||||
if (!m_inserting && !HasData()) return wxEmptyString;
|
||||
LBoundControl* pkControl = GetControl(m_pkColName);
|
||||
if (pkControl == NULL) return wxEmptyString;
|
||||
wxString sql = _T("UPDATE");
|
||||
bool updating = true;
|
||||
if (pkControl->IsNull())
|
||||
{
|
||||
sql = _T("INSERT INTO");
|
||||
updating = false;
|
||||
}
|
||||
wxString columns = wxEmptyString;
|
||||
wxString values = wxEmptyString;
|
||||
wxString singleValue = wxEmptyString;
|
||||
for (unsigned int i = 0; i < m_BoundControls.GetCount(); i++)
|
||||
{
|
||||
LBoundControl * control = static_cast<LBoundControl*> (m_BoundControls.Item(i));
|
||||
if (!control) continue;
|
||||
const wxString colName = control->GetColumnName();
|
||||
if (updating)
|
||||
{
|
||||
if (colName.Lower() != m_pkColName.Lower())
|
||||
{
|
||||
if (control->IsDirty())
|
||||
{
|
||||
// Build "col = value, " string
|
||||
singleValue = wxEmptyString;
|
||||
control->GetData().GetAs<wxString> (&singleValue);
|
||||
EscapeChar(singleValue);
|
||||
const wxString quote = control->IsNull() ? wxString(wxEmptyString) : control->GetSQLQuote();
|
||||
values += colName + _T(" = ") + quote
|
||||
+ singleValue
|
||||
+ quote + _T(", ");
|
||||
}
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
if (colName.Lower() != m_pkColName.Lower())
|
||||
{
|
||||
if (control->IsDirty())
|
||||
{
|
||||
// Construct "col1, col2,..." string
|
||||
columns += colName + _T(", ");
|
||||
singleValue = wxEmptyString;
|
||||
control->GetData().GetAs<wxString> (&singleValue);
|
||||
EscapeChar(singleValue);
|
||||
// Construct "value1, value2,..." string
|
||||
const wxString quote = control->IsNull() ? wxString(wxEmptyString) : control->GetSQLQuote();
|
||||
values += quote
|
||||
+ singleValue
|
||||
+ quote + _T(", ");
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
if (updating)
|
||||
{
|
||||
if (values.IsEmpty())
|
||||
{
|
||||
sql = wxEmptyString;
|
||||
}
|
||||
else
|
||||
{
|
||||
// Append "WHERE pkColName = value"
|
||||
values = values.Mid(0, (values.Len() - 2));
|
||||
singleValue = wxEmptyString;
|
||||
pkControl->GetData().GetAs<wxString> (&singleValue);
|
||||
values += _T(" WHERE ") + m_pkColName + _T(" = ")
|
||||
+ pkControl->GetSQLQuote() + singleValue
|
||||
+ pkControl->GetSQLQuote();
|
||||
// Build final string
|
||||
sql += _T(" ") + m_tblName + _T(" SET ") + values;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
if (values.IsEmpty() || columns.IsEmpty())
|
||||
{
|
||||
sql = wxEmptyString;
|
||||
}
|
||||
else
|
||||
{
|
||||
// Manage master/child columns
|
||||
wxString childColList = wxEmptyString;
|
||||
wxString masterValueList = wxEmptyString;
|
||||
if (IsChild())
|
||||
{
|
||||
childColList = MakeChildInsertColList();
|
||||
masterValueList = MakeMasterValueList();
|
||||
}
|
||||
columns = childColList + columns;
|
||||
columns = columns.Mid(0, (columns.Len() - 2));
|
||||
columns = _T(" (") + columns + _T(") ");
|
||||
values = masterValueList + values;
|
||||
values = values.Mid(0, (values.Len() - 2));
|
||||
values = _T(" (") + values + _T(") ");
|
||||
// Near final string
|
||||
sql = sql + _T(" ") + m_tblName + columns + _T("VALUES") + values;
|
||||
// Add returning primary key clause
|
||||
if (!m_pkColName.IsEmpty())
|
||||
{
|
||||
sql += _T(" RETURNING ") + m_pkColName;
|
||||
}
|
||||
}
|
||||
}
|
||||
return sql;
|
||||
}
|
||||
|
||||
const wxString LPQResultSet::MakeDeleteSQL() const
|
||||
{
|
||||
if (m_readOnly || !m_canDelete) return wxEmptyString;
|
||||
if (!HasData()) return wxEmptyString;
|
||||
if (m_inserting) return wxEmptyString;
|
||||
LBoundControl* pkControl = GetControl(m_pkColName);
|
||||
if (pkControl == NULL) return wxEmptyString;
|
||||
wxString pkValue = wxEmptyString;
|
||||
pkControl->GetData().GetAs<wxString>(&pkValue);
|
||||
EscapeChar(pkValue); // ???
|
||||
wxString sql = _T("DELETE FROM ") + m_tblName
|
||||
+ _T(" WHERE ") + m_pkColName + _T(" = ")
|
||||
+ pkControl->GetSQLQuote()
|
||||
+ pkValue
|
||||
+ pkControl->GetSQLQuote();
|
||||
return sql;
|
||||
}
|
||||
|
||||
const int LPQResultSet::GetRowByPKValue(const wxAny& pkValue) const
|
||||
{
|
||||
if (!HasData()) return -1;
|
||||
PGresult * prs = static_cast<PGresult*> (m_rs);
|
||||
int colIdx = PQfnumber(prs, m_pkColName.c_str());
|
||||
if (colIdx < 0) return -1;
|
||||
for (unsigned int i = 0; i < GetRowCount(); i++)
|
||||
{
|
||||
const wxAny beData = GetData(i, colIdx);
|
||||
if (pkValue.As<wxString>() == beData.As<wxString>()) return i;
|
||||
}
|
||||
return -1;
|
||||
}
|
||||
|
||||
#endif
|
||||
Reference in New Issue
Block a user