feat: Add capability id header for all NerdGraph requests (#1049)

newrelic · Aug 24, 2023 · 5f82f63 · 5f82f63
1 parent bc87cb6
commit 5f82f63
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/internal/http/client.go b/internal/http/client.go
@@ -535,5 +535,11 @@ func (c *Client) NewNerdGraphRequest(query string, vars map[string]interface{},
 	req.SetAuthStrategy(&NerdGraphAuthorizer{})
 	req.SetErrorValue(&GraphQLErrorResponse{})
 
+	if strings.Contains(c.config.ServiceName, "terraform-provider-newrelic") {
+		req.SetHeader("X-Query-Source-Capability-Id", "TERRAFORM")
+	} else {
+		req.SetHeader("X-Query-Source-Component-Id", c.config.ServiceName)
+	}
+
 	return req, nil
 }