Browse Source

raft: fix node bench

Xiang Li 11 years ago
parent
commit
d244e3bf6e
1 changed files with 4 additions and 2 deletions
  1. 4 2
      raft/node_bench_test.go

+ 4 - 2
raft/node_bench_test.go

@@ -27,14 +27,16 @@ func BenchmarkOneNode(b *testing.B) {
 	defer cancel()
 	defer cancel()
 
 
 	n := newNode()
 	n := newNode()
-	r := newRaft(1, []uint64{1}, 10, 1, nil)
+	s := NewMemoryStorage()
+	r := newRaft(1, []uint64{1}, 10, 1, s)
 	go n.run(r)
 	go n.run(r)
 
 
 	defer n.Stop()
 	defer n.Stop()
 
 
 	n.Campaign(ctx)
 	n.Campaign(ctx)
 	for i := 0; i < b.N; i++ {
 	for i := 0; i < b.N; i++ {
-		<-n.Ready()
+		rd := <-n.Ready()
+		s.Append(rd.Entries)
 		n.Advance()
 		n.Advance()
 		n.Propose(ctx, []byte("foo"))
 		n.Propose(ctx, []byte("foo"))
 	}
 	}