Add dql_query to extensions when GraphQL debug super flag is true

hypermodeinc · Jan 17, 2025 · 16af8e2 · 16af8e2
1 parent 053a440
commit 16af8e2
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/graphql/resolve/query.go b/graphql/resolve/query.go
@@ -139,6 +139,9 @@ func (qr *queryResolver) rewriteAndExecute(ctx context.Context, query schema.Que
 	}
 
 	ext.TouchedUids = resp.GetMetrics().GetNumUids()[touchedUidsKey]
+	if x.Config.GraphQL.GetBool("debug") {
+		ext.DQLQuery = qry
+	}
 	resolved := &Resolved{
 		Data:       resp.GetJson(),
 		Field:      query,

diff --git a/graphql/schema/response.go b/graphql/schema/response.go
@@ -182,6 +182,7 @@ func (r *Response) Output() interface{} {
 type Extensions struct {
 	TouchedUids uint64 `json:"touched_uids,omitempty"`
 	Tracing     *Trace `json:"tracing,omitempty"`
+	DQLQuery    string `json:"dql_query,omitempty"`
 }
 
 // GetTouchedUids returns TouchedUids
@@ -205,6 +206,12 @@ func (e *Extensions) Merge(ext *Extensions) {
 	} else {
 		e.Tracing.Merge(ext.Tracing)
 	}
+
+	if e.DQLQuery == "" {
+		e.DQLQuery = ext.DQLQuery
+	} else {
+		e.DQLQuery = e.DQLQuery + "\n" + ext.DQLQuery
+	}
 }
 
 // Trace : Apollo Tracing is a GraphQL extension for tracing resolver performance.Response