library/cpp/actors/util/should_continue.h


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

#pragma once 
#include "defs.h" 
 
class TProgramShouldContinue { 
public: 
    enum EState { 
        Continue, 
        Stop, 
        Restart, 
    }; 
 
    void ShouldRestart(); 
    void ShouldStop(int returnCode = 0); 
 
    EState PollState(); 
    int GetReturnCode(); 
 
    void Reset(); 
private: 
    TAtomic ReturnCode = 0; 
    TAtomic State = Continue; 
};