tdlight-team/tdlight
tdlight-team/tdlight
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Use new/delete instead of ArrayAllocator on external threads.

Browse Source
This commit is contained in:
levlam 2021-11-18 16:22:54 +03:00
parent 32713560b8
commit 6f3dc2afb1
1 changed files with 21 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
tdutils/td/utils/StackAllocator.cpp
View File

@ -49,6 +49,19 @@ class ArrayAllocator final : public StackAllocator::AllocatorImpl {
}
}
};
class NewAllocator final : public StackAllocator::AllocatorImpl {
MutableSlice allocate(size_t size) final {
return {new char[size], size};
}
void free_ptr(char *ptr, size_t size) final {
delete[] ptr;
}
public:
~NewAllocator() final = default;
};
} // namespace
StackAllocator::Ptr::~Ptr() {
@ -58,9 +71,14 @@ StackAllocator::Ptr::~Ptr() {
}
StackAllocator::AllocatorImpl *StackAllocator::impl() {
static TD_THREAD_LOCAL ArrayAllocator *array_allocator; // static zero-initialized
init_thread_local<ArrayAllocator>(array_allocator);
return array_allocator;
if (get_thread_id() != 0) {
static TD_THREAD_LOCAL ArrayAllocator *array_allocator; // static zero-initialized
init_thread_local<ArrayAllocator>(array_allocator);
return array_allocator;
} else {
static NewAllocator new_allocator;
return &new_allocator;
}
}
} // namespace td