36 lines
809 B
Go
36 lines
809 B
Go
|
package torrent
|
||
|
|
|
||
|
|
import "sync"
|
||
|
|
|
||
|
|
// Runs deferred actions on Unlock. Note that actions are assumed to be the results of changes that
|
||
|
|
// would only occur with a write lock at present. The race detector should catch instances of defers
|
||
|
|
// without the write lock being held.
|
||
|
|
type lockWithDeferreds struct {
|
||
|
|
internal sync.RWMutex
|
||
|
|
unlockActions []func()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (me *lockWithDeferreds) Lock() {
|
||
|
|
me.internal.Lock()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (me *lockWithDeferreds) Unlock() {
|
||
|
|
for _, a := range me.unlockActions {
|
||
|
|
a()
|
||
|
|
}
|
||
|
|
me.unlockActions = me.unlockActions[:0]
|
||
|
|
me.internal.Unlock()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (me *lockWithDeferreds) RLock() {
|
||
|
|
me.internal.RLock()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (me *lockWithDeferreds) RUnlock() {
|
||
|
|
me.internal.RUnlock()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (me *lockWithDeferreds) Defer(action func()) {
|
||
|
|
me.unlockActions = append(me.unlockActions, action)
|
||
|
|
}
|