go - Golang - 按值锁定
问题描述
我正在编写一个接受 tableName 值和 updEpoch 值的 golang api,即:
curl -F "tableName=abc" -F "updEpoch=123" myhost:8080/singleroute
curl -F "tableName=abc" -F "updEpoch=456" myhost:8080/singleroute
curl -F "tableName=def" -F "updEpoch=123" myhost:8080/singleroute
curl -F "tableName=def" -F "updEpoch=345" myhost:8080/singleroute
我想允许并行处理多个不同的 tableName 请求,但每个 tableName 只能同时处理 1 个请求。所以在上面的例子中,如果同时触发了以上 4 个请求,那么第 1 和第 3 应该能够同时运行(作为唯一的 tableNames),但是第 2 只会在第 1 完成后开始,而第 4 只会在第 3 完成后开始。当我研究互斥锁时,似乎没有任何示例适合这种情况,我不想在代码中的任何地方硬编码 abc/def.etc,因为相同的规则应该适用于任何任意 tableName。
我的猜测基于 Crowman 的帮助:
package main
import (
"fmt"
"sync"
"time"
"http"
)
km := KeyedMutex{}
type KeyedMutex struct {
mutexes sync.Map // Zero value is empty and ready for use
}
func (m *KeyedMutex) Lock(key string) func() {
value, _ := m.mutexes.LoadOrStore(key, &sync.Mutex{})
mtx := value.(*sync.Mutex)
mtx.Lock()
return func() { mtx.Unlock() }
}
func myFunc(key string, data string) string {
//do some stuff
return "done for key:"+key+", data: "+data
}
func main() {
key := //some form value sent to my api
data := //some form value sent to my api
unlock := km.Lock(key)
defer unlock()
retVal := myFunc(key, data)
}
解决方案
您可以将 async.Map
与您的表名一起用作键,并将 a*sync.Mutex
作为值。
例如:
package main
import (
"fmt"
"sync"
"time"
)
type KeyedMutex struct {
mutexes sync.Map // Zero value is empty and ready for use
}
func (m *KeyedMutex) Lock(key string) func() {
value, _ := m.mutexes.LoadOrStore(key, &sync.Mutex{})
mtx := value.(*sync.Mutex)
mtx.Lock()
return func() { mtx.Unlock() }
}
func main() {
wg := sync.WaitGroup{}
km := KeyedMutex{}
for _, job := range []struct {
key string
data string
}{
{key: "abc", data: "123"},
{key: "abc", data: "456"},
{key: "def", data: "123"},
{key: "def", data: "456"},
} {
var job = job
wg.Add(1)
go func() {
defer wg.Done()
unlock := km.Lock(job.key)
defer unlock()
fmt.Printf("%s:%s mutex acquired\n", job.key, job.data)
time.Sleep(time.Second * 1) // To ensure some goroutines visibly block
fmt.Printf("%s:%s done\n", job.key, job.data)
}()
}
wg.Wait()
}
带有样本输出:
crow@mac:$ ./mut
def:456 mutex acquired
abc:456 mutex acquired
abc:456 done
def:456 done
abc:123 mutex acquired
def:123 mutex acquired
def:123 done
abc:123 done
crow@mac:$
显示具有不同表名的请求立即获取互斥锁,但具有相同表名的请求被序列化。
推荐阅读
- javascript - 使触摸设备上的缩放行为更像在桌面上
- java - java.util.ConcurrentModificationException:空
- c# - 我该怎么做才能停止收到连接池错误
- python - 如何在我的 python anytree graphviz 输出中反转箭头的方向?
- fabricjs - 使用 Fabric.js 中的对象创建剪贴蒙版
- c# - 使用 ASP.NET MVC,是否可以在下载后执行 WPF .exe 文件?
- mysql - MySQL查询的WHERE子句中的CASE
- c++ - gRPC 回调与 C++ 中的流式传输
- c# - 读取 xml 文件包含破坏日语字符的多语言字符
- c++ - 尝试无错误分配内存后程序退出