intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 49 insertions, 0 deletions

diff --git a/library/cpp/actors/util/affinity.h b/library/cpp/actors/util/affinity.h
new file mode 100644
index 0000000000..ae106ed180
--- /dev/null
+++ b/library/cpp/actors/util/affinity.h
@@ -0,0 +1,49 @@
+#pragma once
+
+#include "defs.h"
+#include "cpumask.h"
+
+// Platform-specific class to set or get thread affinity
+class TAffinity: public TThrRefBase, TNonCopyable {
+    class TImpl;
+    THolder<TImpl> Impl;
+
+public:
+    TAffinity();
+    TAffinity(const ui8* cpus, ui32 size);
+    explicit TAffinity(const TCpuMask& mask);
+    ~TAffinity();
+
+    void Current();
+    void Set() const;
+    bool Empty() const;
+
+    operator TCpuMask() const;
+};
+
+// Scoped affinity setter
+class TAffinityGuard : TNonCopyable {
+    bool Stacked;
+    TAffinity OldAffinity;
+
+public:
+    TAffinityGuard(const TAffinity* affinity) {
+        Stacked = false;
+        if (affinity && !affinity->Empty()) {
+            OldAffinity.Current();
+            affinity->Set();
+            Stacked = true;
+        }
+    }
+
+    ~TAffinityGuard() {
+        Release();
+    }
+
+    void Release() {
+        if (Stacked) {
+            OldAffinity.Set();
+            Stacked = false;
+        }
+    }
+};