Merge pull request #637 from rainforestpay/nrpgx5_send_query_params

Add SendQueryParameters flag to nrgpx5 to selectively enable sending database query parameters
newrelic · Mar 30, 2023 · fa50493 · fa50493
2 parents b89dbe0 + ab1d062
commit fa50493
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/v3/integrations/nrpgx5/nrpgx5.go b/v3/integrations/nrpgx5/nrpgx5.go
@@ -68,8 +68,9 @@ func init() {
 
 type (
 	Tracer struct {
-		BaseSegment newrelic.DatastoreSegment
-		ParseQuery  func(segment *newrelic.DatastoreSegment, query string)
+		BaseSegment         newrelic.DatastoreSegment
+		ParseQuery          func(segment *newrelic.DatastoreSegment, query string)
+		SendQueryParameters bool
 	}
 
 	nrPgxSegmentType string
@@ -83,7 +84,8 @@ const (
 
 func NewTracer() *Tracer {
 	return &Tracer{
-		ParseQuery: sqlparse.ParseQuery,
+		ParseQuery:          sqlparse.ParseQuery,
+		SendQueryParameters: true,
 	}
 }
 
@@ -109,7 +111,9 @@ func (t *Tracer) TraceQueryStart(ctx context.Context, conn *pgx.Conn, data pgx.T
 	segment := t.BaseSegment
 	segment.StartTime = newrelic.FromContext(ctx).StartSegmentNow()
 	segment.ParameterizedQuery = data.SQL
-	segment.QueryParameters = t.getQueryParameters(data.Args)
+	if t.SendQueryParameters {
+		segment.QueryParameters = t.getQueryParameters(data.Args)
+	}
 
 	// fill Operation and Collection
 	t.ParseQuery(&segment, data.SQL)