library/cpp/messagebus/actor/what_thread_does_guard.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

#pragma once

#include "what_thread_does.h"

template <class T>
class TWhatThreadDoesAcquireGuard: public TNonCopyable {
public: 
    inline TWhatThreadDoesAcquireGuard(const T& t, const char* acquire) noexcept { 
        Init(&t, acquire); 
    } 

    inline TWhatThreadDoesAcquireGuard(const T* t, const char* acquire) noexcept { 
        Init(t, acquire); 
    } 

    inline ~TWhatThreadDoesAcquireGuard() { 
        Release(); 
    } 

    inline void Release() noexcept { 
        if (WasAcquired()) { 
            const_cast<T*>(T_)->Release(); 
            T_ = nullptr; 
        }
    } 

    inline bool WasAcquired() const noexcept { 
        return T_ != nullptr; 
    } 

private: 
    inline void Init(const T* t, const char* acquire) noexcept { 
        T_ = const_cast<T*>(t); 
        TWhatThreadDoesPushPop pp(acquire); 
        T_->Acquire(); 
    } 

private: 
    T* T_; 
};