1 files changed, 36 insertions, 0 deletions
diff --git a/protocols/Telegram/tdlib/td/tddb/td/db/SqliteKeyValueSafe.h b/protocols/Telegram/tdlib/td/tddb/td/db/SqliteKeyValueSafe.h
new file mode 100644
index 0000000000..d63af3cfb2
--- /dev/null
+++ b/protocols/Telegram/tdlib/td/tddb/td/db/SqliteKeyValueSafe.h
@@ -0,0 +1,36 @@
+//
+// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2018
+//
+// Distributed under the Boost Software License, Version 1.0. (See accompanying
+// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
+//
+#pragma once
+
+#include "td/db/SqliteConnectionSafe.h"
+#include "td/db/SqliteKeyValue.h"
+
+#include <memory>
+
+namespace td {
+
+class SqliteKeyValueSafe {
+ public:
+  SqliteKeyValueSafe(string name, std::shared_ptr<SqliteConnectionSafe> safe_connection)
+      : lsls_kv_([name = std::move(name), safe_connection = std::move(safe_connection)] {
+        SqliteKeyValue kv;
+        kv.init_with_connection(safe_connection->get().clone(), name).ensure();
+        return kv;
+      }) {
+  }
+  SqliteKeyValue &get() {
+    return lsls_kv_.get();
+  }
+  void close() {
+    lsls_kv_.clear_values();
+  }
+
+ private:
+  LazySchedulerLocalStorage<SqliteKeyValue> lsls_kv_;
+};
+
+}  // namespace td