37 lines
957 B
Go
37 lines
957 B
Go
package lock
|
||
|
||
import "sync"
|
||
|
||
// 用sync.Map替代原生map，天然支持并发安全
|
||
var lock sync.Map
|
||
|
||
// GetLock 获取锁（返回true表示已上锁，false表示未上锁）
|
||
func GetLock(key string) bool {
|
||
v, ok := lock.Load(key)
|
||
if !ok {
|
||
return false
|
||
}
|
||
// 断言为bool类型（确保存储的是布尔值）
|
||
locked, ok := v.(bool)
|
||
return ok && locked
|
||
}
|
||
|
||
// SetLock 设置锁（原子操作）
|
||
func SetLock(key string) {
|
||
lock.Store(key, true)
|
||
}
|
||
|
||
// DestroyLock 销毁锁（原子操作）
|
||
func DestroyLock(key string) {
|
||
lock.Delete(key)
|
||
}
|
||
|
||
// TryLock 尝试加锁（核心：检查+设置原子化）
|
||
// 返回true表示加锁成功，false表示已被上锁
|
||
func TryLock(key string) bool {
|
||
// LoadOrStore：如果key不存在则存储值，返回false；如果已存在则返回true
|
||
_, loaded := lock.LoadOrStore(key, true)
|
||
// loaded为true表示已上锁，返回false；loaded为false表示加锁成功，返回true
|
||
return !loaded
|
||
}
|