add ObjectsPool
This commit is contained in:
parent
922a86d2b0
commit
5fdb71c47b
@ -5,8 +5,6 @@
|
||||
#include <queue>
|
||||
#include <vector>
|
||||
|
||||
#include "typedefs.hpp"
|
||||
|
||||
namespace util {
|
||||
/// @brief Thread-safe pool of same-sized buffers
|
||||
/// @tparam T array type
|
||||
|
||||
51
src/util/ObjectsPool.hpp
Normal file
51
src/util/ObjectsPool.hpp
Normal file
@ -0,0 +1,51 @@
|
||||
#pragma once
|
||||
|
||||
#include <memory>
|
||||
#include <mutex>
|
||||
#include <vector>
|
||||
#include <queue>
|
||||
|
||||
namespace util {
|
||||
struct AlignedDeleter {
|
||||
void operator()(void* p) const {
|
||||
std::free(p);
|
||||
}
|
||||
};
|
||||
|
||||
template <class T>
|
||||
class ObjectsPool {
|
||||
public:
|
||||
ObjectsPool(size_t preallocated = 0) {
|
||||
for (size_t i = 0; i < preallocated; i++) {
|
||||
allocateNew();
|
||||
}
|
||||
}
|
||||
|
||||
template<typename... Args>
|
||||
std::shared_ptr<T> create(Args&&... args) {
|
||||
std::lock_guard lock(mutex);
|
||||
if (freeObjects.empty()) {
|
||||
allocateNew();
|
||||
}
|
||||
auto ptr = freeObjects.front();
|
||||
freeObjects.pop();
|
||||
new (ptr)T(std::forward<Args>(args)...);
|
||||
return std::shared_ptr<T>(reinterpret_cast<T*>(ptr), [this](T* ptr) {
|
||||
std::lock_guard lock(mutex);
|
||||
freeObjects.push(ptr);
|
||||
});
|
||||
}
|
||||
private:
|
||||
std::vector<std::unique_ptr<void, AlignedDeleter>> objects;
|
||||
std::queue<void*> freeObjects;
|
||||
std::mutex mutex;
|
||||
|
||||
void allocateNew() {
|
||||
std::unique_ptr<void, AlignedDeleter> ptr(
|
||||
std::aligned_alloc(alignof(T), sizeof(T))
|
||||
);
|
||||
freeObjects.push(ptr.get());
|
||||
objects.push_back(std::move(ptr));
|
||||
}
|
||||
};
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user