@aws-amplify/datastore 3.12.6-next.13 → 3.12.6-next.32

package/lib/datastore/datastore.js

@@ -1,115 +1,25 @@
 "use strict";
-var __assign = (this && this.__assign) || function () {
-    __assign = Object.assign || function(t) {
-        for (var s, i = 1, n = arguments.length; i < n; i++) {
-            s = arguments[i];
-            for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p))
-                t[p] = s[p];
-        }
-        return t;
-    };
-    return __assign.apply(this, arguments);
-};
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
-var __generator = (this && this.__generator) || function (thisArg, body) {
-    var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g;
-    return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
-    function verb(n) { return function (v) { return step([n, v]); }; }
-    function step(op) {
-        if (f) throw new TypeError("Generator is already executing.");
-        while (_) try {
-            if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
-            if (y = 0, t) op = [op[0] & 2, t.value];
-            switch (op[0]) {
-                case 0: case 1: t = op; break;
-                case 4: _.label++; return { value: op[1], done: false };
-                case 5: _.label++; y = op[1]; op = [0]; continue;
-                case 7: op = _.ops.pop(); _.trys.pop(); continue;
-                default:
-                    if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
-                    if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
-                    if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
-                    if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
-                    if (t[2]) _.ops.pop();
-                    _.trys.pop(); continue;
-            }
-            op = body.call(thisArg, _);
-        } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
-        if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
-    }
-};
-var __rest = (this && this.__rest) || function (s, e) {
-    var t = {};
-    for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0)
-        t[p] = s[p];
-    if (s != null && typeof Object.getOwnPropertySymbols === "function")
-        for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
-            if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i]))
-                t[p[i]] = s[p[i]];
-        }
-    return t;
-};
-var __read = (this && this.__read) || function (o, n) {
-    var m = typeof Symbol === "function" && o[Symbol.iterator];
-    if (!m) return o;
-    var i = m.call(o), r, ar = [], e;
-    try {
-        while ((n === void 0 || n-- > 0) && !(r = i.next()).done) ar.push(r.value);
-    }
-    catch (error) { e = { error: error }; }
-    finally {
-        try {
-            if (r && !r.done && (m = i["return"])) m.call(i);
-        }
-        finally { if (e) throw e.error; }
-    }
-    return ar;
-};
-var __values = (this && this.__values) || function(o) {
-    var s = typeof Symbol === "function" && Symbol.iterator, m = s && o[s], i = 0;
-    if (m) return m.call(o);
-    if (o && typeof o.length === "number") return {
-        next: function () {
-            if (o && i >= o.length) o = void 0;
-            return { value: o && o[i++], done: !o };
-        }
-    };
-    throw new TypeError(s ? "Object is not iterable." : "Symbol.iterator is not defined.");
-};
-var __spread = (this && this.__spread) || function () {
-    for (var ar = [], i = 0; i < arguments.length; i++) ar = ar.concat(__read(arguments[i]));
-    return ar;
-};
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
-var api_1 = __importDefault(require("@aws-amplify/api"));
-var core_1 = require("@aws-amplify/core");
+var tslib_1 = require("tslib");
+var api_1 = require("@aws-amplify/api");
 var auth_1 = require("@aws-amplify/auth");
-var cache_1 = __importDefault(require("@aws-amplify/cache"));
+var cache_1 = require("@aws-amplify/cache");
+var core_1 = require("@aws-amplify/core");
 var immer_1 = require("immer");
 var uuid_1 = require("uuid");
-var zen_observable_ts_1 = __importDefault(require("zen-observable-ts"));
+var zen_observable_ts_1 = tslib_1.__importDefault(require("zen-observable-ts"));
 var authModeStrategies_1 = require("../authModeStrategies");
 var predicates_1 = require("../predicates");
 var storage_1 = require("../storage/storage");
 var sync_1 = require("../sync");
 var types_1 = require("../types");
 var util_1 = require("../util");
+var utils_1 = require("../sync/utils");
 immer_1.setAutoFreeze(true);
 immer_1.enablePatches();
 var logger = new core_1.ConsoleLogger('DataStore');
 var ulid = util_1.monotonicUlidFactory(Date.now());
-var isNode = core_1.JS.browserOrNode().isNode;
+var isNode = core_1.browserOrNode().isNode;
 var SETTING_SCHEMA_VERSION = 'schemaVersion';
 var schema;
 var modelNamespaceMap = new WeakMap();
@@ -137,7 +47,7 @@ var initSchema = function (userSchema) {
         return userClasses;
     }
     logger.log('validating schema', { schema: userSchema });
-    var internalUserNamespace = __assign({ name: util_1.USER }, userSchema);
+    var internalUserNamespace = tslib_1.__assign({ name: util_1.USER }, userSchema);
     logger.log('DataStore', 'Init models');
     userClasses = createTypeClasses(internalUserNamespace);
     logger.log('DataStore', 'Models initialized');
@@ -158,7 +68,7 @@ var initSchema = function (userSchema) {
     };
     Object.keys(schema.namespaces).forEach(function (namespace) {
         var e_1, _a;
-        var _b = __read(util_1.establishRelationAndKeys(schema.namespaces[namespace]), 2), relations = _b[0], keys = _b[1];
+        var _b = tslib_1.__read(util_1.establishRelationAndKeys(schema.namespaces[namespace]), 2), relations = _b[0], keys = _b[1];
         schema.namespaces[namespace].relationships = relations;
         schema.namespaces[namespace].keys = keys;
         var modelAssociations = new Map();
@@ -186,7 +96,7 @@ var initSchema = function (userSchema) {
                 throw new Error('Models are not topologically sortable. Please verify your schema.');
             }
             try {
-                for (var _c = (e_1 = void 0, __values(Array.from(modelAssociations.keys()))), _d = _c.next(); !_d.done; _d = _c.next()) {
+                for (var _c = (e_1 = void 0, tslib_1.__values(Array.from(modelAssociations.keys()))), _d = _c.next(); !_d.done; _d = _c.next()) {
                     var modelName = _d.value;
                     var parents = modelAssociations.get(modelName);
                     if (parents.every(function (x) { return result.has(x); })) {
@@ -208,10 +118,14 @@ var initSchema = function (userSchema) {
     return userClasses;
 };
 exports.initSchema = initSchema;
-/* Checks if the schema has been initialized by initSchema().
+/**
+ * Throws an exception if the schema has *not* been initialized
+ * by `initSchema()`.
  *
- * Call this function before accessing schema.
- * Currently this only needs to be called in start() and clear() because all other functions will call start first.
+ * **To be called before trying to access schema.**
+ *
+ * Currently this only needs to be called in `start()` and `clear()` because
+ * all other functions will call start first.
  */
 var checkSchemaInitialized = function () {
     if (schema === undefined) {
@@ -223,18 +137,23 @@ var checkSchemaInitialized = function () {
 var createTypeClasses = function (namespace) {
     var classes = {};
     Object.entries(namespace.models).forEach(function (_a) {
-        var _b = __read(_a, 2), modelName = _b[0], modelDefinition = _b[1];
+        var _b = tslib_1.__read(_a, 2), modelName = _b[0], modelDefinition = _b[1];
         var clazz = createModelClass(modelDefinition);
         classes[modelName] = clazz;
         modelNamespaceMap.set(clazz, namespace.name);
     });
     Object.entries(namespace.nonModels || {}).forEach(function (_a) {
-        var _b = __read(_a, 2), typeName = _b[0], typeDefinition = _b[1];
+        var _b = tslib_1.__read(_a, 2), typeName = _b[0], typeDefinition = _b[1];
         var clazz = createNonModelClass(typeDefinition);
         classes[typeName] = clazz;
     });
     return classes;
 };
+/**
+ * Collection of instantiated models to allow storage of metadata apart from
+ * the model visible to the consuming app -- in case the app doesn't have
+ * metadata fields (_version, _deleted, etc.) exposed on the model itself.
+ */
 var instancesMetadata = new WeakSet();
 function modelInstanceCreator(modelConstructor, init) {
     instancesMetadata.add(init);
@@ -243,15 +162,23 @@ function modelInstanceCreator(modelConstructor, init) {
 var validateModelFields = function (modelDefinition) { return function (k, v) {
     var fieldDefinition = modelDefinition.fields[k];
     if (fieldDefinition !== undefined) {
-        var type = fieldDefinition.type, isRequired_1 = fieldDefinition.isRequired, isArrayNullable = fieldDefinition.isArrayNullable, name_1 = fieldDefinition.name, isArray = fieldDefinition.isArray;
+        var type_1 = fieldDefinition.type, isRequired_1 = fieldDefinition.isRequired, isArrayNullable = fieldDefinition.isArrayNullable, name_1 = fieldDefinition.name, isArray = fieldDefinition.isArray;
         if (((!isArray && isRequired_1) || (isArray && !isArrayNullable)) &&
             (v === null || v === undefined)) {
             throw new Error("Field " + name_1 + " is required");
         }
-        if (types_1.isGraphQLScalarType(type)) {
-            var jsType_1 = types_1.GraphQLScalarType.getJSType(type);
-            var validateScalar_1 = types_1.GraphQLScalarType.getValidationFunction(type);
-            if (type === 'AWSJSON') {
+        if (types_1.isSchemaModelWithAttributes(modelDefinition) &&
+            !util_1.isIdManaged(modelDefinition)) {
+            var keys = util_1.extractPrimaryKeyFieldNames(modelDefinition);
+            if (keys.includes(k) && v === '') {
+                logger.error(util_1.errorMessages.idEmptyString, { k: k, value: v });
+                throw new Error(util_1.errorMessages.idEmptyString);
+            }
+        }
+        if (types_1.isGraphQLScalarType(type_1)) {
+            var jsType_1 = types_1.GraphQLScalarType.getJSType(type_1);
+            var validateScalar_1 = types_1.GraphQLScalarType.getValidationFunction(type_1);
+            if (type_1 === 'AWSJSON') {
                 if (typeof v === jsType_1) {
                     return;
                 }
@@ -295,7 +222,7 @@ var validateModelFields = function (modelDefinition) { return function (k, v) {
                         }
                     });
                     if (!validationStatus.every(function (s) { return s; })) {
-                        throw new Error("All elements in the " + name_1 + " array should be of type " + type + ", validation failed for one or more elements. " + v);
+                        throw new Error("All elements in the " + name_1 + " array should be of type " + type_1 + ", validation failed for one or more elements. " + v);
                     }
                 }
             }
@@ -308,7 +235,42 @@ var validateModelFields = function (modelDefinition) { return function (k, v) {
             else if (!util_1.isNullOrUndefined(v) &&
                 validateScalar_1 &&
                 !validateScalar_1(v)) {
-                throw new Error("Field " + name_1 + " should be of type " + type + ", validation failed. " + v);
+                throw new Error("Field " + name_1 + " should be of type " + type_1 + ", validation failed. " + v);
+            }
+        }
+        else if (types_1.isNonModelFieldType(type_1)) {
+            // do not check non model fields if undefined or null
+            if (!util_1.isNullOrUndefined(v)) {
+                var subNonModelDefinition_1 = schema.namespaces.user.nonModels[type_1.nonModel];
+                var modelValidator_1 = validateModelFields(subNonModelDefinition_1);
+                if (isArray) {
+                    var errorTypeText = type_1.nonModel;
+                    if (!isRequired_1) {
+                        errorTypeText = type_1.nonModel + " | null | undefined";
+                    }
+                    if (!Array.isArray(v)) {
+                        throw new Error("Field " + name_1 + " should be of type [" + errorTypeText + "], " + typeof v + " received. " + v);
+                    }
+                    v.forEach(function (item) {
+                        if ((util_1.isNullOrUndefined(item) && isRequired_1) ||
+                            (typeof item !== 'object' && typeof item !== 'undefined')) {
+                            throw new Error("All elements in the " + name_1 + " array should be of type " + type_1.nonModel + ", [" + typeof item + "] received. " + item);
+                        }
+                        if (!util_1.isNullOrUndefined(item)) {
+                            Object.keys(subNonModelDefinition_1.fields).forEach(function (subKey) {
+                                modelValidator_1(subKey, item[subKey]);
+                            });
+                        }
+                    });
+                }
+                else {
+                    if (typeof v !== 'object') {
+                        throw new Error("Field " + name_1 + " should be of type " + type_1.nonModel + ", " + typeof v + " recieved. " + v);
+                    }
+                    Object.keys(subNonModelDefinition_1.fields).forEach(function (subKey) {
+                        modelValidator_1(subKey, v[subKey]);
+                    });
+                }
             }
         }
     }
@@ -337,7 +299,7 @@ var castInstanceType = function (modelDefinition, k, v) {
 var initializeInstance = function (init, modelDefinition, draft) {
     var modelValidator = validateModelFields(modelDefinition);
     Object.entries(init).forEach(function (_a) {
-        var _b = __read(_a, 2), k = _b[0], v = _b[1];
+        var _b = tslib_1.__read(_a, 2), k = _b[0], v = _b[1];
         var parsedValue = castInstanceType(modelDefinition, k, v);
         modelValidator(k, parsedValue);
         draft[k] = parsedValue;
@@ -348,21 +310,29 @@ var createModelClass = function (modelDefinition) {
         function Model(init) {
             var instance = immer_1.produce(this, function (draft) {
                 initializeInstance(init, modelDefinition, draft);
-                var modelInstanceMetadata = instancesMetadata.has(init)
+                // model is initialized inside a DataStore component (e.g. by Sync Engine, Storage Engine, etc.)
+                var isInternallyInitialized = instancesMetadata.has(init);
+                var modelInstanceMetadata = isInternallyInitialized
                     ? init
                     : {};
-                var _id = modelInstanceMetadata.id, _version = modelInstanceMetadata._version, _lastChangedAt = modelInstanceMetadata._lastChangedAt, _deleted = modelInstanceMetadata._deleted;
-                // instancesIds are set by modelInstanceCreator, it is accessible only internally
-                var isInternal = _id !== null && _id !== undefined;
-                var id = isInternal
-                    ? _id
-                    : modelDefinition.syncable
-                        ? uuid_1.v4()
-                        : ulid();
-                if (!isInternal) {
+                var _id = modelInstanceMetadata.id;
+                if (util_1.isIdManaged(modelDefinition)) {
+                    var isInternalModel = _id !== null && _id !== undefined;
+                    var id = isInternalModel
+                        ? _id
+                        : modelDefinition.syncable
+                            ? uuid_1.v4()
+                            : ulid();
+                    draft.id = id;
+                }
+                else if (util_1.isIdOptionallyManaged(modelDefinition)) {
+                    // only auto-populate if the id was not provided
+                    draft.id = draft.id || uuid_1.v4();
+                }
+                if (!isInternallyInitialized) {
                     checkReadOnlyPropertyOnCreate(draft, modelDefinition);
                 }
-                draft.id = id;
+                var _version = modelInstanceMetadata._version, _lastChangedAt = modelInstanceMetadata._lastChangedAt, _deleted = modelInstanceMetadata._deleted;
                 if (modelDefinition.syncable) {
                     draft._version = _version;
                     draft._lastChangedAt = _lastChangedAt;
@@ -381,10 +351,12 @@ var createModelClass = function (modelDefinition) {
             var patches;
             var model = immer_1.produce(source, function (draft) {
                 fn(draft);
-                draft.id = source.id;
+                var keyNames = util_1.extractPrimaryKeyFieldNames(modelDefinition);
+                // Keys are immutable
+                keyNames.forEach(function (key) { return (draft[key] = source[key]); });
                 var modelValidator = validateModelFields(modelDefinition);
                 Object.entries(draft).forEach(function (_a) {
-                    var _b = __read(_a, 2), k = _b[0], v = _b[1];
+                    var _b = tslib_1.__read(_a, 2), k = _b[0], v = _b[1];
                     var parsedValue = castInstanceType(modelDefinition, k, v);
                     modelValidator(k, parsedValue);
                 });
@@ -392,7 +364,7 @@ var createModelClass = function (modelDefinition) {
             var hasExistingPatches = modelPatchesMap.has(source);
             if (patches.length || hasExistingPatches) {
                 if (hasExistingPatches) {
-                    var _a = __read(modelPatchesMap.get(source), 2), existingPatches = _a[0], existingSource = _a[1];
+                    var _a = tslib_1.__read(modelPatchesMap.get(source), 2), existingPatches = _a[0], existingSource = _a[1];
                     var mergedPatches = util_1.mergePatches(existingSource, existingPatches, patches);
                     modelPatchesMap.set(model, [mergedPatches, existingSource]);
                     checkReadOnlyPropertyOnUpdate(mergedPatches, modelDefinition);
@@ -414,7 +386,7 @@ var createModelClass = function (modelDefinition) {
             var instance = modelInstanceCreator(clazz, json);
             var modelValidator = validateModelFields(modelDefinition);
             Object.entries(instance).forEach(function (_a) {
-                var _b = __read(_a, 2), k = _b[0], v = _b[1];
+                var _b = tslib_1.__read(_a, 2), k = _b[0], v = _b[1];
                 modelValidator(k, v);
             });
             return instance;
@@ -438,7 +410,7 @@ var checkReadOnlyPropertyOnUpdate = function (patches, modelDefinition) {
     var patchArray = patches.map(function (p) { return [p.path[0], p.value]; });
     var fields = modelDefinition.fields;
     patchArray.forEach(function (_a) {
-        var _b = __read(_a, 2), key = _b[0], val = _b[1];
+        var _b = tslib_1.__read(_a, 2), key = _b[0], val = _b[1];
         if (!val || !fields[key])
             return;
         if (fields[key].isReadOnly) {
@@ -467,7 +439,7 @@ function isQueryOne(obj) {
 function defaultConflictHandler(conflictData) {
     var localModel = conflictData.localModel, modelConstructor = conflictData.modelConstructor, remoteModel = conflictData.remoteModel;
     var _version = remoteModel._version;
-    return modelInstanceCreator(modelConstructor, __assign(__assign({}, localModel), { _version: _version }));
+    return modelInstanceCreator(modelConstructor, tslib_1.__assign(tslib_1.__assign({}, localModel), { _version: _version }));
 }
 function defaultErrorHandler(error) {
     logger.warn(error);
@@ -500,25 +472,36 @@ function getModelConstructorByModelName(namespaceName, modelName) {
         throw new Error(msg);
     }
 }
+/**
+ * Queries the DataStore metadata tables to see if they are the expected
+ * version. If not, clobbers the whole DB. If so, leaves them alone.
+ * Otherwise, simply writes the schema version.
+ *
+ * SIDE EFFECT:
+ * 1. Creates a transaction
+ * 1. Updates data.
+ *
+ * @param storage Storage adapter containing the metadata.
+ * @param version The expected schema version.
+ */
 function checkSchemaVersion(storage, version) {
-    return __awaiter(this, void 0, void 0, function () {
+    return tslib_1.__awaiter(this, void 0, void 0, function () {
         var Setting, modelDefinition;
         var _this = this;
-        return __generator(this, function (_a) {
+        return tslib_1.__generator(this, function (_a) {
             switch (_a.label) {
                 case 0:
                     Setting = dataStoreClasses.Setting;
                     modelDefinition = schema.namespaces[util_1.DATASTORE].models.Setting;
-                    return [4 /*yield*/, storage.runExclusive(function (s) { return __awaiter(_this, void 0, void 0, function () {
+                    return [4 /*yield*/, storage.runExclusive(function (s) { return tslib_1.__awaiter(_this, void 0, void 0, function () {
                             var _a, schemaVersionSetting, storedValue;
-                            return __generator(this, function (_b) {
+                            return tslib_1.__generator(this, function (_b) {
                                 switch (_b.label) {
                                     case 0: return [4 /*yield*/, s.query(Setting, predicates_1.ModelPredicateCreator.createFromExisting(modelDefinition, function (c) {
-                                            // @ts-ignore Argument of type '"eq"' is not assignable to parameter of type 'never'.
                                             return c.key('eq', SETTING_SCHEMA_VERSION);
                                         }), { page: 0, limit: 1 })];
                                     case 1:
-                                        _a = __read.apply(void 0, [_b.sent(), 1]), schemaVersionSetting = _a[0];
+                                        _a = tslib_1.__read.apply(void 0, [_b.sent(), 1]), schemaVersionSetting = _a[0];
                                         if (!(schemaVersionSetting !== undefined &&
                                             schemaVersionSetting.value !== undefined)) return [3 /*break*/, 4];
                                         storedValue = JSON.parse(schemaVersionSetting.value);
@@ -583,13 +566,21 @@ function getNamespace() {
     };
     return namespace;
 }
+var DataStoreState;
+(function (DataStoreState) {
+    DataStoreState["NotRunning"] = "Not Running";
+    DataStoreState["Starting"] = "Starting";
+    DataStoreState["Running"] = "Running";
+    DataStoreState["Stopping"] = "Stopping";
+    DataStoreState["Clearing"] = "Clearing";
+})(DataStoreState || (DataStoreState = {}));
 var DataStore = /** @class */ (function () {
     function DataStore() {
         var _this = this;
         // reference to configured category instances. Used for preserving SSR context
         this.Auth = auth_1.Auth;
-        this.API = api_1.default;
-        this.Cache = cache_1.default;
+        this.API = api_1.API;
+        this.Cache = cache_1.BrowserStorageCache;
         this.amplifyConfig = {};
         this.syncPredicates = new WeakMap();
         // object that gets passed to descendent classes. Allows us to pass these down by reference
@@ -598,156 +589,235 @@ var DataStore = /** @class */ (function () {
             API: this.API,
             Cache: this.Cache,
         };
-        this.start = function () { return __awaiter(_this, void 0, void 0, function () {
-            var aws_appsync_graphqlEndpoint, _a, fullSyncIntervalInMilliseconds;
+        /**
+         * **IMPORTANT!**
+         *
+         * Accumulator for background things that can **and MUST** be called when
+         * DataStore stops.
+         *
+         * These jobs **MUST** be *idempotent promises* that resolve ONLY
+         * once the intended jobs are completely finished and/or otherwise destroyed
+         * and cleaned up with ZERO outstanding:
+         *
+         * 1. side effects (e.g., state changes)
+         * 1. callbacks
+         * 1. subscriptions
+         * 1. calls to storage
+         * 1. *etc.*
+         *
+         * Methods that create pending promises, subscriptions, callbacks, or any
+         * type of side effect **MUST** be registered with the manager. And, a new
+         * manager must be created after each `exit()`.
+         *
+         * Failure to comply will put DataStore into a highly unpredictable state
+         * when it needs to stop or clear -- which occurs when restarting with new
+         * sync expressions, during testing, and potentially during app code
+         * recovery handling, etc..
+         *
+         * It is up to the discretion of each disposer whether to wait for job
+         * completion or to cancel operations and issue failures *as long as the
+         * disposer returns in a reasonable amount of time.*
+         *
+         * (Reasonable = *seconds*, not minutes.)
+         */
+        this.runningProcesses = new core_1.BackgroundProcessManager();
+        /**
+         * Indicates what state DataStore is in.
+         *
+         * Not [yet?] used for actual state management; but for messaging
+         * when errors occur, to help troubleshoot.
+         */
+        this.state = DataStoreState.NotRunning;
+        /**
+         * If not already done:
+         * 1. Attaches and initializes storage.
+         * 1. Loads the schema and records metadata.
+         * 1. If `this.amplifyConfig.aws_appsync_graphqlEndpoint` contains a URL,
+         * attaches a sync engine, starts it, and subscribes.
+         */
+        this.start = function () { return tslib_1.__awaiter(_this, void 0, void 0, function () {
             var _this = this;
-            return __generator(this, function (_b) {
-                switch (_b.label) {
-                    case 0:
-                        if (!(this.initialized === undefined)) return [3 /*break*/, 1];
-                        logger.debug('Starting DataStore');
-                        this.initialized = new Promise(function (res, rej) {
-                            _this.initResolve = res;
-                            _this.initReject = rej;
-                        });
-                        return [3 /*break*/, 3];
-                    case 1: return [4 /*yield*/, this.initialized];
-                    case 2:
-                        _b.sent();
-                        return [2 /*return*/];
-                    case 3:
-                        this.storage = new storage_1.ExclusiveStorage(schema, namespaceResolver, getModelConstructorByModelName, modelInstanceCreator, this.storageAdapter, this.sessionId);
-                        return [4 /*yield*/, this.storage.init()];
-                    case 4:
-                        _b.sent();
-                        checkSchemaInitialized();
-                        return [4 /*yield*/, checkSchemaVersion(this.storage, schema.version)];
-                    case 5:
-                        _b.sent();
-                        aws_appsync_graphqlEndpoint = this.amplifyConfig.aws_appsync_graphqlEndpoint;
-                        if (!aws_appsync_graphqlEndpoint) return [3 /*break*/, 7];
-                        logger.debug('GraphQL endpoint available', aws_appsync_graphqlEndpoint);
-                        _a = this;
-                        return [4 /*yield*/, this.processSyncExpressions()];
-                    case 6:
-                        _a.syncPredicates = _b.sent();
-                        this.sync = new sync_1.SyncEngine(schema, namespaceResolver, exports.syncClasses, userClasses, this.storage, modelInstanceCreator, this.conflictHandler, this.errorHandler, this.syncPredicates, this.amplifyConfig, this.authModeStrategy, this.amplifyContext);
-                        fullSyncIntervalInMilliseconds = this.fullSyncInterval * 1000 * 60;
-                        syncSubscription = this.sync
-                            .start({ fullSyncInterval: fullSyncIntervalInMilliseconds })
-                            .subscribe({
-                            next: function (_a) {
-                                var type = _a.type, data = _a.data;
-                                // In Node, we need to wait for queries to be synced to prevent returning empty arrays.
-                                // In the Browser, we can begin returning data once subscriptions are in place.
-                                var readyType = isNode
-                                    ? sync_1.ControlMessage.SYNC_ENGINE_SYNC_QUERIES_READY
-                                    : sync_1.ControlMessage.SYNC_ENGINE_STORAGE_SUBSCRIBED;
-                                if (type === readyType) {
-                                    _this.initResolve();
-                                }
-                                core_1.Hub.dispatch('datastore', {
-                                    event: type,
-                                    data: data,
-                                });
-                            },
-                            error: function (err) {
-                                logger.warn('Sync error', err);
-                                _this.initReject();
-                            },
-                        });
-                        return [3 /*break*/, 8];
-                    case 7:
-                        logger.warn("Data won't be synchronized. No GraphQL endpoint configured. Did you forget `Amplify.configure(awsconfig)`?", {
-                            config: this.amplifyConfig,
+            return tslib_1.__generator(this, function (_a) {
+                return [2 /*return*/, this.runningProcesses
+                        .add(function () { return tslib_1.__awaiter(_this, void 0, void 0, function () {
+                        var aws_appsync_graphqlEndpoint, _a, fullSyncIntervalInMilliseconds;
+                        var _this = this;
+                        return tslib_1.__generator(this, function (_b) {
+                            switch (_b.label) {
+                                case 0:
+                                    this.state = DataStoreState.Starting;
+                                    if (!(this.initialized === undefined)) return [3 /*break*/, 1];
+                                    logger.debug('Starting DataStore');
+                                    this.initialized = new Promise(function (res, rej) {
+                                        _this.initResolve = res;
+                                        _this.initReject = rej;
+                                    });
+                                    return [3 /*break*/, 3];
+                                case 1: return [4 /*yield*/, this.initialized];
+                                case 2:
+                                    _b.sent();
+                                    return [2 /*return*/];
+                                case 3:
+                                    this.storage = new storage_1.ExclusiveStorage(schema, namespaceResolver, getModelConstructorByModelName, modelInstanceCreator, this.storageAdapter, this.sessionId);
+                                    return [4 /*yield*/, this.storage.init()];
+                                case 4:
+                                    _b.sent();
+                                    checkSchemaInitialized();
+                                    return [4 /*yield*/, checkSchemaVersion(this.storage, schema.version)];
+                                case 5:
+                                    _b.sent();
+                                    aws_appsync_graphqlEndpoint = this.amplifyConfig.aws_appsync_graphqlEndpoint;
+                                    if (!aws_appsync_graphqlEndpoint) return [3 /*break*/, 7];
+                                    logger.debug('GraphQL endpoint available', aws_appsync_graphqlEndpoint);
+                                    _a = this;
+                                    return [4 /*yield*/, this.processSyncExpressions()];
+                                case 6:
+                                    _a.syncPredicates = _b.sent();
+                                    this.sync = new sync_1.SyncEngine(schema, namespaceResolver, exports.syncClasses, userClasses, this.storage, modelInstanceCreator, this.conflictHandler, this.errorHandler, this.syncPredicates, this.amplifyConfig, this.authModeStrategy, this.amplifyContext, this.connectivityMonitor);
+                                    fullSyncIntervalInMilliseconds = this.fullSyncInterval * 1000 * 60;
+                                    syncSubscription = this.sync
+                                        .start({ fullSyncInterval: fullSyncIntervalInMilliseconds })
+                                        .subscribe({
+                                        next: function (_a) {
+                                            var type = _a.type, data = _a.data;
+                                            // In Node, we need to wait for queries to be synced to prevent returning empty arrays.
+                                            // In the Browser, we can begin returning data once subscriptions are in place.
+                                            var readyType = isNode
+                                                ? sync_1.ControlMessage.SYNC_ENGINE_SYNC_QUERIES_READY
+                                                : sync_1.ControlMessage.SYNC_ENGINE_STORAGE_SUBSCRIBED;
+                                            if (type === readyType) {
+                                                _this.initResolve();
+                                            }
+                                            core_1.Hub.dispatch('datastore', {
+                                                event: type,
+                                                data: data,
+                                            });
+                                        },
+                                        error: function (err) {
+                                            logger.warn('Sync error', err);
+                                            _this.initReject();
+                                        },
+                                    });
+                                    return [3 /*break*/, 8];
+                                case 7:
+                                    logger.warn("Data won't be synchronized. No GraphQL endpoint configured. Did you forget `Amplify.configure(awsconfig)`?", {
+                                        config: this.amplifyConfig,
+                                    });
+                                    this.initResolve();
+                                    _b.label = 8;
+                                case 8: return [4 /*yield*/, this.initialized];
+                                case 9:
+                                    _b.sent();
+                                    this.state = DataStoreState.Running;
+                                    return [2 /*return*/];
+                            }
                         });
-                        this.initResolve();
-                        _b.label = 8;
-                    case 8: return [4 /*yield*/, this.initialized];
-                    case 9:
-                        _b.sent();
-                        return [2 /*return*/];
-                }
+                    }); }, 'datastore start')
+                        .catch(this.handleAddProcError('DataStore.start()'))];
             });
         }); };
-        this.query = function (modelConstructor, idOrCriteria, paginationProducer) { return __awaiter(_this, void 0, void 0, function () {
-            var msg, modelDefinition, predicate, pagination, result;
-            return __generator(this, function (_a) {
-                switch (_a.label) {
-                    case 0: return [4 /*yield*/, this.start()];
-                    case 1:
-                        _a.sent();
-                        //#region Input validation
-                        if (!isValidModelConstructor(modelConstructor)) {
-                            msg = 'Constructor is not for a valid model';
-                            logger.error(msg, { modelConstructor: modelConstructor });
-                            throw new Error(msg);
-                        }
-                        if (typeof idOrCriteria === 'string') {
-                            if (paginationProducer !== undefined) {
-                                logger.warn('Pagination is ignored when querying by id');
-                            }
-                        }
-                        modelDefinition = getModelDefinition(modelConstructor);
-                        if (isQueryOne(idOrCriteria)) {
-                            predicate = predicates_1.ModelPredicateCreator.createForId(modelDefinition, idOrCriteria);
-                        }
-                        else {
-                            if (predicates_1.isPredicatesAll(idOrCriteria)) {
-                                // Predicates.ALL means "all records", so no predicate (undefined)
-                                predicate = undefined;
-                            }
-                            else {
-                                predicate = predicates_1.ModelPredicateCreator.createFromExisting(modelDefinition, idOrCriteria);
+        this.query = function (modelConstructor, identifierOrCriteria, paginationProducer) { return tslib_1.__awaiter(_this, void 0, void 0, function () {
+            var _this = this;
+            return tslib_1.__generator(this, function (_a) {
+                return [2 /*return*/, this.runningProcesses
+                        .add(function () { return tslib_1.__awaiter(_this, void 0, void 0, function () {
+                        var msg, modelDefinition, keyFields, predicate, msg, pagination, result, returnOne;
+                        return tslib_1.__generator(this, function (_a) {
+                            switch (_a.label) {
+                                case 0: return [4 /*yield*/, this.start()];
+                                case 1:
+                                    _a.sent();
+                                    //#region Input validation
+                                    if (!isValidModelConstructor(modelConstructor)) {
+                                        msg = 'Constructor is not for a valid model';
+                                        logger.error(msg, { modelConstructor: modelConstructor });
+                                        throw new Error(msg);
+                                    }
+                                    if (typeof identifierOrCriteria === 'string') {
+                                        if (paginationProducer !== undefined) {
+                                            logger.warn('Pagination is ignored when querying by id');
+                                        }
+                                    }
+                                    modelDefinition = getModelDefinition(modelConstructor);
+                                    keyFields = util_1.extractPrimaryKeyFieldNames(modelDefinition);
+                                    if (isQueryOne(identifierOrCriteria)) {
+                                        if (keyFields.length > 1) {
+                                            msg = util_1.errorMessages.queryByPkWithCompositeKeyPresent;
+                                            logger.error(msg, { keyFields: keyFields });
+                                            throw new Error(msg);
+                                        }
+                                        predicate = predicates_1.ModelPredicateCreator.createForSingleField(modelDefinition, keyFields[0], identifierOrCriteria);
+                                    }
+                                    else {
+                                        // Object is being queried using object literal syntax
+                                        if (types_1.isIdentifierObject(identifierOrCriteria, modelDefinition)) {
+                                            predicate = predicates_1.ModelPredicateCreator.createForPk(modelDefinition, identifierOrCriteria);
+                                        }
+                                        else if (predicates_1.isPredicatesAll(identifierOrCriteria)) {
+                                            // Predicates.ALL means "all records", so no predicate (undefined)
+                                            predicate = undefined;
+                                        }
+                                        else {
+                                            predicate = predicates_1.ModelPredicateCreator.createFromExisting(modelDefinition, identifierOrCriteria);
+                                        }
+                                    }
+                                    pagination = this.processPagination(modelDefinition, paginationProducer);
+                                    //#endregion
+                                    logger.debug('params ready', {
+                                        modelConstructor: modelConstructor,
+                                        predicate: predicates_1.ModelPredicateCreator.getPredicates(predicate, false),
+                                        pagination: tslib_1.__assign(tslib_1.__assign({}, pagination), { sort: predicates_1.ModelSortPredicateCreator.getPredicates(pagination && pagination.sort, false) }),
+                                    });
+                                    return [4 /*yield*/, this.storage.query(modelConstructor, predicate, pagination)];
+                                case 2:
+                                    result = _a.sent();
+                                    returnOne = isQueryOne(identifierOrCriteria) ||
+                                        types_1.isIdentifierObject(identifierOrCriteria, modelDefinition);
+                                    return [2 /*return*/, returnOne ? result[0] : result];
                             }
-                        }
-                        pagination = this.processPagination(modelDefinition, paginationProducer);
-                        //#endregion
-                        logger.debug('params ready', {
-                            modelConstructor: modelConstructor,
-                            predicate: predicates_1.ModelPredicateCreator.getPredicates(predicate, false),
-                            pagination: __assign(__assign({}, pagination), { sort: predicates_1.ModelSortPredicateCreator.getPredicates(pagination && pagination.sort, false) }),
                         });
-                        return [4 /*yield*/, this.storage.query(modelConstructor, predicate, pagination)];
-                    case 2:
-                        result = _a.sent();
-                        return [2 /*return*/, isQueryOne(idOrCriteria) ? result[0] : result];
-                }
+                    }); }, 'datastore query')
+                        .catch(this.handleAddProcError('DataStore.query()'))];
             });
         }); };
-        this.save = function (model, condition) { return __awaiter(_this, void 0, void 0, function () {
-            var patchesTuple, modelConstructor, msg, modelDefinition, producedCondition, _a, savedModel;
+        this.save = function (model, condition) { return tslib_1.__awaiter(_this, void 0, void 0, function () {
             var _this = this;
-            return __generator(this, function (_b) {
-                switch (_b.label) {
-                    case 0: return [4 /*yield*/, this.start()];
-                    case 1:
-                        _b.sent();
-                        patchesTuple = modelPatchesMap.get(model);
-                        modelConstructor = model
-                            ? model.constructor
-                            : undefined;
-                        if (!isValidModelConstructor(modelConstructor)) {
-                            msg = 'Object is not an instance of a valid model';
-                            logger.error(msg, { model: model });
-                            throw new Error(msg);
-                        }
-                        modelDefinition = getModelDefinition(modelConstructor);
-                        producedCondition = predicates_1.ModelPredicateCreator.createFromExisting(modelDefinition, condition);
-                        return [4 /*yield*/, this.storage.runExclusive(function (s) { return __awaiter(_this, void 0, void 0, function () {
-                                return __generator(this, function (_a) {
-                                    switch (_a.label) {
-                                        case 0: return [4 /*yield*/, s.save(model, producedCondition, undefined, patchesTuple)];
-                                        case 1:
-                                            _a.sent();
-                                            return [2 /*return*/, s.query(modelConstructor, predicates_1.ModelPredicateCreator.createForId(modelDefinition, model.id))];
+            return tslib_1.__generator(this, function (_a) {
+                return [2 /*return*/, this.runningProcesses
+                        .add(function () { return tslib_1.__awaiter(_this, void 0, void 0, function () {
+                        var patchesTuple, modelConstructor, msg, modelDefinition, producedCondition, _a, savedModel;
+                        var _this = this;
+                        return tslib_1.__generator(this, function (_b) {
+                            switch (_b.label) {
+                                case 0: return [4 /*yield*/, this.start()];
+                                case 1:
+                                    _b.sent();
+                                    patchesTuple = modelPatchesMap.get(model);
+                                    modelConstructor = model ? model.constructor : undefined;
+                                    if (!isValidModelConstructor(modelConstructor)) {
+                                        msg = 'Object is not an instance of a valid model';
+                                        logger.error(msg, { model: model });
+                                        throw new Error(msg);
                                     }
-                                });
-                            }); })];
-                    case 2:
-                        _a = __read.apply(void 0, [_b.sent(), 1]), savedModel = _a[0];
-                        return [2 /*return*/, savedModel];
-                }
+                                    modelDefinition = getModelDefinition(modelConstructor);
+                                    producedCondition = predicates_1.ModelPredicateCreator.createFromExisting(modelDefinition, condition);
+                                    return [4 /*yield*/, this.storage.runExclusive(function (s) { 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*/, s.save(model, producedCondition, undefined, patchesTuple)];
+                                                    case 1:
+                                                        _a.sent();
+                                                        return [2 /*return*/, s.query(modelConstructor, predicates_1.ModelPredicateCreator.createForPk(modelDefinition, model))];
+                                                }
+                                            });
+                                        }); })];
+                                case 2:
+                                    _a = tslib_1.__read.apply(void 0, [_b.sent(), 1]), savedModel = _a[0];
+                                    return [2 /*return*/, savedModel];
+                            }
+                        });
+                    }); }, 'datastore save')
+                        .catch(this.handleAddProcError('DataStore.save()'))];
             });
         }); };
         this.setConflictHandler = function (config) {
@@ -776,75 +846,95 @@ var DataStore = /** @class */ (function () {
             }
             return _this.errorHandler || defaultErrorHandler;
         };
-        this.delete = function (modelOrConstructor, idOrCriteria) { return __awaiter(_this, void 0, void 0, function () {
-            var condition, msg, modelConstructor, msg, msg, _a, deleted, model, modelConstructor, msg, modelDefinition, idPredicate, msg, _b, _c, deleted;
-            return __generator(this, function (_d) {
-                switch (_d.label) {
-                    case 0: return [4 /*yield*/, this.start()];
-                    case 1:
-                        _d.sent();
-                        if (!modelOrConstructor) {
-                            msg = 'Model or Model Constructor required';
-                            logger.error(msg, { modelOrConstructor: modelOrConstructor });
-                            throw new Error(msg);
-                        }
-                        if (!isValidModelConstructor(modelOrConstructor)) return [3 /*break*/, 3];
-                        modelConstructor = modelOrConstructor;
-                        if (!idOrCriteria) {
-                            msg = 'Id to delete or criteria required. Do you want to delete all? Pass Predicates.ALL';
-                            logger.error(msg, { idOrCriteria: idOrCriteria });
-                            throw new Error(msg);
-                        }
-                        if (typeof idOrCriteria === 'string') {
-                            condition = predicates_1.ModelPredicateCreator.createForId(getModelDefinition(modelConstructor), idOrCriteria);
-                        }
-                        else {
-                            condition = predicates_1.ModelPredicateCreator.createFromExisting(getModelDefinition(modelConstructor), 
-                            /**
-                             * idOrCriteria is always a ProducerModelPredicate<T>, never a symbol.
-                             * The symbol is used only for typing purposes. e.g. see Predicates.ALL
-                             */
-                            idOrCriteria);
-                            if (!condition || !predicates_1.ModelPredicateCreator.isValidPredicate(condition)) {
-                                msg = 'Criteria required. Do you want to delete all? Pass Predicates.ALL';
-                                logger.error(msg, { condition: condition });
-                                throw new Error(msg);
-                            }
-                        }
-                        return [4 /*yield*/, this.storage.delete(modelConstructor, condition)];
-                    case 2:
-                        _a = __read.apply(void 0, [_d.sent(), 1]), deleted = _a[0];
-                        return [2 /*return*/, deleted];
-                    case 3:
-                        model = modelOrConstructor;
-                        modelConstructor = Object.getPrototypeOf(model || {})
-                            .constructor;
-                        if (!isValidModelConstructor(modelConstructor)) {
-                            msg = 'Object is not an instance of a valid model';
-                            logger.error(msg, { model: model });
-                            throw new Error(msg);
-                        }
-                        modelDefinition = getModelDefinition(modelConstructor);
-                        idPredicate = predicates_1.ModelPredicateCreator.createForId(modelDefinition, model.id);
-                        if (idOrCriteria) {
-                            if (typeof idOrCriteria !== 'function') {
-                                msg = 'Invalid criteria';
-                                logger.error(msg, { idOrCriteria: idOrCriteria });
-                                throw new Error(msg);
+        this.delete = function (modelOrConstructor, identifierOrCriteria) { return tslib_1.__awaiter(_this, void 0, void 0, function () {
+            var _this = this;
+            return tslib_1.__generator(this, function (_a) {
+                return [2 /*return*/, this.runningProcesses
+                        .add(function () { return tslib_1.__awaiter(_this, void 0, void 0, function () {
+                        var condition, msg, modelConstructor, msg, modelDefinition, keyFields, msg, msg, _a, deleted, model, modelConstructor, msg, modelDefinition, pkPredicate, msg, _b, _c, deleted;
+                        return tslib_1.__generator(this, function (_d) {
+                            switch (_d.label) {
+                                case 0: return [4 /*yield*/, this.start()];
+                                case 1:
+                                    _d.sent();
+                                    if (!modelOrConstructor) {
+                                        msg = 'Model or Model Constructor required';
+                                        logger.error(msg, { modelOrConstructor: modelOrConstructor });
+                                        throw new Error(msg);
+                                    }
+                                    if (!isValidModelConstructor(modelOrConstructor)) return [3 /*break*/, 3];
+                                    modelConstructor = modelOrConstructor;
+                                    if (!identifierOrCriteria) {
+                                        msg = 'Id to delete or criteria required. Do you want to delete all? Pass Predicates.ALL';
+                                        logger.error(msg, { identifierOrCriteria: identifierOrCriteria });
+                                        throw new Error(msg);
+                                    }
+                                    modelDefinition = getModelDefinition(modelConstructor);
+                                    if (typeof identifierOrCriteria === 'string') {
+                                        keyFields = util_1.extractPrimaryKeyFieldNames(modelDefinition);
+                                        if (keyFields.length > 1) {
+                                            msg = util_1.errorMessages.deleteByPkWithCompositeKeyPresent;
+                                            logger.error(msg, { keyFields: keyFields });
+                                            throw new Error(msg);
+                                        }
+                                        condition = predicates_1.ModelPredicateCreator.createForSingleField(getModelDefinition(modelConstructor), keyFields[0], identifierOrCriteria);
+                                    }
+                                    else {
+                                        if (types_1.isIdentifierObject(identifierOrCriteria, modelDefinition)) {
+                                            condition = predicates_1.ModelPredicateCreator.createForPk(modelDefinition, identifierOrCriteria);
+                                        }
+                                        else {
+                                            condition = predicates_1.ModelPredicateCreator.createFromExisting(modelDefinition, 
+                                            /**
+                                             * idOrCriteria is always a ProducerModelPredicate<T>, never a symbol.
+                                             * The symbol is used only for typing purposes. e.g. see Predicates.ALL
+                                             */
+                                            identifierOrCriteria);
+                                        }
+                                        if (!condition ||
+                                            !predicates_1.ModelPredicateCreator.isValidPredicate(condition)) {
+                                            msg = 'Criteria required. Do you want to delete all? Pass Predicates.ALL';
+                                            logger.error(msg, { condition: condition });
+                                            throw new Error(msg);
+                                        }
+                                    }
+                                    return [4 /*yield*/, this.storage.delete(modelConstructor, condition)];
+                                case 2:
+                                    _a = tslib_1.__read.apply(void 0, [_d.sent(), 1]), deleted = _a[0];
+                                    return [2 /*return*/, deleted];
+                                case 3:
+                                    model = modelOrConstructor;
+                                    modelConstructor = Object.getPrototypeOf(model || {})
+                                        .constructor;
+                                    if (!isValidModelConstructor(modelConstructor)) {
+                                        msg = 'Object is not an instance of a valid model';
+                                        logger.error(msg, { model: model });
+                                        throw new Error(msg);
+                                    }
+                                    modelDefinition = getModelDefinition(modelConstructor);
+                                    pkPredicate = predicates_1.ModelPredicateCreator.createForPk(modelDefinition, model);
+                                    if (identifierOrCriteria) {
+                                        if (typeof identifierOrCriteria !== 'function') {
+                                            msg = 'Invalid criteria';
+                                            logger.error(msg, { identifierOrCriteria: identifierOrCriteria });
+                                            throw new Error(msg);
+                                        }
+                                        condition = identifierOrCriteria(pkPredicate);
+                                    }
+                                    else {
+                                        condition = pkPredicate;
+                                    }
+                                    return [4 /*yield*/, this.storage.delete(model, condition)];
+                                case 4:
+                                    _b = tslib_1.__read.apply(void 0, [_d.sent(), 1]), _c = tslib_1.__read(_b[0], 1), deleted = _c[0];
+                                    return [2 /*return*/, deleted];
                             }
-                            condition = idOrCriteria(idPredicate);
-                        }
-                        else {
-                            condition = idPredicate;
-                        }
-                        return [4 /*yield*/, this.storage.delete(model, condition)];
-                    case 4:
-                        _b = __read.apply(void 0, [_d.sent(), 1]), _c = __read(_b[0], 1), deleted = _c[0];
-                        return [2 /*return*/, deleted];
-                }
+                        });
+                    }); }, 'datastore delete')
+                        .catch(this.handleAddProcError('DataStore.delete()'))];
             });
         }); };
-        this.observe = function (modelOrConstructor, idOrCriteria) {
+        this.observe = function (modelOrConstructor, identifierOrCriteria) {
             var predicate;
             var modelConstructor = modelOrConstructor && isValidModelConstructor(modelOrConstructor)
                 ? modelOrConstructor
@@ -853,10 +943,10 @@ var DataStore = /** @class */ (function () {
                 var model = modelOrConstructor;
                 var modelConstructor_1 = model && Object.getPrototypeOf(model).constructor;
                 if (isValidModelConstructor(modelConstructor_1)) {
-                    if (idOrCriteria) {
+                    if (identifierOrCriteria) {
                         logger.warn('idOrCriteria is ignored when using a model instance', {
                             model: model,
-                            idOrCriteria: idOrCriteria,
+                            identifierOrCriteria: identifierOrCriteria,
                         });
                     }
                     return _this.observe(modelConstructor_1, model.id);
@@ -867,9 +957,17 @@ var DataStore = /** @class */ (function () {
                     throw new Error(msg);
                 }
             }
-            if (idOrCriteria !== undefined && modelConstructor === undefined) {
+            // observe should not accept object literal syntax
+            if (identifierOrCriteria &&
+                modelConstructor &&
+                types_1.isIdentifierObject(identifierOrCriteria, getModelDefinition(modelConstructor))) {
+                var msg = util_1.errorMessages.observeWithObjectLiteral;
+                logger.error(msg, { objectLiteral: identifierOrCriteria });
+                throw new Error(msg);
+            }
+            if (identifierOrCriteria !== undefined && modelConstructor === undefined) {
                 var msg = 'Cannot provide criteria without a modelConstructor';
-                logger.error(msg, idOrCriteria);
+                logger.error(msg, identifierOrCriteria);
                 throw new Error(msg);
             }
             if (modelConstructor && !isValidModelConstructor(modelConstructor)) {
@@ -877,19 +975,27 @@ var DataStore = /** @class */ (function () {
                 logger.error(msg, { modelConstructor: modelConstructor });
                 throw new Error(msg);
             }
-            if (typeof idOrCriteria === 'string') {
-                predicate = predicates_1.ModelPredicateCreator.createForId(getModelDefinition(modelConstructor), idOrCriteria);
+            if (typeof identifierOrCriteria === 'string') {
+                var modelDefinition = getModelDefinition(modelConstructor);
+                var _a = tslib_1.__read(util_1.extractPrimaryKeyFieldNames(modelDefinition), 1), keyField = _a[0];
+                predicate = predicates_1.ModelPredicateCreator.createForSingleField(getModelDefinition(modelConstructor), keyField, identifierOrCriteria);
             }
             else {
-                predicate =
-                    modelConstructor &&
-                        predicates_1.ModelPredicateCreator.createFromExisting(getModelDefinition(modelConstructor), idOrCriteria);
+                if (predicates_1.isPredicatesAll(identifierOrCriteria)) {
+                    predicate = undefined;
+                }
+                else {
+                    predicate =
+                        modelConstructor &&
+                            predicates_1.ModelPredicateCreator.createFromExisting(getModelDefinition(modelConstructor), identifierOrCriteria);
+                }
             }
             return new zen_observable_ts_1.default(function (observer) {
                 var handle;
-                (function () { return __awaiter(_this, void 0, void 0, function () {
+                _this.runningProcesses
+                    .add(function () { return tslib_1.__awaiter(_this, void 0, void 0, function () {
                     var _this = this;
-                    return __generator(this, function (_a) {
+                    return tslib_1.__generator(this, function (_a) {
                         switch (_a.label) {
                             case 0: return [4 /*yield*/, this.start()];
                             case 1:
@@ -903,36 +1009,52 @@ var DataStore = /** @class */ (function () {
                                     return namespaceResolver(model) === util_1.USER;
                                 })
                                     .subscribe({
-                                    next: function (item) { return __awaiter(_this, void 0, void 0, function () {
-                                        var message, freshElement;
-                                        return __generator(this, function (_a) {
-                                            switch (_a.label) {
-                                                case 0:
-                                                    message = item;
-                                                    if (!(item.opType !== 'DELETE')) return [3 /*break*/, 2];
-                                                    return [4 /*yield*/, this.query(item.model, item.element.id)];
-                                                case 1:
-                                                    freshElement = _a.sent();
-                                                    message = __assign(__assign({}, message), { element: freshElement });
-                                                    _a.label = 2;
-                                                case 2:
-                                                    observer.next(message);
-                                                    return [2 /*return*/];
-                                            }
-                                        });
-                                    }); },
+                                    next: function (item) {
+                                        return _this.runningProcesses.isOpen &&
+                                            _this.runningProcesses.add(function () { return tslib_1.__awaiter(_this, void 0, void 0, function () {
+                                                var message, modelDefinition, keyFields, primaryKeysAndValues, freshElement;
+                                                return tslib_1.__generator(this, function (_a) {
+                                                    switch (_a.label) {
+                                                        case 0:
+                                                            message = item;
+                                                            if (!(item.opType !== 'DELETE')) return [3 /*break*/, 2];
+                                                            modelDefinition = getModelDefinition(item.model);
+                                                            keyFields = util_1.extractPrimaryKeyFieldNames(modelDefinition);
+                                                            primaryKeysAndValues = util_1.extractPrimaryKeysAndValues(item.element, keyFields);
+                                                            return [4 /*yield*/, this.query(item.model, primaryKeysAndValues)];
+                                                        case 1:
+                                                            freshElement = _a.sent();
+                                                            message = tslib_1.__assign(tslib_1.__assign({}, message), { element: freshElement });
+                                                            _a.label = 2;
+                                                        case 2:
+                                                            observer.next(message);
+                                                            return [2 /*return*/];
+                                                    }
+                                                });
+                                            }); }, 'datastore observe message handler');
+                                    },
                                     error: function (err) { return observer.error(err); },
                                     complete: function () { return observer.complete(); },
                                 });
                                 return [2 /*return*/];
                         }
                     });
-                }); })();
-                return function () {
-                    if (handle) {
-                        handle.unsubscribe();
-                    }
-                };
+                }); }, 'datastore observe observable initialization')
+                    .catch(_this.handleAddProcError('DataStore.observe()'))
+                    .catch(function (error) {
+                    observer.error(error);
+                });
+                // better than no cleaner, but if the subscriber is handling the
+                // complete() message async and not registering with the context,
+                // this will still be problematic.
+                return _this.runningProcesses.addCleaner(function () { return tslib_1.__awaiter(_this, void 0, void 0, function () {
+                    return tslib_1.__generator(this, function (_a) {
+                        if (handle) {
+                            handle.unsubscribe();
+                        }
+                        return [2 /*return*/];
+                    });
+                }); }, 'DataStore.observe() cleanup');
             });
         };
         this.observeQuery = function (model, criteria, options) {
@@ -966,8 +1088,9 @@ var DataStore = /** @class */ (function () {
                 var sort = (options || {}).sort;
                 var sortOptions = sort ? { sort: sort } : undefined;
                 var modelDefinition = getModelDefinition(model);
+                var keyFields = util_1.extractPrimaryKeyFieldNames(modelDefinition);
                 if (isQueryOne(criteria)) {
-                    predicate = predicates_1.ModelPredicateCreator.createForId(modelDefinition, criteria);
+                    predicate = predicates_1.ModelPredicateCreator.createForSingleField(modelDefinition, keyFields[0], criteria);
                 }
                 else {
                     if (predicates_1.isPredicatesAll(criteria)) {
@@ -980,10 +1103,11 @@ var DataStore = /** @class */ (function () {
                 }
                 var _a = predicates_1.ModelPredicateCreator.getPredicates(predicate, false) || {}, predicates = _a.predicates, predicateGroupType = _a.type;
                 var hasPredicate = !!predicates;
-                (function () { return __awaiter(_this, void 0, void 0, function () {
+                _this.runningProcesses
+                    .add(function () { return tslib_1.__awaiter(_this, void 0, void 0, function () {
                     var err_1;
                     var _this = this;
-                    return __generator(this, function (_a) {
+                    return tslib_1.__generator(this, function (_a) {
                         switch (_a.label) {
                             case 0:
                                 _a.trys.push([0, 2, , 3]);
@@ -991,7 +1115,9 @@ var DataStore = /** @class */ (function () {
                             case 1:
                                 // first, query and return any locally-available records
                                 (_a.sent()).forEach(function (item) {
-                                    return items.set(item.id, item);
+                                    var itemModelDefinition = getModelDefinition(model);
+                                    var idOrPk = utils_1.getIdentifierValue(itemModelDefinition, item);
+                                    items.set(idOrPk, item);
                                 });
                                 // Observe the model and send a stream of updates (debounced).
                                 // We need to post-filter results instead of passing criteria through
@@ -1000,15 +1126,17 @@ var DataStore = /** @class */ (function () {
                                 handle = this.observe(model).subscribe(function (_a) {
                                     var element = _a.element, model = _a.model, opType = _a.opType;
                                     var _b, _c;
+                                    var itemModelDefinition = getModelDefinition(model);
+                                    var idOrPk = utils_1.getIdentifierValue(itemModelDefinition, element);
                                     if (hasPredicate &&
                                         !util_1.validatePredicate(element, predicateGroupType, predicates)) {
                                         if (opType === 'UPDATE' &&
-                                            (items.has(element.id) || itemsChanged.has(element.id))) {
+                                            (items.has(idOrPk) || itemsChanged.has(idOrPk))) {
                                             // tracking as a "deleted item" will include the item in
                                             // page limit calculations and ensure it is removed from the
                                             // final items collection, regardless of which collection(s)
                                             // it is currently in. (I mean, it could be in both, right!?)
-                                            deletedItemIds.push(element.id);
+                                            deletedItemIds.push(idOrPk);
                                         }
                                         else {
                                             // ignore updates for irrelevant/filtered items.
@@ -1020,13 +1148,14 @@ var DataStore = /** @class */ (function () {
                                     // in the `mergePage` method within src/sync/merger.ts. The final state of a model instance
                                     // depends on the LATEST record (for a given id).
                                     if (opType === 'DELETE') {
-                                        deletedItemIds.push(element.id);
+                                        deletedItemIds.push(idOrPk);
                                     }
                                     else {
-                                        itemsChanged.set(element.id, element);
+                                        itemsChanged.set(idOrPk, element);
                                     }
                                     var isSynced = (_c = (_b = _this.sync) === null || _b === void 0 ? void 0 : _b.getModelSyncedStatus(model)) !== null && _c !== void 0 ? _c : false;
-                                    var limit = itemsChanged.size - deletedItemIds.length >= _this.syncPageSize;
+                                    var limit = itemsChanged.size - deletedItemIds.length >=
+                                        _this.syncPageSize;
                                     if (limit || isSynced) {
                                         limitTimerRace.resolve();
                                     }
@@ -1043,7 +1172,11 @@ var DataStore = /** @class */ (function () {
                             case 3: return [2 /*return*/];
                         }
                     });
-                }); })();
+                }); }, 'datastore observequery startup')
+                    .catch(_this.handleAddProcError('DataStore.observeQuery()'))
+                    .catch(function (error) {
+                    observer.error(error);
+                });
                 /**
                  * Combines the `items`, `itemsChanged`, and `deletedItemIds` collections into
                  * a snapshot in the form of `{ items: T[], isSynced: boolean}`.
@@ -1053,14 +1186,18 @@ var DataStore = /** @class */ (function () {
                 var generateSnapshot = function () {
                     var _a, _b;
                     var isSynced = (_b = (_a = _this.sync) === null || _a === void 0 ? void 0 : _a.getModelSyncedStatus(model)) !== null && _b !== void 0 ? _b : false;
-                    var itemsArray = __spread(Array.from(items.values()), Array.from(itemsChanged.values()));
+                    var itemsArray = tslib_1.__spread(Array.from(items.values()), Array.from(itemsChanged.values()));
                     if (options === null || options === void 0 ? void 0 : options.sort) {
                         sortItems(itemsArray);
                     }
                     items.clear();
-                    itemsArray.forEach(function (item) { return items.set(item.id, item); });
+                    itemsArray.forEach(function (item) {
+                        var itemModelDefinition = getModelDefinition(model);
+                        var idOrPk = utils_1.getIdentifierValue(itemModelDefinition, item);
+                        items.set(idOrPk, item);
+                    });
                     // remove deleted items from the final result set
-                    deletedItemIds.forEach(function (id) { return items.delete(id); });
+                    deletedItemIds.forEach(function (idOrPk) { return items.delete(idOrPk); });
                     return {
                         items: Array.from(items.values()),
                         isSynced: isSynced,
@@ -1075,7 +1212,8 @@ var DataStore = /** @class */ (function () {
                  * @param snapshot The generated items data to emit.
                  */
                 var emitSnapshot = function (snapshot) {
-                    // send the generated snapshot to the primary subscription
+                    // send the generated snapshot to the primary subscription.
+                    // NOTE: This observer's handler *could* be async ...
                     observer.next(snapshot);
                     // reset the changed items sets
                     itemsChanged.clear();
@@ -1111,15 +1249,18 @@ var DataStore = /** @class */ (function () {
                     if (event === sync_1.ControlMessage.SYNC_ENGINE_MODEL_SYNCED &&
                         ((_b = data === null || data === void 0 ? void 0 : data.model) === null || _b === void 0 ? void 0 : _b.name) === model.name) {
                         generateAndEmitSnapshot();
-                        core_1.Hub.remove('api', hubCallback);
+                        core_1.Hub.remove('datastore', hubCallback);
                     }
                 };
                 core_1.Hub.listen('datastore', hubCallback);
-                return function () {
-                    if (handle) {
-                        handle.unsubscribe();
-                    }
-                };
+                return _this.runningProcesses.addCleaner(function () { return tslib_1.__awaiter(_this, void 0, void 0, function () {
+                    return tslib_1.__generator(this, function (_a) {
+                        if (handle) {
+                            handle.unsubscribe();
+                        }
+                        return [2 /*return*/];
+                    });
+                }); }, 'datastore observequery cleaner');
             });
         };
         this.configure = function (config) {
@@ -1127,8 +1268,8 @@ var DataStore = /** @class */ (function () {
             _this.amplifyContext.Auth = _this.Auth;
             _this.amplifyContext.API = _this.API;
             _this.amplifyContext.Cache = _this.Cache;
-            var configDataStore = config.DataStore, configAuthModeStrategyType = config.authModeStrategyType, configConflictHandler = config.conflictHandler, configErrorHandler = config.errorHandler, configMaxRecordsToSync = config.maxRecordsToSync, configSyncPageSize = config.syncPageSize, configFullSyncInterval = config.fullSyncInterval, configSyncExpressions = config.syncExpressions, configAuthProviders = config.authProviders, configStorageAdapter = config.storageAdapter, configFromAmplify = __rest(config, ["DataStore", "authModeStrategyType", "conflictHandler", "errorHandler", "maxRecordsToSync", "syncPageSize", "fullSyncInterval", "syncExpressions", "authProviders", "storageAdapter"]);
-            _this.amplifyConfig = __assign(__assign({}, configFromAmplify), _this.amplifyConfig);
+            var configDataStore = config.DataStore, configAuthModeStrategyType = config.authModeStrategyType, configConflictHandler = config.conflictHandler, configErrorHandler = config.errorHandler, configMaxRecordsToSync = config.maxRecordsToSync, configSyncPageSize = config.syncPageSize, configFullSyncInterval = config.fullSyncInterval, configSyncExpressions = config.syncExpressions, configAuthProviders = config.authProviders, configStorageAdapter = config.storageAdapter, configFromAmplify = tslib_1.__rest(config, ["DataStore", "authModeStrategyType", "conflictHandler", "errorHandler", "maxRecordsToSync", "syncPageSize", "fullSyncInterval", "syncExpressions", "authProviders", "storageAdapter"]);
+            _this.amplifyConfig = tslib_1.__assign(tslib_1.__assign({}, configFromAmplify), _this.amplifyConfig);
             _this.conflictHandler = _this.setConflictHandler(config);
             _this.errorHandler = _this.setErrorHandler(config);
             var authModeStrategyType = (configDataStore && configDataStore.authModeStrategyType) ||
@@ -1178,66 +1319,132 @@ var DataStore = /** @class */ (function () {
                     undefined;
             _this.sessionId = _this.retrieveSessionId();
         };
-        this.clear = function clear() {
-            return __awaiter(this, void 0, void 0, function () {
-                return __generator(this, function (_a) {
-                    switch (_a.label) {
-                        case 0:
-                            checkSchemaInitialized();
-                            if (!(this.storage === undefined)) return [3 /*break*/, 2];
-                            // connect to storage so that it can be cleared without fully starting DataStore
-                            this.storage = new storage_1.ExclusiveStorage(schema, namespaceResolver, getModelConstructorByModelName, modelInstanceCreator, this.storageAdapter, this.sessionId);
-                            return [4 /*yield*/, this.storage.init()];
-                        case 1:
-                            _a.sent();
-                            _a.label = 2;
-                        case 2:
-                            if (syncSubscription && !syncSubscription.closed) {
-                                syncSubscription.unsubscribe();
-                            }
-                            return [4 /*yield*/, this.storage.clear()];
-                        case 3:
-                            _a.sent();
-                            if (this.sync) {
-                                this.sync.unsubscribeConnectivity();
-                            }
-                            this.initialized = undefined; // Should re-initialize when start() is called.
-                            this.storage = undefined;
-                            this.sync = undefined;
-                            this.syncPredicates = new WeakMap();
-                            return [2 /*return*/];
-                    }
-                });
-            });
-        };
-        this.stop = function stop() {
-            return __awaiter(this, void 0, void 0, function () {
-                return __generator(this, function (_a) {
-                    switch (_a.label) {
-                        case 0:
-                            if (!(this.initialized !== undefined)) return [3 /*break*/, 2];
-                            return [4 /*yield*/, this.start()];
-                        case 1:
-                            _a.sent();
-                            _a.label = 2;
-                        case 2:
-                            if (syncSubscription && !syncSubscription.closed) {
-                                syncSubscription.unsubscribe();
-                            }
-                            if (this.sync) {
-                                this.sync.unsubscribeConnectivity();
-                            }
-                            this.initialized = undefined; // Should re-initialize when start() is called.
-                            this.sync = undefined;
-                            return [2 /*return*/];
-                    }
-                });
-            });
-        };
     }
     DataStore.prototype.getModuleName = function () {
         return 'DataStore';
     };
+    /**
+     * Builds a function to capture `BackgroundManagerNotOpenError`'s to produce friendlier,
+     * more instructive errors for customers.
+     *
+     * @param operation The name of the operation (usually a Datastore method) the customer
+     * tried to call.
+     */
+    DataStore.prototype.handleAddProcError = function (operation) {
+        var _this = this;
+        /**
+         * If the tested error is a `BackgroundManagerNotOpenError`, it will be captured
+         * and replaced with a friendlier message that instructs the App Developer.
+         *
+         * @param err An error to test.
+         */
+        var handler = function (err) {
+            if (err.message.startsWith('BackgroundManagerNotOpenError')) {
+                throw new Error([
+                    "DataStoreStateError: Tried to execute `" + operation + "` while DataStore was \"" + _this.state + "\".",
+                    "This can only be done while DataStore is \"Started\" or \"Stopped\". To remedy:",
+                    'Ensure all calls to `stop()` and `clear()` have completed first.',
+                    'If this is not possible, retry the operation until it succeeds.',
+                ].join('\n'));
+            }
+            else {
+                throw err;
+            }
+        };
+        return handler;
+    };
+    /**
+     * Clears all data from storage and removes all data, schema info, other
+     * initialization details, and then stops DataStore.
+     *
+     * That said, reinitialization is required after clearing. This can be done
+     * by explicitiliy calling `start()` or any method that implicitly starts
+     * DataStore, such as `query()`, `save()`, or `delete()`.
+     */
+    DataStore.prototype.clear = function () {
+        return tslib_1.__awaiter(this, void 0, void 0, function () {
+            return tslib_1.__generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        checkSchemaInitialized();
+                        this.state = DataStoreState.Clearing;
+                        return [4 /*yield*/, this.runningProcesses.close()];
+                    case 1:
+                        _a.sent();
+                        if (!(this.storage === undefined)) return [3 /*break*/, 3];
+                        // connect to storage so that it can be cleared without fully starting DataStore
+                        this.storage = new storage_1.ExclusiveStorage(schema, namespaceResolver, getModelConstructorByModelName, modelInstanceCreator, this.storageAdapter, this.sessionId);
+                        return [4 /*yield*/, this.storage.init()];
+                    case 2:
+                        _a.sent();
+                        _a.label = 3;
+                    case 3:
+                        if (syncSubscription && !syncSubscription.closed) {
+                            syncSubscription.unsubscribe();
+                        }
+                        if (!this.sync) return [3 /*break*/, 5];
+                        return [4 /*yield*/, this.sync.stop()];
+                    case 4:
+                        _a.sent();
+                        _a.label = 5;
+                    case 5: return [4 /*yield*/, this.storage.clear()];
+                    case 6:
+                        _a.sent();
+                        this.initialized = undefined; // Should re-initialize when start() is called.
+                        this.storage = undefined;
+                        this.sync = undefined;
+                        this.syncPredicates = new WeakMap();
+                        return [4 /*yield*/, this.runningProcesses.open()];
+                    case 7:
+                        _a.sent();
+                        this.state = DataStoreState.NotRunning;
+                        return [2 /*return*/];
+                }
+            });
+        });
+    };
+    /**
+     * Stops all DataStore sync activities.
+     *
+     * TODO: "Waits for graceful termination of
+     * running queries and terminates subscriptions."
+     */
+    DataStore.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:
+                        this.state = DataStoreState.Stopping;
+                        return [4 /*yield*/, this.runningProcesses.close()];
+                    case 1:
+                        _a.sent();
+                        if (syncSubscription && !syncSubscription.closed) {
+                            syncSubscription.unsubscribe();
+                        }
+                        if (!this.sync) return [3 /*break*/, 3];
+                        return [4 /*yield*/, this.sync.stop()];
+                    case 2:
+                        _a.sent();
+                        _a.label = 3;
+                    case 3:
+                        this.initialized = undefined; // Should re-initialize when start() is called.
+                        this.sync = undefined;
+                        return [4 /*yield*/, this.runningProcesses.open()];
+                    case 4:
+                        _a.sent();
+                        this.state = DataStoreState.NotRunning;
+                        return [2 /*return*/];
+                }
+            });
+        });
+    };
+    /**
+     * Validates given pagination input from a query and creates a pagination
+     * argument for use against the storage layer.
+     *
+     * @param modelDefinition
+     * @param paginationProducer
+     */
     DataStore.prototype.processPagination = function (modelDefinition, paginationProducer) {
         var sortPredicate;
         var _a = paginationProducer || {}, limit = _a.limit, page = _a.page, sort = _a.sort;
@@ -1272,19 +1479,23 @@ var DataStore = /** @class */ (function () {
             sort: sortPredicate,
         };
     };
+    /**
+     * Examines the configured `syncExpressions` and produces a WeakMap of
+     * SchemaModel -> predicate to use during sync.
+     */
     DataStore.prototype.processSyncExpressions = function () {
-        return __awaiter(this, void 0, void 0, function () {
+        return tslib_1.__awaiter(this, void 0, void 0, function () {
             var syncPredicates;
             var _this = this;
-            return __generator(this, function (_a) {
+            return tslib_1.__generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         if (!this.syncExpressions || !this.syncExpressions.length) {
                             return [2 /*return*/, new WeakMap()];
                         }
-                        return [4 /*yield*/, Promise.all(this.syncExpressions.map(function (syncExpression) { return __awaiter(_this, void 0, void 0, function () {
+                        return [4 /*yield*/, Promise.all(this.syncExpressions.map(function (syncExpression) { return tslib_1.__awaiter(_this, void 0, void 0, function () {
                                 var _a, modelConstructor, conditionProducer, modelDefinition, condition, predicate;
-                                return __generator(this, function (_b) {
+                                return tslib_1.__generator(this, function (_b) {
                                     switch (_b.label) {
                                         case 0: return [4 /*yield*/, syncExpression];
                                         case 1:
@@ -1318,9 +1529,9 @@ var DataStore = /** @class */ (function () {
         }
     };
     DataStore.prototype.unwrapPromise = function (conditionProducer) {
-        return __awaiter(this, void 0, void 0, function () {
+        return tslib_1.__awaiter(this, void 0, void 0, function () {
             var condition, error_1;
-            return __generator(this, function (_a) {
+            return tslib_1.__generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         _a.trys.push([0, 2, , 3]);
@@ -1341,7 +1552,7 @@ var DataStore = /** @class */ (function () {
     };
     DataStore.prototype.weakMapFromEntries = function (entries) {
         return entries.reduce(function (map, _a) {
-            var _b = __read(_a, 2), modelDefinition = _b[0], predicate = _b[1];
+            var _b = tslib_1.__read(_a, 2), modelDefinition = _b[0], predicate = _b[1];
             if (map.has(modelDefinition)) {
                 var name_2 = modelDefinition.name;
                 logger.warn("You can only utilize one Sync Expression per model.\n          Subsequent sync expressions for the " + name_2 + " model will be ignored.");
@@ -1353,20 +1564,22 @@ var DataStore = /** @class */ (function () {
             return map;
         }, new WeakMap());
     };
-    // database separation for Amplify Console. Not a public API
+    /**
+     * A session ID to allow CMS to open databases against multiple apps.
+     * This session ID is only expected be set by AWS Amplify Studio.
+     */
     DataStore.prototype.retrieveSessionId = function () {
         try {
             var sessionId = sessionStorage.getItem('datastoreSessionId');
             if (sessionId) {
                 var aws_appsync_graphqlEndpoint = this.amplifyConfig.aws_appsync_graphqlEndpoint;
                 var appSyncUrl = aws_appsync_graphqlEndpoint.split('/')[2];
-                var _a = __read(appSyncUrl.split('.'), 1), appSyncId = _a[0];
+                var _a = tslib_1.__read(appSyncUrl.split('.'), 1), appSyncId = _a[0];
                 return sessionId + "-" + appSyncId;
             }
         }
-        catch (_b) {
-            return undefined;
-        }
+        catch (_b) { }
+        return undefined;
     };
     return DataStore;
 }());