library/cpp/messagebus/local_tasks.h


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

#pragma once 
 
#include <util/system/atomic.h>

class TLocalTasks { 
private: 
    TAtomic GotTasks;
 
public: 
    TLocalTasks() 
        : GotTasks(0)
    {
    }
 
    void AddTask() { 
        AtomicSet(GotTasks, 1);
    } 
 
    bool FetchTask() { 
        bool gotTasks = AtomicCas(&GotTasks, 0, 1);
        return gotTasks; 
    } 
};