91 lines
1.9 KiB
Go
91 lines
1.9 KiB
Go
|
package syncUtil
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"runtime/debug"
|
|||
|
|
"sync"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 写锁对象
|
|||
|
|
type Locker struct {
|
|||
|
|
locking bool
|
|||
|
|
prevStack []byte
|
|||
|
|
mutex sync.Mutex
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 内部锁
|
|||
|
|
// 返回值：
|
|||
|
|
// 加锁是否成功
|
|||
|
|
func (this *Locker) lock() bool {
|
|||
|
|
this.mutex.Lock()
|
|||
|
|
defer this.mutex.Unlock()
|
|||
|
|
|
|||
|
|
// 如果已经被锁定，则返回失败
|
|||
|
|
if this.locking {
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 否则进行锁定，并返回成功
|
|||
|
|
this.locking = true
|
|||
|
|
|
|||
|
|
// 记录Stack信息
|
|||
|
|
if if_record_stack_info {
|
|||
|
|
this.prevStack = debug.Stack()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return true
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 尝试加锁，如果在指定的时间内失败，则会返回失败；否则返回成功
|
|||
|
|
// timeout:指定的毫秒数,timeout<=0则将会死等
|
|||
|
|
// 返回值：
|
|||
|
|
// 成功或失败
|
|||
|
|
// 如果失败，返回上一次成功加锁时的堆栈信息
|
|||
|
|
// 如果失败，返回当前的堆栈信息
|
|||
|
|
func (this *Locker) Lock(timeout int) (successful bool, prevStack string, currStack string) {
|
|||
|
|
timeout = getTimeout(timeout)
|
|||
|
|
|
|||
|
|
// 遍历指定的次数（即指定的超时时间）
|
|||
|
|
for i := 0; i < timeout; i++ {
|
|||
|
|
// 如果锁定成功，则返回成功
|
|||
|
|
if this.lock() {
|
|||
|
|
successful = true
|
|||
|
|
break
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 如果锁定失败，则休眠1ms，然后再重试
|
|||
|
|
time.Sleep(time.Millisecond)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 如果时间结束仍然是失败，则返回上次成功的堆栈信息，以及当前的堆栈信息
|
|||
|
|
if successful == false {
|
|||
|
|
if this.prevStack != nil && len(this.prevStack) > 0 {
|
|||
|
|
prevStack = string(this.prevStack)
|
|||
|
|
}
|
|||
|
|
currStack = string(debug.Stack())
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 锁定（死等方式）
|
|||
|
|
func (this *Locker) WaitLock() {
|
|||
|
|
successful, prevStack, currStack := this.Lock(0)
|
|||
|
|
if successful == false {
|
|||
|
|
fmt.Printf("Locker.WaitLock():{PrevStack:%s, currStack:%s}\n", prevStack, currStack)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 解锁
|
|||
|
|
func (this *Locker) Unlock() {
|
|||
|
|
this.mutex.Lock()
|
|||
|
|
defer this.mutex.Unlock()
|
|||
|
|
this.locking = false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 创建新的锁对象
|
|||
|
|
func NewLocker() *Locker {
|
|||
|
|
return &Locker{}
|
|||
|
|
}
|