소스 검색

raft: fix node bench

Xiang Li 11 년 전
부모
커밋
d244e3bf6e
1개의 변경된 파일4개의 추가작업 그리고 2개의 파일을 삭제
  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()
 
 	n := newNode()
-	r := newRaft(1, []uint64{1}, 10, 1, nil)
+	s := NewMemoryStorage()
+	r := newRaft(1, []uint64{1}, 10, 1, s)
 	go n.run(r)
 
 	defer n.Stop()
 
 	n.Campaign(ctx)
 	for i := 0; i < b.N; i++ {
-		<-n.Ready()
+		rd := <-n.Ready()
+		s.Append(rd.Entries)
 		n.Advance()
 		n.Propose(ctx, []byte("foo"))
 	}