edinburgh 0.5.0 → 0.6.1

package/src/migrate.ts

@@ -1,9 +1,9 @@
 import * as lowlevel from "olmdb/lowlevel";
 import DataPack from "./datapack.js";
-import { modelRegistry, currentTxn, Transaction } from "./models.js";
+import { currentTxn, type Transaction, transact } from "./edinburgh.js";
+import { modelRegistry } from "./models.js";
 import { dbDel, toBuffer, bytesEqual } from "./utils.js";
 import { deserializeType, TypeWrapper } from "./types.js";
-import { transact } from "./edinburgh.js";
 
 const INDEX_ID_PREFIX = -2;
 
@@ -123,7 +123,7 @@ export async function runMigration(options: MigrationOptions = {}): Promise<Migr
         if (options.tables && !options.tables.includes(model.tableName)) continue;
         knownIndexIds.add(model._indexId!);
         modelByPkIndexId.set(model._indexId!, model);
-        for (const sec of model._secondaries || []) {
+        for (const sec of Object.values(model._secondaries || {})) {
             knownIndexIds.add(sec._indexId!);
         }
     }
@@ -157,7 +157,7 @@ export async function runMigration(options: MigrationOptions = {}): Promise<Migr
             let secondaryCount = 0;
             let rewrittenCount = 0;
             const migrateFn = (model as any).migrate as ((record: Record<string, any>) => void) | undefined;
-            const secondaries = model._secondaries || [];
+            const secondaries = Object.values(model._secondaries || {});
 
             await forEachRow(indexId, (txn, keyBuf, valueBuf) => {
                 const valuePack = new DataPack(valueBuf);
@@ -170,7 +170,7 @@ export async function runMigration(options: MigrationOptions = {}): Promise<Migr
                 const record: Record<string, any> = {};
                 const keyPack = new DataPack(keyBuf);
                 keyPack.readNumber(); // skip indexId
-                for (const [name, type] of model._pkFieldTypes.entries()) {
+                for (const [name, type] of model._indexFields.entries()) {
                     record[name] = type.deserialize(keyPack);
                 }
                 for (const [name, type] of versionInfo.nonKeyFields.entries()) {
@@ -189,14 +189,14 @@ export async function runMigration(options: MigrationOptions = {}): Promise<Migr
                             sec._write(txn, keyBuf, record as any);
                             secondaryCount++;
                         } else if (preMigrate) {
-                            if (sec._update(txn, keyBuf, record, preMigrate)) secondaryCount++;
+                            if (sec._update(txn, keyBuf, record as any, preMigrate)) secondaryCount++;
                         }
                     }
                 }
 
                 // Rewrite primary row data to current version
                 if (rewriteData) {
-                    model._writePrimary(txn, keyBuf, record);
+                    model._writePK(txn, keyBuf, record);
                     rewrittenCount++;
                 }
             });
@@ -224,7 +224,6 @@ export async function runMigration(options: MigrationOptions = {}): Promise<Migr
             const failures: Record<string, number> = {};
 
             await forEachRow(oldDef.id, (txn, keyBuf) => {
-                let instance;
                 try {
                     // Deserialize old key
                     const keyPack = new DataPack(keyBuf);
@@ -239,7 +238,7 @@ export async function runMigration(options: MigrationOptions = {}): Promise<Migr
                     if (migrateFn) migrateFn(record);
 
                     // _write validates, checks duplicates, writes primary + secondaries
-                    instance = new (model as any)(record, txn);
+                    let instance = new (model as any)(record, txn);
                     instance._write(txn);
                     dbDel(txn.id, keyBuf);
                     converted++;
@@ -250,7 +249,7 @@ export async function runMigration(options: MigrationOptions = {}): Promise<Migr
                         failures['error'] = (failures['error'] || 0) + 1;
                     }
                 } finally {
-                    if (instance) txn.instances.delete(instance);
+                    txn.instances.clear(); // We've already handled the _write() ourselves
                 }
             });