From f16e393658a3b64bf489b023ec342d48dc77a278 Mon Sep 17 00:00:00 2001
From: heyongqiang <heyongqiang@fb.com>
Date: Wed, 15 Aug 2012 15:44:23 -0700
Subject: [PATCH] add more options to db_ben

Summary: as subject

Test Plan: run db_bench with new options

Reviewers: dhruba

Differential Revision: https://reviews.facebook.net/D4677
---
 db/db_bench.cc            | 38 ++++++++++++++++++++++++++++++++++++++
 include/leveldb/options.h | 18 +++++++++---------
 2 files changed, 47 insertions(+), 9 deletions(-)

diff --git a/db/db_bench.cc b/db/db_bench.cc
index fffbe9928..7ec2ed839 100644
--- a/db/db_bench.cc
+++ b/db/db_bench.cc
@@ -120,9 +120,26 @@ static class leveldb::DBStatistics* dbstats = NULL;
 // Number of write operations to do.  If negative, do FLAGS_num reads.
 static long FLAGS_writes = -1;
 
+// These default values might change if the hardcoded
+
 // Sync all writes to disk
 static bool FLAGS_sync = false;
 
+// If true, do not wait until data is synced to disk.
+static bool FLAGS_disable_data_sync = false;
+
+// Target level-0 file size for compaction
+static int FLAGS_target_file_size_base = 2 * 1048576;
+
+// A multiplier to compute targe level-N file size
+static int FLAGS_target_file_size_multiplier = 1;
+
+// Max bytes for level-0
+static int FLAGS_max_bytes_for_level_base = 10 * 1048576;
+
+// A multiplier to compute max bytes for level-N
+static int FLAGS_max_bytes_for_level_multiplier = 10;
+
 // posix or hdfs environment
 static leveldb::Env* FLAGS_env = leveldb::Env::Default();
 
@@ -745,6 +762,12 @@ class Benchmark {
     options.max_open_files = FLAGS_open_files;
     options.statistics = dbstats;
     options.env = FLAGS_env;
+    options.disableDataSync = FLAGS_disable_data_sync;
+    options.target_file_size_base = FLAGS_target_file_size_base;
+    options.target_file_size_multiplier = FLAGS_target_file_size_multiplier;
+    options.max_bytes_for_level_base = FLAGS_max_bytes_for_level_base;
+    options.max_bytes_for_level_multiplier =
+        FLAGS_max_bytes_for_level_multiplier;
     Status s = DB::Open(options, FLAGS_db, &db_);
     if (!s.ok()) {
       fprintf(stderr, "open error: %s\n", s.ToString().c_str());
@@ -1030,8 +1053,23 @@ int main(int argc, char** argv) {
     } else if (sscanf(argv[i], "--sync=%d%c", &n, &junk) == 1 &&
                (n == 0 || n == 1)) {
       FLAGS_sync = n;
+    } else if (sscanf(argv[i], "--disable_data_sync=%d%c", &n, &junk) == 1 &&
+        (n == 0 || n == 1)) {
+      FLAGS_disable_data_sync = n;
     } else if (sscanf(argv[i], "--hdfs=%s", &hdfsname) == 1) {
       FLAGS_env  = new leveldb::HdfsEnv(hdfsname);
+    } else if (sscanf(argv[i], "--target_file_size_base=%d%c",
+        &n, &junk) == 1) {
+      FLAGS_target_file_size_base = n;
+    } else if ( sscanf(argv[i], "--target_file_size_multiplier=%d%c",
+        &n, &junk) == 1) {
+      FLAGS_target_file_size_multiplier = n;
+    } else if (
+        sscanf(argv[i], "--max_bytes_for_level_base=%d%c", &n, &junk) == 1) {
+      FLAGS_max_bytes_for_level_base = n;
+    } else if (sscanf(argv[i], "--max_bytes_for_level_multiplier=%d%c",
+        &n, &junk) == 1) {
+      FLAGS_max_bytes_for_level_multiplier = n;
     } else {
       fprintf(stderr, "Invalid flag '%s'\n", argv[i]);
       exit(1);
diff --git a/include/leveldb/options.h b/include/leveldb/options.h
index d0b5a4cf9..71b794866 100644
--- a/include/leveldb/options.h
+++ b/include/leveldb/options.h
@@ -152,17 +152,17 @@ struct Options {
   // (max_bytes_for_level_base)^(max_bytes_for_level_multiplier).
 
   int max_bytes_for_level_base;
-	int max_bytes_for_level_multiplier;
+  int max_bytes_for_level_multiplier;
 
-	// Maximum number of bytes in all compacted files.  We avoid expanding
-	// the lower level file set of a compaction if it would make the
-	// total compaction cover more than
-	// (expanded_compaction_factor * targetFileSizeLevel()) many bytes.
-	int expanded_compaction_factor;
+  // Maximum number of bytes in all compacted files.  We avoid expanding
+  // the lower level file set of a compaction if it would make the
+  // total compaction cover more than
+  // (expanded_compaction_factor * targetFileSizeLevel()) many bytes.
+  int expanded_compaction_factor;
 
-	// Control maximum bytes of overlaps in grandparent (i.e., level+2) before we
-	// stop building a single file in a level->level+1 compaction.
-	int max_grandparent_overlap_factor;
+  // Control maximum bytes of overlaps in grandparent (i.e., level+2) before we
+  // stop building a single file in a level->level+1 compaction.
+  int max_grandparent_overlap_factor;
 
   // Compress blocks using the specified compression algorithm.  This
   // parameter can be changed dynamically.