feat: skip value get (#114)

* feat: skip value get
Co-authored-by: Nwangyi <wangyi.ywq@bytedance.com>

db/db_impl.cc

@@ -1637,7 +1637,7 @@ Status DBImpl::Get(const ReadOptions& read_options,
                    ColumnFamilyHandle* column_family, const Slice& key,
                    LazyBuffer* value) {
   auto s = GetImpl(read_options, column_family, key, value);
-  assert(!s.ok() || value->valid());
+  assert(!s.ok() || value == nullptr || value->valid());
   return s;
 }
 
@@ -1648,7 +1648,6 @@ Status DBImpl::GetImpl(const ReadOptions& read_options,
   LatencyHistGuard guard(&read_latency_reporter_);
   read_qps_reporter_.AddCount(1);
 
-  assert(lazy_val != nullptr);
   StopWatch sw(env_, stats_, DB_GET);
   PERF_TIMER_GUARD(get_snapshot_time);
 
@@ -1742,7 +1741,7 @@ Status DBImpl::GetImpl(const ReadOptions& read_options,
     RecordTick(stats_, MEMTABLE_MISS);
   }
 
-  if (s.ok()) {
+  if (s.ok() && lazy_val != nullptr) {
     lazy_val->pin(LazyBufferPinLevel::DB);
     s = lazy_val->fetch();
   }

db/db_test.cc

@@ -5684,6 +5684,38 @@ TEST_F(DBTest, ThreadLocalPtrDeadlock) {
   fprintf(stderr, "Done. Flushed %d times, destroyed %d threads\n",
           flushes_done.load(), threads_destroyed.load());
 }
+
+TEST_F(DBTest, SkipValueGet) {
+  Options options = CurrentOptions();
+  DestroyAndReopen(options);
+
+  ASSERT_OK(Put(Key(0), ""));
+
+  // memtable get
+  auto s = db_->Get(ReadOptions{}, db_->DefaultColumnFamily(), Key(0));
+  ASSERT_OK(s);
+  s = db_->Get(ReadOptions{}, db_->DefaultColumnFamily(), Key(0),
+               static_cast<LazyBuffer*>(nullptr));
+  ASSERT_OK(s);
+  s = db_->Get(ReadOptions{}, Key(0), static_cast<std::string*>(nullptr));
+  ASSERT_OK(s);
+  s = db_->Get(ReadOptions{}, Key(1), static_cast<std::string*>(nullptr));
+  ASSERT_TRUE(s.IsNotFound());
+
+  ASSERT_OK(Flush());
+  dbfull()->TEST_WaitForFlushMemTable();
+
+  // sst get
+  s = db_->Get(ReadOptions{}, db_->DefaultColumnFamily(), Key(0));
+  ASSERT_OK(s);
+  s = db_->Get(ReadOptions{}, db_->DefaultColumnFamily(), Key(0),
+               static_cast<LazyBuffer*>(nullptr));
+  ASSERT_OK(s);
+  s = db_->Get(ReadOptions{}, Key(0), static_cast<std::string*>(nullptr));
+  ASSERT_OK(s);
+  s = db_->Get(ReadOptions{}, Key(1), static_cast<std::string*>(nullptr));
+  ASSERT_TRUE(s.IsNotFound());
+}
 }  // namespace TERARKDB_NAMESPACE
 
 int main(int argc, char** argv) {

db/memtable.cc

@@ -773,7 +773,7 @@ static bool SaveValue(void* arg, const Slice& internal_key, const char* value) {
         }
         *s->status = Status::OK();
         if (*s->merge_in_progress) {
-          if (s->value != nullptr) {
+          if (LIKELY(s->value != nullptr)) {
             LazyBuffer lazy_val(GetLengthPrefixedSlice(value), false);
             *s->status = MergeHelper::TimedFullMerge(
                 merge_operator, s->key->user_key(), &lazy_val,
@@ -783,7 +783,7 @@ static bool SaveValue(void* arg, const Slice& internal_key, const char* value) {
               s->value->pin(LazyBufferPinLevel::Internal);
             }
           }
-        } else if (s->value != nullptr) {
+        } else if (LIKELY(s->value != nullptr)) {
           s->value->reset(GetLengthPrefixedSlice(value),
                           s->inplace_update_support);
         }
@@ -797,7 +797,7 @@ static bool SaveValue(void* arg, const Slice& internal_key, const char* value) {
       case kTypeSingleDeletion:
       case kTypeRangeDeletion: {
         if (*s->merge_in_progress) {
-          if (s->value != nullptr) {
+          if (LIKELY(s->value != nullptr)) {
             *s->status = MergeHelper::TimedFullMerge(
                 merge_operator, s->key->user_key(), nullptr,
                 merge_context->GetOperands(), s->value, s->logger,
@@ -831,13 +831,15 @@ static bool SaveValue(void* arg, const Slice& internal_key, const char* value) {
             LazyBuffer(GetLengthPrefixedSlice(value), Cleanable()));
         if (merge_operator->ShouldMerge(
                 merge_context->GetOperandsDirectionBackward())) {
+          if (LIKELY(s->value != nullptr)) {
             *s->status = MergeHelper::TimedFullMerge(
                 merge_operator, s->key->user_key(), nullptr,
-              merge_context->GetOperands(), s->value, s->logger, s->statistics,
-              s->env_, true);
+                merge_context->GetOperands(), s->value, s->logger,
+                s->statistics, s->env_, true);
             if (s->status->ok()) {
               s->value->pin(LazyBufferPinLevel::Internal);
             }
+          }
           *s->found_final_value = true;
           return false;
         }

db/merge_test.cc

@@ -169,6 +169,9 @@ class Counters {
   bool get(const std::string& key, uint64_t* value) {
     std::string str;
     auto s = db_->Get(get_option_, key, &str);
+    auto x = num_merge_operator_calls;
+    assert(db_->Get(get_option_, key).code() == s.code());
+    num_merge_operator_calls = x;
 
     if (s.IsNotFound()) {
       // return default value if not found;

db/version_set.cc

@@ -1410,12 +1410,14 @@ void Version::Get(const ReadOptions& read_options, const Slice& user_key,
     }
     // merge_operands are in saver and we hit the beginning of the key history
     // do a final merge of nullptr and operands;
+    if (value != nullptr) {
       *status = MergeHelper::TimedFullMerge(
-        merge_operator_, user_key, nullptr, merge_context->GetOperands(), value,
-        info_log_, db_statistics_, env_, true);
+          merge_operator_, user_key, nullptr, merge_context->GetOperands(),
+          value, info_log_, db_statistics_, env_, true);
       if (status->ok()) {
         value->pin(LazyBufferPinLevel::Internal);
       }
+    }
   } else {
     if (key_exists != nullptr) {
       *key_exists = false;

include/rocksdb/db.h

@@ -353,6 +353,7 @@ class DB {
 
   // If the database contains an entry for "key" store the
   // corresponding value in *value and return OK.
+  // If "value" equals to nullptr, the corresponding value will not be loaded.
   //
   // If there is no entry for "key" leave *value unchanged and return
   // a status for which Status::IsNotFound() returns true.
@@ -361,21 +362,33 @@ class DB {
   virtual inline Status Get(const ReadOptions& options,
                             ColumnFamilyHandle* column_family, const Slice& key,
                             std::string* value) {
-    assert(value != nullptr);
+    if (value != nullptr) {
       LazyBuffer lazy_val(value);
       auto s = Get(options, column_family, key, &lazy_val);
       if (s.ok()) {
         s = std::move(lazy_val).dump(value);
       }
       return s;
+    } else {
+      return Get(options, column_family, key,
+                 static_cast<LazyBuffer*>(nullptr));
+    }
   }
   virtual Status Get(const ReadOptions& options,
                      ColumnFamilyHandle* column_family, const Slice& key,
                      LazyBuffer* value) = 0;
+  virtual Status Get(const ReadOptions& options,
+                     ColumnFamilyHandle* column_family, const Slice& key) {
+    return Get(options, column_family, key, static_cast<LazyBuffer*>(nullptr));
+  }
   virtual Status Get(const ReadOptions& options, const Slice& key,
                      std::string* value) {
     return Get(options, DefaultColumnFamily(), key, value);
   }
+  virtual Status Get(const ReadOptions& options, const Slice& key) {
+    return Get(options, DefaultColumnFamily(), key,
+               static_cast<LazyBuffer*>(nullptr));
+  }
 #ifdef BOOSTLIB
   static void CallOnMainStack(const std::function<void()>&);
   static void SubmitAsyncTask(std::function<void()>);

utilities/ttl/db_ttl_impl.cc

@@ -209,6 +209,10 @@ Status DBWithTTLImpl::Put(const WriteOptions& options,
 Status DBWithTTLImpl::Get(const ReadOptions& options,
                           ColumnFamilyHandle* column_family, const Slice& key,
                           LazyBuffer* value) {
+  LazyBuffer buffer;
+  if (value == nullptr) {
+    value = &buffer;
+  }
   Status st = db_->Get(options, column_family, key, value);
   if (!st.ok()) {
     return st;