From afd897619fba10b881c4567d76793c6fdc60e93a Mon Sep 17 00:00:00 2001
From: Gil <mcg@google.com>
Date: Wed, 13 Jun 2018 08:12:09 -0700
Subject: Minor LevelDB cleanups (#1401)

* Add leveldb_transaction* to the CMake build

* Use std::unique_ptr for leveldb::DB pointers
---
 Firestore/Source/Local/FSTLevelDB.mm | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'Firestore/Source/Local/FSTLevelDB.mm')

diff --git a/Firestore/Source/Local/FSTLevelDB.mm b/Firestore/Source/Local/FSTLevelDB.mm
index 9f75a3e..9dc50a2 100644
--- a/Firestore/Source/Local/FSTLevelDB.mm
+++ b/Firestore/Source/Local/FSTLevelDB.mm
@@ -17,6 +17,7 @@
 #import "Firestore/Source/Local/FSTLevelDB.h"
 
 #include <memory>
+#include <utility>
 
 #import "FIRFirestoreErrors.h"
 #import "Firestore/Source/Local/FSTLevelDBMigrations.h"
@@ -60,6 +61,7 @@ using leveldb::WriteOptions;
 
 @implementation FSTLevelDB {
   std::unique_ptr<LevelDbTransaction> _transaction;
+  std::unique_ptr<leveldb::DB> _ptr;
   FSTTransactionRunner _transactionRunner;
 }
 
@@ -82,6 +84,10 @@ using leveldb::WriteOptions;
   return self;
 }
 
+- (leveldb::DB *)ptr {
+  return _ptr.get();
+}
+
 - (const FSTTransactionRunner &)run {
   return _transactionRunner;
 }
-- 
cgit v1.2.3