GO 里面 MAP 如何实现 key 不存在 get 操作等待 直到 key 存在或者超时,保证并发安全,且需要实现以下接口:
~~~go
type sp interface {
Out(key string, val interface{}) //存入key /val,如果该key读取的goroutine挂起,则唤醒。此方法不会阻塞,时刻都可以立即执行并返回
Rd(key string, timeout time.Duration) interface{} //读取一个key,如果key不存在阻塞,等待key存在或者超时
}
~~~
**解析:**
看到阻塞协程第一个想到的就是`channel`,题目中要求并发安全,那么必须用锁,还要实现多个`goroutine`读的时候如果值不存在则阻塞,直到写入值,那么每个键值需要有一个阻塞`goroutine`的`channel`。
实现如下:
~~~
type Map struct {
c map[string]*entry
rmx *sync.RWMutex
}
type entry struct {
ch chan struct{}
value interface{}
isExist bool
}
func (m *Map) Out(key string, val interface{}) {
m.rmx.Lock()
defer m.rmx.Unlock()
if e, ok := m.c[key]; ok {
e.value = val
e.isExist = true
close(e.ch)
} else {
e = &entry{ch: make(chan struct{}), isExist: true,value:val}
m.c[key] = e
close(e.ch)
}
}
func (m *Map) Rd(key string, timeout time.Duration) interface{} {
m.rmx.Lock()
if e, ok := m.c[key]; ok && e.isExist {
m.rmx.Unlock()
return e.value
} else if !ok {
e = &entry{ch: make(chan struct{}), isExist: false}
m.c[key] = e
m.rmx.Unlock()
fmt.Println("协程阻塞 -> ", key)
select {
case <-e.ch:
return e.value
case <-time.After(timeout):
fmt.Println("协程超时 -> ", key)
return nil
}
} else {
m.rmx.Unlock()
fmt.Println("协程阻塞 -> ", key)
select {
case <-e.ch:
return e.value
case <-time.After(timeout):
fmt.Println("协程超时 -> ", key)
return nil
}
}
}
~~~
- Golnag常见面试题目解析
- 交替打印数组和字母
- 判断字符串中字符是否全都不同
- 翻转字符串
- 判断两个给定的字符串排序后是否一致
- 字符串替换问题
- 机器人坐标计算
- 语法题目一
- 语法题目二
- goroutine和channel使用一
- 实现阻塞读的并发安全Map
- 定时与 panic 恢复
- 高并发下的锁与map读写问题
- 为 sync.WaitGroup 中Wait函数支持 WaitTimeout 功能.
- 七道语法找错题目
- golang 并发题目测试
- 记一道字节跳动的算法面试题
- 多协程查询切片问题
- 对已经关闭的的chan进行读写,会怎么样?为什么?
- 简单聊聊内存逃逸?
- 字符串转成byte数组,会发生内存拷贝吗?
- http包的内存泄漏