tyler.durden/icinga2
1
0
Fork 0
mirror of https://github.com/Icinga/icinga2.git synced 2025-07-25 14:44:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

ido: Add acknowledgements.

Browse Source 
refs #4379
This commit is contained in:
Michael Friedrich 2013-09-24 11:49:02 +02:00
parent 2ca6ec42e2
commit ace2c742c1
2 changed files with 71 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

66
lib/ido/servicedbobject.cpp
View File

@ -40,11 +40,16 @@ INITIALIZE_ONCE(ServiceDbObject, &ServiceDbObject::StaticInitialize);
void ServiceDbObject::StaticInitialize(void) void ServiceDbObject::StaticInitialize(void)
{ {
/* Status */
Service::OnCommentAdded.connect(boost::bind(&ServiceDbObject::AddComment, _1, _2)); Service::OnCommentAdded.connect(boost::bind(&ServiceDbObject::AddComment, _1, _2));
Service::OnCommentRemoved.connect(boost::bind(&ServiceDbObject::RemoveComment, _1, _2)); Service::OnCommentRemoved.connect(boost::bind(&ServiceDbObject::RemoveComment, _1, _2));
Service::OnDowntimeAdded.connect(boost::bind(&ServiceDbObject::AddDowntime, _1, _2)); Service::OnDowntimeAdded.connect(boost::bind(&ServiceDbObject::AddDowntime, _1, _2));
Service::OnDowntimeRemoved.connect(boost::bind(&ServiceDbObject::RemoveDowntime, _1, _2)); Service::OnDowntimeRemoved.connect(boost::bind(&ServiceDbObject::RemoveDowntime, _1, _2));
Service::OnDowntimeTriggered.connect(boost::bind(&ServiceDbObject::TriggerDowntime, _1, _2)); Service::OnDowntimeTriggered.connect(boost::bind(&ServiceDbObject::TriggerDowntime, _1, _2));
/* History */
Service::OnAcknowledgementSet.connect(boost::bind(&ServiceDbObject::AddAcknowledgement, _1, _2, _3, _4, _5, _6));
} }
ServiceDbObject::ServiceDbObject(const DbType::Ptr& type, const String& name1, const String& name2) ServiceDbObject::ServiceDbObject(const DbType::Ptr& type, const String& name1, const String& name2)
@ -603,6 +608,7 @@ void ServiceDbObject::TriggerDowntime(const Service::Ptr& service, const Diction
fields1->Set("actual_start_time_usec", actual_start_time_usec); fields1->Set("actual_start_time_usec", actual_start_time_usec);
fields1->Set("is_in_effect", 1); fields1->Set("is_in_effect", 1);
fields1->Set("trigger_time", DbValue::FromTimestamp(downtime->Get("trigger_time"))); fields1->Set("trigger_time", DbValue::FromTimestamp(downtime->Get("trigger_time")));
fields1->Set("instance_id", 0); /* DbConnection class fills in real ID */
query1.WhereCriteria = boost::make_shared<Dictionary>(); query1.WhereCriteria = boost::make_shared<Dictionary>();
query1.WhereCriteria->Set("object_id", service); query1.WhereCriteria->Set("object_id", service);
@ -624,6 +630,7 @@ void ServiceDbObject::TriggerDowntime(const Service::Ptr& service, const Diction
fields2->Set("actual_start_time_usec", actual_start_time_usec); fields2->Set("actual_start_time_usec", actual_start_time_usec);
fields2->Set("is_in_effect", 1); fields2->Set("is_in_effect", 1);
fields2->Set("trigger_time", DbValue::FromTimestamp(downtime->Get("trigger_time"))); fields2->Set("trigger_time", DbValue::FromTimestamp(downtime->Get("trigger_time")));
fields2->Set("instance_id", 0); /* DbConnection class fills in real ID */
query2.WhereCriteria = boost::make_shared<Dictionary>(); query2.WhereCriteria = boost::make_shared<Dictionary>();
query2.WhereCriteria->Set("object_id", host); query2.WhereCriteria->Set("object_id", host);
@ -633,3 +640,62 @@ void ServiceDbObject::TriggerDowntime(const Service::Ptr& service, const Diction
OnQuery(query2); OnQuery(query2);
} }
} }
void ServiceDbObject::AddAcknowledgement(const Service::Ptr& service, const String& author, const String& comment,
AcknowledgementType type, double expiry, const String& authority)
{
Host::Ptr host = service->GetHost();
if (!host)
return;
Log(LogDebug, "ido", "add acknowledgement for '" + service->GetName() + "'");
double now = Utility::GetTime();
unsigned long entry_time = static_cast<long>(now);
unsigned long entry_time_usec = (now - entry_time) * 1000 * 1000;
unsigned long end_time = static_cast<long>(expiry);
DbQuery query1;
query1.Table = "acknowledgements";
query1.Type = DbQueryInsert;
Dictionary::Ptr fields1 = boost::make_shared<Dictionary>();
fields1->Set("entry_time", DbValue::FromTimestamp(entry_time));
fields1->Set("entry_time_usec", entry_time_usec);
fields1->Set("acknowledgement_type", type);
fields1->Set("object_id", service);
fields1->Set("state", service->GetState());
fields1->Set("author_name", author);
fields1->Set("comment_data", comment);
fields1->Set("is_sticky", type == AcknowledgementSticky ? 1 : 0);
fields1->Set("end_time", DbValue::FromTimestamp(end_time));
fields1->Set("instance_id", 0); /* DbConnection class fills in real ID */
query1.Fields = fields1;
OnQuery(query1);
if (host->GetHostCheckService() == service) {
DbQuery query2;
query2.Table = "acknowledgements";
query2.Type = DbQueryInsert;
Dictionary::Ptr fields2 = boost::make_shared<Dictionary>();
fields2->Set("entry_time", DbValue::FromTimestamp(entry_time));
fields2->Set("entry_time_usec", entry_time_usec);
fields2->Set("acknowledgement_type", type);
fields2->Set("object_id", host);
fields2->Set("state", service->GetState());
fields2->Set("author_name", author);
fields2->Set("comment_data", comment);
fields2->Set("is_sticky", type == AcknowledgementSticky ? 1 : 0);
fields2->Set("end_time", DbValue::FromTimestamp(end_time));
fields2->Set("instance_id", 0); /* DbConnection class fills in real ID */
query2.Fields = fields2;
OnQuery(query2);
}
}

5
lib/ido/servicedbobject.h
View File

@ -51,6 +51,7 @@ protected:
virtual void OnStatusUpdate(void); virtual void OnStatusUpdate(void);
private: private:
/* Status */
static void AddComments(const Service::Ptr& service); static void AddComments(const Service::Ptr& service);
static void AddComment(const Service::Ptr& service, const Dictionary::Ptr& comment); static void AddComment(const Service::Ptr& service, const Dictionary::Ptr& comment);
static void AddCommentByType(const DynamicObject::Ptr& object, const Dictionary::Ptr& comment); static void AddCommentByType(const DynamicObject::Ptr& object, const Dictionary::Ptr& comment);
@ -63,6 +64,10 @@ private:
static void RemoveDowntimes(const Service::Ptr& service); static void RemoveDowntimes(const Service::Ptr& service);
static void RemoveDowntime(const Service::Ptr& service, const Dictionary::Ptr& downtime); static void RemoveDowntime(const Service::Ptr& service, const Dictionary::Ptr& downtime);
static void TriggerDowntime(const Service::Ptr& service, const Dictionary::Ptr& downtime); static void TriggerDowntime(const Service::Ptr& service, const Dictionary::Ptr& downtime);
/* History */
static void AddAcknowledgement(const Service::Ptr& service, const String& author, const String& comment, AcknowledgementType type, double expiry, const String& authority);
}; };
} }