go笔记–go协程安全map的四种方法
读写锁
这个应该是最简单的方法,读取的时候加读锁,修改的时候加写锁。
type LockMap struct {
m map[interface{}]interface{}
sync.RWMutex
}
func (lm *LockMap) Get(key interface{}) (value interface{}, ok bool) {
lm.RLock()
defer lm.RUnlock()
value, ok = lm.m[key]
return
}
func (lm *LockMap) Put(key interface{}, value interface{}) {
lm.Lock()
defer lm.Unlock()
lm.m[key] = value
}
func (lm *LockMap) Del(key interface{}) {
lm.Lock()
defer lm.Unlock()
delete(lm.m, key)
}
sync.map
同时sync.Map是不需要初始化的。
var smap sync.Map
func Set (k,v interface{}){
smap.Store(k,v)
}
func Get (k interface{}) interface{}{
v ,exit := smap.Load(k)
if exit {
return v
}
return nil
}
func Del(key interface{}){
smap.Delete(key)
}
func MapRange (funcs func(key, value interface{}) bool) {
smap.Range(funcs)
}
atomic.Value
在kratos 里面看到的一种协程安全map,利用了atomic.Value来封装的。
package paladin
import (
"strings"
"sync/atomic"
)
// KeyNamed key naming to lower case.
func KeyNamed(key string) string {
return strings.ToLower(key)
}
// Map is config map, key(filename) -> value(file).
type Map struct {
values atomic.Value
}
// Store sets the value of the Value to values map.
func (m *Map) Store(values map[string]*Value) {
dst := make(map[string]*Value, len(values))
for k, v := range values {
dst[KeyNamed(k)] = v
}
m.values.Store(dst)
}
// Load returns the value set by the most recent Store.
func (m *Map) Load() map[string]*Value {
src := m.values.Load().(map[string]*Value)
dst := make(map[string]*Value, len(src))
for k, v := range src {
dst[k] = v
}
return dst
}
// Exist check if values map exist a key.
func (m *Map) Exist(key string) bool {
_, ok := m.Load()[KeyNamed(key)]
return ok
}
// Get return get value by key.
func (m *Map) Get(key string) *Value {
v, ok := m.Load()[KeyNamed(key)]
if ok {
return v
}
return &Value{}
}
// Keys return map keys.
func (m *Map) Keys() []string {
values := m.Load()
keys := make([]string, 0, len(values))
for key := range values {
keys = append(keys, key)
}
return keys
}
chan
最后一种方法当然就是管道啦,select:case 万精油的方法嘛, 这里就不放例程了。