File tree 2 files changed +9
-9
lines changed 2 files changed +9
-9
lines changed Original file line number Diff line number Diff line change 1
1
#include " should_continue.h"
2
2
3
3
void TProgramShouldContinue::ShouldRestart () {
4
- AtomicSet ( State, Restart );
4
+ State. store (Restart, std::memory_order_release );
5
5
}
6
6
7
7
void TProgramShouldContinue::ShouldStop (int returnCode) {
8
- AtomicSet ( ReturnCode, returnCode );
9
- AtomicSet ( State, Stop );
8
+ ReturnCode. store (returnCode, std::memory_order_release );
9
+ State. store (Stop, std::memory_order_release );
10
10
}
11
11
12
12
TProgramShouldContinue::EState TProgramShouldContinue::PollState () {
13
- return static_cast <EState>( AtomicGet ( State) );
13
+ return State. load (std::memory_order_acquire );
14
14
}
15
15
16
16
int TProgramShouldContinue::GetReturnCode () {
17
- return static_cast < int >( AtomicGet ( ReturnCode) );
17
+ return ReturnCode. load (std::memory_order_acquire );
18
18
}
19
19
20
20
void TProgramShouldContinue::Reset () {
21
- AtomicSet ( ReturnCode, 0 );
22
- AtomicSet ( State, Continue );
21
+ ReturnCode. store ( 0 , std::memory_order_release );
22
+ State. store (Continue, std::memory_order_release );
23
23
}
Original file line number Diff line number Diff line change @@ -17,6 +17,6 @@ class TProgramShouldContinue {
17
17
18
18
void Reset ();
19
19
private:
20
- TAtomic ReturnCode = 0 ;
21
- TAtomic State = Continue;
20
+ std::atomic< int > ReturnCode = 0 ;
21
+ std::atomic<EState> State = Continue;
22
22
};
You can’t perform that action at this time.
0 commit comments