Fixed map race in concurrency

toomore · May 1, 2021 · fee733c · fee733c
1 parent a75ed5f
commit fee733c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/mails/replace.go b/mails/replace.go
@@ -95,6 +95,7 @@ func filteratags(rg *regexp.Regexp, body *[]byte, cid string) map[string]LinksDa
 		conn     = utils.GetConn()
 		result   = make(map[string]LinksData)
 		wg       sync.WaitGroup
+		lock     = sync.RWMutex{}
 	)
 	wg.Add(len(allATags))
 	for _, v := range allATags {
@@ -110,11 +111,13 @@ func filteratags(rg *regexp.Regexp, body *[]byte, cid string) map[string]LinksDa
 					rows.Scan(&linkID)
 				}
 			}
+			lock.Lock()
 			result[linkID] = LinksData{
 				Md5h:   md5hstr,
 				LinkID: linkID,
 				URL:    url,
 			}
+			lock.Unlock()
 			wg.Done()
 		}(v[1])
 	}