1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103
|
/**
* Orthanc - A Lightweight, RESTful DICOM Store
* Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics
* Department, University Hospital of Liege, Belgium
* Copyright (C) 2017-2023 Osimis S.A., Belgium
* Copyright (C) 2021-2023 Sebastien Jodogne, ICTEAM UCLouvain, 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/>.
**/
#pragma once
#if ORTHANC_ENABLE_POSTGRESQL != 1
# error PostgreSQL support must be enabled to use this file
#endif
#include "PostgreSQLParameters.h"
#include "../Common/IDatabaseFactory.h"
namespace OrthancDatabases
{
class PostgreSQLDatabase : public IDatabase
{
private:
friend class PostgreSQLStatement;
friend class PostgreSQLLargeObject;
class Factory;
PostgreSQLParameters parameters_;
void* pg_; /* Object of type "PGconn*" */
void ThrowException(bool log);
void Close();
bool RunAdvisoryLockStatement(const std::string& statement);
public:
explicit PostgreSQLDatabase(const PostgreSQLParameters& parameters) :
parameters_(parameters),
pg_(NULL)
{
}
~PostgreSQLDatabase();
void Open();
bool AcquireAdvisoryLock(int32_t lock);
bool ReleaseAdvisoryLock(int32_t lock);
void AdvisoryLock(int32_t lock);
void ExecuteMultiLines(const std::string& sql);
bool DoesTableExist(const std::string& name);
bool DoesColumnExist(const std::string& tableName,
const std::string& columnName);
void ClearAll(); // Only for unit tests!
virtual Dialect GetDialect() const ORTHANC_OVERRIDE
{
return Dialect_PostgreSQL;
}
virtual IPrecompiledStatement* Compile(const Query& query) ORTHANC_OVERRIDE;
virtual ITransaction* CreateTransaction(TransactionType type) ORTHANC_OVERRIDE;
class TransientAdvisoryLock
{
private:
PostgreSQLDatabase& database_;
int32_t lock_;
public:
TransientAdvisoryLock(PostgreSQLDatabase& database,
int32_t lock);
~TransientAdvisoryLock();
};
static IDatabaseFactory* CreateDatabaseFactory(const PostgreSQLParameters& parameters);
static PostgreSQLDatabase* CreateDatabaseConnection(const PostgreSQLParameters& parameters);
};
}
|