apps/lkm
1
0
Fork 0
mirror of https://github.com/lkmio/lkm.git synced 2025-09-27 03:26:01 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

等待队列增加锁保护

Browse Source
This commit is contained in:
yangjiechina
2023-12-14 22:35:54 +08:00
parent 54759a1588
commit 4947217197
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
stream/sink_manager.go
View File

@@ -11,10 +11,11 @@ func init() {
}
func AddSinkToWaitingQueue(streamId string, sink ISink) {
mutex.Lock()
defer mutex.Unlock()
m, ok := waitingSinks[streamId]
if !ok {
mutex.Lock()
mutex.Unlock()
if m, ok = waitingSinks[streamId]; !ok {
m = make(map[SinkId]ISink, 64)
waitingSinks[streamId] = m
@@ -25,6 +26,9 @@ func AddSinkToWaitingQueue(streamId string, sink ISink) {
}
func RemoveSinkFromWaitingQueue(sourceId string, sinkId SinkId) (ISink, bool) {
mutex.Lock()
defer mutex.Unlock()
m, ok := waitingSinks[sourceId]
if !ok {
return nil, false
@@ -39,6 +43,9 @@ func RemoveSinkFromWaitingQueue(sourceId string, sinkId SinkId) (ISink, bool) {
}
func PopWaitingSinks(sourceId string) []ISink {
mutex.Lock()
defer mutex.Unlock()
source, ok := waitingSinks[sourceId]
if !ok {
return nil