
Go语言自定义锁:针对特定实体的锁机制
问题:Go语言的sync.Mutex锁适用于全局资源保护,但如何实现针对特定实体(例如用户ID)的自定义锁?
解决方案:
针对不同场景,解决方案有所不同:
立即学习“go语言免费学习笔记(深入)”;
主要特性: 1、支持多种语言 BEES支持多种语言,后台添加自动生成,可为每种语言分配网站风格。 2、功能强大灵活 BEES除内置的文章、产品等模型外,还可以自定义生成其它模型,满足不同的需求 3、自定义表单系统 BEES可自定义表单系统,后台按需要生成,将生成的标签加到模板中便可使用。 4、模板制作方便 采用MVC设计模式实现了程序与模板完全分离,分别适合美工和程序员使用。 5、用户体验好 前台
-
分布式环境: 使用分布式锁服务,例如Redis或Etcd。这些服务提供基于键值对的锁机制,您可以使用用户ID作为键来实现针对特定用户的锁。
-
单进程环境: 可以使用
sync.Map或自定义的map来实现。创建一个map,键为用户ID,值为sync.Mutex锁。 当需要对特定用户加锁时,从map中获取对应的锁,不存在则创建新的锁。
示例代码(单进程环境,使用sync.Map):
package main
import (
"fmt"
"sync"
)
type UserLocker struct {
locks sync.Map
}
func (ul *UserLocker) Lock(uid int) {
lock, _ := ul.locks.LoadOrStore(uid, &sync.Mutex{})
lock.(*sync.Mutex).Lock()
}
func (ul *UserLocker) Unlock(uid int) {
lock, ok := ul.locks.Load(uid)
if ok {
lock.(*sync.Mutex).Unlock()
}
}
func main() {
locker := &UserLocker{}
locker.Lock(1)
fmt.Println("User 1 locked")
// ...操作用户1的数据...
locker.Unlock(1)
fmt.Println("User 1 unlocked")
locker.Lock(2)
fmt.Println("User 2 locked")
// ...操作用户2的数据...
locker.Unlock(2)
fmt.Println("User 2 unlocked")
}
这个示例展示了如何使用sync.Map来管理不同用户ID对应的锁。 LoadOrStore方法保证了线程安全地获取或创建锁。 记住在操作完成后始终调用Unlock释放锁,避免死锁。 对于分布式锁的实现,需要参考具体分布式锁服务的文档。









