tyler.durden
/
icinga2
mirror of https://github.com/Icinga/icinga2.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fine-grained locks (WIP, Part 4).

This commit is contained in:
Gunnar Beutner 2013-02-19 07:26:52 +01:00
parent 3dace35cf1
commit ecc95b3dc0
9 changed files with 41 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
lib/base/dynamicobject.cpp
View File

@ -353,8 +353,8 @@ void DynamicObject::Unregister(void)
OnUnregistered(GetSelf());
}
ScriptTask::Ptr DynamicObject::InvokeMethod(const String& method,
const vector<Value>& arguments, ScriptTask::CompletionCallback callback)
ScriptTask::Ptr DynamicObject::MakeMethodTask(const String& method,
const vector<Value>& arguments)
{
Value value = Get("methods");
@ -377,10 +377,7 @@ ScriptTask::Ptr DynamicObject::InvokeMethod(const String& method,
if (!func)
BOOST_THROW_EXCEPTION(invalid_argument("Function '" + funcName + "' does not exist."));
ScriptTask::Ptr task = boost::make_shared<ScriptTask>(func, arguments);
task->Start(callback);
return task;
return boost::make_shared<ScriptTask>(func, arguments);
}
/*

4
lib/base/dynamicobject.h
View File

@ -100,8 +100,8 @@ public:
static signals2::signal<void (const DynamicObject::Ptr&)> OnUnregistered;
static signals2::signal<void (double, const set<DynamicObject *>&)> OnTransactionClosing;
ScriptTask::Ptr InvokeMethod(const String& method,
const vector<Value>& arguments, ScriptTask::CompletionCallback callback);
ScriptTask::Ptr MakeMethodTask(const String& method,
const vector<Value>& arguments);
shared_ptr<DynamicType> GetType(void) const;
String GetName(void) const;

1
lib/base/scripttask.cpp
View File

@ -29,6 +29,5 @@ ScriptTask::ScriptTask(const ScriptFunction::Ptr& function,
void ScriptTask::Run(void)
{
ObjectLock olock(this);
m_Function->Invoke(GetSelf(), m_Arguments);
}

3
lib/icinga/api.cpp
View File

@ -35,5 +35,8 @@ void API::GetAnswerToEverything(const ScriptTask::Ptr& task, const vector<Value>
Logger::Write(LogInformation, "icinga", "Hello from the Icinga 2 API: " + text);
{
ObjectLock olock(task);
task->FinishResult(42);
}
}

9
lib/icinga/notification.cpp
View File

@ -75,8 +75,8 @@ void Notification::SendNotification(NotificationType type)
vector<Value> arguments;
arguments.push_back(static_cast<Notification::Ptr>(GetSelf()));
arguments.push_back(type);
ScriptTask::Ptr task;
task = InvokeMethod("notify", arguments, boost::bind(&Notification::NotificationCompletedHandler, this, _1));
ScriptTask::Ptr task = MakeMethodTask("notify", arguments);
if (!task) {
Logger::Write(LogWarning, "icinga", "Notification object '" + GetName() + "' doesn't have a 'notify' method.");
@ -84,11 +84,10 @@ void Notification::SendNotification(NotificationType type)
return;
}
if (!task->IsFinished()) {
/* We need to keep the task object alive until the completion handler is called. */
m_Tasks.insert(task);
}
task->Start(boost::bind(&Notification::NotificationCompletedHandler, this, _1));
}
void Notification::NotificationCompletedHandler(const ScriptTask::Ptr& task)

3
lib/icinga/nullchecktask.cpp
View File

@ -31,5 +31,8 @@ void NullCheckTask::ScriptFunc(const ScriptTask::Ptr& task, const vector<Value>&
Dictionary::Ptr cr = boost::make_shared<Dictionary>();
cr->Set("state", StateUnknown);
{
ObjectLock olock(task);
task->FinishResult(cr);
}
}

7
lib/icinga/pluginchecktask.cpp
View File

@ -79,7 +79,11 @@ void PluginCheckTask::ProcessFinishedHandler(PluginCheckTask ct)
try {
pr = ct.m_Process->GetResult();
} catch (...) {
{
ObjectLock olock(ct.m_Task);
ct.m_Task->FinishException(boost::current_exception());
}
return;
}
@ -90,8 +94,11 @@ void PluginCheckTask::ProcessFinishedHandler(PluginCheckTask ct)
result->Set("execution_start", pr.ExecutionStart);
result->Set("execution_end", pr.ExecutionEnd);
{
ObjectLock olock(ct.m_Task);
ct.m_Task->FinishResult(result);
}
}
ServiceState PluginCheckTask::ExitStatusToState(int exitStatus)
{

7
lib/icinga/pluginnotificationtask.cpp
View File

@ -107,9 +107,16 @@ void PluginNotificationTask::ProcessFinishedHandler(PluginNotificationTask ct)
Logger::Write(LogWarning, "icinga", msgbuf.str());
}
{
ObjectLock olock(ct.m_Task);
ct.m_Task->FinishResult(Empty);
}
} catch (...) {
{
ObjectLock olock(ct.m_Task);
ct.m_Task->FinishException(boost::current_exception());
}
return;
}
}

15
lib/icinga/service-check.cpp
View File

@ -435,24 +435,13 @@ void Service::BeginExecuteCheck(const function<void (void)>& callback)
scheduleInfo->Set("schedule_start", GetNextCheck());
scheduleInfo->Set("execution_start", Utility::GetTime());
try {
vector<Value> arguments;
arguments.push_back(static_cast<Service::Ptr>(GetSelf()));
ScriptTask::Ptr task;
task = InvokeMethod("check", arguments, boost::bind(&Service::CheckCompletedHandler, this, scheduleInfo, _1, callback));
if (!task->IsFinished())
ScriptTask::Ptr task = MakeMethodTask("check", arguments);
Set("current_task", task);
} catch (...) {
/* something went wrong while setting up the method call -
* reschedule the service and call the callback anyway. */
UpdateNextCheck();
callback();
throw;
}
task->Start(boost::bind(&Service::CheckCompletedHandler, this, scheduleInfo, _1, callback));
}
void Service::CheckCompletedHandler(const Dictionary::Ptr& scheduleInfo,