feat: set operation name also with persistent queries

Author: alepane21

router/core/operation_processor.go

@@ -897,6 +897,40 @@ func (o *OperationKit) setAndParseOperationDoc() error {
 	return nil
 }
 
+func (o *OperationKit) populateOperationNameFromDoc() {
+	if o.parsedOperation.Request.OperationName != "" {
+		return
+	}
+	if o.kit == nil || o.kit.doc == nil {
+		return
+	}
+
+	ref := o.operationDefinitionRef
+	if ref < 0 || ref >= len(o.kit.doc.OperationDefinitions) {
+		ref = ast.InvalidRef
+	}
+	if ref == ast.InvalidRef {
+		for i := range o.kit.doc.RootNodes {
+			if o.kit.doc.RootNodes[i].Kind == ast.NodeKindOperationDefinition {
+				ref = o.kit.doc.RootNodes[i].Ref
+				break
+			}
+		}
+	}
+	if ref == ast.InvalidRef {
+		return
+	}
+
+	name := o.kit.doc.OperationDefinitionNameString(ref)
+	if name == "" {
+		return
+	}
+
+	o.parsedOperation.Request.OperationName = name
+	o.operationDefinitionRef = ref
+	o.originalOperationNameRef = o.kit.doc.OperationDefinitions[ref].Name
+}
+
 func (o *OperationKit) normalizeVariablesCacheKey() uint64 {
 	_, _ = o.kit.keyGen.Write(o.kit.doc.Input.Variables)
 	_, _ = o.kit.keyGen.WriteString(o.parsedOperation.NormalizedRepresentation)
@@ -1144,6 +1178,7 @@ func (o *OperationKit) handleFoundPersistedOperationEntry(entry NormalizationCac
 	if err != nil {
 		return err
 	}
+	o.populateOperationNameFromDoc()
 	return nil
 }
 

router/core/operation_processor_test.go

@@ -75,6 +75,33 @@ func TestOperationProcessorPersistentOperations(t *testing.T) {
 	}
 }
 
+func TestPersistedOperationCachePopulatesOperationName(t *testing.T) {
+	executor := &Executor{
+		PlanConfig:      plan.Configuration{},
+		RouterSchema:    nil,
+		Resolver:        nil,
+		RenameTypeNames: nil,
+	}
+	processor := NewOperationProcessor(OperationProcessorOptions{
+		Executor:                executor,
+		MaxOperationSizeInBytes: 10 << 20,
+		ParseKitPoolSize:        1,
+	})
+
+	kit, err := processor.NewKit()
+	require.NoError(t, err)
+	defer kit.Free()
+
+	entry := NormalizationCacheEntry{
+		normalizedRepresentation: "query TestOperation { a }",
+		operationType:            "query",
+	}
+
+	err = kit.handleFoundPersistedOperationEntry(entry)
+	require.NoError(t, err)
+	require.Equal(t, "TestOperation", kit.parsedOperation.Request.OperationName)
+}
+
 func TestParseOperationProcessor(t *testing.T) {
 	executor := &Executor{
 		PlanConfig:      plan.Configuration{},