library/cpp/messagebus/test/helper/alloc_counter.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

#pragma once 
 
#include <util/generic/noncopyable.h> 
#include <util/system/atomic.h> 
#include <util/system/yassert.h> 
 
class TAllocCounter : TNonCopyable {
private: 
    TAtomic* CountPtr; 

public: 
    TAllocCounter(TAtomic* countPtr)
        : CountPtr(countPtr)
    {
        AtomicIncrement(*CountPtr); 
    } 
 
    ~TAllocCounter() { 
        Y_VERIFY(AtomicDecrement(*CountPtr) >= 0, "released too many");
    } 
};