2015-09-29 19:19:54 +00:00
|
|
|
|
#include <common/likely.h>
|
|
|
|
|
#include <common/logger_useful.h>
|
2015-10-05 01:35:28 +00:00
|
|
|
|
#include <DB/Common/Exception.h>
|
2014-10-25 18:33:52 +00:00
|
|
|
|
#include <DB/Common/formatReadable.h>
|
2014-05-03 22:57:43 +00:00
|
|
|
|
#include <DB/IO/WriteHelpers.h>
|
2014-05-08 12:00:47 +00:00
|
|
|
|
#include <iomanip>
|
2014-05-03 22:57:43 +00:00
|
|
|
|
|
|
|
|
|
#include <DB/Common/MemoryTracker.h>
|
|
|
|
|
|
|
|
|
|
|
2016-01-11 21:46:36 +00:00
|
|
|
|
namespace DB
|
|
|
|
|
{
|
|
|
|
|
namespace ErrorCodes
|
|
|
|
|
{
|
|
|
|
|
extern const int MEMORY_LIMIT_EXCEEDED;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2014-05-03 22:57:43 +00:00
|
|
|
|
MemoryTracker::~MemoryTracker()
|
|
|
|
|
{
|
2015-12-30 15:39:11 +00:00
|
|
|
|
logPeakMemoryUsage();
|
2014-05-03 22:57:43 +00:00
|
|
|
|
}
|
|
|
|
|
|
2015-12-30 15:39:11 +00:00
|
|
|
|
|
|
|
|
|
void MemoryTracker::logPeakMemoryUsage() const
|
|
|
|
|
{
|
|
|
|
|
LOG_DEBUG(&Logger::get("MemoryTracker"),
|
|
|
|
|
"Peak memory usage" << (description ? " " + std::string(description) : "")
|
|
|
|
|
<< ": " << formatReadableSizeWithBinarySuffix(peak) << ".");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2014-05-08 15:54:51 +00:00
|
|
|
|
void MemoryTracker::alloc(Int64 size)
|
2014-05-03 22:57:43 +00:00
|
|
|
|
{
|
2014-05-08 15:54:51 +00:00
|
|
|
|
Int64 will_be = __sync_add_and_fetch(&amount, size);
|
2014-05-03 22:57:43 +00:00
|
|
|
|
|
2015-12-23 07:39:28 +00:00
|
|
|
|
/// Используется непотокобезопасный генератор случайных чисел. Совместное распределение в разных потоках не будет равномерным.
|
|
|
|
|
/// В данном случае, это нормально.
|
|
|
|
|
if (unlikely(fault_probability && drand48() < fault_probability))
|
|
|
|
|
{
|
|
|
|
|
free(size);
|
2015-12-30 15:39:11 +00:00
|
|
|
|
|
|
|
|
|
std::stringstream message;
|
|
|
|
|
message << "Memory tracker";
|
|
|
|
|
if (description)
|
|
|
|
|
message << " " << description;
|
|
|
|
|
message << ": fault injected. Would use " << formatReadableSizeWithBinarySuffix(will_be)
|
|
|
|
|
<< " (attempt to allocate chunk of " << size << " bytes)"
|
|
|
|
|
<< ", maximum: " << formatReadableSizeWithBinarySuffix(limit);
|
|
|
|
|
|
|
|
|
|
throw DB::Exception(message.str(), DB::ErrorCodes::MEMORY_LIMIT_EXCEEDED);
|
2015-12-23 07:39:28 +00:00
|
|
|
|
}
|
|
|
|
|
|
2014-05-03 22:57:43 +00:00
|
|
|
|
if (unlikely(limit && will_be > limit))
|
|
|
|
|
{
|
|
|
|
|
free(size);
|
2015-12-30 15:39:11 +00:00
|
|
|
|
|
|
|
|
|
std::stringstream message;
|
|
|
|
|
message << "Memory limit";
|
|
|
|
|
if (description)
|
|
|
|
|
message << " " << description;
|
|
|
|
|
message << " exceeded: would use " << formatReadableSizeWithBinarySuffix(will_be)
|
|
|
|
|
<< " (attempt to allocate chunk of " << size << " bytes)"
|
|
|
|
|
<< ", maximum: " << formatReadableSizeWithBinarySuffix(limit);
|
|
|
|
|
|
|
|
|
|
throw DB::Exception(message.str(), DB::ErrorCodes::MEMORY_LIMIT_EXCEEDED);
|
2014-05-03 22:57:43 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (will_be > peak)
|
|
|
|
|
peak = will_be;
|
2015-12-30 15:39:11 +00:00
|
|
|
|
|
|
|
|
|
if (next)
|
|
|
|
|
next->alloc(size);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
void MemoryTracker::free(Int64 size)
|
|
|
|
|
{
|
|
|
|
|
__sync_sub_and_fetch(&amount, size);
|
|
|
|
|
|
|
|
|
|
if (next)
|
|
|
|
|
next->free(size);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
void MemoryTracker::reset()
|
|
|
|
|
{
|
|
|
|
|
amount = 0;
|
|
|
|
|
peak = 0;
|
|
|
|
|
limit = 0;
|
2014-05-03 22:57:43 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
__thread MemoryTracker * current_memory_tracker = nullptr;
|