瀏覽代碼

benchmark: support mvcc txn

Anthony Romano 9 年之前
父節點
當前提交
0ed3c83e49
共有 1 個文件被更改,包括 3 次插入6 次删除
  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)
 		}