Merge pull request #27 from libp2p/feat/inline-pk

add support for extracting inlined public keys
libp2p · Apr 16, 2018 · 72e0087 · 72e0087
2 parents fd4d308 + 6ef12b5
commit 72e0087
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/p2p/host/peerstore/peerstore.go b/p2p/host/peerstore/peerstore.go
@@ -127,6 +127,15 @@ func (kb *keybook) PubKey(p peer.ID) ic.PubKey {
 	kb.RLock()
 	pk := kb.pks[p]
 	kb.RUnlock()
+	if pk != nil {
+		return pk
+	}
+	pk, err := p.ExtractPublicKey()
+	if err == nil && pk != nil {
+		kb.Lock()
+		kb.pks[p] = pk
+		kb.Unlock()
+	}
 	return pk
 }