blob: dc06a9384aa427a88460455137f44cedac67f28a (
plain)
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
|
/* 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 */
|