compactor: Fix data race in revision compactor test

Use atomic functions to manipulate `rev` of `fakeRevGetter`
so that the tester goroutine can update the
value without race with the compactor's goroutine.
This commit is contained in:
Iwasaki Yudai
2017-06-27 13:42:56 -07:00
parent 4dd7ef0a60
commit c678dcbd91
2 changed files with 11 additions and 5 deletions

View File

@ -15,6 +15,8 @@
package compactor
import (
"sync/atomic"
pb "github.com/coreos/etcd/etcdserver/etcdserverpb"
"github.com/coreos/etcd/pkg/testutil"
"golang.org/x/net/context"
@ -36,6 +38,10 @@ type fakeRevGetter struct {
func (fr *fakeRevGetter) Rev() int64 {
fr.Record(testutil.Action{Name: "g"})
fr.rev++
return fr.rev
rev := atomic.AddInt64(&fr.rev, 1)
return rev
}
func (fr *fakeRevGetter) SetRev(rev int64) {
atomic.StoreInt64(&fr.rev, rev)
}