blob: 666fe560e171053557564d676f46d2f2dbd95ae4 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
/* 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);
}
|