Avoid passing columnPresenceBuffer around

ed3889fb0e8fb71a059686f7c4133e9058935718

1 files changed, 28 insertions, 0 deletions

diff --git a/library/cpp/yt/memory/tls_scratch-inl.h b/library/cpp/yt/memory/tls_scratch-inl.h
new file mode 100644
index 0000000000..32c26d67c0
--- /dev/null
+++ b/library/cpp/yt/memory/tls_scratch-inl.h
@@ -0,0 +1,28 @@
+#ifndef TLS_SCRATH_INL_H_
+#error "Direct inclusion of this file is not allowed, include tls_scratch.h"
+// For the sake of sane code completion.
+#include "tls_scratch.h"
+#endif
+
+#include <library/cpp/yt/misc/tls.h>
+
+namespace NYT {
+
+////////////////////////////////////////////////////////////////////////////////
+
+template <class T>
+TMutableRange<T> GetTlsScratchBuffer(size_t size)
+{
+    // This is a workround for std::vector<bool>.
+    using TBoxed = std::array<T, 1>;
+    YT_THREAD_LOCAL(std::vector<TBoxed>) tlsVector;
+    auto& vector = GetTlsRef(tlsVector);
+    vector.reserve(size);
+    auto range = TMutableRange(reinterpret_cast<T*>(vector.data()), size);
+    std::fill(range.begin(), range.end(), T());
+    return range;
+}
+
+////////////////////////////////////////////////////////////////////////////////
+
+} // namespace NYT