summaryrefslogtreecommitdiffstats
path: root/lib/base/context.cpp
blob: 9c0a7810624b524cea7718c1b30afc97fa7b02d0 (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
/* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */

#include "base/context.hpp"
#include <boost/thread/tss.hpp>
#include <iostream>
#include <sstream>
#include <utility>

using namespace icinga;

static boost::thread_specific_ptr<std::vector<std::function<void(std::ostream&)>>> l_Frames;

ContextFrame::ContextFrame(std::function<void(std::ostream&)> message)
{
	GetFrames().emplace_back(std::move(message));
}

ContextFrame::~ContextFrame()
{
	GetFrames().pop_back();
}

std::vector<std::function<void(std::ostream&)>>& ContextFrame::GetFrames()
{
	if (!l_Frames.get())
		l_Frames.reset(new std::vector<std::function<void(std::ostream&)>>());

	return *l_Frames;
}

ContextTrace::ContextTrace()
{
	for (auto frame (ContextFrame::GetFrames().rbegin()); frame != ContextFrame::GetFrames().rend(); ++frame) {
		std::ostringstream oss;

		(*frame)(oss);
		m_Frames.emplace_back(oss.str());
	}
}

void ContextTrace::Print(std::ostream& fp) const
{
	if (m_Frames.empty())
		return;

	fp << "\n";

	int i = 0;
	for (const String& frame : m_Frames) {
		fp << "\t(" << i << ") " << frame << "\n";
		i++;
	}
}

size_t ContextTrace::GetLength() const
{
	return m_Frames.size();
}

std::ostream& icinga::operator<<(std::ostream& stream, const ContextTrace& trace)
{
	trace.Print(stream);
	return stream;
}