1 /* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */
3 #include "db_ido/endpointdbobject.hpp"
4 #include "db_ido/dbtype.hpp"
5 #include "db_ido/dbvalue.hpp"
6 #include "icinga/icingaapplication.hpp"
7 #include "base/objectlock.hpp"
8 #include "base/initialize.hpp"
9 #include "base/configtype.hpp"
10 #include "base/utility.hpp"
11 #include "base/convert.hpp"
12 #include "base/logger.hpp"
14 using namespace icinga;
16 REGISTER_DBTYPE(Endpoint, "endpoint", DbObjectTypeEndpoint, "endpoint_object_id", EndpointDbObject);
18 INITIALIZE_ONCE(&EndpointDbObject::StaticInitialize);
20 void EndpointDbObject::StaticInitialize()
22 Endpoint::OnConnected.connect(std::bind(&EndpointDbObject::UpdateConnectedStatus, _1));
23 Endpoint::OnDisconnected.connect(std::bind(&EndpointDbObject::UpdateConnectedStatus, _1));
26 EndpointDbObject::EndpointDbObject(const DbType::Ptr& type, const String& name1, const String& name2)
27 : DbObject(type, name1, name2)
30 Dictionary::Ptr EndpointDbObject::GetConfigFields() const
32 Endpoint::Ptr endpoint = static_pointer_cast<Endpoint>(GetObject());
34 return new Dictionary({
35 { "identity", endpoint->GetName() },
36 { "node", IcingaApplication::GetInstance()->GetNodeName() },
37 { "zone_object_id", endpoint->GetZone() }
41 Dictionary::Ptr EndpointDbObject::GetStatusFields() const
43 Endpoint::Ptr endpoint = static_pointer_cast<Endpoint>(GetObject());
46 Log(LogDebug, "EndpointDbObject")
47 << "update status for endpoint '" << endpoint->GetName() << "'";
49 return new Dictionary({
50 { "identity", endpoint->GetName() },
51 { "node", IcingaApplication::GetInstance()->GetNodeName() },
52 { "zone_object_id", endpoint->GetZone() },
53 { "is_connected", EndpointIsConnected(endpoint) }
57 void EndpointDbObject::UpdateConnectedStatus(const Endpoint::Ptr& endpoint)
59 bool connected = EndpointIsConnected(endpoint);
61 Log(LogDebug, "EndpointDbObject")
62 << "update is_connected=" << connected << " for endpoint '" << endpoint->GetName() << "'";
65 query1.Table = "endpointstatus";
66 query1.Type = DbQueryUpdate;
67 query1.Category = DbCatState;
69 query1.Fields = new Dictionary({
70 { "is_connected", (connected ? 1 : 0) },
71 { "status_update_time", DbValue::FromTimestamp(Utility::GetTime()) }
74 query1.WhereCriteria = new Dictionary({
75 { "endpoint_object_id", endpoint },
76 { "instance_id", 0 } /* DbConnection class fills in real ID */
82 int EndpointDbObject::EndpointIsConnected(const Endpoint::Ptr& endpoint)
84 unsigned int is_connected = endpoint->GetConnected() ? 1 : 0;
86 /* if identity is equal to node, fake is_connected */
87 if (endpoint->GetName() == IcingaApplication::GetInstance()->GetNodeName())