mirror of https://github.com/Icinga/icinga2.git
69 lines
1.5 KiB
C++
69 lines
1.5 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)
|
|
{
|
|
m_ObjectsPath = filename;
|
|
|
|
auto *fp = new std::fstream();
|
|
try {
|
|
m_ObjectsTempFile = Utility::CreateTempFile(filename + ".XXXXXX", 0600, *fp);
|
|
} catch (const std::exception& ex) {
|
|
Log(LogCritical, "cli", "Could not create temporary objects file: " + DiagnosticInformation(ex, false));
|
|
Application::Exit(1);
|
|
}
|
|
|
|
if (!*fp)
|
|
BOOST_THROW_EXCEPTION(std::runtime_error("Could not open '" + m_ObjectsTempFile + "' file"));
|
|
|
|
m_ObjectsFP = fp;
|
|
}
|
|
|
|
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()
|
|
{
|
|
delete m_ObjectsFP;
|
|
m_ObjectsFP = nullptr;
|
|
|
|
#ifdef _WIN32
|
|
_unlink(m_ObjectsTempFile.CStr());
|
|
#else /* _WIN32 */
|
|
unlink(m_ObjectsTempFile.CStr());
|
|
#endif /* _WIN32 */
|
|
}
|
|
|
|
void ConfigCompilerContext::FinishObjectsFile()
|
|
{
|
|
delete m_ObjectsFP;
|
|
m_ObjectsFP = nullptr;
|
|
|
|
Utility::RenameFile(m_ObjectsTempFile, m_ObjectsPath);
|
|
}
|
|
|