luciano/kinesis-consumer
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

protect counter race condition

Browse source
This commit is contained in:
Harlow Ward 2019-04-08 19:34:47 -07:00
parent dbab92f317
commit d176afa7ec
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
consumer_test.go
View file

@ -321,8 +321,12 @@ func (fc *fakeCheckpoint) Get(streamName, shardID string) (string, error) {
// implementation of counter // implementation of counter
type fakeCounter struct { type fakeCounter struct {
counter int64 counter int64
mu sync.Mutex
} }
func (fc *fakeCounter) Add(streamName string, count int64) { func (fc *fakeCounter) Add(streamName string, count int64) {
fc.mu.Lock()
defer fc.mu.Unlock()
fc.counter += count fc.counter += count
} }