git-sync/crawlab
1
0
Fork 0
mirror of https://github.com/crawlab-team/crawlab.git synced 2026-01-22 17:31:03 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #587 from hantmac/develop

Browse Source 
bug fix: fix concurrent map write error
This commit is contained in:
Marvin Zhang
2020-02-23 18:57:17 +08:00
committed by GitHub
parent c5be4fab03 7d81c013a6
commit 0c9b9ad47e
1 changed files with 12 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
backend/utils/chan.go
View File

@@ -1,29 +1,33 @@
package utils
import (
"sync"
)
var TaskExecChanMap = NewChanMap()
type ChanMap struct {
m map[string]chan string
m sync.Map
}
func NewChanMap() *ChanMap {
return &ChanMap{m: make(map[string]chan string)}
return &ChanMap{m: sync.Map{}}
}
func (cm *ChanMap) Chan(key string) chan string {
if ch, ok := cm.m[key]; ok {
return ch
if ch, ok := cm.m.Load(key); ok {
return ch.(interface{}).(chan string)
}
ch := make(chan string, 10)
cm.m[key] = ch
cm.m.Store(key, ch)
return ch
}
func (cm *ChanMap) ChanBlocked(key string) chan string {
if ch, ok := cm.m[key]; ok {
return ch
if ch, ok := cm.m.Load(key); ok {
return ch.(interface{}).(chan string)
}
ch := make(chan string)
cm.m[key] = ch
cm.m.Store(key, ch)
return ch
}