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

ConfigPackageUtility: Don't reset ongoing package updates on config validation success and process is going to be reloaded

Browse Source
This commit is contained in:
Yonas Habteab 2022-03-07 17:50:15 +01:00
parent 575af4c980
commit 362adcab1a
2 changed files with 23 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
lib/remote/configpackageutility.cpp
View File

@ -167,7 +167,8 @@ void ConfigPackageUtility::ActivateStage(const String& packageName, const String
WritePackageConfig(packageName); WritePackageConfig(packageName);
} }
void ConfigPackageUtility::TryActivateStageCallback(const ProcessResult& pr, const String& packageName, const String& stageName, bool activate, bool reload) void ConfigPackageUtility::TryActivateStageCallback(const ProcessResult& pr, const String& packageName, const String& stageName,
bool activate, bool reload, const Shared<Defer>::Ptr& resetPackageUpdates)
{ {
String logFile = GetPackageDir() + "/" + packageName + "/" + stageName + "/startup.log"; String logFile = GetPackageDir() + "/" + packageName + "/" + stageName + "/startup.log";
std::ofstream fpLog(logFile.CStr(), std::ofstream::out | std::ostream::binary | std::ostream::trunc); std::ofstream fpLog(logFile.CStr(), std::ofstream::out | std::ostream::binary | std::ostream::trunc);
@ -188,8 +189,17 @@ void ConfigPackageUtility::TryActivateStageCallback(const ProcessResult& pr, con
ActivateStage(packageName, stageName); ActivateStage(packageName, stageName);
} }
if (reload) if (reload) {
/*
* Cancel the deferred callback before going out of scope so that the config stages handler
* flag isn't resetting earlier and allowing other clients to submit further requests while
* Icinga2 is reloading. Otherwise, the ongoing request will be cancelled halfway before the
* operation is completed once the new worker becomes ready.
*/
resetPackageUpdates->Cancel();
Application::RequestRestart(); Application::RequestRestart();
}
} }
} else { } else {
Log(LogCritical, "ConfigPackageUtility") Log(LogCritical, "ConfigPackageUtility")
@ -198,7 +208,8 @@ void ConfigPackageUtility::TryActivateStageCallback(const ProcessResult& pr, con
} }
} }
void ConfigPackageUtility::AsyncTryActivateStage(const String& packageName, const String& stageName, bool activate, bool reload) void ConfigPackageUtility::AsyncTryActivateStage(const String& packageName, const String& stageName, bool activate, bool reload,
const Shared<Defer>::Ptr& resetPackageUpdates)
{ {
VERIFY(Application::GetArgC() >= 1); VERIFY(Application::GetArgC() >= 1);
@ -224,7 +235,9 @@ void ConfigPackageUtility::AsyncTryActivateStage(const String& packageName, cons
Process::Ptr process = new Process(Process::PrepareCommand(args)); Process::Ptr process = new Process(Process::PrepareCommand(args));
process->SetTimeout(Application::GetReloadTimeout()); process->SetTimeout(Application::GetReloadTimeout());
process->Run([packageName, stageName, activate, reload](const ProcessResult& pr) { TryActivateStageCallback(pr, packageName, stageName, activate, reload); }); process->Run([packageName, stageName, activate, reload, resetPackageUpdates](const ProcessResult& pr) {
TryActivateStageCallback(pr, packageName, stageName, activate, reload, resetPackageUpdates);
});
} }
void ConfigPackageUtility::DeleteStage(const String& packageName, const String& stageName) void ConfigPackageUtility::DeleteStage(const String& packageName, const String& stageName)

8
lib/remote/configpackageutility.hpp
View File

@ -8,6 +8,8 @@
#include "base/dictionary.hpp" #include "base/dictionary.hpp"
#include "base/process.hpp" #include "base/process.hpp"
#include "base/string.hpp" #include "base/string.hpp"
#include "base/defer.hpp"
#include "base/shared.hpp"
#include <vector> #include <vector>
namespace icinga namespace icinga
@ -37,7 +39,8 @@ public:
static void SetActiveStage(const String& packageName, const String& stageName); static void SetActiveStage(const String& packageName, const String& stageName);
static void SetActiveStageToFile(const String& packageName, const String& stageName); static void SetActiveStageToFile(const String& packageName, const String& stageName);
static void ActivateStage(const String& packageName, const String& stageName); static void ActivateStage(const String& packageName, const String& stageName);
static void AsyncTryActivateStage(const String& packageName, const String& stageName, bool activate, bool reload); static void AsyncTryActivateStage(const String& packageName, const String& stageName, bool activate, bool reload,
const Shared<Defer>::Ptr& resetPackageUpdates);
static std::vector<std::pair<String, bool> > GetFiles(const String& packageName, const String& stageName); static std::vector<std::pair<String, bool> > GetFiles(const String& packageName, const String& stageName);
@ -59,7 +62,8 @@ private:
static void WritePackageConfig(const String& packageName); static void WritePackageConfig(const String& packageName);
static void WriteStageConfig(const String& packageName, const String& stageName); static void WriteStageConfig(const String& packageName, const String& stageName);
static void TryActivateStageCallback(const ProcessResult& pr, const String& packageName, const String& stageName, bool activate, bool reload); static void TryActivateStageCallback(const ProcessResult& pr, const String& packageName, const String& stageName, bool activate,
bool reload, const Shared<Defer>::Ptr& resetPackageUpdates);
static bool ValidateFreshName(const String& name); static bool ValidateFreshName(const String& name);
}; };