summaryrefslogtreecommitdiffstats
path: root/lib/base/shared.hpp
blob: 63b35cb2f131644dba3caac457ac394bf6526fc5 (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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
/* Icinga 2 | (c) 2019 Icinga GmbH | GPLv2+ */

#ifndef SHARED_H
#define SHARED_H

#include "base/atomic.hpp"
#include <boost/smart_ptr/intrusive_ptr.hpp>
#include <cstdint>
#include <utility>

namespace icinga
{

template<class T>
class Shared;

template<class T>
inline void intrusive_ptr_add_ref(Shared<T> *object)
{
	object->m_References.fetch_add(1);
}

template<class T>
inline void intrusive_ptr_release(Shared<T> *object)
{
	if (object->m_References.fetch_sub(1) == 1u) {
		delete object;
	}
}

/**
 * Seamless wrapper for any class to create shared pointers of.
 * Saves a memory allocation compared to std::shared_ptr.
 *
 * @ingroup base
 */
template<class T>
class Shared : public T
{
	friend void intrusive_ptr_add_ref<>(Shared<T> *object);
	friend void intrusive_ptr_release<>(Shared<T> *object);

public:
	typedef boost::intrusive_ptr<Shared> Ptr;

	/**
	 * Like std::make_shared, but for this class.
	 *
	 * @param args Constructor arguments
	 *
	 * @return Ptr
	 */
	template<class... Args>
	static inline
	Ptr Make(Args&&... args)
	{
		return new Shared(std::forward<Args>(args)...);
	}

	inline Shared(const Shared& origin) : Shared((const T&)origin)
	{
	}

	inline Shared(Shared&& origin) : Shared((T&&)origin)
	{
	}

	template<class... Args>
	inline Shared(Args&&... args) : T(std::forward<Args>(args)...), m_References(0)
	{
	}

	inline Shared& operator=(const Shared& rhs)
	{
		return operator=((const T&)rhs);
	}

	inline Shared& operator=(Shared&& rhs)
	{
		return operator=((T&&)rhs);
	}

	inline Shared& operator=(const T& rhs)
	{
		T::operator=(rhs);
		return *this;
	}

	inline Shared& operator=(T&& rhs)
	{
		T::operator=(std::move(rhs));
		return *this;
	}

private:
	Atomic<uint_fast64_t> m_References;
};

}

#endif /* SHARED_H */