From fa88cbc71e52d2c7c4d09c42a5bce24774482d9e Mon Sep 17 00:00:00 2001
From: Vamsi Ponnekanti <nponnekanti@fb.com>
Date: Tue, 3 Dec 2013 23:16:36 -0800
Subject: [PATCH] [Log dumper broken when merge operator is in log]

Summary: $title

Test Plan:
on my dev box

Revert Plan: OK

Task ID: #

Reviewers: emayanke, dhruba, haobo

Reviewed By: haobo

CC: leveldb

Differential Revision: https://reviews.facebook.net/D14451
---
 util/ldb_cmd.cc | 58 ++++++++++++++++++++++++-------------------------
 1 file changed, 28 insertions(+), 30 deletions(-)

diff --git a/util/ldb_cmd.cc b/util/ldb_cmd.cc
index e255f8196..58d81460e 100644
--- a/util/ldb_cmd.cc
+++ b/util/ldb_cmd.cc
@@ -1226,25 +1226,41 @@ void ChangeCompactionStyleCommand::DoCommand() {
 
 class InMemoryHandler : public WriteBatch::Handler {
  public:
+  InMemoryHandler(stringstream& row, bool print_values) : Handler(),row_(row) {
+    print_values_ = print_values;
+  }
 
-  virtual void Put(const Slice& key, const Slice& value) {
-    putMap_[key.ToString()] = value.ToString();
+  void commonPutMerge(const Slice& key, const Slice& value) {
+    string k = LDBCommand::StringToHex(key.ToString());
+    if (print_values_) {
+      string v = LDBCommand::StringToHex(value.ToString());
+      row_ << k << " : ";
+      row_ << v << " ";
+    } else {
+      row_ << k << " ";
+    }
   }
-  virtual void Delete(const Slice& key) {
-    deleteList_.push_back(key.ToString(true));
+
+  virtual void Put(const Slice& key, const Slice& value) {
+    row_ << "PUT : ";
+    commonPutMerge(key, value);
   }
-  virtual ~InMemoryHandler() { };
 
-  map<string, string> PutMap() {
-    return putMap_;
+  virtual void Merge(const Slice& key, const Slice& value) {
+    row_ << "MERGE : ";
+    commonPutMerge(key, value);
   }
-  vector<string> DeleteList() {
-    return deleteList_;
+
+  virtual void Delete(const Slice& key) {
+    row_ <<",DELETE : ";
+    row_ << LDBCommand::StringToHex(key.ToString()) << " ";
   }
 
+  virtual ~InMemoryHandler() { };
+
  private:
-  map<string, string> putMap_;
-  vector<string> deleteList_;
+  stringstream & row_;
+  bool print_values_;
 };
 
 const string WALDumperCommand::ARG_WAL_FILE = "walfile";
@@ -1322,26 +1338,8 @@ void WALDumperCommand::DoCommand() {
         row<<WriteBatchInternal::Count(&batch)<<",";
         row<<WriteBatchInternal::ByteSize(&batch)<<",";
         row<<reader.LastRecordOffset()<<",";
-        InMemoryHandler handler;
+        InMemoryHandler handler(row, print_values_);
         batch.Iterate(&handler);
-        row << "PUT : ";
-        if (print_values_) {
-          for (auto& kv : handler.PutMap()) {
-            string k = StringToHex(kv.first);
-            string v = StringToHex(kv.second);
-            row << k << " : ";
-            row << v << " ";
-          }
-        }
-        else {
-          for(auto& kv : handler.PutMap()) {
-            row << StringToHex(kv.first) << " ";
-          }
-        }
-        row<<",DELETE : ";
-        for(string& s : handler.DeleteList()) {
-          row << StringToHex(s) << " ";
-        }
         row<<"\n";
       }
       cout<<row.str();