Merge pull request #1735 from DimuthuMadushan/server-side-cache

Introduce GraphQL Server-side Caching

ballerina-tests/Ballerina.toml

@@ -1,5 +1,5 @@
 [package]
 org = "ballerina"
 name = "graphql_tests"
-version = "1.10.1"
+version = "1.11.0"
 

ballerina-tests/Dependencies.toml

@@ -33,7 +33,7 @@ dependencies = [
 [[package]]
 org = "ballerina"
 name = "constraint"
-version = "1.4.0"
+version = "1.5.0"
 dependencies = [
 	{org = "ballerina", name = "jballerina.java"}
 ]
@@ -44,7 +44,7 @@ modules = [
 [[package]]
 org = "ballerina"
 name = "crypto"
-version = "2.5.0"
+version = "2.6.2"
 dependencies = [
 	{org = "ballerina", name = "jballerina.java"},
 	{org = "ballerina", name = "time"}
@@ -67,9 +67,11 @@ modules = [
 [[package]]
 org = "ballerina"
 name = "graphql"
-version = "1.10.1"
+version = "1.11.0"
 dependencies = [
 	{org = "ballerina", name = "auth"},
+	{org = "ballerina", name = "cache"},
+	{org = "ballerina", name = "crypto"},
 	{org = "ballerina", name = "file"},
 	{org = "ballerina", name = "http"},
 	{org = "ballerina", name = "io"},
@@ -96,7 +98,7 @@ modules = [
 [[package]]
 org = "ballerina"
 name = "graphql_tests"
-version = "1.10.1"
+version = "1.11.0"
 dependencies = [
 	{org = "ballerina", name = "constraint"},
 	{org = "ballerina", name = "file"},
@@ -119,7 +121,7 @@ modules = [
 [[package]]
 org = "ballerina"
 name = "http"
-version = "2.10.5"
+version = "2.10.6"
 dependencies = [
 	{org = "ballerina", name = "auth"},
 	{org = "ballerina", name = "cache"},

ballerina-tests/build.gradle

@@ -1,5 +1,3 @@
-import org.apache.tools.ant.taskdefs.condition.Os
-
 /*
  * Copyright (c) 2022, WSO2 Inc. (http://www.wso2.org) All Rights Reserved.
  *

ballerina-tests/custom_prefix_without_import.bal

@@ -14,7 +14,14 @@
 // specific language governing permissions and limitations
 // under the License.
 
+import ballerina/graphql;
+
 service /custom_prefix on graphqlListener {
+    @graphql:ResourceConfig {
+        cacheConfig: {
+            enabled: true
+        }
+    }
     resource function get greeting() returns string {
         return "Hello, world";
     }

ballerina-tests/tests/batch_load_functions.bal

@@ -71,3 +71,29 @@ isolated function faultyAuthorLoaderFunction(readonly & anydata[] ids) returns A
         return validKeys.'map(key => authorTable.get(key));
     }
 };
+
+isolated function authorLoaderFunction2(readonly & anydata[] ids) returns AuthorRow[]|error {
+    readonly & int[] keys = check ids.ensureType();
+    // Simulate query: SELECT * FROM authors WHERE id IN (...keys);
+    lock {
+        dispatchCountOfAuthorLoader += 1;
+    }
+    lock {
+        readonly & int[] validKeys = keys.'filter(key => authorTable2.hasKey(key)).cloneReadOnly();
+        return keys.length() != validKeys.length() ? error("Invalid keys found for authors")
+            : validKeys.'map(key => authorTable2.get(key));
+    }
+};
+
+isolated function bookLoaderFunction2(readonly & anydata[] ids) returns BookRow[][]|error {
+    final readonly & int[] keys = check ids.ensureType();
+    // Simulate query: SELECT * FROM books WHERE author IN (...keys);
+    lock {
+        dispatchCountOfBookLoader += 1;
+    }
+    return keys.'map(isolated function(readonly & int key) returns BookRow[] {
+        lock {
+            return bookTable2.'filter(book => book.author == key).toArray().clone();
+        }
+    });
+};

ballerina-tests/tests/interceptors.bal

@@ -847,7 +847,7 @@ readonly service class ServiceLevelInterceptor {
     }
 
     isolated function grantAccess(string fieldName) returns boolean {
-        string[] grantedFields = ["profile", "books", "setName", "person", "setAge", "customer", "newBooks"];
+        string[] grantedFields = ["profile", "books", "setName", "person", "setAge", "customer", "newBooks", "updatePerson", "person"];
         if grantedFields.indexOf(fieldName) is int {
             return true;
         }

ballerina-tests/tests/object_types.bal

@@ -483,3 +483,110 @@ distinct service class NestedField {
             ],
             @graphql:ID string[] j = ["id1"]) returns string? => ();
 }
+
+public type HumanService FriendService|EnemyService;
+
+public isolated distinct service class FriendService {
+    private final string name;
+    private final int age;
+    private final boolean isMarried;
+
+    public isolated function init(string name, int age, boolean isMarried) {
+        self.name = name;
+        self.age = age;
+        self.isMarried = isMarried;
+    }
+
+    @graphql:ResourceConfig {
+        cacheConfig: {
+            maxAge: 180
+        }
+    }
+    isolated resource function get name() returns string {
+        return self.name;
+    }
+
+    @graphql:ResourceConfig {
+        cacheConfig: {
+            maxAge: 180
+        }
+    }
+    isolated resource function get age() returns int {
+        return self.age;
+    }
+
+    @graphql:ResourceConfig {
+        cacheConfig: {
+            maxAge: 180
+        }
+    }
+    isolated resource function get isMarried() returns boolean {
+        return self.isMarried;
+    }
+}
+
+public isolated distinct service class AssociateService {
+    private final string name;
+    private final string status;
+
+    public isolated function init(string name, string status) {
+        self.name = name;
+        self.status = status;
+    }
+
+    isolated resource function get name() returns string {
+        return self.name;
+    }
+
+    isolated resource function get status() returns string {
+        return self.status;
+    }
+}
+
+public isolated distinct service class EnemyService {
+    private final string name;
+    private final int age;
+    private final boolean isMarried;
+
+    public isolated function init(string name, int age, boolean isMarried) {
+        self.name = name;
+        self.age = age;
+        self.isMarried = isMarried;
+    }
+
+    isolated resource function get name() returns string {
+        return self.name;
+    }
+
+    isolated resource function get age() returns int {
+        return self.age;
+    }
+
+    isolated resource function get isMarried() returns boolean {
+        return self.isMarried;
+    }
+}
+
+public isolated distinct service class AuthorData2 {
+    private final readonly & AuthorRow author;
+
+    isolated function init(AuthorRow author) {
+        self.author = author.cloneReadOnly();
+    }
+
+    isolated resource function get name() returns string {
+        return self.author.name;
+    }
+
+    isolated function preBooks(graphql:Context ctx) {
+        dataloader:DataLoader bookLoader = ctx.getDataLoader(BOOK_LOADER_2);
+        bookLoader.add(self.author.id);
+    }
+
+    isolated resource function get books(graphql:Context ctx) returns BookData[]|error {
+        dataloader:DataLoader bookLoader = ctx.getDataLoader(BOOK_LOADER_2);
+        BookRow[] bookrows = check bookLoader.get(self.author.id);
+        return from BookRow bookRow in bookrows
+            select new BookData(bookRow);
+    }
+}

ballerina-tests/tests/records.bal

@@ -377,3 +377,33 @@ type InputObject2 record {|
 type InputObject3 record {|
     int age = 30;
 |};
+
+type Friend record {|
+    readonly string name;
+    int age;
+    boolean isMarried;
+|};
+
+type Enemy record {|
+    readonly string name;
+    int age;
+    boolean isMarried;
+|};
+
+type EnemyInput record {|
+    readonly string name = "Enemy6";
+    int age = 12;
+|};
+
+type Associate record {|
+    readonly string name;
+    string status;
+|};
+
+public type Relationship FriendService|AssociateService;
+
+type User record {|
+    int id?;
+    string name?;
+    int age?;
+|};

ballerina-tests/tests/resources/documents/server_cache.graphql

@@ -0,0 +1,8 @@
+query A {
+    greet
+    name(id:1)
+}
+
+mutation B {
+    updateName(name: "John", enableEvict: false)
+}

ballerina-tests/tests/resources/documents/server_cache_eviction.graphql

@@ -0,0 +1,12 @@
+query A {
+    greet
+    name(id:1)
+}
+
+mutation B {
+    updateName(name: "John", enableEvict: true)
+}
+
+mutation C {
+    updateName(name: "Walter White", enableEvict: false)
+}

ballerina-tests/tests/resources/documents/server_cache_eviction_with_arrays.graphql

@@ -0,0 +1,13 @@
+query A {
+    getFriendServices {
+        name
+        age
+    }
+}
+
+mutation B {
+    updateFriend(name: "Tyler", age: 28, isMarried: true, enableEvict: true) {
+        name
+        age
+    }
+}

ballerina-tests/tests/resources/documents/server_cache_eviction_with_dataloader.graphql

@@ -0,0 +1,26 @@
+query  A {
+  authors(ids: [1, 2, 3]) {
+    name
+    books {
+      id
+      title
+    }
+  }
+}
+
+mutation B {
+  sabthar: updateAuthorName(id: 1, name: "Sabthar", enableEvict: true) {
+    name
+    books {
+      id
+      title
+    }
+  }
+  mahroof: updateAuthorName(id: 2, name: "Mahroof") {
+    name
+    books {
+      id
+      title
+    }
+  }
+}

ballerina-tests/tests/resources/documents/server_cache_eviction_with_dataloader_operational.graphql

@@ -0,0 +1,26 @@
+query  A {
+  authors(ids: [1, 2, 3]) {
+    name
+    books {
+      id
+      title
+    }
+  }
+}
+
+mutation B {
+  harari: updateAuthorName(id: 1, name: "Yual Noah Harari", enableEvict: true) {
+    name
+    books {
+      id
+      title
+    }
+  }
+  manson: updateAuthorName(id: 2, name: "Mark Manson") {
+    name
+    books {
+      id
+      title
+    }
+  }
+}

ballerina-tests/tests/resources/documents/server_cache_eviction_with_interceptors.graphql

@@ -0,0 +1,7 @@
+query A {
+    enemy
+}
+
+mutation B {
+    updateEnemy(name: "Snape", enableEvict: true)
+}

ballerina-tests/tests/resources/documents/server_cache_eviction_with_list_inputs.graphql

@@ -0,0 +1,10 @@
+query A ($names:[String!]!){
+    isAllMarried(names: $names)
+}
+
+mutation B {
+    updateEnemy(enemy: {name: "Enemy3", age: 44}, enableEvict: true) {
+        name
+        isMarried
+    }
+}

ballerina-tests/tests/resources/documents/server_cache_eviction_with_nullable_inputs.graphql

@@ -0,0 +1,13 @@
+query A {
+    enemies(isMarried: null) {
+        name
+        isMarried
+    }
+}
+
+mutation B {
+    removeEnemy(name: "Enemy2", enableEvict: false) {
+        name
+        isMarried
+    }
+}

ballerina-tests/tests/resources/documents/server_cache_eviction_with_service_obj.graphql

@@ -0,0 +1,13 @@
+query A {
+    getFriendService(name: "Jesse Pinkman") {
+        name
+        age
+    }
+}
+
+mutation B {
+    updateFriend(name: "Jesse Pinkman", age: 30, isMarried: false, enableEvict: true) {
+        name
+        age
+    }
+}