Explorar o código

benchmark: support mvcc txn

Anthony Romano %!s(int64=9) %!d(string=hai) anos
pai
achega
0ed3c83e49
Modificáronse 1 ficheiros con 3 adicións e 6 borrados
  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)
 		}