add initialCount support to rust compiler

Author: robrichard

compiler/crates/graphql-ir/src/errors.rs

@@ -356,7 +356,7 @@ pub enum ValidationMessage {
     #[error("Invalid use of @stream on scalar field '{field_name}'")]
     InvalidStreamOnScalarField { field_name: StringKey },
 
-    #[error("Invalid use of @stream, the 'initial_count' argument is required.")]
+    #[error("Invalid use of @stream, the 'initial_count' or 'initialCount' argument is required.")]
     StreamInitialCountRequired,
 
     #[error("{variables_string} never used in operation '{operation_name}'.")]

compiler/crates/graphql-transforms/src/defer_stream/directives.rs

@@ -58,6 +58,8 @@ impl<'a> StreamDirective<'a> {
                 label_arg = Some(arg);
             } else if arg.name.item == DEFER_STREAM_CONSTANTS.initial_count_arg {
                 initial_count_arg = Some(arg);
+            } else if arg.name.item == DEFER_STREAM_CONSTANTS.initial_count_arg_oss {
+                initial_count_arg = Some(arg);
             } else if arg.name.item == DEFER_STREAM_CONSTANTS.use_customized_batch_arg {
                 use_customized_batch_arg = Some(arg);
             } else {

compiler/crates/graphql-transforms/src/defer_stream/mod.rs

@@ -26,6 +26,7 @@ pub struct DeferStreamConstants {
     pub if_arg: StringKey,
     pub label_arg: StringKey,
     pub initial_count_arg: StringKey,
+    pub initial_count_arg_oss: StringKey,
     pub use_customized_batch_arg: StringKey,
 }
 
@@ -37,6 +38,7 @@ impl Default for DeferStreamConstants {
             if_arg: "if".intern(),
             label_arg: "label".intern(),
             initial_count_arg: "initial_count".intern(),
+            initial_count_arg_oss: "initialCount".intern(),
             use_customized_batch_arg: "use_customized_batch".intern(),
         }
     }

compiler/crates/graphql-transforms/src/transform_connections.rs

@@ -125,6 +125,7 @@ impl<'s> ConnectionTransform<'s> {
             for arg in &connection_directive.arguments {
                 if arg.name.item == DEFER_STREAM_CONSTANTS.if_arg
                     || arg.name.item == DEFER_STREAM_CONSTANTS.initial_count_arg
+                    || arg.name.item == DEFER_STREAM_CONSTANTS.initial_count_arg_oss
                     || arg.name.item == DEFER_STREAM_CONSTANTS.use_customized_batch_arg
                 {
                     arguments.push(arg.clone());

compiler/crates/graphql-transforms/tests/defer_stream/fixtures/fragment-with-stream-initial-count-arg-oss.expected

@@ -0,0 +1,31 @@
+==================================== INPUT ====================================
+query QueryWithFragmentWithStream($id: ID!, $initialCount: Int) {
+  node(id: $id) {
+    id
+    ...FeedbackFragment
+  }
+}
+
+fragment FeedbackFragment on Feedback {
+  id
+  actors @stream(initialCount: $initialCount, label: "StreamedActorsLabel") {
+    name
+  }
+}
+==================================== OUTPUT ===================================
+query QueryWithFragmentWithStream(
+  $id: ID!
+  $initialCount: Int
+) {
+  node(id: $id) {
+    id
+    ...FeedbackFragment
+  }
+}
+
+fragment FeedbackFragment on Feedback {
+  id
+  actors @stream(label: "FeedbackFragment$stream$StreamedActorsLabel", initialCount: $initialCount) {
+    name
+  }
+}

compiler/crates/graphql-transforms/tests/defer_stream/fixtures/fragment-with-stream-initial-count-arg-oss.graphql

@@ -0,0 +1,13 @@
+query QueryWithFragmentWithStream($id: ID!, $initialCount: Int) {
+  node(id: $id) {
+    id
+    ...FeedbackFragment
+  }
+}
+
+fragment FeedbackFragment on Feedback {
+  id
+  actors @stream(initialCount: $initialCount, label: "StreamedActorsLabel") {
+    name
+  }
+}

compiler/crates/graphql-transforms/tests/defer_stream/fixtures/fragment-with-stream-missing-initial-count-arg.invalid.expected

@@ -15,6 +15,6 @@ fragment FeedbackFragment on Feedback {
   }
 }
 ==================================== ERROR ====================================
-Invalid use of @stream, the 'initial_count' argument is required.:
+Invalid use of @stream, the 'initial_count' or 'initialCount' argument is required.:
 fragment-with-stream-missing-initial-count-arg.invalid.graphql:10:11:
   actors @stream(label: "StreamedActorsLabel") {

compiler/crates/graphql-transforms/tests/defer_stream_test.rs

@@ -89,6 +89,13 @@ fn fragment_with_stream_initial_count_arg() {
     test_fixture(transform_fixture, "fragment-with-stream-initial-count-arg.graphql", "defer_stream/fixtures/fragment-with-stream-initial-count-arg.expected", input, expected);
 }
 
+#[test]
+fn fragment_with_stream_initial_count_arg_oss() {
+    let input = include_str!("defer_stream/fixtures/fragment-with-stream-initial-count-arg-oss.graphql");
+    let expected = include_str!("defer_stream/fixtures/fragment-with-stream-initial-count-arg-oss.expected");
+    test_fixture(transform_fixture, "fragment-with-stream-initial-count-arg-oss.graphql", "defer_stream/fixtures/fragment-with-stream-initial-count-arg-oss.expected", input, expected);
+}
+
 #[test]
 fn fragment_with_stream_missing_initial_count_arg_invalid() {
     let input = include_str!("defer_stream/fixtures/fragment-with-stream-missing-initial-count-arg.invalid.graphql");

compiler/crates/graphql-transforms/tests/transform_connections/fixtures/stream-connection-oss.expected

@@ -0,0 +1,57 @@
+==================================== INPUT ====================================
+query NodeQuery($id: ID!) {
+  node(id: $id) {
+    id
+    ... on Story {
+      comments(first: 10)
+        @stream_connection(key: "NodeQuery_comments", initialCount: 0) {
+        edges {
+          node {
+            actor {
+              name
+            }
+          }
+        }
+        pageInfo {
+          endCursor
+          hasNextPage
+        }
+      }
+    }
+  }
+}
+==================================== OUTPUT ===================================
+query NodeQuery(
+  $id: ID!
+) @__connectionMetadata(__connectionMetadataArgument: [[["node", "comments"], "forward", null, null, null, null, true]]) {
+  node(id: $id) {
+    id
+    ... on Story {
+      comments(first: 10) @__clientField(key: "NodeQuery_comments", handle: "connection") {
+        edges @stream(label: "NodeQuery_comments", initialCount: 0) {
+          node {
+            actor {
+              name
+            }
+          }
+          ... on CommentsEdge {
+            cursor
+            node {
+              __typename
+            }
+          }
+        }
+        ... @defer(label: "NodeQuery$defer$NodeQuery_comments$pageInfo") {
+          pageInfo {
+            endCursor
+            hasNextPage
+            ... on PageInfo {
+              endCursor
+              hasNextPage
+            }
+          }
+        }
+      }
+    }
+  }
+}

compiler/crates/graphql-transforms/tests/transform_connections/fixtures/stream-connection-oss.graphql

@@ -0,0 +1,21 @@
+query NodeQuery($id: ID!) {
+  node(id: $id) {
+    id
+    ... on Story {
+      comments(first: 10)
+        @stream_connection(key: "NodeQuery_comments", initialCount: 0) {
+        edges {
+          node {
+            actor {
+              name
+            }
+          }
+        }
+        pageInfo {
+          endCursor
+          hasNextPage
+        }
+      }
+    }
+  }
+}