Expose seqno for multi-batch-write (#307)

* expose seqno for multi-batch-write Signed-off-by: 5kbpers <[email protected]> * format Signed-off-by: 5kbpers <[email protected]> Signed-off-by: 5kbpers <[email protected]>
tikv · Aug 19, 2022 · 3cd757c · 3cd757c
1 parent f4cba2f
commit 3cd757c
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 4 deletions.
diff --git a/db/db_impl/db_impl.h b/db/db_impl/db_impl.h
@@ -182,7 +182,8 @@ class DBImpl : public DB {
 
   using DB::MultiBatchWrite;
   virtual Status MultiBatchWrite(const WriteOptions& options,
-                                 std::vector<WriteBatch*>&& updates) override;
+                                 std::vector<WriteBatch*>&& updates,
+                                 uint64_t* seq) override;
 
   using DB::Get;
   virtual Status Get(const ReadOptions& options,

diff --git a/db/db_impl/db_impl_write.cc b/db/db_impl/db_impl_write.cc
@@ -134,9 +134,11 @@ void DBImpl::MultiBatchWriteCommit(CommitRequest* request) {
 }
 
 Status DBImpl::MultiBatchWrite(const WriteOptions& options,
-                               std::vector<WriteBatch*>&& updates) {
+                               std::vector<WriteBatch*>&& updates,
+                               uint64_t* seq) {
   if (immutable_db_options_.enable_multi_batch_write) {
-    return MultiBatchWriteImpl(options, std::move(updates), nullptr, nullptr);
+    return MultiBatchWriteImpl(options, std::move(updates), nullptr, nullptr, 0,
+                               seq);
   } else {
     return Status::NotSupported();
   }

diff --git a/include/rocksdb/db.h b/include/rocksdb/db.h
@@ -478,10 +478,16 @@ class DB {
   }
 
   virtual Status MultiBatchWrite(const WriteOptions& /*options*/,
-                                 std::vector<WriteBatch*>&& /*updates*/) {
+                                 std::vector<WriteBatch*>&& /*updates*/,
+                                 uint64_t* /*seq*/) {
     return Status::NotSupported();
   }
 
+  virtual Status MultiBatchWrite(const WriteOptions& options,
+                                 std::vector<WriteBatch*>&& updates) {
+    return MultiBatchWrite(options, std::move(updates), nullptr);
+  }
+
   // If the database contains an entry for "key" store the
   // corresponding value in *value and return OK.
   //