@aws-amplify/datastore 3.7.3 → 3.7.4-custom-pk.79

package/lib/storage/adapter/IndexedDBAdapter.js

@@ -42,17 +42,6 @@ var __asyncValues = (this && this.__asyncValues) || function (o) {
     function verb(n) { i[n] = o[n] && function (v) { return new Promise(function (resolve, reject) { v = o[n](v), settle(resolve, reject, v.done, v.value); }); }; }
     function settle(resolve, reject, d, v) { Promise.resolve(v).then(function(v) { resolve({ value: v, done: d }); }, reject); }
 };
-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 __read = (this && this.__read) || function (o, n) {
     var m = typeof Symbol === "function" && o[Symbol.iterator];
     if (!m) return o;
@@ -69,6 +58,17 @@ var __read = (this && this.__read) || function (o, n) {
     }
     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;
@@ -123,6 +123,35 @@ var IndexedDBAdapter = /** @class */ (function () {
         var storeName = namespace + "_" + modelName;
         return storeName;
     };
+    IndexedDBAdapter.prototype.getNamespaceAndModelFromStorename = function (storeName) {
+        var _a = __read(storeName.split('_')), namespaceName = _a[0], modelNameArr = _a.slice(1);
+        return {
+            namespaceName: namespaceName,
+            modelName: modelNameArr.join('_'),
+        };
+    };
+    IndexedDBAdapter.prototype.getIndexKeyPath = function (namespaceName, modelName) {
+        var _a, _b;
+        var keyPath = (_b = (_a = this.schema.namespaces[namespaceName]) === null || _a === void 0 ? void 0 : _a.keys[modelName]) === null || _b === void 0 ? void 0 : _b.primaryKey;
+        if (keyPath) {
+            return keyPath;
+        }
+        return ['id'];
+    };
+    IndexedDBAdapter.prototype.getIndexKeyValues = function (model) {
+        var modelConstructor = Object.getPrototypeOf(model)
+            .constructor;
+        var namespaceName = this.namespaceResolver(modelConstructor);
+        var keys = this.getIndexKeyPath(namespaceName, modelConstructor.name);
+        var keyValues = keys.map(function (field) { return model[field]; });
+        return keyValues;
+    };
+    IndexedDBAdapter.prototype.keysEqual = function (keysA, keysB) {
+        if (keysA.length !== keysB.length) {
+            return false;
+        }
+        return keysA.every(function (key, idx) { return key === keysB[idx]; });
+    };
     IndexedDBAdapter.prototype.setUp = function (theSchema, namespaceResolver, modelInstanceCreator, getModelConstructorByModelName, sessionId) {
         return __awaiter(this, void 0, void 0, function () {
             var VERSION, _a, error_1;
@@ -154,40 +183,35 @@ var IndexedDBAdapter = /** @class */ (function () {
                     case 5:
                         _b.trys.push([5, 8, , 9]);
                         if (!!this.db) return [3 /*break*/, 7];
-                        VERSION = 2;
+                        VERSION = 3;
                         _a = this;
                         return [4 /*yield*/, idb.openDB(this.dbName, VERSION, {
                                 upgrade: function (db, oldVersion, newVersion, txn) { return __awaiter(_this, void 0, void 0, function () {
-                                    var _a, _b, storeName, origStore, tmpName, newStore, cursor, count, e_1_1, error_2;
-                                    var e_1, _c;
+                                    var _a, _b, storeName, origStore, tmpName, newStore, origIndexNames, origIndexNames_1, origIndexNames_1_1, idxName, idx, keyPath_1, unique, _c, namespaceName, modelName, keyPath, cursor, count, e_1_1, error_2;
+                                    var e_1, _d, e_2, _e;
                                     var _this = this;
-                                    return __generator(this, function (_d) {
-                                        switch (_d.label) {
+                                    return __generator(this, function (_f) {
+                                        switch (_f.label) {
                                             case 0:
                                                 if (oldVersion === 0) {
                                                     Object.keys(theSchema.namespaces).forEach(function (namespaceName) {
                                                         var namespace = theSchema.namespaces[namespaceName];
                                                         Object.keys(namespace.models).forEach(function (modelName) {
                                                             var storeName = _this.getStorename(namespaceName, modelName);
-                                                            var store = db.createObjectStore(storeName, {
-                                                                autoIncrement: true,
-                                                            });
-                                                            var indexes = _this.schema.namespaces[namespaceName].relationships[modelName].indexes;
-                                                            indexes.forEach(function (index) { return store.createIndex(index, index); });
-                                                            store.createIndex('byId', 'id', { unique: true });
+                                                            _this.createObjectStoreForModel(db, namespaceName, storeName, modelName);
                                                         });
                                                     });
                                                     return [2 /*return*/];
                                                 }
-                                                if (!(oldVersion === 1 && newVersion === 2)) return [3 /*break*/, 16];
-                                                _d.label = 1;
+                                                if (!((oldVersion === 1 || oldVersion === 2) && newVersion === 3)) return [3 /*break*/, 16];
+                                                _f.label = 1;
                                             case 1:
-                                                _d.trys.push([1, 14, , 15]);
-                                                _d.label = 2;
+                                                _f.trys.push([1, 14, , 15]);
+                                                _f.label = 2;
                                             case 2:
-                                                _d.trys.push([2, 11, 12, 13]);
+                                                _f.trys.push([2, 11, 12, 13]);
                                                 _a = __values(txn.objectStoreNames), _b = _a.next();
-                                                _d.label = 3;
+                                                _f.label = 3;
                                             case 3:
                                                 if (!!_b.done) return [3 /*break*/, 10];
                                                 storeName = _b.value;
@@ -198,46 +222,91 @@ var IndexedDBAdapter = /** @class */ (function () {
                                                     keyPath: undefined,
                                                     autoIncrement: true,
                                                 });
-                                                newStore.createIndex('byId', 'id', { unique: true });
+                                                origIndexNames = Object.values(origStore.indexNames);
+                                                try {
+                                                    for (origIndexNames_1 = (e_2 = void 0, __values(origIndexNames)), origIndexNames_1_1 = origIndexNames_1.next(); !origIndexNames_1_1.done; origIndexNames_1_1 = origIndexNames_1.next()) {
+                                                        idxName = origIndexNames_1_1.value;
+                                                        if (idxName === 'byId') {
+                                                            // don't migrate this index. It'll be replaced with byPk
+                                                            continue;
+                                                        }
+                                                        idx = origStore.index(idxName);
+                                                        keyPath_1 = idx.keyPath, unique = idx.unique;
+                                                        newStore.createIndex(idxName, keyPath_1, { unique: unique });
+                                                    }
+                                                }
+                                                catch (e_2_1) { e_2 = { error: e_2_1 }; }
+                                                finally {
+                                                    try {
+                                                        if (origIndexNames_1_1 && !origIndexNames_1_1.done && (_e = origIndexNames_1.return)) _e.call(origIndexNames_1);
+                                                    }
+                                                    finally { if (e_2) throw e_2.error; }
+                                                }
+                                                _c = this.getNamespaceAndModelFromStorename(storeName), namespaceName = _c.namespaceName, modelName = _c.modelName;
+                                                keyPath = this.getIndexKeyPath(namespaceName, modelName);
+                                                newStore.createIndex('byPk', keyPath, { unique: true });
                                                 return [4 /*yield*/, origStore.openCursor()];
                                             case 4:
-                                                cursor = _d.sent();
+                                                cursor = _f.sent();
                                                 count = 0;
-                                                _d.label = 5;
+                                                _f.label = 5;
                                             case 5:
                                                 if (!(cursor && cursor.value)) return [3 /*break*/, 8];
                                                 // we don't pass key, since they are all new entries in the new store
                                                 return [4 /*yield*/, newStore.put(cursor.value)];
                                             case 6:
                                                 // we don't pass key, since they are all new entries in the new store
-                                                _d.sent();
+                                                _f.sent();
                                                 return [4 /*yield*/, cursor.continue()];
                                             case 7:
-                                                cursor = _d.sent();
+                                                cursor = _f.sent();
                                                 count++;
                                                 return [3 /*break*/, 5];
                                             case 8:
                                                 // delete original
                                                 db.deleteObjectStore(tmpName);
                                                 logger.debug(count + " " + storeName + " records migrated");
-                                                _d.label = 9;
+                                                _f.label = 9;
                                             case 9:
                                                 _b = _a.next();
                                                 return [3 /*break*/, 3];
                                             case 10: return [3 /*break*/, 13];
                                             case 11:
-                                                e_1_1 = _d.sent();
+                                                e_1_1 = _f.sent();
                                                 e_1 = { error: e_1_1 };
                                                 return [3 /*break*/, 13];
                                             case 12:
                                                 try {
-                                                    if (_b && !_b.done && (_c = _a.return)) _c.call(_a);
+                                                    if (_b && !_b.done && (_d = _a.return)) _d.call(_a);
                                                 }
                                                 finally { if (e_1) throw e_1.error; }
                                                 return [7 /*endfinally*/];
-                                            case 13: return [3 /*break*/, 15];
+                                            case 13:
+                                                // add new models created after IndexedDB, but before migration
+                                                // this case may happen when a user has not opened an app for
+                                                // some time and a new model is added during that time
+                                                Object.keys(theSchema.namespaces).forEach(function (namespaceName) {
+                                                    var namespace = theSchema.namespaces[namespaceName];
+                                                    var objectStoreNames = new Set(txn.objectStoreNames);
+                                                    Object.keys(namespace.models)
+                                                        .map(function (modelName) {
+                                                        return [
+                                                            modelName,
+                                                            _this.getStorename(namespaceName, modelName),
+                                                        ];
+                                                    })
+                                                        .filter(function (_a) {
+                                                        var _b = __read(_a, 2), storeName = _b[1];
+                                                        return !objectStoreNames.has(storeName);
+                                                    })
+                                                        .forEach(function (_a) {
+                                                        var _b = __read(_a, 2), modelName = _b[0], storeName = _b[1];
+                                                        _this.createObjectStoreForModel(db, namespaceName, storeName, modelName);
+                                                    });
+                                                });
+                                                return [3 /*break*/, 15];
                                             case 14:
-                                                error_2 = _d.sent();
+                                                error_2 = _f.sent();
                                                 logger.error('Error migrating IndexedDB data', error_2);
                                                 txn.abort();
                                                 throw error_2;
@@ -261,7 +330,7 @@ var IndexedDBAdapter = /** @class */ (function () {
             });
         });
     };
-    IndexedDBAdapter.prototype._get = function (storeOrStoreName, id) {
+    IndexedDBAdapter.prototype._get = function (storeOrStoreName, keyArr) {
         return __awaiter(this, void 0, void 0, function () {
             var index, storeName, store, result;
             return __generator(this, function (_a) {
@@ -269,13 +338,13 @@ var IndexedDBAdapter = /** @class */ (function () {
                     case 0:
                         if (typeof storeOrStoreName === 'string') {
                             storeName = storeOrStoreName;
-                            index = this.db.transaction(storeName, 'readonly').store.index('byId');
+                            index = this.db.transaction(storeName, 'readonly').store.index('byPk');
                         }
                         else {
                             store = storeOrStoreName;
-                            index = store.index('byId');
+                            index = store.index('byPk');
                         }
-                        return [4 /*yield*/, index.get(id)];
+                        return [4 /*yield*/, index.get(keyArr)];
                     case 1:
                         result = _a.sent();
                         return [2 /*return*/, result];
@@ -284,9 +353,9 @@ var IndexedDBAdapter = /** @class */ (function () {
         });
     };
     IndexedDBAdapter.prototype.save = function (model, condition) {
-        var e_2, _a;
+        var e_3, _a;
         return __awaiter(this, void 0, void 0, function () {
-            var modelConstructor, storeName, connectedModels, namespaceName, set, connectionStoreNames, tx, store, fromDB, predicates, predicateObjs, type, isValid, msg, result, connectionStoreNames_1, connectionStoreNames_1_1, resItem, storeName_1, item, instance, store_1, id, fromDB_1, opType, key, e_2_1;
+            var modelConstructor, storeName, namespaceName, connectedModels, set, connectionStoreNames, tx, store, keyValues, fromDB, predicates, predicateObjs, type, isValid, msg, result, connectionStoreNames_1, connectionStoreNames_1_1, resItem, storeName_1, item, instance, keys, store_1, itemKeyValues, fromDB_1, opType, modelKeyValues, keysEqual, key, e_3_1;
             var _this = this;
             return __generator(this, function (_b) {
                 switch (_b.label) {
@@ -296,18 +365,20 @@ var IndexedDBAdapter = /** @class */ (function () {
                         modelConstructor = Object.getPrototypeOf(model)
                             .constructor;
                         storeName = this.getStorenameForModel(modelConstructor);
-                        connectedModels = util_1.traverseModel(modelConstructor.name, model, this.schema.namespaces[this.namespaceResolver(modelConstructor)], this.modelInstanceCreator, this.getModelConstructorByModelName);
                         namespaceName = this.namespaceResolver(modelConstructor);
+                        connectedModels = util_1.traverseModel(modelConstructor.name, model, this.schema.namespaces[namespaceName], this.modelInstanceCreator, this.getModelConstructorByModelName);
                         set = new Set();
                         connectionStoreNames = Object.values(connectedModels).map(function (_a) {
                             var modelName = _a.modelName, item = _a.item, instance = _a.instance;
                             var storeName = _this.getStorename(namespaceName, modelName);
                             set.add(storeName);
-                            return { storeName: storeName, item: item, instance: instance };
+                            var keys = _this.getIndexKeyPath(namespaceName, modelName);
+                            return { storeName: storeName, item: item, instance: instance, keys: keys };
                         });
                         tx = this.db.transaction(__spread([storeName], Array.from(set.values())), 'readwrite');
                         store = tx.objectStore(storeName);
-                        return [4 /*yield*/, this._get(store, model.id)];
+                        keyValues = this.getIndexKeyValues(model);
+                        return [4 /*yield*/, this._get(store, keyValues)];
                     case 2:
                         fromDB = _b.sent();
                         if (condition && fromDB) {
@@ -330,15 +401,20 @@ var IndexedDBAdapter = /** @class */ (function () {
                     case 5:
                         if (!(connectionStoreNames_1_1 = _b.sent(), !connectionStoreNames_1_1.done)) return [3 /*break*/, 10];
                         resItem = connectionStoreNames_1_1.value;
-                        storeName_1 = resItem.storeName, item = resItem.item, instance = resItem.instance;
+                        storeName_1 = resItem.storeName, item = resItem.item, instance = resItem.instance, keys = resItem.keys;
                         store_1 = tx.objectStore(storeName_1);
-                        id = item.id;
-                        return [4 /*yield*/, this._get(store_1, id)];
+                        itemKeyValues = keys.map(function (key) {
+                            var value = item[key];
+                            return value;
+                        });
+                        return [4 /*yield*/, this._get(store_1, itemKeyValues)];
                     case 6:
                         fromDB_1 = _b.sent();
                         opType = fromDB_1 === undefined ? types_1.OpType.INSERT : types_1.OpType.UPDATE;
-                        if (!(id === model.id || opType === types_1.OpType.INSERT)) return [3 /*break*/, 9];
-                        return [4 /*yield*/, store_1.index('byId').getKey(item.id)];
+                        modelKeyValues = this.getIndexKeyValues(model);
+                        keysEqual = this.keysEqual(itemKeyValues, modelKeyValues);
+                        if (!(keysEqual || opType === types_1.OpType.INSERT)) return [3 /*break*/, 9];
+                        return [4 /*yield*/, store_1.index('byPk').getKey(itemKeyValues)];
                     case 7:
                         key = _b.sent();
                         return [4 /*yield*/, store_1.put(item, key)];
@@ -349,8 +425,8 @@ var IndexedDBAdapter = /** @class */ (function () {
                     case 9: return [3 /*break*/, 4];
                     case 10: return [3 /*break*/, 17];
                     case 11:
-                        e_2_1 = _b.sent();
-                        e_2 = { error: e_2_1 };
+                        e_3_1 = _b.sent();
+                        e_3 = { error: e_3_1 };
                         return [3 /*break*/, 17];
                     case 12:
                         _b.trys.push([12, , 15, 16]);
@@ -361,7 +437,7 @@ var IndexedDBAdapter = /** @class */ (function () {
                         _b.label = 14;
                     case 14: return [3 /*break*/, 16];
                     case 15:
-                        if (e_2) throw e_2.error;
+                        if (e_3) throw e_3.error;
                         return [7 /*endfinally*/];
                     case 16: return [7 /*endfinally*/];
                     case 17: return [4 /*yield*/, tx.done];
@@ -374,9 +450,9 @@ var IndexedDBAdapter = /** @class */ (function () {
     };
     IndexedDBAdapter.prototype.load = function (namespaceName, srcModelName, records) {
         var records_1, records_1_1, records_2, records_2_1;
-        var e_3, _a, e_4, _b, e_5, _c;
+        var e_4, _a, e_5, _b, e_6, _c;
         return __awaiter(this, void 0, void 0, function () {
-            var namespace, relations, connectionStoreNames, modelConstructor, tx, relations_1, relations_1_1, relation, fieldName, modelName, targetName, storeName, store, modelConstructor_1, _d, recordItem, getByfield, connectionRecord, e_4_1, recordItem, connectionRecord, e_5_1, e_3_1;
+            var namespace, relations, connectionStoreNames, modelConstructor, tx, relations_1, relations_1_1, relation, fieldName, modelName, targetName, targetNames, storeName, store, modelConstructor_1, _d, recordItem, getByFields, allPresent, keys, connectionRecord, getByfield, key, connectionRecord, e_5_1, recordItem, allPresent, keys, connectionRecord, key, connectionRecord, e_6_1, e_4_1;
             var _this = this;
             return __generator(this, function (_e) {
                 switch (_e.label) {
@@ -396,122 +472,170 @@ var IndexedDBAdapter = /** @class */ (function () {
                         tx = this.db.transaction(__spread(connectionStoreNames), 'readonly');
                         _e.label = 1;
                     case 1:
-                        _e.trys.push([1, 34, 35, 40]);
+                        _e.trys.push([1, 38, 39, 44]);
                         relations_1 = __asyncValues(relations);
                         _e.label = 2;
                     case 2: return [4 /*yield*/, relations_1.next()];
                     case 3:
-                        if (!(relations_1_1 = _e.sent(), !relations_1_1.done)) return [3 /*break*/, 33];
+                        if (!(relations_1_1 = _e.sent(), !relations_1_1.done)) return [3 /*break*/, 37];
                         relation = relations_1_1.value;
-                        fieldName = relation.fieldName, modelName = relation.modelName, targetName = relation.targetName;
+                        fieldName = relation.fieldName, modelName = relation.modelName, targetName = relation.targetName, targetNames = relation.targetNames;
                         storeName = this.getStorename(namespaceName, modelName);
                         store = tx.objectStore(storeName);
                         modelConstructor_1 = this.getModelConstructorByModelName(namespaceName, modelName);
                         _d = relation.relationType;
                         switch (_d) {
                             case 'HAS_ONE': return [3 /*break*/, 4];
-                            case 'BELONGS_TO': return [3 /*break*/, 17];
-                            case 'HAS_MANY': return [3 /*break*/, 30];
+                            case 'BELONGS_TO': return [3 /*break*/, 19];
+                            case 'HAS_MANY': return [3 /*break*/, 34];
                         }
-                        return [3 /*break*/, 31];
+                        return [3 /*break*/, 35];
                     case 4:
-                        _e.trys.push([4, 10, 11, 16]);
+                        _e.trys.push([4, 12, 13, 18]);
                         records_1 = __asyncValues(records);
                         _e.label = 5;
                     case 5: return [4 /*yield*/, records_1.next()];
                     case 6:
-                        if (!(records_1_1 = _e.sent(), !records_1_1.done)) return [3 /*break*/, 9];
+                        if (!(records_1_1 = _e.sent(), !records_1_1.done)) return [3 /*break*/, 11];
                         recordItem = records_1_1.value;
-                        getByfield = recordItem[targetName] ? targetName : fieldName;
-                        if (!recordItem[getByfield])
-                            return [3 /*break*/, 9];
-                        return [4 /*yield*/, this._get(store, recordItem[getByfield])];
+                        if (!(targetNames === null || targetNames === void 0 ? void 0 : targetNames.length)) return [3 /*break*/, 8];
+                        getByFields = [];
+                        allPresent = void 0;
+                        // iterate through all targetnames to make sure they are all present in the recordItem
+                        allPresent = targetNames.every(function (targetName) {
+                            return recordItem[targetName] != null;
+                        });
+                        if (!allPresent) {
+                            return [3 /*break*/, 11];
+                        }
+                        getByFields = targetNames;
+                        keys = getByFields.map(function (getByField) { return recordItem[getByField]; });
+                        return [4 /*yield*/, this._get(store, keys)];
                     case 7:
                         connectionRecord = _e.sent();
                         recordItem[fieldName] =
                             connectionRecord &&
                                 this.modelInstanceCreator(modelConstructor_1, connectionRecord);
-                        _e.label = 8;
-                    case 8: return [3 /*break*/, 5];
-                    case 9: return [3 /*break*/, 16];
-                    case 10:
-                        e_4_1 = _e.sent();
-                        e_4 = { error: e_4_1 };
-                        return [3 /*break*/, 16];
-                    case 11:
-                        _e.trys.push([11, , 14, 15]);
-                        if (!(records_1_1 && !records_1_1.done && (_b = records_1.return))) return [3 /*break*/, 13];
-                        return [4 /*yield*/, _b.call(records_1)];
+                        return [3 /*break*/, 10];
+                    case 8:
+                        getByfield = recordItem[targetName]
+                            ? targetName
+                            : fieldName;
+                        // We break here, because the recordItem does not have 'team', the `getByField`
+                        // extract the keys on the related model.
+                        if (!recordItem[getByfield])
+                            return [3 /*break*/, 11];
+                        key = [recordItem[getByfield]];
+                        return [4 /*yield*/, this._get(store, key)];
+                    case 9:
+                        connectionRecord = _e.sent();
+                        recordItem[fieldName] =
+                            connectionRecord &&
+                                this.modelInstanceCreator(modelConstructor_1, connectionRecord);
+                        _e.label = 10;
+                    case 10: return [3 /*break*/, 5];
+                    case 11: return [3 /*break*/, 18];
                     case 12:
-                        _e.sent();
-                        _e.label = 13;
-                    case 13: return [3 /*break*/, 15];
+                        e_5_1 = _e.sent();
+                        e_5 = { error: e_5_1 };
+                        return [3 /*break*/, 18];
+                    case 13:
+                        _e.trys.push([13, , 16, 17]);
+                        if (!(records_1_1 && !records_1_1.done && (_b = records_1.return))) return [3 /*break*/, 15];
+                        return [4 /*yield*/, _b.call(records_1)];
                     case 14:
-                        if (e_4) throw e_4.error;
+                        _e.sent();
+                        _e.label = 15;
+                    case 15: return [3 /*break*/, 17];
+                    case 16:
+                        if (e_5) throw e_5.error;
                         return [7 /*endfinally*/];
-                    case 15: return [7 /*endfinally*/];
-                    case 16: return [3 /*break*/, 32];
-                    case 17:
-                        _e.trys.push([17, 23, 24, 29]);
-                        records_2 = __asyncValues(records);
-                        _e.label = 18;
-                    case 18: return [4 /*yield*/, records_2.next()];
+                    case 17: return [7 /*endfinally*/];
+                    case 18: return [3 /*break*/, 36];
                     case 19:
-                        if (!(records_2_1 = _e.sent(), !records_2_1.done)) return [3 /*break*/, 22];
+                        _e.trys.push([19, 27, 28, 33]);
+                        records_2 = __asyncValues(records);
+                        _e.label = 20;
+                    case 20: return [4 /*yield*/, records_2.next()];
+                    case 21:
+                        if (!(records_2_1 = _e.sent(), !records_2_1.done)) return [3 /*break*/, 26];
                         recordItem = records_2_1.value;
-                        if (!recordItem[targetName]) return [3 /*break*/, 21];
-                        return [4 /*yield*/, this._get(store, recordItem[targetName])];
-                    case 20:
+                        if (!(targetNames === null || targetNames === void 0 ? void 0 : targetNames.length)) return [3 /*break*/, 23];
+                        allPresent = void 0;
+                        // iterate through all targetnames to make sure they are all present in the recordItem
+                        allPresent = targetNames.every(function (targetName) {
+                            return recordItem[targetName] != null;
+                        });
+                        // If not present, there is not yet a connected record
+                        if (!allPresent) {
+                            return [3 /*break*/, 26];
+                        }
+                        keys = targetNames.map(function (targetName) { return recordItem[targetName]; });
+                        return [4 /*yield*/, this._get(store, keys)];
+                    case 22:
                         connectionRecord = _e.sent();
                         recordItem[fieldName] =
                             connectionRecord &&
                                 this.modelInstanceCreator(modelConstructor_1, connectionRecord);
-                        delete recordItem[targetName];
-                        _e.label = 21;
-                    case 21: return [3 /*break*/, 18];
-                    case 22: return [3 /*break*/, 29];
+                        targetNames === null || targetNames === void 0 ? void 0 : targetNames.map(function (targetName) {
+                            delete recordItem[targetName];
+                        });
+                        return [3 /*break*/, 25];
                     case 23:
-                        e_5_1 = _e.sent();
-                        e_5 = { error: e_5_1 };
-                        return [3 /*break*/, 29];
+                        if (!recordItem[targetName]) return [3 /*break*/, 25];
+                        key = [recordItem[targetName]];
+                        return [4 /*yield*/, this._get(store, key)];
                     case 24:
-                        _e.trys.push([24, , 27, 28]);
-                        if (!(records_2_1 && !records_2_1.done && (_c = records_2.return))) return [3 /*break*/, 26];
+                        connectionRecord = _e.sent();
+                        recordItem[fieldName] =
+                            connectionRecord &&
+                                this.modelInstanceCreator(modelConstructor_1, connectionRecord);
+                        delete recordItem[targetName];
+                        _e.label = 25;
+                    case 25: return [3 /*break*/, 20];
+                    case 26: return [3 /*break*/, 33];
+                    case 27:
+                        e_6_1 = _e.sent();
+                        e_6 = { error: e_6_1 };
+                        return [3 /*break*/, 33];
+                    case 28:
+                        _e.trys.push([28, , 31, 32]);
+                        if (!(records_2_1 && !records_2_1.done && (_c = records_2.return))) return [3 /*break*/, 30];
                         return [4 /*yield*/, _c.call(records_2)];
-                    case 25:
+                    case 29:
                         _e.sent();
-                        _e.label = 26;
-                    case 26: return [3 /*break*/, 28];
-                    case 27:
-                        if (e_5) throw e_5.error;
+                        _e.label = 30;
+                    case 30: return [3 /*break*/, 32];
+                    case 31:
+                        if (e_6) throw e_6.error;
                         return [7 /*endfinally*/];
-                    case 28: return [7 /*endfinally*/];
-                    case 29: return [3 /*break*/, 32];
-                    case 30: 
+                    case 32: return [7 /*endfinally*/];
+                    case 33: return [3 /*break*/, 36];
+                    case 34: 
                     // TODO: Lazy loading
-                    return [3 /*break*/, 32];
-                    case 31:
-                        util_1.exhaustiveCheck(relation.relationType);
-                        return [3 /*break*/, 32];
-                    case 32: return [3 /*break*/, 2];
-                    case 33: return [3 /*break*/, 40];
-                    case 34:
-                        e_3_1 = _e.sent();
-                        e_3 = { error: e_3_1 };
-                        return [3 /*break*/, 40];
+                    return [3 /*break*/, 36];
                     case 35:
-                        _e.trys.push([35, , 38, 39]);
-                        if (!(relations_1_1 && !relations_1_1.done && (_a = relations_1.return))) return [3 /*break*/, 37];
+                        util_1.exhaustiveCheck(relation.relationType);
+                        return [3 /*break*/, 36];
+                    case 36: return [3 /*break*/, 2];
+                    case 37: return [3 /*break*/, 44];
+                    case 38:
+                        e_4_1 = _e.sent();
+                        e_4 = { error: e_4_1 };
+                        return [3 /*break*/, 44];
+                    case 39:
+                        _e.trys.push([39, , 42, 43]);
+                        if (!(relations_1_1 && !relations_1_1.done && (_a = relations_1.return))) return [3 /*break*/, 41];
                         return [4 /*yield*/, _a.call(relations_1)];
-                    case 36:
+                    case 40:
                         _e.sent();
-                        _e.label = 37;
-                    case 37: return [3 /*break*/, 39];
-                    case 38:
-                        if (e_3) throw e_3.error;
+                        _e.label = 41;
+                    case 41: return [3 /*break*/, 43];
+                    case 42:
+                        if (e_4) throw e_4.error;
                         return [7 /*endfinally*/];
-                    case 39: return [7 /*endfinally*/];
-                    case 40: return [2 /*return*/, records.map(function (record) {
+                    case 43: return [7 /*endfinally*/];
+                    case 44: return [2 /*return*/, records.map(function (record) {
                             return _this.modelInstanceCreator(modelConstructor, record);
                         })];
                 }
@@ -520,7 +644,7 @@ var IndexedDBAdapter = /** @class */ (function () {
     };
     IndexedDBAdapter.prototype.query = function (modelConstructor, predicate, pagination) {
         return __awaiter(this, void 0, void 0, function () {
-            var storeName, namespaceName, predicates, queryById, hasSort, hasPagination, records;
+            var storeName, namespaceName, predicates, keyPath, queryByKey, hasSort, hasPagination, records;
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
@@ -530,7 +654,8 @@ var IndexedDBAdapter = /** @class */ (function () {
                         storeName = this.getStorenameForModel(modelConstructor);
                         namespaceName = this.namespaceResolver(modelConstructor);
                         predicates = predicate && predicates_1.ModelPredicateCreator.getPredicates(predicate);
-                        queryById = predicates && this.idFromPredicate(predicates);
+                        keyPath = this.getIndexKeyPath(namespaceName, modelConstructor.name);
+                        queryByKey = predicates && this.keyValueFromPredicate(predicates, keyPath);
                         hasSort = pagination && pagination.sort;
                         hasPagination = pagination && pagination.limit;
                         return [4 /*yield*/, (function () { return __awaiter(_this, void 0, void 0, function () {
@@ -538,8 +663,8 @@ var IndexedDBAdapter = /** @class */ (function () {
                                 return __generator(this, function (_a) {
                                     switch (_a.label) {
                                         case 0:
-                                            if (!queryById) return [3 /*break*/, 2];
-                                            return [4 /*yield*/, this.getById(storeName, queryById)];
+                                            if (!queryByKey) return [3 /*break*/, 2];
+                                            return [4 /*yield*/, this.getByKey(storeName, queryByKey)];
                                         case 1:
                                             record = _a.sent();
                                             return [2 /*return*/, record ? [record] : []];
@@ -571,12 +696,12 @@ var IndexedDBAdapter = /** @class */ (function () {
             });
         });
     };
-    IndexedDBAdapter.prototype.getById = function (storeName, id) {
+    IndexedDBAdapter.prototype.getByKey = function (storeName, keyValue) {
         return __awaiter(this, void 0, void 0, function () {
             var record;
             return __generator(this, function (_a) {
                 switch (_a.label) {
-                    case 0: return [4 /*yield*/, this._get(storeName, id)];
+                    case 0: return [4 /*yield*/, this._get(storeName, keyValue)];
                     case 1:
                         record = _a.sent();
                         return [2 /*return*/, record];
@@ -594,11 +719,31 @@ var IndexedDBAdapter = /** @class */ (function () {
             });
         });
     };
-    IndexedDBAdapter.prototype.idFromPredicate = function (predicates) {
+    IndexedDBAdapter.prototype.keyValueFromPredicate = function (predicates, keyPath) {
+        var e_7, _a;
         var predicateObjs = predicates.predicates;
-        var idPredicate = predicateObjs.length === 1 &&
-            predicateObjs.find(function (p) { return types_1.isPredicateObj(p) && p.field === 'id' && p.operator === 'eq'; });
-        return idPredicate && idPredicate.operand;
+        if (predicateObjs.length !== keyPath.length) {
+            return;
+        }
+        var keyValues = [];
+        var _loop_1 = function (key) {
+            var predicateObj = predicateObjs.find(function (p) { return types_1.isPredicateObj(p) && p.field === key && p.operator === 'eq'; });
+            predicateObj && keyValues.push(predicateObj.operand);
+        };
+        try {
+            for (var keyPath_2 = __values(keyPath), keyPath_2_1 = keyPath_2.next(); !keyPath_2_1.done; keyPath_2_1 = keyPath_2.next()) {
+                var key = keyPath_2_1.value;
+                _loop_1(key);
+            }
+        }
+        catch (e_7_1) { e_7 = { error: e_7_1 }; }
+        finally {
+            try {
+                if (keyPath_2_1 && !keyPath_2_1.done && (_a = keyPath_2.return)) _a.call(keyPath_2);
+            }
+            finally { if (e_7) throw e_7.error; }
+        }
+        return keyValues.length === keyPath.length ? keyValues : undefined;
     };
     IndexedDBAdapter.prototype.filterOnPredicate = function (storeName, predicates) {
         return __awaiter(this, void 0, void 0, function () {
@@ -704,7 +849,7 @@ var IndexedDBAdapter = /** @class */ (function () {
     };
     IndexedDBAdapter.prototype.delete = function (modelOrModelConstructor, condition) {
         return __awaiter(this, void 0, void 0, function () {
-            var deleteQueue, modelConstructor, nameSpace, storeName, models, relations, deletedModels, deletedModels, model, modelConstructor, nameSpace, storeName, tx, store, fromDB, msg, predicates, predicateObjs, type, isValid, msg, relations, relations, deletedModels;
+            var deleteQueue, modelConstructor, nameSpace, storeName, models, relations, deletedModels, deletedModels, model, modelConstructor, namespaceName, storeName, tx, store, keyValues, fromDB, msg, predicates, predicateObjs, type, isValid, msg, relations, relations, deletedModels;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0: return [4 /*yield*/, this.checkPrivate()];
@@ -753,12 +898,13 @@ var IndexedDBAdapter = /** @class */ (function () {
                         model = modelOrModelConstructor;
                         modelConstructor = Object.getPrototypeOf(model)
                             .constructor;
-                        nameSpace = this.namespaceResolver(modelConstructor);
+                        namespaceName = this.namespaceResolver(modelConstructor);
                         storeName = this.getStorenameForModel(modelConstructor);
                         if (!condition) return [3 /*break*/, 13];
                         tx = this.db.transaction([storeName], 'readwrite');
                         store = tx.objectStore(storeName);
-                        return [4 /*yield*/, this._get(store, model.id)];
+                        keyValues = this.getIndexKeyValues(model);
+                        return [4 /*yield*/, this._get(store, keyValues)];
                     case 10:
                         fromDB = _a.sent();
                         if (fromDB === undefined) {
@@ -777,16 +923,14 @@ var IndexedDBAdapter = /** @class */ (function () {
                         return [4 /*yield*/, tx.done];
                     case 11:
                         _a.sent();
-                        relations = this.schema.namespaces[nameSpace].relationships[modelConstructor.name]
-                            .relationTypes;
-                        return [4 /*yield*/, this.deleteTraverse(relations, [model], modelConstructor.name, nameSpace, deleteQueue)];
+                        relations = this.schema.namespaces[namespaceName].relationships[modelConstructor.name].relationTypes;
+                        return [4 /*yield*/, this.deleteTraverse(relations, [model], modelConstructor.name, namespaceName, deleteQueue)];
                     case 12:
                         _a.sent();
                         return [3 /*break*/, 15];
                     case 13:
-                        relations = this.schema.namespaces[nameSpace].relationships[modelConstructor.name]
-                            .relationTypes;
-                        return [4 /*yield*/, this.deleteTraverse(relations, [model], modelConstructor.name, nameSpace, deleteQueue)];
+                        relations = this.schema.namespaces[namespaceName].relationships[modelConstructor.name].relationTypes;
+                        return [4 /*yield*/, this.deleteTraverse(relations, [model], modelConstructor.name, namespaceName, deleteQueue)];
                     case 14:
                         _a.sent();
                         _a.label = 15;
@@ -805,9 +949,9 @@ var IndexedDBAdapter = /** @class */ (function () {
     };
     IndexedDBAdapter.prototype.deleteItem = function (deleteQueue) {
         var deleteQueue_1, deleteQueue_1_1;
-        var e_6, _a, e_7, _b;
+        var e_8, _a, e_9, _b;
         return __awaiter(this, void 0, void 0, function () {
-            var connectionStoreNames, tx, deleteItem, storeName, items, store, items_1, items_1_1, item, key, e_7_1, e_6_1;
+            var connectionStoreNames, tx, deleteItem, storeName, items, store, items_1, items_1_1, item, key, keyValues, itemKey, e_9_1, e_8_1;
             return __generator(this, function (_c) {
                 switch (_c.label) {
                     case 0:
@@ -839,11 +983,14 @@ var IndexedDBAdapter = /** @class */ (function () {
                         if (!item) return [3 /*break*/, 12];
                         key = void 0;
                         if (!(typeof item === 'object')) return [3 /*break*/, 8];
-                        return [4 /*yield*/, store.index('byId').getKey(item['id'])];
+                        keyValues = this.getIndexKeyValues(item);
+                        return [4 /*yield*/, store.index('byPk').getKey(keyValues)];
                     case 7:
                         key = _c.sent();
                         return [3 /*break*/, 10];
-                    case 8: return [4 /*yield*/, store.index('byId').getKey(item.toString())];
+                    case 8:
+                        itemKey = [item.toString()];
+                        return [4 /*yield*/, store.index('byPk').getKey([itemKey])];
                     case 9:
                         key = _c.sent();
                         _c.label = 10;
@@ -856,8 +1003,8 @@ var IndexedDBAdapter = /** @class */ (function () {
                     case 12: return [3 /*break*/, 5];
                     case 13: return [3 /*break*/, 20];
                     case 14:
-                        e_7_1 = _c.sent();
-                        e_7 = { error: e_7_1 };
+                        e_9_1 = _c.sent();
+                        e_9 = { error: e_9_1 };
                         return [3 /*break*/, 20];
                     case 15:
                         _c.trys.push([15, , 18, 19]);
@@ -868,14 +1015,14 @@ var IndexedDBAdapter = /** @class */ (function () {
                         _c.label = 17;
                     case 17: return [3 /*break*/, 19];
                     case 18:
-                        if (e_7) throw e_7.error;
+                        if (e_9) throw e_9.error;
                         return [7 /*endfinally*/];
                     case 19: return [7 /*endfinally*/];
                     case 20: return [3 /*break*/, 2];
                     case 21: return [3 /*break*/, 28];
                     case 22:
-                        e_6_1 = _c.sent();
-                        e_6 = { error: e_6_1 };
+                        e_8_1 = _c.sent();
+                        e_8 = { error: e_8_1 };
                         return [3 /*break*/, 28];
                     case 23:
                         _c.trys.push([23, , 26, 27]);
@@ -886,7 +1033,7 @@ var IndexedDBAdapter = /** @class */ (function () {
                         _c.label = 25;
                     case 25: return [3 /*break*/, 27];
                     case 26:
-                        if (e_6) throw e_6.error;
+                        if (e_8) throw e_8.error;
                         return [7 /*endfinally*/];
                     case 27: return [7 /*endfinally*/];
                     case 28: return [2 /*return*/];
@@ -896,21 +1043,21 @@ var IndexedDBAdapter = /** @class */ (function () {
     };
     IndexedDBAdapter.prototype.deleteTraverse = function (relations, models, srcModel, nameSpace, deleteQueue) {
         var relations_2, relations_2_1, models_1, models_1_1, models_2, models_2_1;
-        var e_8, _a, e_9, _b, e_10, _c;
+        var e_10, _a, e_11, _b, e_12, _c;
         return __awaiter(this, void 0, void 0, function () {
-            var rel, relationType, fieldName, modelName, targetName, storeName, index, _d, model, hasOneIndex, hasOneCustomField, value, recordToDelete, e_9_1, model, childrenArray, e_10_1, e_8_1;
+            var rel, relationType, fieldName, modelName, targetName, targetNames, storeName, index, _d, model, hasOneIndex, hasConnectedModelFields, keyValues, values, recordToDelete, hasOneIndex_1, hasOneCustomField, keyValues, value, recordToDelete, e_11_1, model, keyValues, hasManyIndex, childrenArray, e_12_1, e_10_1;
             var _this = this;
             return __generator(this, function (_e) {
                 switch (_e.label) {
                     case 0:
-                        _e.trys.push([0, 35, 36, 41]);
+                        _e.trys.push([0, 38, 39, 44]);
                         relations_2 = __asyncValues(relations);
                         _e.label = 1;
                     case 1: return [4 /*yield*/, relations_2.next()];
                     case 2:
-                        if (!(relations_2_1 = _e.sent(), !relations_2_1.done)) return [3 /*break*/, 34];
+                        if (!(relations_2_1 = _e.sent(), !relations_2_1.done)) return [3 /*break*/, 37];
                         rel = relations_2_1.value;
-                        relationType = rel.relationType, fieldName = rel.fieldName, modelName = rel.modelName, targetName = rel.targetName;
+                        relationType = rel.relationType, fieldName = rel.fieldName, modelName = rel.modelName, targetName = rel.targetName, targetNames = rel.targetNames;
                         storeName = this.getStorename(nameSpace, modelName);
                         index = util_1.getIndex(this.schema.namespaces[nameSpace].relationships[modelName]
                             .relationTypes, srcModel) ||
@@ -921,118 +1068,157 @@ var IndexedDBAdapter = /** @class */ (function () {
                         _d = relationType;
                         switch (_d) {
                             case 'HAS_ONE': return [3 /*break*/, 3];
-                            case 'HAS_MANY': return [3 /*break*/, 17];
-                            case 'BELONGS_TO': return [3 /*break*/, 31];
+                            case 'HAS_MANY': return [3 /*break*/, 20];
+                            case 'BELONGS_TO': return [3 /*break*/, 34];
                         }
-                        return [3 /*break*/, 32];
+                        return [3 /*break*/, 35];
                     case 3:
-                        _e.trys.push([3, 10, 11, 16]);
+                        _e.trys.push([3, 13, 14, 19]);
                         models_1 = __asyncValues(models);
                         _e.label = 4;
                     case 4: return [4 /*yield*/, models_1.next()];
                     case 5:
-                        if (!(models_1_1 = _e.sent(), !models_1_1.done)) return [3 /*break*/, 9];
+                        if (!(models_1_1 = _e.sent(), !models_1_1.done)) return [3 /*break*/, 12];
                         model = models_1_1.value;
-                        hasOneIndex = index || 'byId';
-                        hasOneCustomField = targetName in model;
-                        value = hasOneCustomField ? model[targetName] : model.id;
-                        if (!value)
-                            return [3 /*break*/, 9];
+                        hasOneIndex = index || 'byPk';
+                        if (!(targetNames && targetNames.length > 0)) return [3 /*break*/, 8];
+                        hasConnectedModelFields = targetNames.every(function (targetName) {
+                            return model.hasOwnProperty(targetName);
+                        });
+                        keyValues = this.getIndexKeyValues(model);
+                        values = [];
+                        if (hasConnectedModelFields) {
+                            // Values will be that of the child model
+                            values = targetNames.map(function (targetName) { return model[targetName]; });
+                        }
+                        else {
+                            // values will be that of the parent model
+                            values = keyValues;
+                        }
+                        if (values.length === 0)
+                            return [3 /*break*/, 12];
                         return [4 /*yield*/, this.db
                                 .transaction(storeName, 'readwrite')
                                 .objectStore(storeName)
                                 .index(hasOneIndex)
-                                .get(value)];
+                                .get(values)];
                     case 6:
-                        recordToDelete = (_e.sent());
+                        recordToDelete = _e.sent();
                         return [4 /*yield*/, this.deleteTraverse(this.schema.namespaces[nameSpace].relationships[modelName]
                                 .relationTypes, recordToDelete ? [recordToDelete] : [], modelName, nameSpace, deleteQueue)];
                     case 7:
                         _e.sent();
-                        _e.label = 8;
-                    case 8: return [3 /*break*/, 4];
-                    case 9: return [3 /*break*/, 16];
+                        return [3 /*break*/, 12];
+                    case 8:
+                        hasOneIndex_1 = index || 'byPk';
+                        hasOneCustomField = targetName in model;
+                        keyValues = this.getIndexKeyValues(model);
+                        value = hasOneCustomField ? model[targetName] : keyValues[0];
+                        if (hasOneIndex_1 === 'byPk') {
+                            // byPk requires an array keyValue
+                            value = [value];
+                        }
+                        if (!value)
+                            return [3 /*break*/, 12];
+                        return [4 /*yield*/, this.db
+                                .transaction(storeName, 'readwrite')
+                                .objectStore(storeName)
+                                .index(hasOneIndex_1)
+                                .get(value)];
+                    case 9:
+                        recordToDelete = _e.sent();
+                        return [4 /*yield*/, this.deleteTraverse(this.schema.namespaces[nameSpace].relationships[modelName]
+                                .relationTypes, recordToDelete ? [recordToDelete] : [], modelName, nameSpace, deleteQueue)];
                     case 10:
-                        e_9_1 = _e.sent();
-                        e_9 = { error: e_9_1 };
-                        return [3 /*break*/, 16];
-                    case 11:
-                        _e.trys.push([11, , 14, 15]);
-                        if (!(models_1_1 && !models_1_1.done && (_b = models_1.return))) return [3 /*break*/, 13];
-                        return [4 /*yield*/, _b.call(models_1)];
-                    case 12:
                         _e.sent();
-                        _e.label = 13;
-                    case 13: return [3 /*break*/, 15];
+                        _e.label = 11;
+                    case 11: return [3 /*break*/, 4];
+                    case 12: return [3 /*break*/, 19];
+                    case 13:
+                        e_11_1 = _e.sent();
+                        e_11 = { error: e_11_1 };
+                        return [3 /*break*/, 19];
                     case 14:
-                        if (e_9) throw e_9.error;
-                        return [7 /*endfinally*/];
-                    case 15: return [7 /*endfinally*/];
-                    case 16: return [3 /*break*/, 33];
+                        _e.trys.push([14, , 17, 18]);
+                        if (!(models_1_1 && !models_1_1.done && (_b = models_1.return))) return [3 /*break*/, 16];
+                        return [4 /*yield*/, _b.call(models_1)];
+                    case 15:
+                        _e.sent();
+                        _e.label = 16;
+                    case 16: return [3 /*break*/, 18];
                     case 17:
-                        _e.trys.push([17, 24, 25, 30]);
+                        if (e_11) throw e_11.error;
+                        return [7 /*endfinally*/];
+                    case 18: return [7 /*endfinally*/];
+                    case 19: return [3 /*break*/, 36];
+                    case 20:
+                        _e.trys.push([20, 27, 28, 33]);
                         models_2 = __asyncValues(models);
-                        _e.label = 18;
-                    case 18: return [4 /*yield*/, models_2.next()];
-                    case 19:
-                        if (!(models_2_1 = _e.sent(), !models_2_1.done)) return [3 /*break*/, 23];
+                        _e.label = 21;
+                    case 21: return [4 /*yield*/, models_2.next()];
+                    case 22:
+                        if (!(models_2_1 = _e.sent(), !models_2_1.done)) return [3 /*break*/, 26];
                         model = models_2_1.value;
+                        keyValues = this.getIndexKeyValues(model);
+                        hasManyIndex = index || 'byPk';
                         return [4 /*yield*/, this.db
                                 .transaction(storeName, 'readwrite')
                                 .objectStore(storeName)
-                                .index(index)
-                                .getAll(model['id'])];
-                    case 20:
+                                .index(hasManyIndex)
+                                .getAll(keyValues[0])];
+                    case 23:
                         childrenArray = _e.sent();
+                        // .getAll(keyValues); // TODO
                         return [4 /*yield*/, this.deleteTraverse(this.schema.namespaces[nameSpace].relationships[modelName]
                                 .relationTypes, childrenArray, modelName, nameSpace, deleteQueue)];
-                    case 21:
-                        _e.sent();
-                        _e.label = 22;
-                    case 22: return [3 /*break*/, 18];
-                    case 23: return [3 /*break*/, 30];
                     case 24:
-                        e_10_1 = _e.sent();
-                        e_10 = { error: e_10_1 };
-                        return [3 /*break*/, 30];
-                    case 25:
-                        _e.trys.push([25, , 28, 29]);
-                        if (!(models_2_1 && !models_2_1.done && (_c = models_2.return))) return [3 /*break*/, 27];
-                        return [4 /*yield*/, _c.call(models_2)];
-                    case 26:
+                        // .getAll(keyValues); // TODO
                         _e.sent();
-                        _e.label = 27;
-                    case 27: return [3 /*break*/, 29];
+                        _e.label = 25;
+                    case 25: return [3 /*break*/, 21];
+                    case 26: return [3 /*break*/, 33];
+                    case 27:
+                        e_12_1 = _e.sent();
+                        e_12 = { error: e_12_1 };
+                        return [3 /*break*/, 33];
                     case 28:
-                        if (e_10) throw e_10.error;
+                        _e.trys.push([28, , 31, 32]);
+                        if (!(models_2_1 && !models_2_1.done && (_c = models_2.return))) return [3 /*break*/, 30];
+                        return [4 /*yield*/, _c.call(models_2)];
+                    case 29:
+                        _e.sent();
+                        _e.label = 30;
+                    case 30: return [3 /*break*/, 32];
+                    case 31:
+                        if (e_12) throw e_12.error;
                         return [7 /*endfinally*/];
-                    case 29: return [7 /*endfinally*/];
-                    case 30: return [3 /*break*/, 33];
-                    case 31: 
+                    case 32: return [7 /*endfinally*/];
+                    case 33: return [3 /*break*/, 36];
+                    case 34: 
                     // Intentionally blank
-                    return [3 /*break*/, 33];
-                    case 32:
-                        util_1.exhaustiveCheck(relationType);
-                        return [3 /*break*/, 33];
-                    case 33: return [3 /*break*/, 1];
-                    case 34: return [3 /*break*/, 41];
+                    return [3 /*break*/, 36];
                     case 35:
-                        e_8_1 = _e.sent();
-                        e_8 = { error: e_8_1 };
-                        return [3 /*break*/, 41];
-                    case 36:
-                        _e.trys.push([36, , 39, 40]);
-                        if (!(relations_2_1 && !relations_2_1.done && (_a = relations_2.return))) return [3 /*break*/, 38];
+                        util_1.exhaustiveCheck(relationType);
+                        return [3 /*break*/, 36];
+                    case 36: return [3 /*break*/, 1];
+                    case 37: return [3 /*break*/, 44];
+                    case 38:
+                        e_10_1 = _e.sent();
+                        e_10 = { error: e_10_1 };
+                        return [3 /*break*/, 44];
+                    case 39:
+                        _e.trys.push([39, , 42, 43]);
+                        if (!(relations_2_1 && !relations_2_1.done && (_a = relations_2.return))) return [3 /*break*/, 41];
                         return [4 /*yield*/, _a.call(relations_2)];
-                    case 37:
+                    case 40:
                         _e.sent();
-                        _e.label = 38;
-                    case 38: return [3 /*break*/, 40];
-                    case 39:
-                        if (e_8) throw e_8.error;
+                        _e.label = 41;
+                    case 41: return [3 /*break*/, 43];
+                    case 42:
+                        if (e_10) throw e_10.error;
                         return [7 /*endfinally*/];
-                    case 40: return [7 /*endfinally*/];
-                    case 41:
+                    case 43: return [7 /*endfinally*/];
+                    case 44:
                         deleteQueue.push({
                             storeName: this.getStorename(nameSpace, srcModel),
                             items: models.map(function (record) {
@@ -1064,8 +1250,9 @@ var IndexedDBAdapter = /** @class */ (function () {
     };
     IndexedDBAdapter.prototype.batchSave = function (modelConstructor, items) {
         return __awaiter(this, void 0, void 0, function () {
-            var result, storeName, txn, store, _loop_1, this_1, items_2, items_2_1, item, e_11_1;
-            var e_11, _a;
+            var result, storeName, txn, store, _loop_2, this_1, items_2, items_2_1, item, e_13_1;
+            var e_13, _a;
+            var _this = this;
             return __generator(this, function (_b) {
                 switch (_b.label) {
                     case 0:
@@ -1079,21 +1266,26 @@ var IndexedDBAdapter = /** @class */ (function () {
                         storeName = this.getStorenameForModel(modelConstructor);
                         txn = this.db.transaction(storeName, 'readwrite');
                         store = txn.store;
-                        _loop_1 = function (item) {
-                            var connectedModels, id, _deleted, index, key, instance;
+                        _loop_2 = function (item) {
+                            var namespaceName, modelName, model, connectedModels, keyValues, _deleted, index, key, instance;
                             return __generator(this, function (_a) {
                                 switch (_a.label) {
                                     case 0:
-                                        connectedModels = util_1.traverseModel(modelConstructor.name, this_1.modelInstanceCreator(modelConstructor, item), this_1.schema.namespaces[this_1.namespaceResolver(modelConstructor)], this_1.modelInstanceCreator, this_1.getModelConstructorByModelName);
-                                        id = item.id, _deleted = item._deleted;
-                                        index = store.index('byId');
-                                        return [4 /*yield*/, index.getKey(id)];
+                                        namespaceName = this_1.namespaceResolver(modelConstructor);
+                                        modelName = modelConstructor.name;
+                                        model = this_1.modelInstanceCreator(modelConstructor, item);
+                                        connectedModels = util_1.traverseModel(modelName, model, this_1.schema.namespaces[namespaceName], this_1.modelInstanceCreator, this_1.getModelConstructorByModelName);
+                                        keyValues = this_1.getIndexKeyValues(model);
+                                        _deleted = item._deleted;
+                                        index = store.index('byPk');
+                                        return [4 /*yield*/, index.getKey(keyValues)];
                                     case 1:
                                         key = _a.sent();
                                         if (!!_deleted) return [3 /*break*/, 3];
                                         instance = connectedModels.find(function (_a) {
                                             var instance = _a.instance;
-                                            return instance.id === id;
+                                            var instanceKeyValues = _this.getIndexKeyValues(instance);
+                                            return _this.keysEqual(instanceKeyValues, keyValues);
                                         }).instance;
                                         result.push([
                                             instance,
@@ -1123,7 +1315,7 @@ var IndexedDBAdapter = /** @class */ (function () {
                     case 3:
                         if (!!items_2_1.done) return [3 /*break*/, 6];
                         item = items_2_1.value;
-                        return [5 /*yield**/, _loop_1(item)];
+                        return [5 /*yield**/, _loop_2(item)];
                     case 4:
                         _b.sent();
                         _b.label = 5;
@@ -1132,14 +1324,14 @@ var IndexedDBAdapter = /** @class */ (function () {
                         return [3 /*break*/, 3];
                     case 6: return [3 /*break*/, 9];
                     case 7:
-                        e_11_1 = _b.sent();
-                        e_11 = { error: e_11_1 };
+                        e_13_1 = _b.sent();
+                        e_13 = { error: e_13_1 };
                         return [3 /*break*/, 9];
                     case 8:
                         try {
                             if (items_2_1 && !items_2_1.done && (_a = items_2.return)) _a.call(items_2);
                         }
-                        finally { if (e_11) throw e_11.error; }
+                        finally { if (e_13) throw e_13.error; }
                         return [7 /*endfinally*/];
                     case 9: return [4 /*yield*/, txn.done];
                     case 10:
@@ -1149,6 +1341,21 @@ var IndexedDBAdapter = /** @class */ (function () {
             });
         });
     };
+    IndexedDBAdapter.prototype.createObjectStoreForModel = function (db, namespaceName, storeName, modelName) {
+        return __awaiter(this, void 0, void 0, function () {
+            var store, indexes, keyPath;
+            return __generator(this, function (_a) {
+                store = db.createObjectStore(storeName, {
+                    autoIncrement: true,
+                });
+                indexes = this.schema.namespaces[namespaceName].relationships[modelName].indexes;
+                indexes.forEach(function (index) { return store.createIndex(index, index); });
+                keyPath = this.getIndexKeyPath(namespaceName, modelName);
+                store.createIndex('byPk', keyPath, { unique: true });
+                return [2 /*return*/];
+            });
+        });
+    };
     return IndexedDBAdapter;
 }());
 exports.default = new IndexedDBAdapter();