2014-05-03 22:57:43 +00:00
|
|
|
|
#pragma once
|
|
|
|
|
|
|
|
|
|
#include <Yandex/Common.h>
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/** Отслеживает потребление памяти.
|
|
|
|
|
* Кидает исключение, если оно стало бы больше некоторого предельного значения.
|
|
|
|
|
* Один объект может использоваться одновременно в разных потоках.
|
|
|
|
|
*/
|
|
|
|
|
class MemoryTracker
|
|
|
|
|
{
|
2014-05-08 15:54:51 +00:00
|
|
|
|
Int64 amount = 0;
|
|
|
|
|
Int64 peak = 0;
|
|
|
|
|
Int64 limit = 0;
|
2014-05-03 22:57:43 +00:00
|
|
|
|
|
|
|
|
|
public:
|
2014-05-08 15:54:51 +00:00
|
|
|
|
MemoryTracker(Int64 limit_) : limit(limit_) {}
|
2014-05-03 22:57:43 +00:00
|
|
|
|
|
|
|
|
|
~MemoryTracker();
|
|
|
|
|
|
|
|
|
|
/** Вызывайте эти функции перед соответствующими операциями с памятью.
|
|
|
|
|
*/
|
2014-05-08 15:54:51 +00:00
|
|
|
|
void alloc(Int64 size);
|
2014-05-03 22:57:43 +00:00
|
|
|
|
|
2014-05-08 15:54:51 +00:00
|
|
|
|
void realloc(Int64 old_size, Int64 new_size)
|
2014-05-03 22:57:43 +00:00
|
|
|
|
{
|
|
|
|
|
alloc(new_size - old_size);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/** А эту функцию имеет смысл вызывать после освобождения памяти.
|
|
|
|
|
*/
|
2014-05-08 15:54:51 +00:00
|
|
|
|
void free(Int64 size)
|
2014-05-03 22:57:43 +00:00
|
|
|
|
{
|
|
|
|
|
__sync_sub_and_fetch(&amount, size);
|
|
|
|
|
}
|
|
|
|
|
|
2014-05-08 15:54:51 +00:00
|
|
|
|
Int64 get() const
|
2014-05-03 22:57:43 +00:00
|
|
|
|
{
|
|
|
|
|
return amount;
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/** Объект MemoryTracker довольно трудно протащить во все места, где выделяются существенные объёмы памяти.
|
|
|
|
|
* Поэтому, используется thread-local указатель на используемый MemoryTracker или nullptr, если его не нужно использовать.
|
|
|
|
|
* Этот указатель выставляется, когда в данном потоке следует отслеживать потребление памяти.
|
|
|
|
|
* Таким образом, его нужно всего-лишь протащить во все потоки, в которых обрабатывается один запрос.
|
|
|
|
|
*/
|
|
|
|
|
extern __thread MemoryTracker * current_memory_tracker;
|