Merge pull request #417 from ethpandaops/fix/hermes-host-event-payloa…

…d-changes refactor: Update field name 'Transient' to 'Limited'
ethpandaops · Dec 2, 2024 · dedce7f · dedce7f
2 parents c1c55be + 375a14b
commit dedce7f
Show file tree

Hide file tree

Showing 3 changed files with 151 additions and 112 deletions.
diff --git a/pkg/proto/libp2p/trace.go b/pkg/proto/libp2p/trace.go
@@ -9,8 +9,8 @@ import (
 	ma "github.com/multiformats/go-multiaddr"
 	"github.com/probe-lab/hermes/host"
 	"github.com/prysmaticlabs/prysm/v5/consensus-types/primitives"
-	timestamppb "google.golang.org/protobuf/types/known/timestamppb"
-	wrapperspb "google.golang.org/protobuf/types/known/wrapperspb"
+	"google.golang.org/protobuf/types/known/timestamppb"
+	"google.golang.org/protobuf/types/known/wrapperspb"
 )
 
 // Helper function to convert a Hermes TraceEvent to a libp2p AddPeer
@@ -242,7 +242,7 @@ func TraceEventToConnected(event *host.TraceEvent) (*Connected, error) {
 		AgentVersion string
 		Direction    string
 		Opened       time.Time
-		Transient    bool
+		Limited      bool
 	})
 	if !ok {
 		return nil, fmt.Errorf("invalid payload type for Connected")
@@ -254,7 +254,8 @@ func TraceEventToConnected(event *host.TraceEvent) (*Connected, error) {
 		AgentVersion: wrapperspb.String(payload.AgentVersion),
 		Direction:    wrapperspb.String(payload.Direction),
 		Opened:       timestamppb.New(payload.Opened),
-		Transient:    wrapperspb.Bool(payload.Transient),
+		Limited:      wrapperspb.Bool(payload.Limited),
+		Transient:    wrapperspb.Bool(payload.Limited),
 	}, nil
 }
 
@@ -265,7 +266,7 @@ func TraceEventToDisconnected(event *host.TraceEvent) (*Disconnected, error) {
 		AgentVersion string
 		Direction    string
 		Opened       time.Time
-		Transient    bool
+		Limited      bool
 	})
 	if !ok {
 		return nil, fmt.Errorf("invalid payload type for Disconnected")
@@ -277,7 +278,8 @@ func TraceEventToDisconnected(event *host.TraceEvent) (*Disconnected, error) {
 		AgentVersion: wrapperspb.String(payload.AgentVersion),
 		Direction:    wrapperspb.String(payload.Direction),
 		Opened:       timestamppb.New(payload.Opened),
-		Transient:    wrapperspb.Bool(payload.Transient),
+		Limited:      wrapperspb.Bool(payload.Limited),
+		Transient:    wrapperspb.Bool(payload.Limited),
 	}, nil
 }