Browse Source

benchmark: support mvcc txn

Anthony Romano 9 năm trước cách đây
mục cha
commit
0ed3c83e49
1 tập tin đã thay đổi với 3 bổ sung6 xóa
  1. 3 6
      tools/benchmark/cmd/mvcc-put.go

+ 3 - 6
tools/benchmark/cmd/mvcc-put.go

@@ -109,12 +109,9 @@ func mvccPutFunc(cmd *cobra.Command, args []string) {
 	for i := 0; i < totalNrKeys; i++ {
 		st := time.Now()
 		if txn {
-			id := s.TxnBegin()
-			if _, err := s.TxnPut(id, keys[i], vals[i], lease.NoLease); err != nil {
-				fmt.Fprintln(os.Stderr, "txn put error:", err)
-				os.Exit(1)
-			}
-			s.TxnEnd(id)
+			tw := s.Write()
+			tw.Put(keys[i], vals[i], lease.NoLease)
+			tw.End()
 		} else {
 			s.Put(keys[i], vals[i], lease.NoLease)
 		}