@aws-amplify/datastore 3.12.6-next.20 → 3.12.6-next.36

package/lib/sync/processors/mutation.js

@@ -1,7 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 var tslib_1 = require("tslib");
-var api_1 = tslib_1.__importDefault(require("@aws-amplify/api"));
+var api_1 = require("@aws-amplify/api");
 var core_1 = require("@aws-amplify/core");
 var zen_observable_ts_1 = tslib_1.__importDefault(require("zen-observable-ts"));
 var types_1 = require("../../types");
@@ -26,7 +26,8 @@ var MutationProcessor = /** @class */ (function () {
         this.amplifyContext = amplifyContext;
         this.typeQuery = new WeakMap();
         this.processing = false;
-        this.amplifyContext.API = this.amplifyContext.API || api_1.default;
+        this.runningProcesses = new core_1.BackgroundProcessManager();
+        this.amplifyContext.API = this.amplifyContext.API || api_1.API;
         this.generateQueries();
     }
     MutationProcessor.prototype.generateQueries = function () {
@@ -56,164 +57,203 @@ var MutationProcessor = /** @class */ (function () {
         var _this = this;
         var observable = new zen_observable_ts_1.default(function (observer) {
             _this.observer = observer;
-            _this.resume();
-            return function () {
-                _this.pause();
-            };
+            try {
+                _this.resume();
+            }
+            catch (error) {
+                logger.error('mutations processor start error', error);
+                throw error;
+            }
+            return _this.runningProcesses.addCleaner(function () { return tslib_1.__awaiter(_this, void 0, void 0, function () {
+                return tslib_1.__generator(this, function (_a) {
+                    this.pause();
+                    return [2 /*return*/];
+                });
+            }); });
         });
         return observable;
     };
+    MutationProcessor.prototype.stop = function () {
+        return tslib_1.__awaiter(this, void 0, void 0, function () {
+            return tslib_1.__generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0: return [4 /*yield*/, this.runningProcesses.close()];
+                    case 1:
+                        _a.sent();
+                        return [4 /*yield*/, this.runningProcesses.open()];
+                    case 2:
+                        _a.sent();
+                        return [2 /*return*/];
+                }
+            });
+        });
+    };
     MutationProcessor.prototype.resume = function () {
         return tslib_1.__awaiter(this, void 0, void 0, function () {
-            var head, namespaceName, _loop_1, this_1, _a;
             var _this = this;
-            return tslib_1.__generator(this, function (_b) {
-                switch (_b.label) {
-                    case 0:
-                        if (this.processing || !this.isReady()) {
-                            return [2 /*return*/];
-                        }
-                        this.processing = true;
-                        namespaceName = util_1.USER;
-                        _loop_1 = function () {
-                            var model, operation, data, condition, modelConstructor, result, opName, modelDefinition, modelAuthModes, operationAuthModes_1, authModeAttempts_1, authModeRetry_1, error_1, record, hasMore;
-                            var _a;
-                            return tslib_1.__generator(this, function (_b) {
-                                switch (_b.label) {
-                                    case 0:
-                                        model = head.model, operation = head.operation, data = head.data, condition = head.condition;
-                                        modelConstructor = this_1.userClasses[model];
-                                        result = void 0;
-                                        opName = void 0;
-                                        modelDefinition = void 0;
-                                        _b.label = 1;
-                                    case 1:
-                                        _b.trys.push([1, 4, , 5]);
-                                        return [4 /*yield*/, utils_1.getModelAuthModes({
-                                                authModeStrategy: this_1.authModeStrategy,
-                                                defaultAuthMode: this_1.amplifyConfig.aws_appsync_authenticationType,
-                                                modelName: model,
-                                                schema: this_1.schema,
-                                            })];
-                                    case 2:
-                                        modelAuthModes = _b.sent();
-                                        operationAuthModes_1 = modelAuthModes[operation.toUpperCase()];
-                                        authModeAttempts_1 = 0;
-                                        authModeRetry_1 = function () { return tslib_1.__awaiter(_this, void 0, void 0, function () {
-                                            var response, error_2;
-                                            return tslib_1.__generator(this, function (_a) {
-                                                switch (_a.label) {
-                                                    case 0:
-                                                        _a.trys.push([0, 2, , 4]);
-                                                        logger.debug("Attempting mutation with authMode: " + operationAuthModes_1[authModeAttempts_1]);
-                                                        return [4 /*yield*/, this.jitteredRetry(namespaceName, model, operation, data, condition, modelConstructor, this.MutationEvent, head, operationAuthModes_1[authModeAttempts_1])];
-                                                    case 1:
-                                                        response = _a.sent();
-                                                        logger.debug("Mutation sent successfully with authMode: " + operationAuthModes_1[authModeAttempts_1]);
-                                                        return [2 /*return*/, response];
-                                                    case 2:
-                                                        error_2 = _a.sent();
-                                                        authModeAttempts_1++;
-                                                        if (authModeAttempts_1 >= operationAuthModes_1.length) {
-                                                            logger.debug("Mutation failed with authMode: " + operationAuthModes_1[authModeAttempts_1 - 1]);
-                                                            throw error_2;
-                                                        }
-                                                        logger.debug("Mutation failed with authMode: " + operationAuthModes_1[authModeAttempts_1 - 1] + ". Retrying with authMode: " + operationAuthModes_1[authModeAttempts_1]);
-                                                        return [4 /*yield*/, authModeRetry_1()];
-                                                    case 3: return [2 /*return*/, _a.sent()];
-                                                    case 4: return [2 /*return*/];
-                                                }
-                                            });
-                                        }); };
-                                        return [4 /*yield*/, authModeRetry_1()];
-                                    case 3:
-                                        _a = tslib_1.__read.apply(void 0, [_b.sent(), 3]), result = _a[0], opName = _a[1], modelDefinition = _a[2];
-                                        return [3 /*break*/, 5];
-                                    case 4:
-                                        error_1 = _b.sent();
-                                        if (error_1.message === 'Offline' || error_1.message === 'RetryMutation') {
-                                            return [2 /*return*/, "continue"];
-                                        }
-                                        return [3 /*break*/, 5];
-                                    case 5:
-                                        if (!(result === undefined)) return [3 /*break*/, 7];
-                                        logger.debug('done retrying');
-                                        return [4 /*yield*/, this_1.storage.runExclusive(function (storage) { return tslib_1.__awaiter(_this, void 0, void 0, function () {
-                                                return tslib_1.__generator(this, function (_a) {
-                                                    switch (_a.label) {
-                                                        case 0: return [4 /*yield*/, this.outbox.dequeue(storage)];
-                                                        case 1:
-                                                            _a.sent();
-                                                            return [2 /*return*/];
-                                                    }
-                                                });
-                                            }); })];
-                                    case 6:
-                                        _b.sent();
-                                        return [2 /*return*/, "continue"];
-                                    case 7:
-                                        record = result.data[opName];
-                                        hasMore = false;
-                                        return [4 /*yield*/, this_1.storage.runExclusive(function (storage) { return tslib_1.__awaiter(_this, void 0, void 0, function () {
-                                                return tslib_1.__generator(this, function (_a) {
-                                                    switch (_a.label) {
-                                                        case 0: 
-                                                        // using runExclusive to prevent possible race condition
-                                                        // when another record gets enqueued between dequeue and peek
-                                                        return [4 /*yield*/, this.outbox.dequeue(storage, record, operation)];
+            return tslib_1.__generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0: return [4 /*yield*/, (this.runningProcesses.isOpen &&
+                            this.runningProcesses.add(function (onTerminate) { return tslib_1.__awaiter(_this, void 0, void 0, function () {
+                                var head, namespaceName, _loop_1, this_1, _a;
+                                var _this = this;
+                                return tslib_1.__generator(this, function (_b) {
+                                    switch (_b.label) {
+                                        case 0:
+                                            if (this.processing ||
+                                                !this.isReady() ||
+                                                !this.runningProcesses.isOpen) {
+                                                return [2 /*return*/];
+                                            }
+                                            this.processing = true;
+                                            namespaceName = util_1.USER;
+                                            _loop_1 = function () {
+                                                var model, operation, data, condition, modelConstructor, result, opName, modelDefinition, modelAuthModes, operationAuthModes_1, authModeAttempts_1, authModeRetry_1, error_1, record, hasMore;
+                                                var _a;
+                                                return tslib_1.__generator(this, function (_b) {
+                                                    switch (_b.label) {
+                                                        case 0:
+                                                            model = head.model, operation = head.operation, data = head.data, condition = head.condition;
+                                                            modelConstructor = this_1.userClasses[model];
+                                                            result = void 0;
+                                                            opName = void 0;
+                                                            modelDefinition = void 0;
+                                                            _b.label = 1;
                                                         case 1:
-                                                            // using runExclusive to prevent possible race condition
-                                                            // when another record gets enqueued between dequeue and peek
-                                                            _a.sent();
-                                                            return [4 /*yield*/, this.outbox.peek(storage)];
+                                                            _b.trys.push([1, 4, , 5]);
+                                                            return [4 /*yield*/, utils_1.getModelAuthModes({
+                                                                    authModeStrategy: this_1.authModeStrategy,
+                                                                    defaultAuthMode: this_1.amplifyConfig.aws_appsync_authenticationType,
+                                                                    modelName: model,
+                                                                    schema: this_1.schema,
+                                                                })];
                                                         case 2:
-                                                            hasMore = (_a.sent()) !== undefined;
+                                                            modelAuthModes = _b.sent();
+                                                            operationAuthModes_1 = modelAuthModes[operation.toUpperCase()];
+                                                            authModeAttempts_1 = 0;
+                                                            authModeRetry_1 = function () { return tslib_1.__awaiter(_this, void 0, void 0, function () {
+                                                                var response, error_2;
+                                                                return tslib_1.__generator(this, function (_a) {
+                                                                    switch (_a.label) {
+                                                                        case 0:
+                                                                            _a.trys.push([0, 2, , 4]);
+                                                                            logger.debug("Attempting mutation with authMode: " + operationAuthModes_1[authModeAttempts_1]);
+                                                                            return [4 /*yield*/, this.jitteredRetry(namespaceName, model, operation, data, condition, modelConstructor, this.MutationEvent, head, operationAuthModes_1[authModeAttempts_1], onTerminate)];
+                                                                        case 1:
+                                                                            response = _a.sent();
+                                                                            logger.debug("Mutation sent successfully with authMode: " + operationAuthModes_1[authModeAttempts_1]);
+                                                                            return [2 /*return*/, response];
+                                                                        case 2:
+                                                                            error_2 = _a.sent();
+                                                                            authModeAttempts_1++;
+                                                                            if (authModeAttempts_1 >= operationAuthModes_1.length) {
+                                                                                logger.debug("Mutation failed with authMode: " + operationAuthModes_1[authModeAttempts_1 - 1]);
+                                                                                throw error_2;
+                                                                            }
+                                                                            logger.debug("Mutation failed with authMode: " + operationAuthModes_1[authModeAttempts_1 - 1] + ". Retrying with authMode: " + operationAuthModes_1[authModeAttempts_1]);
+                                                                            return [4 /*yield*/, authModeRetry_1()];
+                                                                        case 3: return [2 /*return*/, _a.sent()];
+                                                                        case 4: return [2 /*return*/];
+                                                                    }
+                                                                });
+                                                            }); };
+                                                            return [4 /*yield*/, authModeRetry_1()];
+                                                        case 3:
+                                                            _a = tslib_1.__read.apply(void 0, [_b.sent(), 3]), result = _a[0], opName = _a[1], modelDefinition = _a[2];
+                                                            return [3 /*break*/, 5];
+                                                        case 4:
+                                                            error_1 = _b.sent();
+                                                            if (error_1.message === 'Offline' ||
+                                                                error_1.message === 'RetryMutation') {
+                                                                return [2 /*return*/, "continue"];
+                                                            }
+                                                            return [3 /*break*/, 5];
+                                                        case 5:
+                                                            if (!(result === undefined)) return [3 /*break*/, 7];
+                                                            logger.debug('done retrying');
+                                                            return [4 /*yield*/, this_1.storage.runExclusive(function (storage) { return tslib_1.__awaiter(_this, void 0, void 0, function () {
+                                                                    return tslib_1.__generator(this, function (_a) {
+                                                                        switch (_a.label) {
+                                                                            case 0: return [4 /*yield*/, this.outbox.dequeue(storage)];
+                                                                            case 1:
+                                                                                _a.sent();
+                                                                                return [2 /*return*/];
+                                                                        }
+                                                                    });
+                                                                }); })];
+                                                        case 6:
+                                                            _b.sent();
+                                                            return [2 /*return*/, "continue"];
+                                                        case 7:
+                                                            record = result.data[opName];
+                                                            hasMore = false;
+                                                            return [4 /*yield*/, this_1.storage.runExclusive(function (storage) { return tslib_1.__awaiter(_this, void 0, void 0, function () {
+                                                                    return tslib_1.__generator(this, function (_a) {
+                                                                        switch (_a.label) {
+                                                                            case 0: 
+                                                                            // using runExclusive to prevent possible race condition
+                                                                            // when another record gets enqueued between dequeue and peek
+                                                                            return [4 /*yield*/, this.outbox.dequeue(storage, record, operation)];
+                                                                            case 1:
+                                                                                // using runExclusive to prevent possible race condition
+                                                                                // when another record gets enqueued between dequeue and peek
+                                                                                _a.sent();
+                                                                                return [4 /*yield*/, this.outbox.peek(storage)];
+                                                                            case 2:
+                                                                                hasMore = (_a.sent()) !== undefined;
+                                                                                return [2 /*return*/];
+                                                                        }
+                                                                    });
+                                                                }); })];
+                                                        case 8:
+                                                            _b.sent();
+                                                            this_1.observer.next({
+                                                                operation: operation,
+                                                                modelDefinition: modelDefinition,
+                                                                model: record,
+                                                                hasMore: hasMore,
+                                                            });
                                                             return [2 /*return*/];
                                                     }
                                                 });
-                                            }); })];
-                                    case 8:
-                                        _b.sent();
-                                        this_1.observer.next({
-                                            operation: operation,
-                                            modelDefinition: modelDefinition,
-                                            model: record,
-                                            hasMore: hasMore,
-                                        });
-                                        return [2 /*return*/];
-                                }
-                            });
-                        };
-                        this_1 = this;
-                        _b.label = 1;
+                                            };
+                                            this_1 = this;
+                                            _b.label = 1;
+                                        case 1:
+                                            _a = this.processing &&
+                                                this.runningProcesses.isOpen;
+                                            if (!_a) return [3 /*break*/, 3];
+                                            return [4 /*yield*/, this.outbox.peek(this.storage)];
+                                        case 2:
+                                            _a = (head = _b.sent()) !== undefined;
+                                            _b.label = 3;
+                                        case 3:
+                                            if (!_a) return [3 /*break*/, 5];
+                                            return [5 /*yield**/, _loop_1()];
+                                        case 4:
+                                            _b.sent();
+                                            return [3 /*break*/, 1];
+                                        case 5:
+                                            // pauses itself
+                                            this.pause();
+                                            return [2 /*return*/];
+                                    }
+                                });
+                            }); }, 'mutation resume loop'))];
                     case 1:
-                        _a = this.processing;
-                        if (!_a) return [3 /*break*/, 3];
-                        return [4 /*yield*/, this.outbox.peek(this.storage)];
-                    case 2:
-                        _a = (head = _b.sent()) !== undefined;
-                        _b.label = 3;
-                    case 3:
-                        if (!_a) return [3 /*break*/, 5];
-                        return [5 /*yield**/, _loop_1()];
-                    case 4:
-                        _b.sent();
-                        return [3 /*break*/, 1];
-                    case 5:
-                        // pauses itself
-                        this.pause();
+                        _a.sent();
                         return [2 /*return*/];
                 }
             });
         });
     };
-    MutationProcessor.prototype.jitteredRetry = function (namespaceName, model, operation, data, condition, modelConstructor, MutationEvent, mutationEvent, authMode) {
+    MutationProcessor.prototype.jitteredRetry = function (namespaceName, model, operation, data, condition, modelConstructor, MutationEvent, mutationEvent, authMode, onTerminate) {
         return tslib_1.__awaiter(this, void 0, void 0, function () {
             var _this = this;
             return tslib_1.__generator(this, function (_a) {
                 switch (_a.label) {
                     case 0: return [4 /*yield*/, core_1.retry(function (model, operation, data, condition, modelConstructor, MutationEvent, mutationEvent) { return tslib_1.__awaiter(_this, void 0, void 0, function () {
-                            var _a, query, variables, graphQLCondition, opName, modelDefinition, authToken, tryWith, attempt, opType, result, err_1, _b, error, _c, _d, code, retryWith, err_2, _e, _f, opName_1, query_1, authToken_1, serverData, namespace, updatedMutation, err_3;
+                            var _a, query, variables, graphQLCondition, opName, modelDefinition, authToken, tryWith, attempt, opType, result, err_1, _b, error, _c, _d, code, retryWith, err_2, _e, _f, opName_1, query_1, authToken_1, serverData, namespace, updatedMutation;
                             var _g;
                             return tslib_1.__generator(this, function (_h) {
                                 switch (_h.label) {
@@ -233,14 +273,16 @@ var MutationProcessor = /** @class */ (function () {
                                         opType = this.opTypeFromTransformerOperation(operation);
                                         _h.label = 2;
                                     case 2:
-                                        _h.trys.push([2, 4, , 20]);
+                                        _h.trys.push([2, 4, , 17]);
                                         return [4 /*yield*/, this.amplifyContext.API.graphql(tryWith)];
                                     case 3:
                                         result = (_h.sent());
+                                        // `as any` because TypeScript doesn't seem to like passing tuples
+                                        // through generic params???
                                         return [2 /*return*/, [result, opName, modelDefinition]];
                                     case 4:
                                         err_1 = _h.sent();
-                                        if (!(err_1.errors && err_1.errors.length > 0)) return [3 /*break*/, 18];
+                                        if (!(err_1.errors && err_1.errors.length > 0)) return [3 /*break*/, 15];
                                         _b = tslib_1.__read(err_1.errors, 1), error = _b[0];
                                         _c = error.originalError, _d = (_c === void 0 ? {} : _c).code, code = _d === void 0 ? null : _d;
                                         if (error.errorType === 'Unauthorized') {
@@ -277,7 +319,7 @@ var MutationProcessor = /** @class */ (function () {
                                     case 7:
                                         err_2 = _h.sent();
                                         logger.warn('conflict trycatch', err_2);
-                                        return [3 /*break*/, 20];
+                                        return [3 /*break*/, 17];
                                     case 8:
                                         if (!(retryWith === types_1.DISCARD)) return [3 /*break*/, 11];
                                         _e = tslib_1.__read(utils_1.buildGraphQLOperation(this.schema.namespaces[namespaceName], modelDefinition, 'GET'), 1), _f = tslib_1.__read(_e[0], 3), opName_1 = _f[1], query_1 = _f[2];
@@ -293,6 +335,7 @@ var MutationProcessor = /** @class */ (function () {
                                             })];
                                     case 10:
                                         serverData = _h.sent();
+                                        // onTerminate cancel graphql()
                                         return [2 /*return*/, [serverData, opName_1, modelDefinition]];
                                     case 11:
                                         namespace = this.schema.namespaces[namespaceName];
@@ -302,8 +345,8 @@ var MutationProcessor = /** @class */ (function () {
                                         _h.sent();
                                         throw new core_1.NonRetryableError('RetryMutation');
                                     case 13:
-                                        _h.trys.push([13, 15, 16, 17]);
-                                        return [4 /*yield*/, this.errorHandler({
+                                        try {
+                                            this.errorHandler({
                                                 recoverySuggestion: 'Ensure app code is up to date, auth directives exist and are correct on each model, and that server-side data has not been invalidated by a schema change. If the problem persists, search for or create an issue: https://github.com/aws-amplify/amplify-js/issues',
                                                 localModel: variables.input,
                                                 message: error.message,
@@ -315,33 +358,32 @@ var MutationProcessor = /** @class */ (function () {
                                                 remoteModel: error.data
                                                     ? this.modelInstanceCreator(modelConstructor, error.data)
                                                     : null,
-                                            })];
-                                    case 14:
-                                        _h.sent();
-                                        return [3 /*break*/, 17];
-                                    case 15:
-                                        err_3 = _h.sent();
-                                        logger.warn('Mutation error handler failed with:', err_3);
-                                        return [3 /*break*/, 17];
-                                    case 16: 
-                                    // Return empty tuple, dequeues the mutation
-                                    return [2 /*return*/, error.data
-                                            ? [
-                                                { data: (_g = {}, _g[opName] = error.data, _g) },
-                                                opName,
-                                                modelDefinition,
-                                            ]
-                                            : []];
-                                    case 17: return [3 /*break*/, 19];
-                                    case 18: 
+                                            });
+                                        }
+                                        catch (err) {
+                                            logger.warn('Mutation error handler failed with:', err);
+                                        }
+                                        finally {
+                                            // Return empty tuple, dequeues the mutation
+                                            return [2 /*return*/, error.data
+                                                    ? [
+                                                        { data: (_g = {}, _g[opName] = error.data, _g) },
+                                                        opName,
+                                                        modelDefinition,
+                                                    ]
+                                                    : []];
+                                        }
+                                        _h.label = 14;
+                                    case 14: return [3 /*break*/, 16];
+                                    case 15: 
                                     // Catch-all for client-side errors that don't come back in the `GraphQLError` format.
                                     // These errors should not be retried.
                                     throw new core_1.NonRetryableError(err_1);
-                                    case 19: return [3 /*break*/, 20];
-                                    case 20:
+                                    case 16: return [3 /*break*/, 17];
+                                    case 17:
                                         if (tryWith) return [3 /*break*/, 2];
-                                        _h.label = 21;
-                                    case 21: return [2 /*return*/];
+                                        _h.label = 18;
+                                    case 18: return [2 /*return*/];
                                 }
                             });
                         }); }, [
@@ -352,25 +394,25 @@ var MutationProcessor = /** @class */ (function () {
                             modelConstructor,
                             MutationEvent,
                             mutationEvent,
-                        ], exports.safeJitteredBackoff)];
+                        ], exports.safeJitteredBackoff, onTerminate)];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
             });
         });
     };
     MutationProcessor.prototype.createQueryVariables = function (namespaceName, model, operation, data, condition) {
-        var e_1, _a;
+        var e_1, _a, e_2, _b, e_3, _c;
         var modelDefinition = this.schema.namespaces[namespaceName].models[model];
         var primaryKey = this.schema.namespaces[namespaceName].keys[model].primaryKey;
         var queriesTuples = this.typeQuery.get(modelDefinition);
-        var _b = tslib_1.__read(queriesTuples.find(function (_a) {
+        var _d = tslib_1.__read(queriesTuples.find(function (_a) {
             var _b = tslib_1.__read(_a, 1), transformerMutationType = _b[0];
             return transformerMutationType === operation;
-        }), 3), opName = _b[1], query = _b[2];
-        var _c = JSON.parse(data), _version = _c._version, parsedData = tslib_1.__rest(_c, ["_version"]);
+        }), 3), opName = _d[1], query = _d[2];
+        var _e = JSON.parse(data), _version = _e._version, parsedData = tslib_1.__rest(_e, ["_version"]);
         // include all the fields that comprise a custom PK if one is specified
         var deleteInput = {};
-        if (primaryKey && primaryKey.length) {
+        if (primaryKey === null || primaryKey === void 0 ? void 0 : primaryKey.length) {
             try {
                 for (var primaryKey_1 = tslib_1.__values(primaryKey), primaryKey_1_1 = primaryKey_1.next(); !primaryKey_1_1.done; primaryKey_1_1 = primaryKey_1.next()) {
                     var pkField = primaryKey_1_1.value;
@@ -386,48 +428,66 @@ var MutationProcessor = /** @class */ (function () {
             }
         }
         else {
-            deleteInput['id'] = parsedData.id;
+            deleteInput[util_1.ID] = parsedData.id;
         }
-        var filteredData = operation === utils_1.TransformerMutationType.DELETE
-            ? deleteInput // For DELETE mutations, only PK is sent
-            : Object.values(modelDefinition.fields)
-                .filter(function (_a) {
-                var name = _a.name, type = _a.type, association = _a.association;
-                // connections
-                if (types_1.isModelFieldType(type)) {
-                    // BELONGS_TO
-                    if (types_1.isTargetNameAssociation(association) &&
-                        association.connectionType === 'BELONGS_TO') {
-                        return true;
+        var mutationInput;
+        if (operation === utils_1.TransformerMutationType.DELETE) {
+            // For DELETE mutations, only the key(s) are included in the input
+            mutationInput = deleteInput;
+        }
+        else {
+            // Otherwise, we construct the mutation input with the following logic
+            mutationInput = {};
+            var modelFields = Object.values(modelDefinition.fields);
+            try {
+                for (var modelFields_1 = tslib_1.__values(modelFields), modelFields_1_1 = modelFields_1.next(); !modelFields_1_1.done; modelFields_1_1 = modelFields_1.next()) {
+                    var _f = modelFields_1_1.value, name_1 = _f.name, type = _f.type, association = _f.association;
+                    // model fields should be stripped out from the input
+                    if (types_1.isModelFieldType(type)) {
+                        // except for belongs to relations - we need to replace them with the correct foreign key(s)
+                        if (types_1.isTargetNameAssociation(association) &&
+                            association.connectionType === 'BELONGS_TO') {
+                            var targetNames = util_1.extractTargetNamesFromSrc(association);
+                            if (targetNames) {
+                                try {
+                                    // instead of including the connected model itself, we add its key(s) to the mutation input
+                                    for (var targetNames_1 = (e_3 = void 0, tslib_1.__values(targetNames)), targetNames_1_1 = targetNames_1.next(); !targetNames_1_1.done; targetNames_1_1 = targetNames_1.next()) {
+                                        var targetName = targetNames_1_1.value;
+                                        mutationInput[targetName] = parsedData[targetName];
+                                    }
+                                }
+                                catch (e_3_1) { e_3 = { error: e_3_1 }; }
+                                finally {
+                                    try {
+                                        if (targetNames_1_1 && !targetNames_1_1.done && (_c = targetNames_1.return)) _c.call(targetNames_1);
+                                    }
+                                    finally { if (e_3) throw e_3.error; }
+                                }
+                            }
+                        }
+                        continue;
                     }
-                    // All other connections
-                    return false;
-                }
-                if (operation === utils_1.TransformerMutationType.UPDATE) {
-                    // this limits the update mutation input to changed fields only
-                    return parsedData.hasOwnProperty(name);
+                    // scalar fields / non-model types
+                    if (operation === utils_1.TransformerMutationType.UPDATE) {
+                        if (!parsedData.hasOwnProperty(name_1)) {
+                            // for update mutations - strip out a field if it's unchanged
+                            continue;
+                        }
+                    }
+                    // all other fields are added to the input object
+                    mutationInput[name_1] = parsedData[name_1];
                 }
-                // scalars and non-model types
-                return true;
-            })
-                .map(function (_a) {
-                var name = _a.name, type = _a.type, association = _a.association;
-                var fieldName = name;
-                var val = parsedData[name];
-                if (types_1.isModelFieldType(type) &&
-                    types_1.isTargetNameAssociation(association)) {
-                    fieldName = association.targetName;
-                    val = parsedData[fieldName];
+            }
+            catch (e_2_1) { e_2 = { error: e_2_1 }; }
+            finally {
+                try {
+                    if (modelFields_1_1 && !modelFields_1_1.done && (_b = modelFields_1.return)) _b.call(modelFields_1);
                 }
-                return [fieldName, val];
-            })
-                .reduce(function (acc, _a) {
-                var _b = tslib_1.__read(_a, 2), k = _b[0], v = _b[1];
-                acc[k] = v;
-                return acc;
-            }, {});
+                finally { if (e_2) throw e_2.error; }
+            }
+        }
         // Build mutation variables input object
-        var input = tslib_1.__assign(tslib_1.__assign({}, filteredData), { _version: _version });
+        var input = tslib_1.__assign(tslib_1.__assign({}, mutationInput), { _version: _version });
         var graphQLCondition = JSON.parse(condition);
         var variables = tslib_1.__assign({ input: input }, (operation === utils_1.TransformerMutationType.CREATE
             ? {}