Add client headers for events stream.

attestantio · Jul 22, 2024 · 0138132 · 0138132
1 parent 5f15ea2
commit 0138132
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,3 +1,6 @@
+dev:
+  - add client headers for events stream
+
 0.21.7:
   - use POST for specific validator and validator balance information
 

diff --git a/http/events.go b/http/events.go
@@ -62,6 +62,12 @@ func (s *Service) Events(ctx context.Context, topics []string, handler consensus
 	log.Trace().Str("url", callURL).Msg("GET request to events stream")
 
 	client := sse.NewClient(callURL)
+	for k, v := range s.extraHeaders {
+		client.Headers[k] = v
+	}
+	if _, exists := client.Headers["User-Agent"]; !exists {
+		client.Headers["User-Agent"] = defaultUserAgent
+	}
 	client.Connection.Transport = &http.Transport{
 		Dial: (&net.Dialer{
 			Timeout:   2 * time.Second,