mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-13 01:41:59 +00:00
23 lines
788 B
C++
23 lines
788 B
C++
#pragma once
|
|
|
|
#include <base/types.h>
|
|
#include <Common/AllocationTrace.h>
|
|
|
|
/// Convenience methods, that use current thread's memory_tracker if it is available.
|
|
struct CurrentMemoryTracker
|
|
{
|
|
/// Call the following functions before calling of corresponding operations with memory allocators.
|
|
[[nodiscard]] static AllocationTrace alloc(Int64 size);
|
|
[[nodiscard]] static AllocationTrace allocNoThrow(Int64 size);
|
|
|
|
/// This function should be called after memory deallocation.
|
|
[[nodiscard]] static AllocationTrace free(Int64 size);
|
|
static void check();
|
|
|
|
/// Throws MEMORY_LIMIT_EXCEEDED (if it's allowed to throw exceptions)
|
|
static void injectFault();
|
|
|
|
private:
|
|
[[nodiscard]] static AllocationTrace allocImpl(Int64 size, bool throw_if_memory_exceeded);
|
|
};
|