@bunnykit/orm 0.1.18 → 0.1.19

package/dist/src/index.d.ts

@@ -20,6 +20,7 @@ export { ObserverRegistry, type ObserverContract } from "./model/Observer.js";
 export { MorphMap } from "./model/MorphMap.js";
 export { MorphTo, MorphOne, MorphMany, MorphToMany } from "./model/MorphRelations.js";
 export { BelongsToMany } from "./model/BelongsToMany.js";
+export { IdentityMap } from "./model/IdentityMap.js";
 export { Migration } from "./migration/Migration.js";
 export { Migrator } from "./migration/Migrator.js";
 export type { MigrationEvent, MigrationEventListener, MigrationEventPayload } from "./migration/Migrator.js";

package/dist/src/index.js

@@ -15,6 +15,7 @@ export { ObserverRegistry } from "./model/Observer.js";
 export { MorphMap } from "./model/MorphMap.js";
 export { MorphTo, MorphOne, MorphMany, MorphToMany } from "./model/MorphRelations.js";
 export { BelongsToMany } from "./model/BelongsToMany.js";
+export { IdentityMap } from "./model/IdentityMap.js";
 export { Migration } from "./migration/Migration.js";
 export { Migrator } from "./migration/Migrator.js";
 export { MigrationCreator } from "./migration/MigrationCreator.js";

package/dist/src/model/IdentityMap.d.ts

@@ -0,0 +1,8 @@
+import type { Model } from "./Model.js";
+export declare class IdentityMap {
+    static current(): Map<string, Model> | undefined;
+    static run<T>(callback: () => T | Promise<T>): Promise<T>;
+    static get(table: string, key: string | number): Model | undefined;
+    static set(table: string, key: string | number, model: Model): void;
+    static clear(): void;
+}

package/dist/src/model/IdentityMap.js

@@ -0,0 +1,28 @@
+import { AsyncLocalStorage } from "node:async_hooks";
+const storage = new AsyncLocalStorage();
+export class IdentityMap {
+    static current() {
+        return storage.getStore();
+    }
+    static async run(callback) {
+        return await storage.run(new Map(), callback);
+    }
+    static get(table, key) {
+        const map = this.current();
+        if (!map)
+            return undefined;
+        return map.get(`${table}:${String(key)}`);
+    }
+    static set(table, key, model) {
+        const map = this.current();
+        if (!map)
+            return;
+        map.set(`${table}:${String(key)}`, model);
+    }
+    static clear() {
+        const map = this.current();
+        if (!map)
+            return;
+        map.clear();
+    }
+}

package/dist/src/model/Model.d.ts

@@ -115,6 +115,7 @@ export declare class Model<T extends Record<string, any> = Record<string, any>>
     static getTable(): string;
     static getConnection(): Connection;
     static setConnection(connection: Connection): void;
+    static useIdentityMap<T>(callback: () => T | Promise<T>): Promise<T>;
     static on<M extends ModelConstructor>(this: M, connection: string | Connection): Builder<InstanceType<M>>;
     static forTenant<M extends ModelConstructor>(this: M, tenantId: string): Builder<InstanceType<M>>;
     static query<M extends ModelConstructor>(this: M): Builder<InstanceType<M>>;

package/dist/src/model/Model.js

@@ -7,6 +7,7 @@ import { Schema } from "../schema/Schema.js";
 import { ModelNotFoundError } from "./ModelNotFoundError.js";
 import { ConnectionManager } from "../connection/ConnectionManager.js";
 import { TenantContext } from "../connection/TenantContext.js";
+import { IdentityMap } from "./IdentityMap.js";
 const globalScopes = new WeakMap();
 function getGlobalScopes(model) {
     const scopes = new Map();
@@ -353,6 +354,9 @@ export class Model {
         this.connection = connection;
         ConnectionManager.setDefault(connection);
     }
+    static useIdentityMap(callback) {
+        return IdentityMap.run(callback);
+    }
     static on(connection) {
         const resolved = typeof connection === "string" ? ConnectionManager.require(connection) : connection;
         const builder = new Builder(resolved, resolved.qualifyTable(this.getTable()));
@@ -813,6 +817,13 @@ export class Model {
             await ObserverRegistry.dispatch("created", this);
             await ObserverRegistry.dispatch("saved", this);
         }
+        const identityMap = IdentityMap.current();
+        if (identityMap) {
+            const pk = this.getAttribute(constructor.primaryKey);
+            if (pk !== null && pk !== undefined && pk !== "") {
+                IdentityMap.set(constructor.getTable(), pk, this);
+            }
+        }
         return this;
     }
     updateTimestamps() {

package/dist/src/query/Builder.js

@@ -1,4 +1,5 @@
 import { ModelNotFoundError } from "../model/ModelNotFoundError.js";
+import { IdentityMap } from "../model/IdentityMap.js";
 export class Builder {
     connection;
     tableName;
@@ -589,13 +590,31 @@ export class Builder {
         const results = await this.connection.query(sql, this.bindings);
         const rows = Array.from(results);
         if (this.model) {
+            const identityMap = IdentityMap.current();
+            const table = this.model.getTable();
+            const primaryKey = this.model.primaryKey || "id";
             const models = rows.map((row) => {
+                if (identityMap) {
+                    const pk = row[primaryKey];
+                    if (pk !== null && pk !== undefined) {
+                        const cached = IdentityMap.get(table, pk);
+                        if (cached) {
+                            return cached;
+                        }
+                    }
+                }
                 const instance = new this.model(row);
                 instance.$exists = true;
                 instance.$original = { ...row };
                 if (typeof instance.setConnection === "function") {
                     instance.setConnection(this.connection);
                 }
+                if (identityMap) {
+                    const pk = row[primaryKey];
+                    if (pk !== null && pk !== undefined) {
+                        IdentityMap.set(table, pk, instance);
+                    }
+                }
                 return instance;
             });
             if (this.eagerLoads.length > 0) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bunnykit/orm",
-  "version": "0.1.18",
+  "version": "0.1.19",
   "description": "An Eloquent-inspired ORM for Bun's native SQL client supporting SQLite, MySQL, and PostgreSQL.",
   "license": "MIT",
   "packageManager": "bun@1.3.12",