37 lines
709 B
Go
37 lines
709 B
Go
package domainrouter
|
|
|
|
import "sync"
|
|
|
|
// ThreadMap for disallowing change of elements during runtime, for threadsafty
|
|
type ThreadMap[K comparable, V any] struct {
|
|
dirty map[K]V
|
|
rwMutex sync.RWMutex
|
|
}
|
|
|
|
func NewThreadMap[K comparable, V any](m map[K]V) *ThreadMap[K, V] {
|
|
return &ThreadMap[K, V]{
|
|
dirty: m,
|
|
rwMutex: sync.RWMutex{},
|
|
}
|
|
}
|
|
|
|
func (m *ThreadMap[K, V]) Get(key K) (value V, ok bool) {
|
|
m.rwMutex.RLock()
|
|
defer m.rwMutex.RUnlock()
|
|
value, ok = m.dirty[key]
|
|
return value, ok
|
|
}
|
|
|
|
func (m *ThreadMap[K, V]) SetValue(key K, change func(old V) V) bool {
|
|
m.rwMutex.Lock()
|
|
defer m.rwMutex.Unlock()
|
|
|
|
value, ok := m.dirty[key]
|
|
if !ok {
|
|
return ok
|
|
}
|
|
|
|
m.dirty[key] = change(value)
|
|
return ok
|
|
}
|