Compare commits
No commits in common. "436bcad0e16bd484e01abf46355568dc6290ad09" and "a0bad4d6e955f824624a16984d545fdae7c455c5" have entirely different histories.
436bcad0e1
...
a0bad4d6e9
@ -1,22 +0,0 @@
|
||||
package counter
|
||||
|
||||
import "sync"
|
||||
|
||||
type Counter struct {
|
||||
count int
|
||||
mutex sync.Mutex
|
||||
}
|
||||
|
||||
func NewCounter() *Counter {
|
||||
return &Counter{}
|
||||
}
|
||||
|
||||
func (c *Counter) Inc() {
|
||||
c.mutex.Lock()
|
||||
defer c.mutex.Unlock()
|
||||
c.count++
|
||||
}
|
||||
|
||||
func (c *Counter) Value() int {
|
||||
return c.count
|
||||
}
|
@ -1,44 +0,0 @@
|
||||
package counter
|
||||
|
||||
import (
|
||||
"sync"
|
||||
"testing"
|
||||
)
|
||||
|
||||
// assertCounter
|
||||
// XXX: a copy of Counter is created, but we shouldn't copy a mutex. Use pointer instead.
|
||||
func assertCounter(t testing.TB, got *Counter, want int) {
|
||||
t.Helper()
|
||||
if got.Value() != want {
|
||||
t.Errorf("got %d, want %d", got.Value(), want)
|
||||
}
|
||||
}
|
||||
|
||||
func TestCounter(t *testing.T) {
|
||||
t.Run("incrementing the counter 3 times leaves it at 3", func(t *testing.T) {
|
||||
counter := NewCounter()
|
||||
|
||||
counter.Inc()
|
||||
counter.Inc()
|
||||
counter.Inc()
|
||||
|
||||
assertCounter(t, counter, 3)
|
||||
})
|
||||
|
||||
t.Run("it runs safely concurrently", func(t *testing.T) {
|
||||
wantedCount := 1000
|
||||
counter := NewCounter()
|
||||
|
||||
var wg sync.WaitGroup
|
||||
wg.Add(wantedCount)
|
||||
|
||||
for i := 0; i < wantedCount; i++ {
|
||||
go func() {
|
||||
counter.Inc()
|
||||
wg.Done()
|
||||
}()
|
||||
}
|
||||
wg.Wait()
|
||||
assertCounter(t, counter, wantedCount)
|
||||
})
|
||||
}
|
Loading…
x
Reference in New Issue
Block a user