Progressive override: fix query planner cache warmup

.circleci/config.yml

@@ -1,5 +1,7 @@
 version: 2.1
 
+# Cache key bump: 1
+
 # These "CircleCI Orbs" are reusable bits of configuration that can be shared
 # across projects.  See https://circleci.com/orbs/ for more information.
 orbs:

apollo-router/src/query_planner/caching_query_planner.rs

@@ -336,6 +336,11 @@ where
                     lock.insert(caching_key.metadata)
                 });
 
+                let _ = context.insert(
+                    LABELS_TO_OVERRIDE_KEY,
+                    caching_key.plan_options.override_conditions.clone(),
+                );
+
                 let request = QueryPlannerRequest {
                     query,
                     operation_name,

apollo-router/tests/samples/enterprise/progressive-override/basic/README.md

@@ -0,0 +1,3 @@
+# Progressive override
+
+This tests subgraph field migration: https://www.apollographql.com/docs/federation/entities/migrate-fields/

apollo-router/tests/samples/enterprise/progressive-override/basic/configuration.yaml

@@ -0,0 +1,13 @@
+include_subgraph_errors:
+  all: true
+
+telemetry:
+  exporters:
+    logging:
+      stdout:
+        format: text
+
+experimental_query_planner_mode: legacy
+
+plugins:
+  experimental.expose_query_plan: true

apollo-router/tests/samples/enterprise/progressive-override/basic/configuration2.yaml

@@ -0,0 +1,17 @@
+include_subgraph_errors:
+  all: true
+
+telemetry:
+  exporters:
+    logging:
+      stdout:
+        format: text
+
+experimental_query_planner_mode: legacy
+
+rhai:
+  scripts: "tests/samples/enterprise/progressive-override/basic/rhai"
+  main: "main.rhai"
+
+plugins:
+  experimental.expose_query_plan: true

apollo-router/tests/samples/enterprise/progressive-override/basic/plan.json

@@ -0,0 +1,226 @@
+{
+    "enterprise": true,
+    "actions": [
+        {
+            "type": "Start",
+            "schema_path": "./supergraph.graphql",
+            "configuration_path": "./configuration.yaml",
+            "subgraphs": {
+                "Subgraph1": {
+                    "requests": [
+                        {
+                            "request": {
+                                "body": {
+                                    "query": "query progressive1__Subgraph1__0{percent100{__typename id}}",
+                                    "operationName": "progressive1__Subgraph1__0"
+                                }
+                            },
+                            "response": {
+                                "body": {
+                                    "data": {
+                                        "percent100": {
+                                            "__typename": "T",
+                                            "id": "1"
+                                        }
+                                    }
+                                }
+                            }
+                        }
+                    ]
+                },
+                "Subgraph2": {
+                    "requests": [
+                        {
+                            "request": {
+                                "body": {
+                                    "query": "query progressive1__Subgraph2__1($representations:[_Any!]!){_entities(representations:$representations){...on T{foo}}}",
+                                    "operationName": "progressive1__Subgraph2__1",
+                                    "variables": {
+                                        "representations": [
+                                            {
+                                                "__typename": "T",
+                                                "id": "1"
+                                            }
+                                        ]
+                                    }
+                                }
+                            },
+                            "response": {
+                                "body": {
+                                    "data": {
+                                        "_entities": [
+                                            {
+                                                "foo": 1
+                                            }
+                                        ]
+                                    }
+                                }
+                            }
+                        },
+                        {
+                            "request": {
+                                "body": {
+                                    "query": "query progressive2__Subgraph2__0{percent0{foo}}",
+                                    "operationName": "progressive2__Subgraph2__0"
+                                }
+                            },
+                            "response": {
+                                "body": {
+                                    "data": {
+                                        "percent0": {
+                                            "foo": 2
+                                        }
+                                    }
+                                }
+                            }
+                        }
+                    ]
+                }
+            }
+        },
+        {
+            "type": "Request",
+            "request": {
+                "query": "query progressive1 { percent100 { foo } }"
+            },
+            "headers": {
+                "apollo-expose-query-plan": "false"
+            },
+            "expected_response": {
+                "data": {
+                    "percent100": {
+                        "foo": 1
+                    }
+                }
+            }
+        },
+        {
+            "type": "Request",
+            "request": {
+                "query": "query progressive2 { percent0 { foo } }"
+            },
+            "expected_response": {
+                "data": {
+                    "percent0": {
+                        "foo": 2
+                    }
+                }
+            }
+        },
+        {
+            "type": "ReloadConfiguration",
+            "configuration_path": "./configuration2.yaml"
+        },
+        {
+            "type": "ReloadSubgraphs",
+            "subgraphs": {
+                "Subgraph1": {
+                    "requests": [
+                        {
+                            "request": {
+                                "body": {
+                                    "query": "query progressive3__Subgraph1__0{percent100{__typename id}}",
+                                    "operationName": "progressive3__Subgraph1__0"
+                                }
+                            },
+                            "response": {
+                                "body": {
+                                    "data": {
+                                        "percent100": {
+                                            "__typename": "T",
+                                            "id": "1"
+                                        }
+                                    }
+                                }
+                            }
+                        },
+                        {
+                            "request": {
+                                "body": {
+                                    "query": "query progressive4__Subgraph1__0{percent100{bar}}",
+                                    "operationName": "progressive4__Subgraph1__0"
+                                }
+                            },
+                            "response": {
+                                "body": {
+                                    "data": {
+                                        "percent100": {
+                                            "bar": 2
+                                        }
+                                    }
+                                }
+                            }
+                        }
+                    ]
+                },
+                "Subgraph2": {
+                    "requests": [
+                        {
+                            "request": {
+                                "body": {
+                                    "query": "query progressive3__Subgraph2__1($representations:[_Any!]!){_entities(representations:$representations){...on T{bar}}}",
+                                    "operationName": "progressive3__Subgraph2__1",
+                                    "variables": {
+                                        "representations": [
+                                            {
+                                                "__typename": "T",
+                                                "id": "1"
+                                            }
+                                        ]
+                                    }
+                                }
+                            },
+                            "response": {
+                                "body": {
+                                    "data": {
+                                        "_entities": [
+                                            {
+                                                "bar": 1
+                                            }
+                                        ]
+                                    }
+                                }
+                            }
+                        }
+                    ]
+                }
+            }
+        },
+        {
+            "type": "Request",
+            "request": {
+                "query": "query progressive3 { percent100 { bar } }"
+            },
+            "headers": {
+                "apollo-expose-query-plan": "false"
+            },
+            "expected_response": {
+                "data": {
+                    "percent100": {
+                        "bar": 1
+                    }
+                }
+            }
+        },
+        {
+            "type": "Request",
+            "request": {
+                "query": "query progressive4 { percent100 { bar } }"
+            },
+            "headers": {
+                "apollo-expose-query-plan": "false",
+                "x-override": "true"
+            },
+            "expected_response": {
+                "data": {
+                    "percent100": {
+                        "bar": 2
+                    }
+                }
+            }
+        },
+        {
+            "type": "Stop"
+        }
+    ]
+}

apollo-router/tests/samples/enterprise/progressive-override/basic/rhai/main.rhai

@@ -0,0 +1,22 @@
+fn supergraph_service(service) {
+    const request_callback = Fn("process_request");
+    service.map_request(request_callback);
+}
+
+// Add a timestamp to context which we'll use in the response.
+fn process_request(request) {
+    request.context["request_start"] = Router.APOLLO_START.elapsed;
+    let labels = request.context["apollo_override::unresolved_labels"];
+    print(`unresolved: ${labels}`);
+
+    let override = request.context["apollo_override::labels_to_override"];
+    print(`override: ${override}`);
+
+
+    if "x-override" in request.headers {
+        if request.headers["x-override"] == "true" {
+            request.context["apollo_override::labels_to_override"] += "bar";
+        }
+    }
+}
+