subscriptions: Store the validated GraphQL document in subscriptions

Author: Jannis Pohlmann

subscriptions.go

@@ -5,6 +5,7 @@ import (
 
 	"github.com/graphql-go/graphql"
 	"github.com/graphql-go/graphql/gqlerrors"
+	"github.com/graphql-go/graphql/language/ast"
 	"github.com/graphql-go/graphql/language/parser"
 	log "github.com/sirupsen/logrus"
 )
@@ -31,6 +32,7 @@ type Subscription struct {
 	Variables     map[string]interface{}
 	OperationName string
 	SendData      SubscriptionSendDataFunc
+	Document      *ast.Document
 }
 
 // ConnectionSubscriptions defines a map of all subscriptions of
@@ -108,6 +110,9 @@ func (m *subscriptionManager) AddSubscription(
 		return ErrorsFromGraphQLErrors(validation.Errors)
 	}
 
+	// Remember the query document for later
+	subscription.Document = document
+
 	// Allocate the connection's map of subscription IDs to
 	// subscriptions on demand
 	if m.subscriptions[conn] == nil {