Try extracting public key from peer identity

See ipfs/kubo#3896. Depends on libp2p/go-libp2p-peer#13.
JustinDrake · May 12, 2017 · b77c03a · b77c03a
1 parent f02c72d
commit b77c03a
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/records.go b/records.go
@@ -24,8 +24,14 @@ const MaxRecordAge = time.Hour * 36
 func (dht *IpfsDHT) GetPublicKey(ctx context.Context, p peer.ID) (ci.PubKey, error) {
 	log.Debugf("getPublicKey for: %s", p)
 
+	// try extracting from identity.
+	pk := p.ExtractPublicKey()
+	if pk != nil {
+		return pk, nil
+	}
+
 	// check locally.
-	pk := dht.peerstore.PubKey(p)
+	pk = dht.peerstore.PubKey(p)
 	if pk != nil {
 		return pk, nil
 	}