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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
|
/* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */
#include "db_ido/dbtype.hpp"
#include "db_ido/dbconnection.hpp"
#include "base/objectlock.hpp"
#include "base/debug.hpp"
#include <boost/thread/once.hpp>
using namespace icinga;
DbType::DbType(String name, String table, long tid, String idcolumn, DbType::ObjectFactory factory)
: m_Name(std::move(name)), m_Table(std::move(table)), m_TypeID(tid), m_IDColumn(std::move(idcolumn)), m_ObjectFactory(std::move(factory))
{ }
String DbType::GetName() const
{
return m_Name;
}
String DbType::GetTable() const
{
return m_Table;
}
long DbType::GetTypeID() const
{
return m_TypeID;
}
String DbType::GetIDColumn() const
{
return m_IDColumn;
}
void DbType::RegisterType(const DbType::Ptr& type)
{
std::unique_lock<std::mutex> lock(GetStaticMutex());
GetTypes()[type->GetName()] = type;
}
DbType::Ptr DbType::GetByName(const String& name)
{
String typeName;
if (name == "CheckCommand" || name == "NotificationCommand" || name == "EventCommand")
typeName = "Command";
else
typeName = name;
std::unique_lock<std::mutex> lock(GetStaticMutex());
auto it = GetTypes().find(typeName);
if (it == GetTypes().end())
return nullptr;
return it->second;
}
DbType::Ptr DbType::GetByID(long tid)
{
std::unique_lock<std::mutex> lock(GetStaticMutex());
for (const TypeMap::value_type& kv : GetTypes()) {
if (kv.second->GetTypeID() == tid)
return kv.second;
}
return nullptr;
}
DbObject::Ptr DbType::GetOrCreateObjectByName(const String& name1, const String& name2)
{
ObjectLock olock(this);
auto it = m_Objects.find(std::make_pair(name1, name2));
if (it != m_Objects.end())
return it->second;
DbObject::Ptr dbobj = m_ObjectFactory(this, name1, name2);
m_Objects[std::make_pair(name1, name2)] = dbobj;
String objName = name1;
if (!name2.IsEmpty())
objName += "!" + name2;
String objType = m_Name;
if (m_TypeID == DbObjectTypeCommand) {
if (objName.SubStr(0, 6) == "check_") {
objType = "CheckCommand";
objName = objName.SubStr(6);
} else if (objName.SubStr(0, 13) == "notification_") {
objType = "NotificationCommand";
objName = objName.SubStr(13);
} else if (objName.SubStr(0, 6) == "event_") {
objType = "EventCommand";
objName = objName.SubStr(6);
}
}
dbobj->SetObject(ConfigObject::GetObject(objType, objName));
return dbobj;
}
std::mutex& DbType::GetStaticMutex()
{
static std::mutex mutex;
return mutex;
}
/**
* Caller must hold static mutex.
*/
DbType::TypeMap& DbType::GetTypes()
{
static DbType::TypeMap tm;
return tm;
}
std::set<DbType::Ptr> DbType::GetAllTypes()
{
std::set<DbType::Ptr> result;
{
std::unique_lock<std::mutex> lock(GetStaticMutex());
for (const auto& kv : GetTypes()) {
result.insert(kv.second);
}
}
return result;
}
DbTypeRegistry *DbTypeRegistry::GetInstance()
{
return Singleton<DbTypeRegistry>::GetInstance();
}
|
