aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/kernels/lmdb_reader_op.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/core/kernels/lmdb_reader_op.cc')
-rwxr-xr-xtensorflow/core/kernels/lmdb_reader_op.cc7
1 files changed, 5 insertions, 2 deletions
diff --git a/tensorflow/core/kernels/lmdb_reader_op.cc b/tensorflow/core/kernels/lmdb_reader_op.cc
index 3bb07301b5..31a427f2c9 100755
--- a/tensorflow/core/kernels/lmdb_reader_op.cc
+++ b/tensorflow/core/kernels/lmdb_reader_op.cc
@@ -36,7 +36,7 @@ class LMDBReader : public ReaderBase {
Status OnWorkStartedLocked() override {
MDB_CHECK(mdb_env_create(&mdb_env_));
- int flags = MDB_RDONLY | MDB_NOTLS;
+ int flags = MDB_RDONLY | MDB_NOTLS | MDB_NOLOCK;
// Check if the LMDB filename is actually a file instead of a directory.
// If so, set appropriate flags so we can open it.
@@ -57,10 +57,13 @@ class LMDBReader : public ReaderBase {
if (mdb_env_ != nullptr) {
if (mdb_cursor_) {
mdb_cursor_close(mdb_cursor_);
+ mdb_cursor_ = nullptr;
}
- mdb_txn_abort(mdb_txn_);
mdb_dbi_close(mdb_env_, mdb_dbi_);
+ mdb_txn_abort(mdb_txn_);
mdb_env_close(mdb_env_);
+ mdb_txn_ = nullptr;
+ mdb_dbi_ = 0;
mdb_env_ = nullptr;
}
return Status::OK();