add syncx.Guard func (#620)

This commit is contained in:
Kevin Wan 2021-04-13 00:04:19 +08:00 committed by GitHub
parent b94d7aa532
commit 08fb980ad2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 23 additions and 2 deletions

View File

@ -9,7 +9,12 @@ type Barrier struct {
// Guard guards the given fn on the resource.
func (b *Barrier) Guard(fn func()) {
b.lock.Lock()
defer b.lock.Unlock()
Guard(&b.lock, fn)
}
// Guard guards the given fn with lock.
func Guard(lock sync.Locker, fn func()) {
lock.Lock()
defer lock.Unlock()
fn()
}

View File

@ -38,3 +38,19 @@ func TestBarrierPtr_Guard(t *testing.T) {
wg.Wait()
assert.Equal(t, total, count)
}
func TestGuard(t *testing.T) {
const total = 10000
var count int
var lock sync.Mutex
wg := new(sync.WaitGroup)
wg.Add(total)
for i := 0; i < total; i++ {
go Guard(&lock, func() {
count++
wg.Done()
})
}
wg.Wait()
assert.Equal(t, total, count)
}