From e2ddb615d276df01709d212aa0b3ca1ff3c01abb Mon Sep 17 00:00:00 2001
From: cyq <1056374449@qq.com>
Date: Sun, 29 Dec 2024 22:14:05 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8E=A5=E4=B8=8A=E4=B8=80=E4=B8=AAcommit?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 benchmarks/db_bench_FieldDB.cc | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/benchmarks/db_bench_FieldDB.cc b/benchmarks/db_bench_FieldDB.cc
index fc5e6f4..9b032a7 100644
--- a/benchmarks/db_bench_FieldDB.cc
+++ b/benchmarks/db_bench_FieldDB.cc
@@ -79,6 +79,8 @@ static const char* FLAGS_benchmarks =
     "ReadSeqWhileDeleting,"
     "ReadRandomWhileCreating,"
     "ReadRandomWhileDeleting,"
+    "WriteRandomWithIndex,"
+    "WriteSeqWithIndex,"
     "snappycomp,"
     "snappyuncomp,"
     "zstdcomp,"
@@ -683,6 +685,12 @@ class Benchmark {
       } else if (name == Slice("ReadRandomWhileDeleting")) {
         num_threads++;
         method = &Benchmark::ReadRandomWhileDeleting;
+      } else if (name == Slice("WriteRandomWithIndex")) {
+        fresh_db = true;
+        method = &Benchmark::WriteRandomWithIndex;
+      } else if (name == Slice("WriteSeqWithIndex")) {
+        fresh_db = true;
+        method = &Benchmark::WriteSeqWithIndex;
       } else if (name == Slice("snappycomp")) {
         method = &Benchmark::SnappyCompress;
       } else if (name == Slice("snappyuncomp")) {
@@ -1146,6 +1154,18 @@ class Benchmark {
     db_->DeleteIndex("age", write_options_);
   }
   
+  void WriteSeqWithIndex(ThreadState* thread) {
+    CreateIndex(thread);
+    thread->stats.Start();
+    WriteSeq(thread);
+  }
+
+  void WriteRandomWithIndex(ThreadState* thread) {
+    CreateIndex(thread);
+    thread->stats.Start();
+    WriteRandom(thread);
+  }
+  
   void WriteSeqWhileCreating(ThreadState* thread) {
     if (thread->tid > 0) {
       WriteSeq(thread);