feat: store additional ENR fields

dennis-tra · Feb 13, 2024 · 904cb5d · 904cb5d
1 parent 55a4a2e
commit 904cb5d
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/discv5/crawler.go b/discv5/crawler.go
@@ -120,9 +120,21 @@ func (c *Crawler) Work(ctx context.Context, task PeerInfo) (core.CrawlResult[Pee
 func (c *Crawler) PeerProperties(node *enode.Node) map[string]any {
 	properties := map[string]any{}
 
+	if ip := node.IP(); ip != nil {
+		properties["ip"] = ip.String()
+	}
+
 	properties["seq"] = node.Record().Seq()
 	properties["signature"] = node.Record().Signature()
 
+	if node.UDP() != 0 {
+		properties["udp"] = node.UDP()
+	}
+
+	if node.TCP() != 0 {
+		properties["tcp"] = node.TCP()
+	}
+
 	var enrEntryEth2 ENREntryEth2
 	if err := node.Load(&enrEntryEth2); err == nil {
 		properties["fork_digest"] = enrEntryEth2.ForkDigest.String()