feat: add storage option is_use_raft

OpenAtomFoundation · Mar 16, 2024 · 512e4f2 · 512e4f2
1 parent 478fe1f
commit 512e4f2
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 2 deletions.
diff --git a/src/storage/include/storage/storage.h b/src/storage/include/storage/storage.h
@@ -61,6 +61,7 @@ struct StorageOptions {
   size_t small_compaction_threshold = 5000;
   size_t small_compaction_duration_threshold = 10000;
   size_t db_instance_num = 3;  // default = 3
+  bool is_use_raft = true;
   Status ResetOptions(const OptionType& option_type, const std::unordered_map<std::string, std::string>& options_map);
 };
 

diff --git a/src/storage/src/redis.cc b/src/storage/src/redis.cc
@@ -52,6 +52,7 @@ Redis::~Redis() {
 }
 
 Status Redis::Open(const StorageOptions& storage_options, const std::string& db_path) {
+  is_use_raft_ = storage_options.is_use_raft;
   statistics_store_->SetCapacity(storage_options.statistics_max_size);
   small_compaction_threshold_ = storage_options.small_compaction_threshold;
 

diff --git a/src/storage/src/redis.h b/src/storage/src/redis.h
@@ -356,6 +356,9 @@ class Redis {
   std::atomic_uint64_t small_compaction_duration_threshold_;
   std::unique_ptr<LRUCache<std::string, KeyStatistics>> statistics_store_;
 
+  // For raft
+  bool is_use_raft_;
+
   Status UpdateSpecificKeyStatistics(const DataType& dtype, const std::string& key, uint64_t count);
   Status UpdateSpecificKeyDuration(const DataType& dtype, const std::string& key, uint64_t duration);
   Status AddCompactKeyTaskIfNeeded(const DataType& dtype, const std::string& key, uint64_t count, uint64_t duration);

diff --git a/src/storage/src/redis_hashes.cc b/src/storage/src/redis_hashes.cc
@@ -119,7 +119,7 @@ Status Redis::HDel(const Slice& key, const std::vector<std::string>& fields, int
     }
   }
 
-	auto batch = Batch::CreateBatch(this, true);
+	auto batch = Batch::CreateBatch(this, is_use_raft_);
   rocksdb::ReadOptions read_options;
   const rocksdb::Snapshot* snapshot;
 
@@ -617,7 +617,7 @@ Status Redis::HMSet(const Slice& key, const std::vector<FieldValue>& fvs) {
 }
 
 Status Redis::HSet(const Slice& key, const Slice& field, const Slice& value, int32_t* res) {
-  auto batch = Batch::CreateBatch(this, true);
+  auto batch = Batch::CreateBatch(this, is_use_raft_);
   ScopeRecordLock l(lock_mgr_, key);
 
   uint64_t version = 0;