granicus.if.org Git - icinga2/blob - lib/db_ido/endpointdbobject.cpp

   1 /* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */
   2
   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"
  13
  14 using namespace icinga;
  15
  16 REGISTER_DBTYPE(Endpoint, "endpoint", DbObjectTypeEndpoint, "endpoint_object_id", EndpointDbObject);
  17
  18 INITIALIZE_ONCE(&EndpointDbObject::StaticInitialize);
  19
  20 void EndpointDbObject::StaticInitialize()
  21 {
  22         Endpoint::OnConnected.connect(std::bind(&EndpointDbObject::UpdateConnectedStatus, _1));
  23         Endpoint::OnDisconnected.connect(std::bind(&EndpointDbObject::UpdateConnectedStatus, _1));
  24 }
  25
  26 EndpointDbObject::EndpointDbObject(const DbType::Ptr& type, const String& name1, const String& name2)
  27         : DbObject(type, name1, name2)
  28 { }
  29
  30 Dictionary::Ptr EndpointDbObject::GetConfigFields() const
  31 {
  32         Endpoint::Ptr endpoint = static_pointer_cast<Endpoint>(GetObject());
  33
  34         return new Dictionary({
  35                 { "identity", endpoint->GetName() },
  36                 { "node", IcingaApplication::GetInstance()->GetNodeName() },
  37                 { "zone_object_id", endpoint->GetZone() }
  38         });
  39 }
  40
  41 Dictionary::Ptr EndpointDbObject::GetStatusFields() const
  42 {
  43         Endpoint::Ptr endpoint = static_pointer_cast<Endpoint>(GetObject());
  44
  45
  46         Log(LogDebug, "EndpointDbObject")
  47                 << "update status for endpoint '" << endpoint->GetName() << "'";
  48
  49         return new Dictionary({
  50                 { "identity", endpoint->GetName() },
  51                 { "node", IcingaApplication::GetInstance()->GetNodeName() },
  52                 { "zone_object_id", endpoint->GetZone() },
  53                 { "is_connected", EndpointIsConnected(endpoint) }
  54         });
  55 }
  56
  57 void EndpointDbObject::UpdateConnectedStatus(const Endpoint::Ptr& endpoint)
  58 {
  59         bool connected = EndpointIsConnected(endpoint);
  60
  61         Log(LogDebug, "EndpointDbObject")
  62                 << "update is_connected=" << connected << " for endpoint '" << endpoint->GetName() << "'";
  63
  64         DbQuery query1;
  65         query1.Table = "endpointstatus";
  66         query1.Type = DbQueryUpdate;
  67         query1.Category = DbCatState;
  68
  69         query1.Fields = new Dictionary({
  70                 { "is_connected", (connected ? 1 : 0) },
  71                 { "status_update_time", DbValue::FromTimestamp(Utility::GetTime()) }
  72         });
  73
  74         query1.WhereCriteria = new Dictionary({
  75                 { "endpoint_object_id", endpoint },
  76                 { "instance_id", 0 } /* DbConnection class fills in real ID */
  77         });
  78
  79         OnQuery(query1);
  80 }
  81
  82 int EndpointDbObject::EndpointIsConnected(const Endpoint::Ptr& endpoint)
  83 {
  84         unsigned int is_connected = endpoint->GetConnected() ? 1 : 0;
  85
  86         /* if identity is equal to node, fake is_connected */
  87         if (endpoint->GetName() == IcingaApplication::GetInstance()->GetNodeName())
  88                 is_connected = 1;
  89
  90         return is_connected;
  91 }