fix: map race in GS transport (#208)

ipfs · May 18, 2021 · 717c6d9 · 717c6d9
1 parent e69af10
commit 717c6d9
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/transport/graphsync/graphsync.go b/transport/graphsync/graphsync.go
@@ -1084,7 +1084,7 @@ func newGSKeyToChannelIDMap() *gsKeyToChannelIDMap {
 }
 
 // get the value for a key
-func (m gsKeyToChannelIDMap) load(key graphsyncKey) (datatransfer.ChannelID, bool) {
+func (m *gsKeyToChannelIDMap) load(key graphsyncKey) (datatransfer.ChannelID, bool) {
 	m.lk.RLock()
 	defer m.lk.RUnlock()
 
@@ -1093,7 +1093,7 @@ func (m gsKeyToChannelIDMap) load(key graphsyncKey) (datatransfer.ChannelID, boo
 }
 
 // get the value if any of the keys exists in the map
-func (m gsKeyToChannelIDMap) any(ks ...graphsyncKey) (datatransfer.ChannelID, bool) {
+func (m *gsKeyToChannelIDMap) any(ks ...graphsyncKey) (datatransfer.ChannelID, bool) {
 	m.lk.RLock()
 	defer m.lk.RUnlock()
 
@@ -1107,15 +1107,15 @@ func (m gsKeyToChannelIDMap) any(ks ...graphsyncKey) (datatransfer.ChannelID, bo
 }
 
 // set the value for a key
-func (m gsKeyToChannelIDMap) set(key graphsyncKey, val datatransfer.ChannelID) {
+func (m *gsKeyToChannelIDMap) set(key graphsyncKey, val datatransfer.ChannelID) {
 	m.lk.Lock()
 	defer m.lk.Unlock()
 
 	m.m[key] = val
 }
 
 // call f for each key / value in the map
-func (m gsKeyToChannelIDMap) forEach(f func(k graphsyncKey, chid datatransfer.ChannelID)) {
+func (m *gsKeyToChannelIDMap) forEach(f func(k graphsyncKey, chid datatransfer.ChannelID)) {
 	m.lk.RLock()
 	defer m.lk.RUnlock()
 
@@ -1125,7 +1125,7 @@ func (m gsKeyToChannelIDMap) forEach(f func(k graphsyncKey, chid datatransfer.Ch
 }
 
 // delete any keys that reference this value
-func (m gsKeyToChannelIDMap) deleteRefs(id datatransfer.ChannelID) {
+func (m *gsKeyToChannelIDMap) deleteRefs(id datatransfer.ChannelID) {
 	m.lk.Lock()
 	defer m.lk.Unlock()