88 lines
1.7 KiB
C++
88 lines
1.7 KiB
C++
#include "QueryResultModel.h"
|
|
#include "ResultTableModelUtil.h"
|
|
#include "Pgsql_declare.h"
|
|
#include <QBrush>
|
|
#include <QColor>
|
|
|
|
using namespace Pgsql;
|
|
|
|
QueryResultModel::QueryResultModel(QObject *parent, std::shared_ptr<Pgsql::Result> r)
|
|
: BaseTableModel(parent)
|
|
, result(std::move(r))
|
|
{}
|
|
|
|
|
|
int QueryResultModel::rowCount(const QModelIndex &) const
|
|
{
|
|
int r = result->rows();
|
|
return r;
|
|
}
|
|
|
|
int QueryResultModel::columnCount(const QModelIndex &) const
|
|
{
|
|
int r = result->cols();
|
|
return r;
|
|
}
|
|
|
|
Oid QueryResultModel::getType(int column) const
|
|
{
|
|
return result->type(column);
|
|
}
|
|
|
|
QVariant QueryResultModel::getData(const QModelIndex &index) const
|
|
{
|
|
QVariant r;
|
|
int rij = index.row();
|
|
int col = index.column();
|
|
if (result->null(col, rij)) {
|
|
r = "null";
|
|
}
|
|
else {
|
|
Oid o = result->type(col);
|
|
auto value = result->get(col, rij);
|
|
switch (o) {
|
|
case BOOLOID:
|
|
r = bool(value); //s = (s == "t") ? "TRUE" : "FALSE";
|
|
break;
|
|
default: {
|
|
QString s = value;
|
|
if (s.length() > 256) {
|
|
s.truncate(256);
|
|
}
|
|
r = s;
|
|
}
|
|
}
|
|
}
|
|
return r;
|
|
}
|
|
|
|
QVariant QueryResultModel::data(const QModelIndex &index, int role) const
|
|
{
|
|
if (role == Qt::ForegroundRole) {
|
|
int rij = index.row();
|
|
int col = index.column();
|
|
if (result->null(col, rij)) {
|
|
return QBrush(Qt::gray);
|
|
}
|
|
}
|
|
return BaseTableModel::data(index, role);
|
|
}
|
|
|
|
QVariant QueryResultModel::headerData(int section, Qt::Orientation orientation, int role) const
|
|
{
|
|
QVariant r;
|
|
if (role == Qt::DisplayRole) {
|
|
if (orientation == Qt::Horizontal) {
|
|
r = QString(result->getColName(section));
|
|
}
|
|
else {
|
|
r = QString::number(section + 1);
|
|
}
|
|
}
|
|
return r;
|
|
}
|
|
|
|
//Qt::ItemFlags QueryResultModel::flags(const QModelIndex &) const
|
|
//{
|
|
// return Qt::ItemIsUserCheckable;
|
|
//}
|