aboutsummaryrefslogtreecommitdiffstats
path: root/library/cpp/messagebus/local_tasks.h
diff options
context:
space:
mode:
authorDevtools Arcadia <arcadia-devtools@yandex-team.ru>2022-02-07 18:08:42 +0300
committerDevtools Arcadia <arcadia-devtools@mous.vla.yp-c.yandex.net>2022-02-07 18:08:42 +0300
commit1110808a9d39d4b808aef724c861a2e1a38d2a69 (patch)
treee26c9fed0de5d9873cce7e00bc214573dc2195b7 /library/cpp/messagebus/local_tasks.h
downloadydb-1110808a9d39d4b808aef724c861a2e1a38d2a69.tar.gz
intermediate changes
ref:cde9a383711a11544ce7e107a78147fb96cc4029
Diffstat (limited to 'library/cpp/messagebus/local_tasks.h')
-rw-r--r--library/cpp/messagebus/local_tasks.h23
1 files changed, 23 insertions, 0 deletions
diff --git a/library/cpp/messagebus/local_tasks.h b/library/cpp/messagebus/local_tasks.h
new file mode 100644
index 0000000000..d8e801a457
--- /dev/null
+++ b/library/cpp/messagebus/local_tasks.h
@@ -0,0 +1,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;
+ }
+};