From 96be2c4ee0b362d8fc94741abace82dc6d1e0ac7 Mon Sep 17 00:00:00 2001
From: Haobo Xu <haobo@fb.com>
Date: Mon, 17 Jun 2013 16:13:32 -0700
Subject: [PATCH] [RocksDB] Add mmap_read option for db_stress

Summary: as title, also removed an incorrect assertion

Test Plan: make check; db_stress --mmap_read=1; db_stress --mmap_read=0

Reviewers: dhruba, emayanke

CC: leveldb

Differential Revision: https://reviews.facebook.net/D11367
---
 include/leveldb/options.h |  2 +-
 tools/db_stress.cc        | 10 ++++++++++
 util/env_posix.cc         |  1 -
 3 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/include/leveldb/options.h b/include/leveldb/options.h
index 79d0946ec..fa69a7eff 100644
--- a/include/leveldb/options.h
+++ b/include/leveldb/options.h
@@ -427,7 +427,7 @@ struct Options {
   // Note: Deprecated
   bool allow_readahead_compactions;
 
-  // Allow the OS to mmap file for reading. Default: false
+  // Allow the OS to mmap file for reading sst tables. Default: false
   bool allow_mmap_reads;
 
   // Allow the OS to mmap file for writing. Default: true
diff --git a/tools/db_stress.cc b/tools/db_stress.cc
index a5cb5b35e..02afe306c 100644
--- a/tools/db_stress.cc
+++ b/tools/db_stress.cc
@@ -110,6 +110,9 @@ static const char* FLAGS_db = nullptr;
 // Verify checksum for every block read from storage
 static bool FLAGS_verify_checksum = false;
 
+// Allow reads to occur via mmap-ing files
+static bool FLAGS_use_mmap_reads = leveldb::EnvOptions().use_mmap_reads;
+
 // Database statistics
 static std::shared_ptr<leveldb::Statistics> dbstats;
 
@@ -934,6 +937,7 @@ class StressTest {
     options.env = FLAGS_env;
     options.disableDataSync = FLAGS_disable_data_sync;
     options.use_fsync = FLAGS_use_fsync;
+    options.allow_mmap_reads = FLAGS_use_mmap_reads;
     leveldb_kill_odds = FLAGS_kill_random_test;
     options.target_file_size_base = FLAGS_target_file_size_base;
     options.target_file_size_multiplier = FLAGS_target_file_size_multiplier;
@@ -955,6 +959,9 @@ class StressTest {
     if (purge_percent.Uniform(100) < FLAGS_purge_redundant_percent - 1) {
       options.purge_redundant_kvs_while_flush = false;
     }
+
+    fprintf(stdout, "DB path: [%s]\n", FLAGS_db);
+
     Status s;
     if (FLAGS_ttl == -1) {
       s = DB::Open(options, FLAGS_db, &db_);
@@ -1076,6 +1083,9 @@ int main(int argc, char** argv) {
     } else if (sscanf(argv[i], "--verify_checksum=%d%c", &n, &junk) == 1 &&
                (n == 0 || n == 1)) {
       FLAGS_verify_checksum = n;
+    } else if (sscanf(argv[i], "--mmap_read=%d%c", &n, &junk) == 1 &&
+               (n == 0 || n == 1)) {
+      FLAGS_use_mmap_reads = n;
     } else if (sscanf(argv[i], "--statistics=%d%c", &n, &junk) == 1 &&
                (n == 0 || n == 1)) {
       if (n == 1) {
diff --git a/util/env_posix.cc b/util/env_posix.cc
index 851030981..4e24163a8 100644
--- a/util/env_posix.cc
+++ b/util/env_posix.cc
@@ -108,7 +108,6 @@ class PosixSequentialFile: public SequentialFile {
       const EnvOptions& options)
       : filename_(fname), file_(f), fd_(fileno(f)),
         use_os_buffer_(options.use_os_buffer) {
-    assert(!options.use_mmap_reads);
   }
   virtual ~PosixSequentialFile() { fclose(file_); }