Mercurial > hg > orthanc-databases
view Framework/Common/DatabaseManager.h @ 73:aa81c1c80c75 db-changes
fix compatibility with older sdk
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Fri, 04 Jan 2019 17:06:34 +0100 |
parents | e6c13ddd26d9 |
children | 52c70007bb87 |
line wrap: on
line source
/** * Orthanc - A Lightweight, RESTful DICOM Store * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics * Department, University Hospital of Liege, Belgium * Copyright (C) 2017-2019 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/>. **/ #pragma once #include "IDatabaseFactory.h" #include "StatementLocation.h" #include <Core/Enumerations.h> #include <boost/thread/recursive_mutex.hpp> #include <memory> namespace OrthancDatabases { class DatabaseManager : public boost::noncopyable { private: typedef std::map<StatementLocation, IPrecompiledStatement*> CachedStatements; boost::recursive_mutex mutex_; std::auto_ptr<IDatabaseFactory> factory_; std::auto_ptr<IDatabase> database_; std::auto_ptr<ITransaction> transaction_; CachedStatements cachedStatements_; Dialect dialect_; IDatabase& GetDatabase(); void CloseIfUnavailable(Orthanc::ErrorCode e); IPrecompiledStatement* LookupCachedStatement(const StatementLocation& location) const; IPrecompiledStatement& CacheStatement(const StatementLocation& location, const Query& query); ITransaction& GetTransaction(); void ReleaseImplicitTransaction(); public: explicit DatabaseManager(IDatabaseFactory* factory); // Takes ownership ~DatabaseManager() { Close(); } Dialect GetDialect() const { return dialect_; } void Open() { GetDatabase(); } void Close(); void StartTransaction(); void CommitTransaction(); void RollbackTransaction(); // This class is only used in the "StorageBackend" class Transaction : public boost::noncopyable { private: boost::recursive_mutex::scoped_lock lock_; DatabaseManager& manager_; IDatabase& database_; bool committed_; public: explicit Transaction(DatabaseManager& manager); ~Transaction(); void Commit(); DatabaseManager& GetManager() { return manager_; } IDatabase& GetDatabase() { return database_; } }; class StatementBase : public boost::noncopyable { private: DatabaseManager& manager_; boost::recursive_mutex::scoped_lock lock_; ITransaction& transaction_; std::auto_ptr<Query> query_; std::auto_ptr<IResult> result_; IResult& GetResult() const; protected: DatabaseManager& GetManager() const { return manager_; } ITransaction& GetTransaction() const { return transaction_; } void SetQuery(Query* query); void SetResult(IResult* result); void ClearResult() { result_.reset(); } Query* ReleaseQuery() { return query_.release(); } public: StatementBase(DatabaseManager& manager); virtual ~StatementBase(); // Used only by SQLite IDatabase& GetDatabase() { return manager_.GetDatabase(); } void SetReadOnly(bool readOnly); void SetParameterType(const std::string& parameter, ValueType type); bool IsDone() const; void Next(); size_t GetResultFieldsCount() const; void SetResultFieldType(size_t field, ValueType type); const IValue& GetResultField(size_t index) const; }; class CachedStatement : public StatementBase { private: StatementLocation location_; IPrecompiledStatement* statement_; public: CachedStatement(const StatementLocation& location, DatabaseManager& manager, const std::string& sql); void Execute() { Dictionary parameters; Execute(parameters); } void Execute(const Dictionary& parameters); }; class StandaloneStatement : public StatementBase { private: std::auto_ptr<IPrecompiledStatement> statement_; public: StandaloneStatement(DatabaseManager& manager, const std::string& sql); virtual ~StandaloneStatement(); void Execute(const Dictionary& parameters); }; }; }