blob: 27a8e9394b421c6edd375dff1299bc83636bbac2 (
plain) (
blame)
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
|
#pragma once
#include "stack.h"
#include "stack_common.h"
#include <util/generic/noncopyable.h>
#include <util/generic/ptr.h>
#include <util/generic/vector.h>
namespace NCoro::NStack {
class IGuard;
class TStorage;
struct TPoolAllocatorSettings;
template<typename TGuard>
class TPool final : private TMoveOnly {
struct TMemory {
char* Raw = nullptr;
char* Aligned = nullptr; // points to aligned memory, which includes space for first page guard
};
public:
TPool(uint64_t stackSize, const TPoolAllocatorSettings& settings, const TGuard& guard);
TPool(TPool&& other) noexcept;
~TPool();
NDetails::TStack AllocStack(const char* name);
void FreeStack(NDetails::TStack& stack);
uint64_t GetReleasedSize() const noexcept;
uint64_t GetFullSize() const noexcept;
uint64_t GetNumOfAllocated() const noexcept { return NumOfAllocated_; }
private:
void AllocNewMemoryChunk();
bool IsSmallStack() const noexcept;
bool IsStackFromThisPool(const NDetails::TStack& stack) const noexcept;
NDetails::TStack AllocNewStack(const char* name);
private:
const uint64_t StackSize_ = 0;
uint64_t RssPagesToKeep_ = 0;
const TGuard& Guard_;
TVector<TMemory> Memory_; // memory chunks
THolder<TStorage> Storage_;
char* NextToAlloc_ = nullptr; // points to next available stack in the last memory chunk
const uint64_t ChunkSize_ = 0;
uint64_t NumOfAllocated_ = 0;
};
}
#include "stack_pool.inl"
|
