metal-orm 1.1.11 → 1.1.16
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +758 -769
- package/dist/index.cjs.map +1 -1
- package/dist/index.js.map +1 -1
- package/package.json +100 -96
- package/scripts/generate-entities/schema.mjs +122 -122
- package/scripts/generate-entities/tree-detection.mjs +92 -92
- package/scripts/inflection/pt-br.mjs +468 -468
- package/scripts/run-eslint.mjs +33 -34
- package/scripts/show-sql-seed.sql +102 -0
- package/scripts/show-sql.mjs +321 -325
- package/src/codegen/naming-strategy.ts +54 -54
- package/src/codegen/typescript.ts +482 -482
- package/src/core/ast/adapters.ts +1 -1
- package/src/core/ast/builders.ts +123 -123
- package/src/core/ast/expression-builders.ts +745 -745
- package/src/core/ast/expression-nodes.ts +304 -304
- package/src/core/ast/expression-visitor.ts +212 -212
- package/src/core/ast/expression.ts +20 -20
- package/src/core/ast/helpers.ts +14 -14
- package/src/core/ast/join.ts +18 -18
- package/src/core/ast/query.ts +200 -200
- package/src/core/ddl/dialects/base-schema-dialect.ts +1 -1
- package/src/core/ddl/dialects/mssql-schema-dialect.ts +1 -1
- package/src/core/ddl/dialects/mysql-schema-dialect.ts +1 -1
- package/src/core/ddl/dialects/postgres-schema-dialect.ts +1 -1
- package/src/core/ddl/dialects/render-reference.test.ts +70 -70
- package/src/core/ddl/dialects/sqlite-schema-dialect.ts +1 -1
- package/src/core/ddl/introspect/catalogs/index.ts +5 -5
- package/src/core/ddl/introspect/catalogs/postgres.ts +146 -146
- package/src/core/ddl/introspect/context.ts +15 -15
- package/src/core/ddl/introspect/functions/postgres.ts +35 -35
- package/src/core/ddl/introspect/registry.ts +40 -40
- package/src/core/ddl/introspect/run-select.ts +35 -35
- package/src/core/ddl/introspect/utils.ts +56 -56
- package/src/core/ddl/naming-strategy.ts +16 -16
- package/src/core/ddl/schema-dialect.ts +56 -56
- package/src/core/ddl/schema-diff.ts +234 -234
- package/src/core/ddl/schema-generator.ts +1 -1
- package/src/core/ddl/schema-introspect.ts +25 -25
- package/src/core/ddl/sql-writing.ts +171 -171
- package/src/core/dialect/abstract.ts +541 -541
- package/src/core/dialect/base/cte-compiler.ts +33 -33
- package/src/core/dialect/base/function-table-formatter.ts +103 -103
- package/src/core/dialect/base/groupby-compiler.ts +21 -21
- package/src/core/dialect/base/join-compiler.ts +25 -25
- package/src/core/dialect/base/orderby-compiler.ts +35 -35
- package/src/core/dialect/base/pagination-strategy.ts +32 -32
- package/src/core/dialect/base/returning-strategy.ts +57 -57
- package/src/core/dialect/base/sql-dialect.ts +294 -294
- package/src/core/dialect/dialect-factory.ts +91 -91
- package/src/core/dialect/mssql/functions.ts +120 -120
- package/src/core/dialect/mssql/index.ts +317 -317
- package/src/core/dialect/mysql/functions.ts +103 -103
- package/src/core/dialect/mysql/index.ts +105 -105
- package/src/core/dialect/postgres/functions.ts +108 -108
- package/src/core/dialect/postgres/index.ts +91 -91
- package/src/core/dialect/sqlite/functions.ts +129 -129
- package/src/core/dialect/sqlite/index.ts +67 -67
- package/src/core/driver/database-driver.ts +19 -19
- package/src/core/driver/mssql-driver.ts +23 -23
- package/src/core/driver/mysql-driver.ts +23 -23
- package/src/core/driver/postgres-driver.ts +23 -23
- package/src/core/driver/sqlite-driver.ts +23 -23
- package/src/core/execution/db-executor.ts +92 -92
- package/src/core/execution/executors/better-sqlite3-executor.ts +90 -90
- package/src/core/execution/executors/mssql-executor.ts +184 -184
- package/src/core/execution/executors/mysql-executor.ts +88 -88
- package/src/core/execution/executors/postgres-executor.ts +26 -26
- package/src/core/execution/executors/sqlite-executor.ts +42 -42
- package/src/core/execution/pooling/pool-types.ts +30 -30
- package/src/core/execution/pooling/pool.ts +275 -275
- package/src/core/functions/array.ts +35 -35
- package/src/core/functions/control-flow.ts +79 -79
- package/src/core/functions/datetime.ts +237 -237
- package/src/core/functions/json.ts +70 -70
- package/src/core/functions/numeric.ts +297 -297
- package/src/core/functions/text.ts +334 -334
- package/src/core/functions/types.ts +33 -33
- package/src/core/hydration/types.ts +51 -51
- package/src/core/sql/sql-operator-config.ts +39 -39
- package/src/core/sql/sql.ts +152 -152
- package/src/decorators/column-decorator.ts +87 -87
- package/src/decorators/decorator-metadata.ts +143 -143
- package/src/decorators/entity.ts +93 -93
- package/src/decorators/index.ts +39 -39
- package/src/decorators/relations.ts +231 -231
- package/src/decorators/transformers/built-in/string-transformers.ts +231 -231
- package/src/decorators/transformers/transformer-decorators.ts +134 -134
- package/src/decorators/transformers/transformer-executor.ts +171 -171
- package/src/decorators/transformers/transformer-metadata.ts +73 -73
- package/src/decorators/validators/built-in/br-cep-validator.ts +55 -55
- package/src/decorators/validators/built-in/br-cnpj-validator.ts +105 -105
- package/src/decorators/validators/built-in/br-cpf-validator.ts +103 -103
- package/src/decorators/validators/country-validator-registry.ts +64 -64
- package/src/decorators/validators/country-validators-decorators.ts +203 -203
- package/src/decorators/validators/country-validators.ts +105 -105
- package/src/decorators/validators/validator-metadata.ts +59 -59
- package/src/dto/dto-types.ts +226 -226
- package/src/dto/filter-types.ts +141 -141
- package/src/dto/index.ts +95 -95
- package/src/dto/openapi/generators/base.ts +29 -29
- package/src/dto/openapi/generators/column.ts +37 -37
- package/src/dto/openapi/generators/dto.ts +91 -91
- package/src/dto/openapi/generators/filter.ts +75 -75
- package/src/dto/openapi/generators/nested-dto.ts +618 -618
- package/src/dto/openapi/generators/pagination.ts +111 -111
- package/src/dto/openapi/generators/relation-filter.ts +228 -228
- package/src/dto/openapi/index.ts +12 -12
- package/src/dto/openapi/types.ts +101 -101
- package/src/dto/openapi/utilities.ts +95 -95
- package/src/dto/pagination-utils.ts +150 -150
- package/src/dto/transform.ts +197 -197
- package/src/index.ts +86 -86
- package/src/orm/als.ts +58 -58
- package/src/orm/domain-event-bus.ts +119 -119
- package/src/orm/entity-context.ts +86 -86
- package/src/orm/entity-materializer.ts +159 -159
- package/src/orm/entity-meta.ts +97 -97
- package/src/orm/entity-registry.ts +39 -39
- package/src/orm/entity-relation-cache.ts +39 -39
- package/src/orm/execute.ts +194 -194
- package/src/orm/execution-context.ts +18 -18
- package/src/orm/hydration-context.ts +26 -26
- package/src/orm/hydration.ts +112 -112
- package/src/orm/identity-map.ts +61 -61
- package/src/orm/interceptor-pipeline.ts +16 -16
- package/src/orm/lazy-batch/belongs-to-many.ts +119 -119
- package/src/orm/lazy-batch/belongs-to.ts +108 -108
- package/src/orm/lazy-batch/has-many.ts +69 -69
- package/src/orm/lazy-batch/has-one.ts +68 -68
- package/src/orm/lazy-batch/shared.ts +125 -125
- package/src/orm/orm-session.ts +620 -620
- package/src/orm/orm.ts +34 -34
- package/src/orm/pooled-executor-factory.ts +121 -121
- package/src/orm/query-logger.ts +40 -40
- package/src/orm/relation-change-processor.ts +420 -420
- package/src/orm/relations/belongs-to.ts +143 -143
- package/src/orm/relations/has-many.ts +204 -204
- package/src/orm/relations/has-one.ts +174 -174
- package/src/orm/relations/many-to-many.ts +288 -288
- package/src/orm/relations/morph-many.ts +156 -156
- package/src/orm/relations/morph-one.ts +151 -151
- package/src/orm/relations/morph-to.ts +162 -162
- package/src/orm/runtime-types.ts +92 -92
- package/src/orm/save-graph.ts +561 -561
- package/src/orm/transaction-runner.ts +24 -24
- package/src/orm/unit-of-work.ts +426 -426
- package/src/query/index.ts +69 -69
- package/src/query/target.ts +46 -46
- package/src/query-builder/column-selector.ts +81 -81
- package/src/query-builder/delete.ts +141 -141
- package/src/query-builder/insert-query-state.ts +149 -149
- package/src/query-builder/insert.ts +117 -117
- package/src/query-builder/query-ast-service.ts +287 -287
- package/src/query-builder/query-resolution.ts +78 -78
- package/src/query-builder/raw-column-parser.ts +38 -38
- package/src/query-builder/relation-filter-utils.ts +153 -153
- package/src/query-builder/relation-join-planner.ts +10 -10
- package/src/query-builder/relation-manager.ts +87 -87
- package/src/query-builder/relation-projection-helper.ts +102 -102
- package/src/query-builder/relation-types.ts +9 -9
- package/src/query-builder/relation-utils.ts +15 -15
- package/src/query-builder/select/cte-facet.ts +40 -40
- package/src/query-builder/select/from-facet.ts +80 -80
- package/src/query-builder/select/join-facet.ts +62 -62
- package/src/query-builder/select/predicate-facet.ts +104 -104
- package/src/query-builder/select/projection-facet.ts +70 -70
- package/src/query-builder/select/relation-facet.ts +81 -81
- package/src/query-builder/select/setop-facet.ts +36 -36
- package/src/query-builder/select-helpers.ts +1 -1
- package/src/query-builder/select-query-builder-deps.ts +130 -130
- package/src/query-builder/select-query-state.ts +207 -207
- package/src/query-builder/update.ts +151 -151
- package/src/schema/column-types.ts +322 -322
- package/src/schema/table-guards.ts +37 -37
- package/src/schema/table.ts +281 -281
- package/src/tree/index.ts +13 -13
- package/src/tree/nested-set-strategy.ts +545 -545
- package/src/tree/tree-decorator.ts +397 -397
- package/src/tree/tree-manager.ts +754 -754
- package/src/tree/tree-query.ts +427 -427
- package/src/tree/tree-types.ts +299 -299
|
@@ -1,275 +1,275 @@
|
|
|
1
|
-
import type { PoolAdapter, PoolLease, PoolOptions } from './pool-types.js';
|
|
2
|
-
|
|
3
|
-
/**
|
|
4
|
-
* Node.js Timer with optional unref method (for preventing event loop from staying alive)
|
|
5
|
-
*/
|
|
6
|
-
type NodeTimer = ReturnType<typeof setInterval> & {
|
|
7
|
-
unref?: () => void;
|
|
8
|
-
};
|
|
9
|
-
|
|
10
|
-
type Deferred<T> = {
|
|
11
|
-
promise: Promise<T>;
|
|
12
|
-
resolve: (value: T) => void;
|
|
13
|
-
reject: (err: unknown) => void;
|
|
14
|
-
};
|
|
15
|
-
|
|
16
|
-
const deferred = <T>(): Deferred<T> => {
|
|
17
|
-
let resolve!: (value: T) => void;
|
|
18
|
-
let reject!: (err: unknown) => void;
|
|
19
|
-
const promise = new Promise<T>((res, rej) => {
|
|
20
|
-
resolve = res;
|
|
21
|
-
reject = rej;
|
|
22
|
-
});
|
|
23
|
-
return { promise, resolve, reject };
|
|
24
|
-
};
|
|
25
|
-
|
|
26
|
-
type IdleEntry<T> = {
|
|
27
|
-
resource: T;
|
|
28
|
-
lastUsedAt: number;
|
|
29
|
-
};
|
|
30
|
-
|
|
31
|
-
export class Pool<TResource> {
|
|
32
|
-
private readonly adapter: PoolAdapter<TResource>;
|
|
33
|
-
private readonly options: Required<Pick<PoolOptions, 'max'>> & PoolOptions;
|
|
34
|
-
|
|
35
|
-
private destroyed = false;
|
|
36
|
-
private creating = 0;
|
|
37
|
-
private leased = 0;
|
|
38
|
-
private readonly idle: IdleEntry<TResource>[] = [];
|
|
39
|
-
private readonly waiters: Array<Deferred<PoolLease<TResource>>> = [];
|
|
40
|
-
private reapTimer: ReturnType<typeof setInterval> | null = null;
|
|
41
|
-
|
|
42
|
-
constructor(adapter: PoolAdapter<TResource>, options: PoolOptions) {
|
|
43
|
-
if (!Number.isFinite(options.max) || options.max <= 0) {
|
|
44
|
-
throw new Error('Pool options.max must be a positive number');
|
|
45
|
-
}
|
|
46
|
-
|
|
47
|
-
this.adapter = adapter;
|
|
48
|
-
this.options = { max: options.max, ...options };
|
|
49
|
-
|
|
50
|
-
const idleTimeout = this.options.idleTimeoutMillis;
|
|
51
|
-
if (idleTimeout && idleTimeout > 0) {
|
|
52
|
-
const interval =
|
|
53
|
-
this.options.reapIntervalMillis ?? Math.max(1_000, Math.floor(idleTimeout / 2));
|
|
54
|
-
this.reapTimer = setInterval(() => {
|
|
55
|
-
void this.reapIdle();
|
|
56
|
-
}, interval);
|
|
57
|
-
|
|
58
|
-
// Best-effort: avoid keeping the event loop alive.
|
|
59
|
-
(this.reapTimer as NodeTimer).unref?.();
|
|
60
|
-
}
|
|
61
|
-
|
|
62
|
-
// Best-effort warmup.
|
|
63
|
-
const min = this.options.min ?? 0;
|
|
64
|
-
if (min > 0) {
|
|
65
|
-
void this.warm(min);
|
|
66
|
-
}
|
|
67
|
-
}
|
|
68
|
-
|
|
69
|
-
/**
|
|
70
|
-
* Acquire a resource lease.
|
|
71
|
-
* The returned lease MUST be released or destroyed.
|
|
72
|
-
*/
|
|
73
|
-
async acquire(): Promise<PoolLease<TResource>> {
|
|
74
|
-
if (this.destroyed) {
|
|
75
|
-
throw new Error('Pool is destroyed');
|
|
76
|
-
}
|
|
77
|
-
|
|
78
|
-
// 1) Prefer idle.
|
|
79
|
-
const idle = await this.takeIdleValidated();
|
|
80
|
-
if (idle) {
|
|
81
|
-
this.leased++;
|
|
82
|
-
return this.makeLease(idle);
|
|
83
|
-
}
|
|
84
|
-
|
|
85
|
-
// 2) Create if capacity allows.
|
|
86
|
-
if (this.totalLive() < this.options.max) {
|
|
87
|
-
this.creating++;
|
|
88
|
-
try {
|
|
89
|
-
const created = await this.adapter.create();
|
|
90
|
-
this.leased++;
|
|
91
|
-
return this.makeLease(created);
|
|
92
|
-
} finally {
|
|
93
|
-
this.creating--;
|
|
94
|
-
}
|
|
95
|
-
}
|
|
96
|
-
|
|
97
|
-
// 3) Wait.
|
|
98
|
-
const waiter = deferred<PoolLease<TResource>>();
|
|
99
|
-
this.waiters.push(waiter);
|
|
100
|
-
|
|
101
|
-
const timeout = this.options.acquireTimeoutMillis;
|
|
102
|
-
let timer: NodeTimer | null = null;
|
|
103
|
-
if (timeout && timeout > 0) {
|
|
104
|
-
timer = setTimeout(() => {
|
|
105
|
-
// Remove from queue if still waiting.
|
|
106
|
-
const idx = this.waiters.indexOf(waiter);
|
|
107
|
-
if (idx >= 0) this.waiters.splice(idx, 1);
|
|
108
|
-
waiter.reject(new Error('Pool acquire timeout'));
|
|
109
|
-
}, timeout) as NodeTimer;
|
|
110
|
-
// Best-effort: avoid keeping the event loop alive.
|
|
111
|
-
timer.unref?.();
|
|
112
|
-
}
|
|
113
|
-
|
|
114
|
-
try {
|
|
115
|
-
return await waiter.promise;
|
|
116
|
-
} finally {
|
|
117
|
-
if (timer) clearTimeout(timer);
|
|
118
|
-
}
|
|
119
|
-
}
|
|
120
|
-
|
|
121
|
-
/** Destroy pool and all idle resources; waits for in-flight creations to settle. */
|
|
122
|
-
async destroy(): Promise<void> {
|
|
123
|
-
if (this.destroyed) return;
|
|
124
|
-
this.destroyed = true;
|
|
125
|
-
|
|
126
|
-
if (this.reapTimer) {
|
|
127
|
-
clearInterval(this.reapTimer);
|
|
128
|
-
this.reapTimer = null;
|
|
129
|
-
}
|
|
130
|
-
|
|
131
|
-
// Reject all waiters.
|
|
132
|
-
while (this.waiters.length) {
|
|
133
|
-
this.waiters.shift()!.reject(new Error('Pool destroyed'));
|
|
134
|
-
}
|
|
135
|
-
|
|
136
|
-
// Destroy idle resources.
|
|
137
|
-
while (this.idle.length) {
|
|
138
|
-
const entry = this.idle.shift()!;
|
|
139
|
-
await this.adapter.destroy(entry.resource);
|
|
140
|
-
}
|
|
141
|
-
}
|
|
142
|
-
|
|
143
|
-
private totalLive(): number {
|
|
144
|
-
return this.idle.length + this.leased + this.creating;
|
|
145
|
-
}
|
|
146
|
-
|
|
147
|
-
private makeLease(resource: TResource): PoolLease<TResource> {
|
|
148
|
-
let done = false;
|
|
149
|
-
return {
|
|
150
|
-
resource,
|
|
151
|
-
release: async () => {
|
|
152
|
-
if (done) return;
|
|
153
|
-
done = true;
|
|
154
|
-
await this.releaseResource(resource);
|
|
155
|
-
},
|
|
156
|
-
destroy: async () => {
|
|
157
|
-
if (done) return;
|
|
158
|
-
done = true;
|
|
159
|
-
await this.destroyResource(resource);
|
|
160
|
-
},
|
|
161
|
-
};
|
|
162
|
-
}
|
|
163
|
-
|
|
164
|
-
private async releaseResource(resource: TResource): Promise<void> {
|
|
165
|
-
this.leased = Math.max(0, this.leased - 1);
|
|
166
|
-
if (this.destroyed) {
|
|
167
|
-
await this.adapter.destroy(resource);
|
|
168
|
-
return;
|
|
169
|
-
}
|
|
170
|
-
|
|
171
|
-
// Prefer handing directly to waiters.
|
|
172
|
-
const next = this.waiters.shift();
|
|
173
|
-
if (next) {
|
|
174
|
-
this.leased++;
|
|
175
|
-
next.resolve(this.makeLease(resource));
|
|
176
|
-
return;
|
|
177
|
-
}
|
|
178
|
-
|
|
179
|
-
this.idle.push({ resource, lastUsedAt: Date.now() });
|
|
180
|
-
await this.trimToMinMax();
|
|
181
|
-
}
|
|
182
|
-
|
|
183
|
-
private async destroyResource(resource: TResource): Promise<void> {
|
|
184
|
-
this.leased = Math.max(0, this.leased - 1);
|
|
185
|
-
await this.adapter.destroy(resource);
|
|
186
|
-
|
|
187
|
-
// If there are waiters and we have capacity, create a replacement.
|
|
188
|
-
if (!this.destroyed && this.waiters.length && this.totalLive() < this.options.max) {
|
|
189
|
-
const waiter = this.waiters.shift()!;
|
|
190
|
-
this.creating++;
|
|
191
|
-
try {
|
|
192
|
-
const created = await this.adapter.create();
|
|
193
|
-
this.leased++;
|
|
194
|
-
waiter.resolve(this.makeLease(created));
|
|
195
|
-
} catch (err) {
|
|
196
|
-
waiter.reject(err);
|
|
197
|
-
} finally {
|
|
198
|
-
this.creating--;
|
|
199
|
-
}
|
|
200
|
-
}
|
|
201
|
-
}
|
|
202
|
-
|
|
203
|
-
private async takeIdleValidated(): Promise<TResource | null> {
|
|
204
|
-
while (this.idle.length) {
|
|
205
|
-
const entry = this.idle.pop()!;
|
|
206
|
-
if (!this.adapter.validate) {
|
|
207
|
-
return entry.resource;
|
|
208
|
-
}
|
|
209
|
-
const ok = await this.adapter.validate(entry.resource);
|
|
210
|
-
if (ok) {
|
|
211
|
-
return entry.resource;
|
|
212
|
-
}
|
|
213
|
-
await this.adapter.destroy(entry.resource);
|
|
214
|
-
}
|
|
215
|
-
return null;
|
|
216
|
-
}
|
|
217
|
-
|
|
218
|
-
private async reapIdle(): Promise<void> {
|
|
219
|
-
if (this.destroyed) return;
|
|
220
|
-
const idleTimeout = this.options.idleTimeoutMillis;
|
|
221
|
-
if (!idleTimeout || idleTimeout <= 0) return;
|
|
222
|
-
|
|
223
|
-
const now = Date.now();
|
|
224
|
-
const min = this.options.min ?? 0;
|
|
225
|
-
|
|
226
|
-
// Remove expired resources beyond min.
|
|
227
|
-
const keep: IdleEntry<TResource>[] = [];
|
|
228
|
-
const kill: IdleEntry<TResource>[] = [];
|
|
229
|
-
for (const entry of this.idle) {
|
|
230
|
-
const expired = now - entry.lastUsedAt >= idleTimeout;
|
|
231
|
-
if (expired) kill.push(entry);
|
|
232
|
-
else keep.push(entry);
|
|
233
|
-
}
|
|
234
|
-
|
|
235
|
-
// Keep at least `min`.
|
|
236
|
-
while (keep.length < min && kill.length) {
|
|
237
|
-
keep.push(kill.pop()!);
|
|
238
|
-
}
|
|
239
|
-
|
|
240
|
-
this.idle.length = 0;
|
|
241
|
-
this.idle.push(...keep);
|
|
242
|
-
|
|
243
|
-
for (const entry of kill) {
|
|
244
|
-
await this.adapter.destroy(entry.resource);
|
|
245
|
-
}
|
|
246
|
-
}
|
|
247
|
-
|
|
248
|
-
private async warm(targetMin: number): Promise<void> {
|
|
249
|
-
const min = Math.max(0, targetMin);
|
|
250
|
-
while (!this.destroyed && this.idle.length < min && this.totalLive() < this.options.max) {
|
|
251
|
-
this.creating++;
|
|
252
|
-
try {
|
|
253
|
-
const created = await this.adapter.create();
|
|
254
|
-
this.idle.push({ resource: created, lastUsedAt: Date.now() });
|
|
255
|
-
} catch {
|
|
256
|
-
// If warmup fails, stop trying.
|
|
257
|
-
break;
|
|
258
|
-
} finally {
|
|
259
|
-
this.creating--;
|
|
260
|
-
}
|
|
261
|
-
}
|
|
262
|
-
}
|
|
263
|
-
|
|
264
|
-
private async trimToMinMax(): Promise<void> {
|
|
265
|
-
const max = this.options.max;
|
|
266
|
-
const min = this.options.min ?? 0;
|
|
267
|
-
|
|
268
|
-
// Ensure we don't exceed max in idle (best-effort; leased/creating already counted elsewhere).
|
|
269
|
-
while (this.totalLive() > max && this.idle.length > min) {
|
|
270
|
-
const entry = this.idle.shift();
|
|
271
|
-
if (!entry) break;
|
|
272
|
-
await this.adapter.destroy(entry.resource);
|
|
273
|
-
}
|
|
274
|
-
}
|
|
275
|
-
}
|
|
1
|
+
import type { PoolAdapter, PoolLease, PoolOptions } from './pool-types.js';
|
|
2
|
+
|
|
3
|
+
/**
|
|
4
|
+
* Node.js Timer with optional unref method (for preventing event loop from staying alive)
|
|
5
|
+
*/
|
|
6
|
+
type NodeTimer = ReturnType<typeof setInterval> & {
|
|
7
|
+
unref?: () => void;
|
|
8
|
+
};
|
|
9
|
+
|
|
10
|
+
type Deferred<T> = {
|
|
11
|
+
promise: Promise<T>;
|
|
12
|
+
resolve: (value: T) => void;
|
|
13
|
+
reject: (err: unknown) => void;
|
|
14
|
+
};
|
|
15
|
+
|
|
16
|
+
const deferred = <T>(): Deferred<T> => {
|
|
17
|
+
let resolve!: (value: T) => void;
|
|
18
|
+
let reject!: (err: unknown) => void;
|
|
19
|
+
const promise = new Promise<T>((res, rej) => {
|
|
20
|
+
resolve = res;
|
|
21
|
+
reject = rej;
|
|
22
|
+
});
|
|
23
|
+
return { promise, resolve, reject };
|
|
24
|
+
};
|
|
25
|
+
|
|
26
|
+
type IdleEntry<T> = {
|
|
27
|
+
resource: T;
|
|
28
|
+
lastUsedAt: number;
|
|
29
|
+
};
|
|
30
|
+
|
|
31
|
+
export class Pool<TResource> {
|
|
32
|
+
private readonly adapter: PoolAdapter<TResource>;
|
|
33
|
+
private readonly options: Required<Pick<PoolOptions, 'max'>> & PoolOptions;
|
|
34
|
+
|
|
35
|
+
private destroyed = false;
|
|
36
|
+
private creating = 0;
|
|
37
|
+
private leased = 0;
|
|
38
|
+
private readonly idle: IdleEntry<TResource>[] = [];
|
|
39
|
+
private readonly waiters: Array<Deferred<PoolLease<TResource>>> = [];
|
|
40
|
+
private reapTimer: ReturnType<typeof setInterval> | null = null;
|
|
41
|
+
|
|
42
|
+
constructor(adapter: PoolAdapter<TResource>, options: PoolOptions) {
|
|
43
|
+
if (!Number.isFinite(options.max) || options.max <= 0) {
|
|
44
|
+
throw new Error('Pool options.max must be a positive number');
|
|
45
|
+
}
|
|
46
|
+
|
|
47
|
+
this.adapter = adapter;
|
|
48
|
+
this.options = { max: options.max, ...options };
|
|
49
|
+
|
|
50
|
+
const idleTimeout = this.options.idleTimeoutMillis;
|
|
51
|
+
if (idleTimeout && idleTimeout > 0) {
|
|
52
|
+
const interval =
|
|
53
|
+
this.options.reapIntervalMillis ?? Math.max(1_000, Math.floor(idleTimeout / 2));
|
|
54
|
+
this.reapTimer = setInterval(() => {
|
|
55
|
+
void this.reapIdle();
|
|
56
|
+
}, interval);
|
|
57
|
+
|
|
58
|
+
// Best-effort: avoid keeping the event loop alive.
|
|
59
|
+
(this.reapTimer as NodeTimer).unref?.();
|
|
60
|
+
}
|
|
61
|
+
|
|
62
|
+
// Best-effort warmup.
|
|
63
|
+
const min = this.options.min ?? 0;
|
|
64
|
+
if (min > 0) {
|
|
65
|
+
void this.warm(min);
|
|
66
|
+
}
|
|
67
|
+
}
|
|
68
|
+
|
|
69
|
+
/**
|
|
70
|
+
* Acquire a resource lease.
|
|
71
|
+
* The returned lease MUST be released or destroyed.
|
|
72
|
+
*/
|
|
73
|
+
async acquire(): Promise<PoolLease<TResource>> {
|
|
74
|
+
if (this.destroyed) {
|
|
75
|
+
throw new Error('Pool is destroyed');
|
|
76
|
+
}
|
|
77
|
+
|
|
78
|
+
// 1) Prefer idle.
|
|
79
|
+
const idle = await this.takeIdleValidated();
|
|
80
|
+
if (idle) {
|
|
81
|
+
this.leased++;
|
|
82
|
+
return this.makeLease(idle);
|
|
83
|
+
}
|
|
84
|
+
|
|
85
|
+
// 2) Create if capacity allows.
|
|
86
|
+
if (this.totalLive() < this.options.max) {
|
|
87
|
+
this.creating++;
|
|
88
|
+
try {
|
|
89
|
+
const created = await this.adapter.create();
|
|
90
|
+
this.leased++;
|
|
91
|
+
return this.makeLease(created);
|
|
92
|
+
} finally {
|
|
93
|
+
this.creating--;
|
|
94
|
+
}
|
|
95
|
+
}
|
|
96
|
+
|
|
97
|
+
// 3) Wait.
|
|
98
|
+
const waiter = deferred<PoolLease<TResource>>();
|
|
99
|
+
this.waiters.push(waiter);
|
|
100
|
+
|
|
101
|
+
const timeout = this.options.acquireTimeoutMillis;
|
|
102
|
+
let timer: NodeTimer | null = null;
|
|
103
|
+
if (timeout && timeout > 0) {
|
|
104
|
+
timer = setTimeout(() => {
|
|
105
|
+
// Remove from queue if still waiting.
|
|
106
|
+
const idx = this.waiters.indexOf(waiter);
|
|
107
|
+
if (idx >= 0) this.waiters.splice(idx, 1);
|
|
108
|
+
waiter.reject(new Error('Pool acquire timeout'));
|
|
109
|
+
}, timeout) as NodeTimer;
|
|
110
|
+
// Best-effort: avoid keeping the event loop alive.
|
|
111
|
+
timer.unref?.();
|
|
112
|
+
}
|
|
113
|
+
|
|
114
|
+
try {
|
|
115
|
+
return await waiter.promise;
|
|
116
|
+
} finally {
|
|
117
|
+
if (timer) clearTimeout(timer);
|
|
118
|
+
}
|
|
119
|
+
}
|
|
120
|
+
|
|
121
|
+
/** Destroy pool and all idle resources; waits for in-flight creations to settle. */
|
|
122
|
+
async destroy(): Promise<void> {
|
|
123
|
+
if (this.destroyed) return;
|
|
124
|
+
this.destroyed = true;
|
|
125
|
+
|
|
126
|
+
if (this.reapTimer) {
|
|
127
|
+
clearInterval(this.reapTimer);
|
|
128
|
+
this.reapTimer = null;
|
|
129
|
+
}
|
|
130
|
+
|
|
131
|
+
// Reject all waiters.
|
|
132
|
+
while (this.waiters.length) {
|
|
133
|
+
this.waiters.shift()!.reject(new Error('Pool destroyed'));
|
|
134
|
+
}
|
|
135
|
+
|
|
136
|
+
// Destroy idle resources.
|
|
137
|
+
while (this.idle.length) {
|
|
138
|
+
const entry = this.idle.shift()!;
|
|
139
|
+
await this.adapter.destroy(entry.resource);
|
|
140
|
+
}
|
|
141
|
+
}
|
|
142
|
+
|
|
143
|
+
private totalLive(): number {
|
|
144
|
+
return this.idle.length + this.leased + this.creating;
|
|
145
|
+
}
|
|
146
|
+
|
|
147
|
+
private makeLease(resource: TResource): PoolLease<TResource> {
|
|
148
|
+
let done = false;
|
|
149
|
+
return {
|
|
150
|
+
resource,
|
|
151
|
+
release: async () => {
|
|
152
|
+
if (done) return;
|
|
153
|
+
done = true;
|
|
154
|
+
await this.releaseResource(resource);
|
|
155
|
+
},
|
|
156
|
+
destroy: async () => {
|
|
157
|
+
if (done) return;
|
|
158
|
+
done = true;
|
|
159
|
+
await this.destroyResource(resource);
|
|
160
|
+
},
|
|
161
|
+
};
|
|
162
|
+
}
|
|
163
|
+
|
|
164
|
+
private async releaseResource(resource: TResource): Promise<void> {
|
|
165
|
+
this.leased = Math.max(0, this.leased - 1);
|
|
166
|
+
if (this.destroyed) {
|
|
167
|
+
await this.adapter.destroy(resource);
|
|
168
|
+
return;
|
|
169
|
+
}
|
|
170
|
+
|
|
171
|
+
// Prefer handing directly to waiters.
|
|
172
|
+
const next = this.waiters.shift();
|
|
173
|
+
if (next) {
|
|
174
|
+
this.leased++;
|
|
175
|
+
next.resolve(this.makeLease(resource));
|
|
176
|
+
return;
|
|
177
|
+
}
|
|
178
|
+
|
|
179
|
+
this.idle.push({ resource, lastUsedAt: Date.now() });
|
|
180
|
+
await this.trimToMinMax();
|
|
181
|
+
}
|
|
182
|
+
|
|
183
|
+
private async destroyResource(resource: TResource): Promise<void> {
|
|
184
|
+
this.leased = Math.max(0, this.leased - 1);
|
|
185
|
+
await this.adapter.destroy(resource);
|
|
186
|
+
|
|
187
|
+
// If there are waiters and we have capacity, create a replacement.
|
|
188
|
+
if (!this.destroyed && this.waiters.length && this.totalLive() < this.options.max) {
|
|
189
|
+
const waiter = this.waiters.shift()!;
|
|
190
|
+
this.creating++;
|
|
191
|
+
try {
|
|
192
|
+
const created = await this.adapter.create();
|
|
193
|
+
this.leased++;
|
|
194
|
+
waiter.resolve(this.makeLease(created));
|
|
195
|
+
} catch (err) {
|
|
196
|
+
waiter.reject(err);
|
|
197
|
+
} finally {
|
|
198
|
+
this.creating--;
|
|
199
|
+
}
|
|
200
|
+
}
|
|
201
|
+
}
|
|
202
|
+
|
|
203
|
+
private async takeIdleValidated(): Promise<TResource | null> {
|
|
204
|
+
while (this.idle.length) {
|
|
205
|
+
const entry = this.idle.pop()!;
|
|
206
|
+
if (!this.adapter.validate) {
|
|
207
|
+
return entry.resource;
|
|
208
|
+
}
|
|
209
|
+
const ok = await this.adapter.validate(entry.resource);
|
|
210
|
+
if (ok) {
|
|
211
|
+
return entry.resource;
|
|
212
|
+
}
|
|
213
|
+
await this.adapter.destroy(entry.resource);
|
|
214
|
+
}
|
|
215
|
+
return null;
|
|
216
|
+
}
|
|
217
|
+
|
|
218
|
+
private async reapIdle(): Promise<void> {
|
|
219
|
+
if (this.destroyed) return;
|
|
220
|
+
const idleTimeout = this.options.idleTimeoutMillis;
|
|
221
|
+
if (!idleTimeout || idleTimeout <= 0) return;
|
|
222
|
+
|
|
223
|
+
const now = Date.now();
|
|
224
|
+
const min = this.options.min ?? 0;
|
|
225
|
+
|
|
226
|
+
// Remove expired resources beyond min.
|
|
227
|
+
const keep: IdleEntry<TResource>[] = [];
|
|
228
|
+
const kill: IdleEntry<TResource>[] = [];
|
|
229
|
+
for (const entry of this.idle) {
|
|
230
|
+
const expired = now - entry.lastUsedAt >= idleTimeout;
|
|
231
|
+
if (expired) kill.push(entry);
|
|
232
|
+
else keep.push(entry);
|
|
233
|
+
}
|
|
234
|
+
|
|
235
|
+
// Keep at least `min`.
|
|
236
|
+
while (keep.length < min && kill.length) {
|
|
237
|
+
keep.push(kill.pop()!);
|
|
238
|
+
}
|
|
239
|
+
|
|
240
|
+
this.idle.length = 0;
|
|
241
|
+
this.idle.push(...keep);
|
|
242
|
+
|
|
243
|
+
for (const entry of kill) {
|
|
244
|
+
await this.adapter.destroy(entry.resource);
|
|
245
|
+
}
|
|
246
|
+
}
|
|
247
|
+
|
|
248
|
+
private async warm(targetMin: number): Promise<void> {
|
|
249
|
+
const min = Math.max(0, targetMin);
|
|
250
|
+
while (!this.destroyed && this.idle.length < min && this.totalLive() < this.options.max) {
|
|
251
|
+
this.creating++;
|
|
252
|
+
try {
|
|
253
|
+
const created = await this.adapter.create();
|
|
254
|
+
this.idle.push({ resource: created, lastUsedAt: Date.now() });
|
|
255
|
+
} catch {
|
|
256
|
+
// If warmup fails, stop trying.
|
|
257
|
+
break;
|
|
258
|
+
} finally {
|
|
259
|
+
this.creating--;
|
|
260
|
+
}
|
|
261
|
+
}
|
|
262
|
+
}
|
|
263
|
+
|
|
264
|
+
private async trimToMinMax(): Promise<void> {
|
|
265
|
+
const max = this.options.max;
|
|
266
|
+
const min = this.options.min ?? 0;
|
|
267
|
+
|
|
268
|
+
// Ensure we don't exceed max in idle (best-effort; leased/creating already counted elsewhere).
|
|
269
|
+
while (this.totalLive() > max && this.idle.length > min) {
|
|
270
|
+
const entry = this.idle.shift();
|
|
271
|
+
if (!entry) break;
|
|
272
|
+
await this.adapter.destroy(entry.resource);
|
|
273
|
+
}
|
|
274
|
+
}
|
|
275
|
+
}
|
|
@@ -1,35 +1,35 @@
|
|
|
1
|
-
// Pure AST Builders - No Dialect Logic Here!
|
|
2
|
-
|
|
3
|
-
import { ColumnDef } from '../../schema/column-types.js';
|
|
4
|
-
import { columnOperand, valueToOperand } from '../ast/expression-builders.js';
|
|
5
|
-
import { FunctionNode, OperandNode, isOperandNode, TypedExpression, asType } from '../ast/expression.js';
|
|
6
|
-
|
|
7
|
-
type OperandInput = OperandNode | ColumnDef | string | number | boolean | null;
|
|
8
|
-
|
|
9
|
-
const isColumnDef = (val: unknown): val is ColumnDef => !!val && typeof val === 'object' && 'type' in val && 'name' in val;
|
|
10
|
-
|
|
11
|
-
const toOperand = (input: OperandInput): OperandNode => {
|
|
12
|
-
if (isOperandNode(input)) return input;
|
|
13
|
-
if (isColumnDef(input)) return columnOperand(input);
|
|
14
|
-
|
|
15
|
-
return valueToOperand(input);
|
|
16
|
-
};
|
|
17
|
-
|
|
18
|
-
const fn = (key: string, args: OperandInput[]): FunctionNode => ({
|
|
19
|
-
type: 'Function',
|
|
20
|
-
name: key,
|
|
21
|
-
fn: key,
|
|
22
|
-
args: args.map(toOperand)
|
|
23
|
-
});
|
|
24
|
-
|
|
25
|
-
const afn = <T = unknown[]>(key: string, args: OperandInput[]): TypedExpression<T> => asType<T>(fn(key, args));
|
|
26
|
-
|
|
27
|
-
/**
|
|
28
|
-
* Appends a value to the end of an array.
|
|
29
|
-
*
|
|
30
|
-
* @param array - Array column or value.
|
|
31
|
-
* @param value - Value to append.
|
|
32
|
-
* @returns A `TypedExpression<unknown[]>` representing the `ARRAY_APPEND` SQL function.
|
|
33
|
-
*/
|
|
34
|
-
export const arrayAppend = (array: OperandInput, value: OperandInput): TypedExpression<unknown[]> =>
|
|
35
|
-
afn<unknown[]>('ARRAY_APPEND', [array, value]);
|
|
1
|
+
// Pure AST Builders - No Dialect Logic Here!
|
|
2
|
+
|
|
3
|
+
import { ColumnDef } from '../../schema/column-types.js';
|
|
4
|
+
import { columnOperand, valueToOperand } from '../ast/expression-builders.js';
|
|
5
|
+
import { FunctionNode, OperandNode, isOperandNode, TypedExpression, asType } from '../ast/expression.js';
|
|
6
|
+
|
|
7
|
+
type OperandInput = OperandNode | ColumnDef | string | number | boolean | null;
|
|
8
|
+
|
|
9
|
+
const isColumnDef = (val: unknown): val is ColumnDef => !!val && typeof val === 'object' && 'type' in val && 'name' in val;
|
|
10
|
+
|
|
11
|
+
const toOperand = (input: OperandInput): OperandNode => {
|
|
12
|
+
if (isOperandNode(input)) return input;
|
|
13
|
+
if (isColumnDef(input)) return columnOperand(input);
|
|
14
|
+
|
|
15
|
+
return valueToOperand(input);
|
|
16
|
+
};
|
|
17
|
+
|
|
18
|
+
const fn = (key: string, args: OperandInput[]): FunctionNode => ({
|
|
19
|
+
type: 'Function',
|
|
20
|
+
name: key,
|
|
21
|
+
fn: key,
|
|
22
|
+
args: args.map(toOperand)
|
|
23
|
+
});
|
|
24
|
+
|
|
25
|
+
const afn = <T = unknown[]>(key: string, args: OperandInput[]): TypedExpression<T> => asType<T>(fn(key, args));
|
|
26
|
+
|
|
27
|
+
/**
|
|
28
|
+
* Appends a value to the end of an array.
|
|
29
|
+
*
|
|
30
|
+
* @param array - Array column or value.
|
|
31
|
+
* @param value - Value to append.
|
|
32
|
+
* @returns A `TypedExpression<unknown[]>` representing the `ARRAY_APPEND` SQL function.
|
|
33
|
+
*/
|
|
34
|
+
export const arrayAppend = (array: OperandInput, value: OperandInput): TypedExpression<unknown[]> =>
|
|
35
|
+
afn<unknown[]>('ARRAY_APPEND', [array, value]);
|