34 lines
580 B
Go
34 lines
580 B
Go
package core
|
|
|
|
import (
|
|
"sync/atomic"
|
|
"testing"
|
|
)
|
|
|
|
func (lp *Loadpoint) RLock() {
|
|
if testing.Testing() && atomic.AddInt64(&lp.rwMutex, 1) > 1 {
|
|
panic("reentrant RLock")
|
|
}
|
|
lp.RWMutex.RLock()
|
|
}
|
|
|
|
func (lp *Loadpoint) RUnlock() {
|
|
if testing.Testing() {
|
|
atomic.AddInt64(&lp.rwMutex, -1)
|
|
}
|
|
lp.RWMutex.RUnlock()
|
|
}
|
|
|
|
func (lp *Loadpoint) Lock() {
|
|
if testing.Testing() && atomic.AddInt64(&lp.rwMutex, 1) > 1 {
|
|
panic("reentrant Lock")
|
|
}
|
|
lp.RWMutex.Lock()
|
|
}
|
|
|
|
func (lp *Loadpoint) Unlock() {
|
|
if testing.Testing() {
|
|
atomic.AddInt64(&lp.rwMutex, -1)
|
|
}
|
|
lp.RWMutex.Unlock()
|
|
}
|