mirror of
https://gitee.com/johng/gf
synced 2026-07-05 13:22:16 +08:00
132 lines
3.9 KiB
Go
132 lines
3.9 KiB
Go
// Copyright 2018 gf Author(https://gitee.com/johng/gf). All Rights Reserved.
|
||
//
|
||
// This Source Code Form is subject to the terms of the MIT License.
|
||
// If a copy of the MIT was not distributed with this file,
|
||
// You can obtain one at https://gitee.com/johng/gf.
|
||
|
||
package gmlock
|
||
|
||
import (
|
||
"gitee.com/johng/gf/g/container/gmap"
|
||
"gitee.com/johng/gf/g/os/gtime"
|
||
"time"
|
||
)
|
||
|
||
// 内存锁管理对象
|
||
type Locker struct {
|
||
m *gmap.StringInterfaceMap
|
||
}
|
||
|
||
// 创建一把内存锁使用的底层RWLocker
|
||
func New() *Locker {
|
||
return &Locker{
|
||
m : gmap.NewStringInterfaceMap(),
|
||
}
|
||
}
|
||
|
||
// 内存写锁，如果锁成功返回true，失败则返回false;过期时间单位为毫秒，默认为0表示不过期
|
||
func (l *Locker) TryLock(key string, expire...int) bool {
|
||
return l.doLock(key, l.getExpire(expire...), true)
|
||
}
|
||
|
||
// 内存写锁，锁成功返回true，失败时阻塞，当失败时表示有其他写锁存在;过期时间单位为毫秒，默认为0表示不过期
|
||
func (l *Locker) Lock(key string, expire...int) {
|
||
l.doLock(key, l.getExpire(expire...), false)
|
||
}
|
||
|
||
// 解除基于内存锁的写锁
|
||
func (l *Locker) Unlock(key string) {
|
||
if v := l.m.Get(key); v != nil {
|
||
v.(*Mutex).Unlock()
|
||
}
|
||
}
|
||
|
||
// 内存读锁，如果锁成功返回true，失败则返回false;过期时间单位为毫秒，默认为0表示不过期
|
||
func (l *Locker) TryRLock(key string, expire...int) bool {
|
||
return l.doRLock(key, l.getExpire(expire...), true)
|
||
}
|
||
|
||
// 内存写锁，锁成功返回true，失败时阻塞，当失败时表示有写锁存在;过期时间单位为毫秒，默认为0表示不过期
|
||
func (l *Locker) RLock(key string, expire...int) {
|
||
l.doRLock(key, l.getExpire(expire...), false)
|
||
}
|
||
|
||
// 解除基于内存锁的读锁
|
||
func (l *Locker) RUnlock(key string) {
|
||
if v := l.m.Get(key); v != nil {
|
||
v.(*Mutex).RUnlock()
|
||
}
|
||
}
|
||
|
||
// 获得过期时间，没有设置时默认为0不过期
|
||
func (l *Locker) getExpire(expire...int) int {
|
||
e := 0
|
||
if len(expire) > 0 {
|
||
e = expire[0]
|
||
}
|
||
return e
|
||
}
|
||
|
||
// 内存写锁，当try==true时，如果锁成功返回true，失败则返回false；try==false时，成功时立即返回，否则阻塞等待
|
||
func (l *Locker) doLock(key string, expire int, try bool) bool {
|
||
mu := l.getOrNewMutex(key)
|
||
ok := true
|
||
if try {
|
||
ok = mu.TryLock()
|
||
} else {
|
||
mu.Lock()
|
||
}
|
||
if ok && expire > 0 {
|
||
// 异步goroutine计时处理
|
||
wid := mu.wid.Val()
|
||
gtime.SetTimeout(time.Duration(expire)*time.Millisecond, func() {
|
||
if wid == mu.wid.Val() {
|
||
mu.Unlock()
|
||
}
|
||
})
|
||
}
|
||
return ok
|
||
}
|
||
|
||
// 内存读锁，当try==true时，如果锁成功返回true，失败则返回false；try==false时，成功时立即返回，否则阻塞等待
|
||
func (l *Locker) doRLock(key string, expire int, try bool) bool {
|
||
mu := l.getOrNewMutex(key)
|
||
ok := true
|
||
if try {
|
||
ok = mu.TryRLock()
|
||
} else {
|
||
mu.RLock()
|
||
}
|
||
if ok && expire > 0 {
|
||
// 异步goroutine计时处理
|
||
rid := mu.rid.Val()
|
||
gtime.SetTimeout(time.Duration(expire)*time.Millisecond, func() {
|
||
if rid == mu.rid.Val() {
|
||
mu.RUnlock()
|
||
}
|
||
})
|
||
}
|
||
return ok
|
||
}
|
||
|
||
// 根据指定key查询或者创建新的Mutex
|
||
func (l *Locker) getOrNewMutex(key string) (mu *Mutex) {
|
||
// 优先进行读取检查，提高读取效率
|
||
if v := l.m.Get(key); v != nil {
|
||
mu = v.(*Mutex)
|
||
}
|
||
if mu == nil {
|
||
l.m.LockFunc(func(m map[string]interface{}) {
|
||
// 这里必须再进行一次查询，上面那一次使用的是读锁，支持并发效率高；这里面的是写锁，只支持1个goroutine操作
|
||
if v, ok := m[key]; ok {
|
||
mu = v.(*Mutex)
|
||
}
|
||
if mu == nil {
|
||
mu = NewMutex()
|
||
m[key] = mu
|
||
}
|
||
})
|
||
}
|
||
return
|
||
}
|