icinga2/lib/config/configcompilercontext.cpp
Alexander A. Klimov 34844c146d Deduplicate and stabilize fragile filesystem transactions
by using AtomicFile so they ensure all or nothing of a file gets replaced.
2023-02-15 17:19:57 +01:00

52 lines
1.1 KiB
C++

/* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */
#include "config/configcompilercontext.hpp"
#include "base/singleton.hpp"
#include "base/json.hpp"
#include "base/netstring.hpp"
#include "base/exception.hpp"
#include "base/application.hpp"
#include "base/utility.hpp"
using namespace icinga;
ConfigCompilerContext *ConfigCompilerContext::GetInstance()
{
return Singleton<ConfigCompilerContext>::GetInstance();
}
void ConfigCompilerContext::OpenObjectsFile(const String& filename)
{
try {
m_ObjectsFP.reset(new AtomicFile(filename, 0600));
} catch (const std::exception& ex) {
Log(LogCritical, "cli", "Could not create temporary objects file: " + DiagnosticInformation(ex, false));
Application::Exit(1);
}
}
void ConfigCompilerContext::WriteObject(const Dictionary::Ptr& object)
{
if (!m_ObjectsFP)
return;
String json = JsonEncode(object);
{
std::unique_lock<std::mutex> lock(m_Mutex);
NetString::WriteStringToStream(*m_ObjectsFP, json);
}
}
void ConfigCompilerContext::CancelObjectsFile()
{
m_ObjectsFP.reset(nullptr);
}
void ConfigCompilerContext::FinishObjectsFile()
{
m_ObjectsFP->Commit();
m_ObjectsFP.reset(nullptr);
}