Added ability to directly update the Apollo cache using write methods

Cartfile

@@ -1,2 +1,2 @@
 github "stephencelis/SQLite.swift" ~> 0.11.5
-github "daltoniam/Starscream" ~> 3.0.5
+github "daltoniam/Starscream" ~> 3.0.6

Cartfile.resolved

@@ -1,2 +1,2 @@
-github "daltoniam/Starscream" "3.0.5"
+github "daltoniam/Starscream" "3.0.6"
 github "stephencelis/SQLite.swift" "0.11.5"

Sources/Apollo/ApolloClient.swift

@@ -28,8 +28,9 @@ public typealias OperationResultHandler<Operation: GraphQLOperation> = (_ result
 
 /// The `ApolloClient` class provides the core API for Apollo. This API provides methods to fetch and watch queries, and to perform mutations.
 public class ApolloClient {
-  let networkTransport: NetworkTransport
-  let store: ApolloStore
+  public let networkTransport: NetworkTransport
+  public let store: ApolloStore
+
   public var cacheKeyForObject: CacheKeyForObject? {
     get {
       return store.cacheKeyForObject

Sources/Apollo/ApolloStore.swift

@@ -118,7 +118,7 @@ public final class ApolloStore {
     }
   }
 
-  func load<Query: GraphQLQuery>(query: Query) -> Promise<GraphQLResult<Query.Data>> {
+  public func load<Query: GraphQLQuery>(query: Query) -> Promise<GraphQLResult<Query.Data>> {
     return withinReadTransaction { transaction in
       let mapper = GraphQLSelectionSetMapper<Query.Data>()
       let dependencyTracker = GraphQLDependencyTracker()
@@ -129,7 +129,7 @@ public final class ApolloStore {
     }
   }
 
-  func load<Query: GraphQLQuery>(query: Query, resultHandler: @escaping OperationResultHandler<Query>) {
+  public func load<Query: GraphQLQuery>(query: Query, resultHandler: @escaping OperationResultHandler<Query>) {
     load(query: query).andThen { result in
       resultHandler(result, nil)
     }.catch { error in
@@ -143,9 +143,10 @@ public final class ApolloStore {
 
     fileprivate lazy var loader: DataLoader<CacheKey, Record?> = DataLoader(self.cache.loadRecords)
 
-    fileprivate func makeExecutor() -> GraphQLExecutor {
+    fileprivate func makeExecutor(keySelector: @escaping (GraphQLResolveInfo) -> String = { $0.cacheKeyForField }) -> GraphQLExecutor {
       let executor = GraphQLExecutor { object, info in
-        let value = object[info.cacheKeyForField]
+        let key = keySelector(info)
+        let value = object[key]
         return self.complete(value: value)
       }
 
@@ -206,8 +207,8 @@ public final class ApolloStore {
     fileprivate var updateChangedKeysFunc: DidChangeKeysFunc?
 
     init(cache: NormalizedCache, cacheKeyForObject: CacheKeyForObject?, updateChangedKeysFunc: @escaping DidChangeKeysFunc) {
-        self.updateChangedKeysFunc = updateChangedKeysFunc
-        super.init(cache: cache, cacheKeyForObject: cacheKeyForObject)
+      self.updateChangedKeysFunc = updateChangedKeysFunc
+      super.init(cache: cache, cacheKeyForObject: cacheKeyForObject)
     }
 
     public func update<Query: GraphQLQuery>(query: Query, _ body: (inout Query.Data) throws -> Void) throws {
@@ -232,12 +233,13 @@ public final class ApolloStore {
 
     private func write(object: JSONObject, forSelections selections: [GraphQLSelection], withKey key: CacheKey, variables: GraphQLMap?) throws {
       let normalizer = GraphQLResultNormalizer()
-      try self.makeExecutor().execute(selections: selections, on: object, withKey: key, variables: variables, accumulator: normalizer)
+      let executor = makeExecutor { $0.responseKeyForField }
+      try executor.execute(selections: selections, on: object, withKey: key, variables: variables, accumulator: normalizer)
       .flatMap {
         self.cache.merge(records: $0)
       }.andThen { changedKeys in
         if let didChangeKeysFunc = self.updateChangedKeysFunc {
-            didChangeKeysFunc(changedKeys, nil)
+          didChangeKeysFunc(changedKeys, nil)
         }
       }.wait()
     }

Tests/ApolloCacheDependentTests/ReadWriteFromStoreTests.swift

@@ -43,7 +43,7 @@ class ReadWriteFromStoreTests: XCTestCase {
       })
     }
   }
-  
+
   func testReadHeroNameQueryWithMissingName() throws {
     let initialRecords: RecordSet = [
       "QUERY_ROOT": ["hero": Reference(key: "hero")],
@@ -160,7 +160,44 @@ class ReadWriteFromStoreTests: XCTestCase {
       XCTAssertEqual(friendsNames, ["Luke Skywalker", "Han Solo", "Leia Organa", "C-3PO"])
     }
   }
-
+
+  func testUpdateHeroAndFriendsNamesQueryWithVariable() throws {
+    let initialRecords: RecordSet = [
+      "QUERY_ROOT": ["hero(episode:NEWHOPE)": Reference(key: "2001")],
+      "2001": [
+        "name": "R2-D2",
+        "__typename": "Droid",
+        "friends": [
+          Reference(key: "1000"),
+          Reference(key: "1002"),
+          Reference(key: "1003")
+        ]
+      ],
+      "1000": ["__typename": "Human", "name": "Luke Skywalker"],
+      "1002": ["__typename": "Human", "name": "Han Solo"],
+      "1003": ["__typename": "Human", "name": "Leia Organa"],
+      ]
+
+    try withCache(initialRecords: initialRecords) { (cache) in
+      let store = ApolloStore(cache: cache)
+
+      let query = HeroAndFriendsNamesQuery(episode: Episode.newhope)
+
+      try await(store.withinReadWriteTransaction { transaction in
+        try transaction.update(query: query) { (data: inout HeroAndFriendsNamesQuery.Data) in
+          data.hero?.friends?.append(.makeDroid(name: "C-3PO"))
+        }
+      })
+
+      let result = try await(store.load(query: query))
+      guard let data = result.data else { XCTFail(); return }
+
+      XCTAssertEqual(data.hero?.name, "R2-D2")
+      let friendsNames = data.hero?.friends?.compactMap { $0?.name }
+      XCTAssertEqual(friendsNames, ["Luke Skywalker", "Han Solo", "Leia Organa", "C-3PO"])
+    }
+  }
+
   func testReadHeroDetailsFragmentWithTypeSpecificProperty() throws {
     let initialRecords: RecordSet = [
       "2001": ["name": "R2-D2", "__typename": "Droid", "primaryFunction": "Protocol"]