libp2p/go-libp2p-peerstore
1
0
Fork 0
mirror of https://github.com/libp2p/go-libp2p-peerstore.git synced 2025-01-13 02:20:07 +08:00
Code Issues Projects Releases Wiki Activity

remove unnecessary map lookup memeoryPeerMetadata.Put

Browse Source
This commit is contained in:
Marten Seemann 2021-10-23 11:46:51 +02:00
parent a66346d0c1
commit 83f5e556c3
1 changed files with 3 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
pstoremem/metadata.go
View File

@ -3,15 +3,10 @@ package pstoremem
import (
"sync"
peer "github.com/libp2p/go-libp2p-core/peer"
"github.com/libp2p/go-libp2p-core/peer"
pstore "github.com/libp2p/go-libp2p-core/peerstore"
)
var internKeys = map[string]bool{
"AgentVersion": true,
"ProtocolVersion": true,
}
type metakey struct {
id peer.ID
key string
@ -19,7 +14,7 @@ type metakey struct {
type memoryPeerMetadata struct {
// store other data, like versions
//ds ds.ThreadSafeDatastore
// ds ds.ThreadSafeDatastore
ds map[metakey]interface{}
dslock sync.RWMutex
interned map[string]interface{}
@ -40,7 +35,7 @@ func (ps *memoryPeerMetadata) Put(p peer.ID, key string, val interface{}) error
}
ps.dslock.Lock()
defer ps.dslock.Unlock()
if vals, ok := val.(string); ok && internKeys[key] {
if vals, ok := val.(string); ok && (key == "AgentVersion" || key == "ProtocolVersion") {
if interned, ok := ps.interned[vals]; ok {
val = interned
} else {