Mercurial > hg > orthanc-databases
diff Framework/PostgreSQL/PostgreSQLResult.cpp @ 0:7cea966b6829
initial commit
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Wed, 04 Jul 2018 08:16:29 +0200 |
parents | |
children | 9774802fd05f |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/Framework/PostgreSQL/PostgreSQLResult.cpp Wed Jul 04 08:16:29 2018 +0200 @@ -0,0 +1,233 @@ +/** + * Orthanc - A Lightweight, RESTful DICOM Store + * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics + * Department, University Hospital of Liege, Belgium + * Copyright (C) 2017-2018 Osimis S.A., Belgium + * + * This program is free software: you can redistribute it and/or + * modify it under the terms of the GNU Affero General Public License + * as published by the Free Software Foundation, either version 3 of + * the License, or (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, but + * WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Affero General Public License for more details. + * + * You should have received a copy of the GNU Affero General Public License + * along with this program. If not, see <http://www.gnu.org/licenses/>. + **/ + + +#include "PostgreSQLResult.h" + +#include "../Common/BinaryStringValue.h" +#include "../Common/Integer64Value.h" +#include "../Common/NullValue.h" +#include "../Common/Utf8StringValue.h" + +#include <Core/OrthancException.h> +#include <Core/Logging.h> + +#include <cassert> +#include <boost/lexical_cast.hpp> + +// PostgreSQL includes +#include <libpq-fe.h> +#include <c.h> +#include <catalog/pg_type.h> + +#include <Core/Endianness.h> + + +namespace OrthancDatabases +{ + void PostgreSQLResult::Clear() + { + if (result_ != NULL) + { + PQclear(reinterpret_cast<PGresult*>(result_)); + result_ = NULL; + } + } + + + void PostgreSQLResult::CheckDone() + { + if (position_ >= PQntuples(reinterpret_cast<PGresult*>(result_))) + { + // We are at the end of the result set + Clear(); + } + } + + + void PostgreSQLResult::CheckColumn(unsigned int column, unsigned int /*Oid*/ expectedType) const + { + if (IsDone()) + { + throw Orthanc::OrthancException(Orthanc::ErrorCode_BadSequenceOfCalls); + } + + if (column >= static_cast<unsigned int>(PQnfields(reinterpret_cast<PGresult*>(result_)))) + { + throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange); + } + + if (expectedType != 0 && + expectedType != PQftype(reinterpret_cast<PGresult*>(result_), column)) + { + throw Orthanc::OrthancException(Orthanc::ErrorCode_BadParameterType); + } + } + + + PostgreSQLResult::PostgreSQLResult(PostgreSQLStatement& statement) : + position_(0), + database_(statement.GetDatabase()) + { + result_ = statement.Execute(); + assert(result_ != NULL); // An exception would have been thrown otherwise + + // This is the first call to "Next()" + if (PQresultStatus(reinterpret_cast<PGresult*>(result_)) == PGRES_TUPLES_OK) + { + CheckDone(); + columnsCount_ = static_cast<unsigned int>(PQnfields(reinterpret_cast<PGresult*>(result_))); + } + else + { + // This is not a SELECT request, we're done + Clear(); + columnsCount_ = 0; + } + } + + + void PostgreSQLResult::Next() + { + position_++; + CheckDone(); + } + + + bool PostgreSQLResult::IsNull(unsigned int column) const + { + CheckColumn(column, 0); + return PQgetisnull(reinterpret_cast<PGresult*>(result_), position_, column) != 0; + } + + + bool PostgreSQLResult::GetBoolean(unsigned int column) const + { + CheckColumn(column, BOOLOID); + assert(PQfsize(reinterpret_cast<PGresult*>(result_), column) == 1); + const uint8_t *v = reinterpret_cast<const uint8_t*> + (PQgetvalue(reinterpret_cast<PGresult*>(result_), position_, column)); + return (v[0] != 0); + } + + + int PostgreSQLResult::GetInteger(unsigned int column) const + { + CheckColumn(column, INT4OID); + assert(PQfsize(reinterpret_cast<PGresult*>(result_), column) == 4); + char *v = PQgetvalue(reinterpret_cast<PGresult*>(result_), position_, column); + return htobe32(*reinterpret_cast<int32_t*>(v)); + } + + + int64_t PostgreSQLResult::GetInteger64(unsigned int column) const + { + CheckColumn(column, INT8OID); + assert(PQfsize(reinterpret_cast<PGresult*>(result_), column) == 8); + char *v = PQgetvalue(reinterpret_cast<PGresult*>(result_), position_, column); + return htobe64(*reinterpret_cast<int64_t*>(v)); + } + + + std::string PostgreSQLResult::GetString(unsigned int column) const + { + CheckColumn(column, 0); + + Oid oid = PQftype(reinterpret_cast<PGresult*>(result_), column); + if (oid != TEXTOID && oid != VARCHAROID && oid != BYTEAOID) + { + throw Orthanc::OrthancException(Orthanc::ErrorCode_BadParameterType); + } + + return std::string(PQgetvalue(reinterpret_cast<PGresult*>(result_), position_, column)); + } + + + void PostgreSQLResult::GetLargeObject(std::string& result, + unsigned int column) const + { + CheckColumn(column, OIDOID); + + Oid oid; + assert(PQfsize(reinterpret_cast<PGresult*>(result_), column) == sizeof(oid)); + + oid = *(const Oid*) PQgetvalue(reinterpret_cast<PGresult*>(result_), position_, column); + oid = ntohl(oid); + + PostgreSQLLargeObject::Read(result, database_, boost::lexical_cast<std::string>(oid)); + } + + + void PostgreSQLResult::GetLargeObject(void*& result, + size_t& size, + unsigned int column) const + { + CheckColumn(column, OIDOID); + + Oid oid; + assert(PQfsize(reinterpret_cast<PGresult*>(result_), column) == sizeof(oid)); + + oid = *(const Oid*) PQgetvalue(reinterpret_cast<PGresult*>(result_), position_, column); + oid = ntohl(oid); + + PostgreSQLLargeObject::Read(result, size, database_, boost::lexical_cast<std::string>(oid)); + } + + + IValue* PostgreSQLResult::GetValue(unsigned int column) const + { + if (IsNull(column)) + { + return new NullValue; + } + + Oid type = PQftype(reinterpret_cast<PGresult*>(result_), column); + + switch (type) + { + case BOOLOID: + // Convert Boolean values as integers + return new Integer64Value(GetBoolean(column) ? 1 : 0); + + case INT4OID: + return new Integer64Value(GetInteger(column)); + + case INT8OID: + return new Integer64Value(GetInteger64(column)); + + case TEXTOID: + case VARCHAROID: + return new Utf8StringValue(GetString(column)); + + case BYTEAOID: + return new BinaryStringValue(GetString(column)); + + case OIDOID: + { + std::auto_ptr<BinaryStringValue> value(new BinaryStringValue); + GetLargeObject(value->GetContent(), column); + return value.release(); + } + + default: + throw Orthanc::OrthancException(Orthanc::ErrorCode_NotImplemented); + } + } +}