mirror of
https://github.com/levinsv/pgadmin3.git
synced 2026-05-17 06:05:50 -06:00
support PG11
Поддержка PostgreSQL 11 только для Windows
This commit is contained in:
levinsv
commit
4af765213c
1765 changed files with 407959 additions and 0 deletions
480
ctl/ctlSQLResult.cpp
Normal file
480
ctl/ctlSQLResult.cpp
Normal file
|
|
@ -0,0 +1,480 @@
|
|||
//////////////////////////////////////////////////////////////////////////
|
||||
//
|
||||
// pgAdmin III - PostgreSQL Tools
|
||||
//
|
||||
// Copyright (C) 2002 - 2016, The pgAdmin Development Team
|
||||
// This software is released under the PostgreSQL Licence
|
||||
//
|
||||
// ctlSQLResult.cpp - SQL Query result window
|
||||
//
|
||||
//////////////////////////////////////////////////////////////////////////
|
||||
|
||||
#include "pgAdmin3.h"
|
||||
|
||||
// wxWindows headers
|
||||
#include <wx/wx.h>
|
||||
#include <wx/clipbrd.h>
|
||||
|
||||
#include "db/pgConn.h"
|
||||
#include "db/pgQueryThread.h"
|
||||
#include "ctl/ctlSQLResult.h"
|
||||
#include "utils/sysSettings.h"
|
||||
#include "frm/frmExport.h"
|
||||
|
||||
|
||||
|
||||
ctlSQLResult::ctlSQLResult(wxWindow *parent, pgConn *_conn, wxWindowID id, const wxPoint &pos, const wxSize &size)
|
||||
: ctlSQLGrid(parent, id, pos, size)
|
||||
{
|
||||
conn = _conn;
|
||||
thread = NULL;
|
||||
|
||||
SetTable(new sqlResultTable(), true);
|
||||
|
||||
EnableEditing(false);
|
||||
SetSizer(new wxBoxSizer(wxVERTICAL));
|
||||
|
||||
Connect(wxID_ANY, wxEVT_GRID_RANGE_SELECT, wxGridRangeSelectEventHandler(ctlSQLResult::OnGridSelect));
|
||||
}
|
||||
|
||||
|
||||
|
||||
ctlSQLResult::~ctlSQLResult()
|
||||
{
|
||||
Abort();
|
||||
|
||||
if (thread)
|
||||
{
|
||||
if (thread->IsRunning())
|
||||
{
|
||||
thread->CancelExecution();
|
||||
thread->Wait();
|
||||
}
|
||||
|
||||
delete thread;
|
||||
thread = NULL;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
void ctlSQLResult::SetConnection(pgConn *_conn)
|
||||
{
|
||||
conn = _conn;
|
||||
}
|
||||
|
||||
|
||||
bool ctlSQLResult::Export()
|
||||
{
|
||||
if (NumRows() > 0)
|
||||
{
|
||||
frmExport dlg(this);
|
||||
if (dlg.ShowModal() == wxID_OK)
|
||||
return dlg.Export(NULL);
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
bool ctlSQLResult::ToFile()
|
||||
{
|
||||
if (NumRows() > 0)
|
||||
{
|
||||
frmExport dlg(this);
|
||||
if (dlg.ShowModal() == wxID_OK)
|
||||
return dlg.Export(thread->DataSet());
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
bool ctlSQLResult::ToFile(frmExport *frm)
|
||||
{
|
||||
if (NumRows() > 0)
|
||||
{
|
||||
return frm->Export(thread->DataSet());
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
bool ctlSQLResult::IsColText(int col)
|
||||
{
|
||||
switch (colTypClasses.Item(col))
|
||||
{
|
||||
case PGTYPCLASS_NUMERIC:
|
||||
case PGTYPCLASS_BOOL:
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
|
||||
int ctlSQLResult::Execute(const wxString &query, int resultToRetrieve, wxWindow *caller, long eventId, void *data)
|
||||
{
|
||||
wxGridTableMessage *msg;
|
||||
sqlResultTable *table = (sqlResultTable *)GetTable();
|
||||
msg = new wxGridTableMessage(table, wxGRIDTABLE_NOTIFY_ROWS_DELETED, 0, GetNumberRows());
|
||||
ProcessTableMessage(*msg);
|
||||
delete msg;
|
||||
msg = new wxGridTableMessage(table, wxGRIDTABLE_NOTIFY_COLS_DELETED, 0, GetNumberCols());
|
||||
ProcessTableMessage(*msg);
|
||||
delete msg;
|
||||
|
||||
Abort();
|
||||
|
||||
colNames.Empty();
|
||||
colTypes.Empty();
|
||||
colTypClasses.Empty();
|
||||
|
||||
thread = new pgQueryThread(conn, query, resultToRetrieve, caller, eventId, data);
|
||||
|
||||
if (thread->Create() != wxTHREAD_NO_ERROR)
|
||||
{
|
||||
Abort();
|
||||
return -1;
|
||||
}
|
||||
|
||||
((sqlResultTable *)GetTable())->SetThread(thread);
|
||||
sqlquerytext= wxString(query);
|
||||
thread->Run();
|
||||
return RunStatus();
|
||||
}
|
||||
|
||||
|
||||
int ctlSQLResult::Abort()
|
||||
{
|
||||
if (thread)
|
||||
{
|
||||
((sqlResultTable *)GetTable())->SetThread(0);
|
||||
|
||||
if (thread->IsRunning())
|
||||
{
|
||||
thread->CancelExecution();
|
||||
thread->Wait();
|
||||
}
|
||||
|
||||
delete thread;
|
||||
thread = NULL;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
|
||||
void ctlSQLResult::DisplayData(bool single)
|
||||
{
|
||||
if (!thread || !thread->DataValid())
|
||||
return;
|
||||
|
||||
if (thread->ReturnCode() != PGRES_TUPLES_OK)
|
||||
return;
|
||||
|
||||
rowcountSuppressed = single;
|
||||
Freeze();
|
||||
|
||||
/*
|
||||
* Resize and repopulate by informing it to delete all the rows and
|
||||
* columns, then append the correct number of them. Probably is a
|
||||
* better way to do this.
|
||||
*/
|
||||
wxGridTableMessage *msg;
|
||||
sqlResultTable *table = (sqlResultTable *)GetTable();
|
||||
msg = new wxGridTableMessage(table, wxGRIDTABLE_NOTIFY_ROWS_DELETED, 0, GetNumberRows());
|
||||
ProcessTableMessage(*msg);
|
||||
delete msg;
|
||||
msg = new wxGridTableMessage(table, wxGRIDTABLE_NOTIFY_COLS_DELETED, 0, GetNumberCols());
|
||||
ProcessTableMessage(*msg);
|
||||
delete msg;
|
||||
msg = new wxGridTableMessage(table, wxGRIDTABLE_NOTIFY_ROWS_APPENDED, NumRows());
|
||||
ProcessTableMessage(*msg);
|
||||
delete msg;
|
||||
msg = new wxGridTableMessage(table, wxGRIDTABLE_NOTIFY_COLS_APPENDED, thread->DataSet()->NumCols());
|
||||
ProcessTableMessage(*msg);
|
||||
delete msg;
|
||||
if (NumRows()<1000) {
|
||||
wxGridCellAttr* pAttr = new wxGridCellAttr;
|
||||
pAttr->SetBackgroundColour(wxColour(224,255,224));
|
||||
for(int row = 0; row < NumRows(); ++row) {
|
||||
if (row%2==0) SetRowAttr(row,pAttr);
|
||||
}
|
||||
}
|
||||
if (single)
|
||||
{
|
||||
colNames.Add(thread->DataSet()->ColName(0));
|
||||
colTypes.Add(wxT(""));
|
||||
colTypClasses.Add(0L);
|
||||
|
||||
AutoSizeColumn(0, false, false);
|
||||
}
|
||||
else
|
||||
{
|
||||
long col, nCols = thread->DataSet()->NumCols();
|
||||
|
||||
AutoSizeColumns(false);
|
||||
|
||||
for (col = 0 ; col < nCols ; col++)
|
||||
{
|
||||
colNames.Add(thread->DataSet()->ColName(col));
|
||||
colTypes.Add(thread->DataSet()->ColFullType(col));
|
||||
colTypClasses.Add(thread->DataSet()->ColTypClass(col));
|
||||
|
||||
if (thread->DataSet()->ColTypClass(col) == PGTYPCLASS_NUMERIC)
|
||||
{
|
||||
/*
|
||||
* For numeric columns, set alignment to right.
|
||||
*/
|
||||
wxGridCellAttr *attr = new wxGridCellAttr();
|
||||
attr->SetAlignment(wxALIGN_RIGHT, wxALIGN_TOP);
|
||||
SetColAttr(col, attr);
|
||||
}
|
||||
}
|
||||
}
|
||||
Thaw();
|
||||
}
|
||||
|
||||
|
||||
|
||||
wxString ctlSQLResult::GetMessagesAndClear()
|
||||
{
|
||||
if (thread)
|
||||
return thread->GetMessagesAndClear();
|
||||
return wxEmptyString;
|
||||
}
|
||||
|
||||
|
||||
wxString ctlSQLResult::GetErrorMessage()
|
||||
{
|
||||
return conn->GetLastError();
|
||||
}
|
||||
|
||||
pgError ctlSQLResult::GetResultError()
|
||||
{
|
||||
if (thread)
|
||||
return thread->GetResultError();
|
||||
|
||||
pgError dummy;
|
||||
memset(&dummy, 0, sizeof(dummy));
|
||||
return dummy;
|
||||
}
|
||||
|
||||
long ctlSQLResult::NumRows() const
|
||||
{
|
||||
if (thread && thread->DataValid())
|
||||
return thread->DataSet()->NumRows();
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
long ctlSQLResult::InsertedCount() const
|
||||
{
|
||||
if (thread)
|
||||
return thread->RowsInserted();
|
||||
return -1;
|
||||
}
|
||||
|
||||
|
||||
OID ctlSQLResult::InsertedOid() const
|
||||
{
|
||||
if (thread)
|
||||
return thread->InsertedOid();
|
||||
return (OID) - 1;
|
||||
}
|
||||
|
||||
|
||||
int ctlSQLResult::RunStatus()
|
||||
{
|
||||
if (!thread)
|
||||
return -1;
|
||||
|
||||
if (thread->IsRunning())
|
||||
return CTLSQL_RUNNING;
|
||||
|
||||
return thread->ReturnCode();
|
||||
}
|
||||
|
||||
|
||||
wxString ctlSQLResult::OnGetItemText(long item, long col) const
|
||||
{
|
||||
if (thread && thread->DataValid())
|
||||
{
|
||||
if (!rowcountSuppressed)
|
||||
{
|
||||
if (col)
|
||||
col--;
|
||||
else
|
||||
return NumToStr(item + 1L);
|
||||
}
|
||||
if (item >= 0)
|
||||
{
|
||||
thread->DataSet()->Locate(item + 1);
|
||||
return thread->DataSet()->GetVal(col);
|
||||
}
|
||||
else
|
||||
return thread->DataSet()->ColName(col);
|
||||
}
|
||||
return wxEmptyString;
|
||||
}
|
||||
wxString ctlSQLResult::SummaryColumn()
|
||||
{
|
||||
//ce=cells.Item(0);
|
||||
double sum=0;
|
||||
size_t i;
|
||||
if (GetSelectedRows().GetCount())
|
||||
{
|
||||
//AppendColumnHeader(str, 0, (GetNumberCols() - 1));
|
||||
|
||||
wxArrayInt rows = GetSelectedRows();
|
||||
|
||||
for (i = 0 ; i < rows.GetCount() ; i++)
|
||||
{
|
||||
//str.Append(GetExportLine(rows.Item(i)));
|
||||
//if (rows.GetCount() > 1)
|
||||
// str.Append(END_OF_LINE);
|
||||
}
|
||||
|
||||
//copied = rows.GetCount();
|
||||
}
|
||||
else if (GetSelectedCols().GetCount())
|
||||
{
|
||||
wxArrayInt cols = GetSelectedCols();
|
||||
size_t numRows = GetNumberRows();
|
||||
|
||||
//AppendColumnHeader(str, cols);
|
||||
for (i = 0 ; i < numRows ; i++)
|
||||
{
|
||||
//str.Append(GetExportLine(i, cols));
|
||||
for (size_t col = 0 ; col < cols.Count() ; col++)
|
||||
{
|
||||
wxString text = GetCellValue(i, cols[col]);
|
||||
sum=sum+wxAtof(text);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
//copied = numRows;
|
||||
}
|
||||
else if (GetSelectionBlockTopLeft().GetCount() > 0 &&
|
||||
GetSelectionBlockBottomRight().GetCount() > 0)
|
||||
{
|
||||
unsigned int x1, x2, y1, y2;
|
||||
|
||||
x1 = GetSelectionBlockTopLeft()[0].GetCol();
|
||||
x2 = GetSelectionBlockBottomRight()[0].GetCol();
|
||||
y1 = GetSelectionBlockTopLeft()[0].GetRow();
|
||||
y2 = GetSelectionBlockBottomRight()[0].GetRow();
|
||||
|
||||
//AppendColumnHeader(str, x1, x2);
|
||||
|
||||
for (i = y1; i <= y2; i++)
|
||||
{
|
||||
// str.Append(GetExportLine(i, x1, x2));
|
||||
wxString text = GetCellValue(i, x1);
|
||||
sum=sum+wxAtof(text);
|
||||
|
||||
//copied = y2 - y1 + 1;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
int row, col;
|
||||
|
||||
row = GetGridCursorRow();
|
||||
col = GetGridCursorCol();
|
||||
|
||||
//AppendColumnHeader(str, col, col);
|
||||
|
||||
// str.Append(GetExportLine(row, col, col));
|
||||
//copied = 1;
|
||||
}
|
||||
|
||||
|
||||
wxString result;
|
||||
result.Printf(wxT("%f"), sum);
|
||||
return result;
|
||||
}
|
||||
|
||||
void ctlSQLResult::OnGridSelect(wxGridRangeSelectEvent &event)
|
||||
{
|
||||
SetFocus();
|
||||
}
|
||||
|
||||
wxString sqlResultTable::GetValue(int row, int col)
|
||||
{
|
||||
if (thread && thread->DataValid())
|
||||
{
|
||||
if (col >= 0)
|
||||
{
|
||||
thread->DataSet()->Locate(row + 1);
|
||||
if (settings->GetIndicateNull() && thread->DataSet()->IsNull(col))
|
||||
return wxT("<NULL>");
|
||||
else
|
||||
{
|
||||
|
||||
wxString decimalMark = wxT(".");
|
||||
wxString s = thread->DataSet()->GetVal(col);
|
||||
|
||||
if(thread->DataSet()->ColTypClass(col) == PGTYPCLASS_NUMERIC &&
|
||||
settings->GetDecimalMark().Length() > 0)
|
||||
{
|
||||
decimalMark = settings->GetDecimalMark();
|
||||
s.Replace(wxT("."), decimalMark);
|
||||
|
||||
}
|
||||
if (thread->DataSet()->ColTypClass(col) == PGTYPCLASS_NUMERIC &&
|
||||
settings->GetThousandsSeparator().Length() > 0)
|
||||
{
|
||||
/* Add thousands separator */
|
||||
size_t pos = s.find(decimalMark);
|
||||
if (pos == wxString::npos)
|
||||
pos = s.length();
|
||||
while (pos > 3)
|
||||
{
|
||||
pos -= 3;
|
||||
if (pos > 1 || !s.StartsWith(wxT("-")))
|
||||
s.insert(pos, settings->GetThousandsSeparator());
|
||||
}
|
||||
return s;
|
||||
}
|
||||
else
|
||||
{
|
||||
wxString data = thread->DataSet()->GetVal(col);
|
||||
|
||||
if (data.Length() > (size_t)settings->GetMaxColSize())
|
||||
return thread->DataSet()->GetVal(col).Left(settings->GetMaxColSize()) + wxT(" (...)");
|
||||
else
|
||||
return thread->DataSet()->GetVal(col);
|
||||
}
|
||||
}
|
||||
}
|
||||
else
|
||||
return thread->DataSet()->ColName(col);
|
||||
}
|
||||
return wxEmptyString;
|
||||
}
|
||||
|
||||
sqlResultTable::sqlResultTable()
|
||||
{
|
||||
thread = NULL;
|
||||
}
|
||||
|
||||
int sqlResultTable::GetNumberRows()
|
||||
{
|
||||
if (thread && thread->DataValid())
|
||||
return thread->DataSet()->NumRows();
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
wxString sqlResultTable::GetColLabelValue(int col)
|
||||
{
|
||||
if (thread && thread->DataValid())
|
||||
return thread->DataSet()->ColName(col) + wxT("\n") +
|
||||
thread->DataSet()->ColFullType(col);
|
||||
return wxEmptyString;
|
||||
}
|
||||
|
||||
int sqlResultTable::GetNumberCols()
|
||||
{
|
||||
if (thread && thread->DataValid())
|
||||
return thread->DataSet()->NumCols();
|
||||
return 0;
|
||||
}
|
||||
|
||||
Loading…
Add table
Add a link
Reference in a new issue