diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-13 11:32:39 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-13 11:32:39 +0000 |
commit | 56ae875861ab260b80a030f50c4aff9f9dc8fff0 (patch) | |
tree | 531412110fc901a5918c7f7442202804a83cada9 /lib/db_ido_pgsql/idopgsqlconnection.hpp | |
parent | Initial commit. (diff) | |
download | icinga2-56ae875861ab260b80a030f50c4aff9f9dc8fff0.tar.xz icinga2-56ae875861ab260b80a030f50c4aff9f9dc8fff0.zip |
Adding upstream version 2.14.2.upstream/2.14.2upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'lib/db_ido_pgsql/idopgsqlconnection.hpp')
-rw-r--r-- | lib/db_ido_pgsql/idopgsqlconnection.hpp | 99 |
1 files changed, 99 insertions, 0 deletions
diff --git a/lib/db_ido_pgsql/idopgsqlconnection.hpp b/lib/db_ido_pgsql/idopgsqlconnection.hpp new file mode 100644 index 0000000..dc06a93 --- /dev/null +++ b/lib/db_ido_pgsql/idopgsqlconnection.hpp @@ -0,0 +1,99 @@ +/* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */ + +#ifndef IDOPGSQLCONNECTION_H +#define IDOPGSQLCONNECTION_H + +#include "db_ido_pgsql/idopgsqlconnection-ti.hpp" +#include "pgsql_shim/pgsqlinterface.hpp" +#include "base/array.hpp" +#include "base/timer.hpp" +#include "base/workqueue.hpp" +#include "base/library.hpp" + +namespace icinga +{ + +typedef std::shared_ptr<PGresult> IdoPgsqlResult; + +/** + * An IDO pgSQL database connection. + * + * @ingroup ido + */ +class IdoPgsqlConnection final : public ObjectImpl<IdoPgsqlConnection> +{ +public: + DECLARE_OBJECT(IdoPgsqlConnection); + DECLARE_OBJECTNAME(IdoPgsqlConnection); + + IdoPgsqlConnection(); + + static void StatsFunc(const Dictionary::Ptr& status, const Array::Ptr& perfdata); + + const char * GetLatestSchemaVersion() const noexcept override; + const char * GetCompatSchemaVersion() const noexcept override; + + int GetPendingQueryCount() const override; + +protected: + void OnConfigLoaded() override; + void Resume() override; + void Pause() override; + + void ActivateObject(const DbObject::Ptr& dbobj) override; + void DeactivateObject(const DbObject::Ptr& dbobj) override; + void ExecuteQuery(const DbQuery& query) override; + void ExecuteMultipleQueries(const std::vector<DbQuery>& queries) override; + void CleanUpExecuteQuery(const String& table, const String& time_key, double time_value) override; + void FillIDCache(const DbType::Ptr& type) override; + void NewTransaction() override; + void Disconnect() override; + +private: + DbReference m_InstanceID; + + Library m_Library; + std::unique_ptr<PgsqlInterface, PgsqlInterfaceDeleter> m_Pgsql; + + PGconn *m_Connection; + int m_AffectedRows; + + Timer::Ptr m_ReconnectTimer; + Timer::Ptr m_TxTimer; + + IdoPgsqlResult Query(const String& query); + DbReference GetSequenceValue(const String& table, const String& column); + int GetAffectedRows(); + String Escape(const String& s); + Dictionary::Ptr FetchRow(const IdoPgsqlResult& result, int row); + + bool FieldToEscapedString(const String& key, const Value& value, Value *result); + void InternalActivateObject(const DbObject::Ptr& dbobj); + void InternalDeactivateObject(const DbObject::Ptr& dbobj); + + void InternalNewTransaction(); + void Reconnect(); + + void AssertOnWorkQueue(); + + void ReconnectTimerHandler(); + + void StatsLoggerTimerHandler(); + + bool CanExecuteQuery(const DbQuery& query); + + void InternalExecuteQuery(const DbQuery& query, int typeOverride = -1); + void InternalExecuteMultipleQueries(const std::vector<DbQuery>& queries); + void InternalCleanUpExecuteQuery(const String& table, const String& time_key, double time_value); + + void ClearTableBySession(const String& table); + void ClearTablesBySession(); + + void ExceptionHandler(boost::exception_ptr exp); + + void FinishConnect(double startTime); +}; + +} + +#endif /* IDOPGSQLCONNECTION_H */ |