mirror of
https://github.com/asmogo/nws.git
synced 2025-01-18 18:11:33 +00:00
41 lines
623 B
Go
41 lines
623 B
Go
package exit
|
|
|
|
import (
|
|
"log/slog"
|
|
"sync"
|
|
)
|
|
|
|
type MutexMap struct {
|
|
mu sync.Mutex // a separate mutex to protect the map
|
|
m map[string]*sync.Mutex // map from IDs to mutexes
|
|
}
|
|
|
|
func NewMutexMap() *MutexMap {
|
|
return &MutexMap{
|
|
m: make(map[string]*sync.Mutex),
|
|
}
|
|
}
|
|
|
|
func (mm *MutexMap) Lock(id string) {
|
|
mm.mu.Lock()
|
|
mutex, ok := mm.m[id]
|
|
if !ok {
|
|
mutex = &sync.Mutex{}
|
|
mm.m[id] = mutex
|
|
}
|
|
mm.mu.Unlock()
|
|
|
|
mutex.Lock()
|
|
}
|
|
|
|
func (mm *MutexMap) Unlock(id string) {
|
|
mm.mu.Lock()
|
|
mutex, ok := mm.m[id]
|
|
mm.mu.Unlock()
|
|
if !ok {
|
|
slog.Error("mutex not found", "id", id)
|
|
return
|
|
}
|
|
mutex.Unlock()
|
|
}
|