@rvoh/dream 2.13.0 → 2.14.0-beta.1
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/dist/cjs/src/Dream.js +4 -4
- package/dist/cjs/src/db/DreamDbConnection.js +9 -3
- package/dist/cjs/src/db/dbConnectionLeakDiagnostics.js +0 -14
- package/dist/cjs/src/db/migration-helpers/DreamMigrationHelpers.js +131 -0
- package/dist/cjs/src/db/testDatabasePool.js +182 -0
- package/dist/cjs/src/dream/QueryDriver/Base.js +29 -0
- package/dist/cjs/src/dream/QueryDriver/Kysely.js +56 -22
- package/dist/cjs/src/dream/QueryDriver/Postgres.js +76 -1
- package/dist/cjs/src/dream/QueryDriver/helpers/pg/dropDb.js +3 -1
- package/dist/cjs/src/dream-app/index.js +113 -6
- package/dist/cjs/src/helpers/areEqual.js +1 -1
- package/dist/cjs/src/helpers/loadRepl.js +4 -0
- package/dist/cjs/src/helpers/range.js +1 -1
- package/dist/esm/src/Dream.js +4 -4
- package/dist/esm/src/db/DreamDbConnection.js +9 -3
- package/dist/esm/src/db/dbConnectionLeakDiagnostics.js +0 -14
- package/dist/esm/src/db/migration-helpers/DreamMigrationHelpers.js +131 -0
- package/dist/esm/src/db/testDatabasePool.js +182 -0
- package/dist/esm/src/dream/QueryDriver/Base.js +29 -0
- package/dist/esm/src/dream/QueryDriver/Kysely.js +56 -22
- package/dist/esm/src/dream/QueryDriver/Postgres.js +76 -1
- package/dist/esm/src/dream/QueryDriver/helpers/pg/dropDb.js +3 -1
- package/dist/esm/src/dream-app/index.js +113 -6
- package/dist/esm/src/helpers/areEqual.js +1 -1
- package/dist/esm/src/helpers/loadRepl.js +4 -0
- package/dist/esm/src/helpers/range.js +1 -1
- package/dist/types/src/Dream.d.ts +4 -4
- package/dist/types/src/db/migration-helpers/DreamMigrationHelpers.d.ts +108 -4
- package/dist/types/src/db/testDatabasePool.d.ts +42 -0
- package/dist/types/src/dream/QueryDriver/Base.d.ts +47 -0
- package/dist/types/src/dream/QueryDriver/Kysely.d.ts +11 -0
- package/dist/types/src/dream/QueryDriver/Postgres.d.ts +19 -0
- package/dist/types/src/dream-app/index.d.ts +64 -1
- package/dist/types/src/helpers/areEqual.d.ts +1 -1
- package/dist/types/src/helpers/range.d.ts +8 -4
- package/dist/types/src/types/utils.d.ts +1 -1
- package/dist/types/src/types/utils.ts +1 -1
- package/dist/types/src/types/variadic.d.ts +1 -1
- package/dist/types/src/types/variadic.ts +1 -1
- package/docs/assets/hierarchy.js +1 -1
- package/docs/assets/search.js +1 -1
- package/docs/classes/db.DreamMigrationHelpers.html +65 -9
- package/docs/classes/db.KyselyQueryDriver.html +47 -32
- package/docs/classes/db.PostgresQueryDriver.html +52 -33
- package/docs/classes/db.QueryDriverBase.html +46 -31
- package/docs/classes/errors.CheckConstraintViolation.html +3 -3
- package/docs/classes/errors.ColumnOverflow.html +3 -3
- package/docs/classes/errors.CreateOrFindByFailedToCreateAndFind.html +3 -3
- package/docs/classes/errors.DataIncompatibleWithDatabaseField.html +3 -3
- package/docs/classes/errors.DataTypeColumnTypeMismatch.html +3 -3
- package/docs/classes/errors.DecryptionError.html +2 -2
- package/docs/classes/errors.DecryptionParseError.html +2 -2
- package/docs/classes/errors.DecryptionRotationError.html +3 -3
- package/docs/classes/errors.GlobalNameNotSet.html +3 -3
- package/docs/classes/errors.InvalidCalendarDate.html +2 -2
- package/docs/classes/errors.InvalidClockTime.html +2 -2
- package/docs/classes/errors.InvalidClockTimeTz.html +2 -2
- package/docs/classes/errors.InvalidDateTime.html +2 -2
- package/docs/classes/errors.MissingSerializersDefinition.html +3 -3
- package/docs/classes/errors.NonLoadedAssociation.html +3 -3
- package/docs/classes/errors.NotNullViolation.html +3 -3
- package/docs/classes/errors.RecordNotFound.html +3 -3
- package/docs/classes/errors.ValidationError.html +3 -3
- package/docs/classes/index.CalendarDate.html +33 -33
- package/docs/classes/index.ClockTime.html +32 -32
- package/docs/classes/index.ClockTimeTz.html +35 -35
- package/docs/classes/index.DateTime.html +86 -86
- package/docs/classes/index.Decorators.html +19 -19
- package/docs/classes/index.Dream.html +118 -118
- package/docs/classes/index.DreamApp.html +42 -6
- package/docs/classes/index.DreamTransaction.html +2 -2
- package/docs/classes/index.Env.html +2 -2
- package/docs/classes/index.Query.html +57 -57
- package/docs/classes/system.CliFileWriter.html +4 -4
- package/docs/classes/system.DreamBin.html +2 -2
- package/docs/classes/system.DreamCLI.html +7 -7
- package/docs/classes/system.DreamImporter.html +2 -2
- package/docs/classes/system.DreamLogos.html +2 -2
- package/docs/classes/system.DreamSerializerBuilder.html +11 -11
- package/docs/classes/system.ObjectSerializerBuilder.html +8 -8
- package/docs/classes/system.PathHelpers.html +3 -3
- package/docs/classes/utils.Encrypt.html +3 -3
- package/docs/classes/utils.Range.html +2 -2
- package/docs/functions/db.closeAllDbConnections.html +1 -1
- package/docs/functions/db.dreamDbConnections.html +1 -1
- package/docs/functions/db.untypedDb.html +1 -1
- package/docs/functions/db.validateColumn.html +1 -1
- package/docs/functions/db.validateTable.html +1 -1
- package/docs/functions/errors.pgErrorType.html +1 -1
- package/docs/functions/index.DreamSerializer.html +1 -1
- package/docs/functions/index.ObjectSerializer.html +1 -1
- package/docs/functions/index.ReplicaSafe.html +1 -1
- package/docs/functions/index.STI.html +1 -1
- package/docs/functions/index.SoftDelete.html +1 -1
- package/docs/functions/utils.camelize.html +1 -1
- package/docs/functions/utils.capitalize.html +1 -1
- package/docs/functions/utils.cloneDeepSafe.html +1 -1
- package/docs/functions/utils.compact.html +1 -1
- package/docs/functions/utils.groupBy.html +1 -1
- package/docs/functions/utils.hyphenize.html +1 -1
- package/docs/functions/utils.intersection.html +1 -1
- package/docs/functions/utils.isEmpty.html +1 -1
- package/docs/functions/utils.normalizeUnicode.html +1 -1
- package/docs/functions/utils.pascalize.html +1 -1
- package/docs/functions/utils.percent.html +1 -1
- package/docs/functions/utils.range.html +1 -1
- package/docs/functions/utils.round.html +1 -1
- package/docs/functions/utils.sanitizeString.html +1 -1
- package/docs/functions/utils.snakeify.html +1 -1
- package/docs/functions/utils.sort.html +1 -1
- package/docs/functions/utils.sortBy.html +1 -1
- package/docs/functions/utils.sortObjectByKey.html +1 -1
- package/docs/functions/utils.sortObjectByValue.html +1 -1
- package/docs/functions/utils.uncapitalize.html +1 -1
- package/docs/functions/utils.uniq.html +1 -1
- package/docs/hierarchy.html +1 -1
- package/docs/interfaces/openapi.OpenapiDescription.html +2 -2
- package/docs/interfaces/openapi.OpenapiSchemaProperties.html +1 -1
- package/docs/interfaces/openapi.OpenapiSchemaPropertiesShorthand.html +1 -1
- package/docs/interfaces/openapi.OpenapiTypeFieldObject.html +1 -1
- package/docs/interfaces/types.BelongsToStatement.html +2 -2
- package/docs/interfaces/types.DecoratorContext.html +2 -2
- package/docs/interfaces/types.DreamAppInitOptions.html +2 -2
- package/docs/interfaces/types.DreamAppOpts.html +2 -2
- package/docs/interfaces/types.DreamDbConfig.html +5 -5
- package/docs/interfaces/types.DurationObject.html +2 -2
- package/docs/interfaces/types.EncryptOptions.html +2 -2
- package/docs/interfaces/types.InternalAnyTypedSerializerRendersMany.html +2 -2
- package/docs/interfaces/types.InternalAnyTypedSerializerRendersOne.html +2 -2
- package/docs/interfaces/types.SerializerRendererOpts.html +2 -2
- package/docs/types/openapi.CommonOpenapiSchemaObjectFields.html +1 -1
- package/docs/types/openapi.OpenapiAllTypes.html +1 -1
- package/docs/types/openapi.OpenapiFormats.html +1 -1
- package/docs/types/openapi.OpenapiNumberFormats.html +1 -1
- package/docs/types/openapi.OpenapiPrimitiveBaseTypes.html +1 -1
- package/docs/types/openapi.OpenapiPrimitiveTypes.html +1 -1
- package/docs/types/openapi.OpenapiSchemaArray.html +1 -1
- package/docs/types/openapi.OpenapiSchemaArrayShorthand.html +1 -1
- package/docs/types/openapi.OpenapiSchemaBase.html +1 -1
- package/docs/types/openapi.OpenapiSchemaBody.html +1 -1
- package/docs/types/openapi.OpenapiSchemaBodyShorthand.html +1 -1
- package/docs/types/openapi.OpenapiSchemaCommonFields.html +1 -1
- package/docs/types/openapi.OpenapiSchemaExpressionAllOf.html +2 -2
- package/docs/types/openapi.OpenapiSchemaExpressionAnyOf.html +2 -2
- package/docs/types/openapi.OpenapiSchemaExpressionOneOf.html +2 -2
- package/docs/types/openapi.OpenapiSchemaExpressionRef.html +2 -2
- package/docs/types/openapi.OpenapiSchemaExpressionRefSchemaShorthand.html +2 -2
- package/docs/types/openapi.OpenapiSchemaInteger.html +1 -1
- package/docs/types/openapi.OpenapiSchemaNull.html +2 -2
- package/docs/types/openapi.OpenapiSchemaNumber.html +1 -1
- package/docs/types/openapi.OpenapiSchemaObject.html +1 -1
- package/docs/types/openapi.OpenapiSchemaObjectAllOf.html +1 -1
- package/docs/types/openapi.OpenapiSchemaObjectAllOfShorthand.html +1 -1
- package/docs/types/openapi.OpenapiSchemaObjectAnyOf.html +1 -1
- package/docs/types/openapi.OpenapiSchemaObjectAnyOfShorthand.html +1 -1
- package/docs/types/openapi.OpenapiSchemaObjectBase.html +1 -1
- package/docs/types/openapi.OpenapiSchemaObjectBaseShorthand.html +1 -1
- package/docs/types/openapi.OpenapiSchemaObjectOneOf.html +1 -1
- package/docs/types/openapi.OpenapiSchemaObjectOneOfShorthand.html +1 -1
- package/docs/types/openapi.OpenapiSchemaObjectShorthand.html +1 -1
- package/docs/types/openapi.OpenapiSchemaPrimitiveGeneric.html +1 -1
- package/docs/types/openapi.OpenapiSchemaShorthandExpressionAllOf.html +2 -2
- package/docs/types/openapi.OpenapiSchemaShorthandExpressionAnyOf.html +2 -2
- package/docs/types/openapi.OpenapiSchemaShorthandExpressionOneOf.html +2 -2
- package/docs/types/openapi.OpenapiSchemaShorthandExpressionSerializableRef.html +2 -2
- package/docs/types/openapi.OpenapiSchemaShorthandExpressionSerializerRef.html +2 -2
- package/docs/types/openapi.OpenapiSchemaShorthandPrimitiveGeneric.html +1 -1
- package/docs/types/openapi.OpenapiSchemaString.html +1 -1
- package/docs/types/openapi.OpenapiShorthandAllTypes.html +1 -1
- package/docs/types/openapi.OpenapiShorthandPrimitiveBaseTypes.html +1 -1
- package/docs/types/openapi.OpenapiShorthandPrimitiveTypes.html +1 -1
- package/docs/types/openapi.OpenapiTypeField.html +1 -1
- package/docs/types/system.DreamAppAllowedPackageManagersEnum.html +1 -1
- package/docs/types/types.CalendarDateDurationUnit.html +1 -1
- package/docs/types/types.CalendarDateObject.html +1 -1
- package/docs/types/types.Camelized.html +1 -1
- package/docs/types/types.ClockTimeObject.html +1 -1
- package/docs/types/types.DbConnectionType.html +1 -1
- package/docs/types/types.DbTypes.html +1 -1
- package/docs/types/types.DreamAssociationMetadata.html +1 -1
- package/docs/types/types.DreamAttributes.html +1 -1
- package/docs/types/types.DreamClassAssociationAndStatement.html +1 -1
- package/docs/types/types.DreamClassColumn.html +1 -1
- package/docs/types/types.DreamColumn.html +1 -1
- package/docs/types/types.DreamColumnNames.html +1 -1
- package/docs/types/types.DreamLogLevel.html +1 -1
- package/docs/types/types.DreamLogger.html +2 -2
- package/docs/types/types.DreamModelSerializerType.html +1 -1
- package/docs/types/types.DreamOrViewModelClassSerializerKey.html +1 -1
- package/docs/types/types.DreamOrViewModelSerializerKey.html +1 -1
- package/docs/types/types.DreamParamSafeAttributes.html +1 -1
- package/docs/types/types.DreamParamSafeColumnNames.html +1 -1
- package/docs/types/types.DreamSerializable.html +1 -1
- package/docs/types/types.DreamSerializableArray.html +1 -1
- package/docs/types/types.DreamSerializerKey.html +1 -1
- package/docs/types/types.DreamSerializers.html +1 -1
- package/docs/types/types.DreamVirtualColumns.html +1 -1
- package/docs/types/types.DurationUnit.html +1 -1
- package/docs/types/types.EncryptAlgorithm.html +1 -1
- package/docs/types/types.HasManyStatement.html +1 -1
- package/docs/types/types.HasOneStatement.html +1 -1
- package/docs/types/types.Hyphenized.html +1 -1
- package/docs/types/types.Pascalized.html +1 -1
- package/docs/types/types.PrimaryKeyType.html +1 -1
- package/docs/types/types.RoundingPrecision.html +1 -1
- package/docs/types/types.SerializerCasing.html +1 -1
- package/docs/types/types.SimpleObjectSerializerType.html +1 -1
- package/docs/types/types.Snakeified.html +1 -1
- package/docs/types/types.StrictInterface.html +1 -1
- package/docs/types/types.UpdateableAssociationProperties.html +1 -1
- package/docs/types/types.UpdateableProperties.html +1 -1
- package/docs/types/types.ValidationType.html +1 -1
- package/docs/types/types.ViewModel.html +2 -2
- package/docs/types/types.ViewModelClass.html +1 -1
- package/docs/types/types.WeekdayName.html +1 -1
- package/docs/types/types.WhereStatementForDream.html +1 -1
- package/docs/types/types.WhereStatementForDreamClass.html +1 -1
- package/docs/variables/index.DreamConst.html +1 -1
- package/docs/variables/index.ops.html +1 -1
- package/docs/variables/openapi.openapiPrimitiveTypes.html +1 -1
- package/docs/variables/openapi.openapiShorthandPrimitiveTypes.html +1 -1
- package/docs/variables/system.DreamAppAllowedPackageManagersEnumValues.html +1 -1
- package/docs/variables/system.primaryKeyTypes.html +1 -1
- package/package.json +4 -3
|
@@ -13,12 +13,35 @@ import { isPrimitiveDataType } from '../../db/dataTypes.js';
|
|
|
13
13
|
import DreamApp from '../../dream-app/index.js';
|
|
14
14
|
import camelize from '../../helpers/camelize.js';
|
|
15
15
|
import { findCitextArrayOid, findCorrespondingArrayOid, findEnumArrayOids, parsePostgresBigint, parsePostgresDate, parsePostgresDatetime, parsePostgresDatetimeTz, parsePostgresDecimal, parsePostgresTime, parsePostgresTimeTz, } from '../../helpers/customPgParsers.js';
|
|
16
|
+
import { testDatabasePoolSize } from '../../db/testDatabasePool.js';
|
|
16
17
|
import EnvInternal from '../../helpers/EnvInternal.js';
|
|
17
18
|
import createDb from './helpers/pg/createDb.js';
|
|
18
19
|
import _dropDb from './helpers/pg/dropDb.js';
|
|
19
20
|
import loadPgClient from './helpers/pg/loadPgClient.js';
|
|
20
21
|
import KyselyQueryDriver from './Kysely.js';
|
|
21
22
|
const pgTypes = pg.types;
|
|
23
|
+
// Fixed dream-pool namespace constant for the advisory-lock `key1`. Chosen to
|
|
24
|
+
// sit high in the int4 range and be recognizable in `pg_locks` (classid/objid)
|
|
25
|
+
// when debugging. XORed with a per-namespace hash below so two apps sharing one
|
|
26
|
+
// Postgres cluster reserve disjoint index spaces.
|
|
27
|
+
const DREAM_TEST_POOL_LOCK_NAMESPACE = 0x44_52_4d_00 | 0; // "DRM\0"
|
|
28
|
+
// Aggressive TCP keepalive on the dedicated lock-holding connection. Insurance
|
|
29
|
+
// for remote / CI Postgres and abrupt kills where a FIN might be lost: it lets
|
|
30
|
+
// Postgres reap the dead backend (and release the lock) even when the OS never
|
|
31
|
+
// delivered a clean socket close. For a local kill the lock releases on EOF
|
|
32
|
+
// regardless of this.
|
|
33
|
+
const LOCK_CONNECTION_KEEPALIVE_INITIAL_DELAY_MS = 1_000;
|
|
34
|
+
function int32Hash(value) {
|
|
35
|
+
let hash = 0;
|
|
36
|
+
for (let i = 0; i < value.length; i++) {
|
|
37
|
+
hash = (Math.imul(hash, 31) + value.charCodeAt(i)) | 0;
|
|
38
|
+
}
|
|
39
|
+
return hash;
|
|
40
|
+
}
|
|
41
|
+
function advisoryLockKey1(namespace) {
|
|
42
|
+
// Stays within int4 via the `| 0` coercions.
|
|
43
|
+
return (DREAM_TEST_POOL_LOCK_NAMESPACE ^ int32Hash(namespace)) | 0;
|
|
44
|
+
}
|
|
22
45
|
export default class PostgresQueryDriver extends KyselyQueryDriver {
|
|
23
46
|
/**
|
|
24
47
|
* create the database. Must respond to the NODE_ENV value.
|
|
@@ -161,7 +184,12 @@ export default class PostgresQueryDriver extends KyselyQueryDriver {
|
|
|
161
184
|
await client.query(`DROP DATABASE IF EXISTS ${replicaTestWorkerDatabaseName};`);
|
|
162
185
|
await client.query(`CREATE DATABASE ${replicaTestWorkerDatabaseName} TEMPLATE ${dbConf.name};`);
|
|
163
186
|
}
|
|
164
|
-
|
|
187
|
+
// The pool spans `<base>` (the migration template, also claimable) plus
|
|
188
|
+
// `<base>_2 .. <base>_K`. Each live test worker claims one member for its
|
|
189
|
+
// lifetime via an advisory lock, so the pool must be wide enough to cover
|
|
190
|
+
// active + still-terminating workers — see src/db/testDatabasePool.ts.
|
|
191
|
+
const poolSize = testDatabasePoolSize(parallelTests);
|
|
192
|
+
for (let i = 2; i <= poolSize; i++) {
|
|
165
193
|
const workerDatabaseName = `${dbConf.name}_${i}`;
|
|
166
194
|
DreamCLI.logger.logContinueProgress(`creating duplicate test database ${workerDatabaseName} for concurrent tests...`, { logPrefix: ' ├ [db]', logPrefixColor: 'greenBright' });
|
|
167
195
|
await client.query(`DROP DATABASE IF EXISTS ${workerDatabaseName};`);
|
|
@@ -170,4 +198,51 @@ export default class PostgresQueryDriver extends KyselyQueryDriver {
|
|
|
170
198
|
await client.end();
|
|
171
199
|
DreamCLI.logger.logEndProgress();
|
|
172
200
|
}
|
|
201
|
+
static supportsParallelTestDatabases = true;
|
|
202
|
+
/**
|
|
203
|
+
* @internal
|
|
204
|
+
*
|
|
205
|
+
* Postgres implementation of the per-worker test-database claim seam (see
|
|
206
|
+
* {@link TestDatabaseLockSession} and `src/db/testDatabasePool.ts`).
|
|
207
|
+
*
|
|
208
|
+
* Postgres advisory locks are global to the cluster (not scoped to a single
|
|
209
|
+
* database), so one dedicated connection — to the `postgres` maintenance
|
|
210
|
+
* database — can reserve a pool index for the whole cluster. We use the
|
|
211
|
+
* two-`int4` form `pg_try_advisory_lock(key1, key2)`, whose lock space is
|
|
212
|
+
* distinct from the single-`bigint` form `pg_advisory_lock(bigint)` that
|
|
213
|
+
* application code typically uses, so a claim can never clash with an
|
|
214
|
+
* app-level advisory lock. `key1` folds a hash of the lock namespace (the
|
|
215
|
+
* base database name) into a fixed dream-pool constant; `key2` is the pool
|
|
216
|
+
* index. The lock auto-releases when the connection drops on process exit.
|
|
217
|
+
*/
|
|
218
|
+
static async openTestDatabaseLockSession(connectionName) {
|
|
219
|
+
const dreamApp = DreamApp.getOrFail();
|
|
220
|
+
const creds = dreamApp.dbCredentialsFor(connectionName)?.primary ?? dreamApp.dbCredentialsFor(connectionName)?.replica;
|
|
221
|
+
if (!creds?.name)
|
|
222
|
+
throw new Error(`[dream] cannot open a test-database lock session: the "${connectionName}" connection has no resolvable primary db name`);
|
|
223
|
+
const client = new pg.Client({
|
|
224
|
+
host: creds.host || 'localhost',
|
|
225
|
+
port: creds.port,
|
|
226
|
+
// The maintenance database always exists and we never lock it —
|
|
227
|
+
// advisory locks are cluster-wide.
|
|
228
|
+
database: 'postgres',
|
|
229
|
+
user: creds.user,
|
|
230
|
+
password: creds.password,
|
|
231
|
+
keepAlive: true,
|
|
232
|
+
keepAliveInitialDelayMillis: LOCK_CONNECTION_KEEPALIVE_INITIAL_DELAY_MS,
|
|
233
|
+
});
|
|
234
|
+
await client.connect();
|
|
235
|
+
return {
|
|
236
|
+
async tryAcquire(namespace, index) {
|
|
237
|
+
const result = await client.query('SELECT pg_try_advisory_lock($1::int4, $2::int4) AS locked', [
|
|
238
|
+
advisoryLockKey1(namespace),
|
|
239
|
+
index,
|
|
240
|
+
]);
|
|
241
|
+
return result.rows[0]?.locked === true;
|
|
242
|
+
},
|
|
243
|
+
async release() {
|
|
244
|
+
await client.end().catch(() => undefined);
|
|
245
|
+
},
|
|
246
|
+
};
|
|
247
|
+
}
|
|
173
248
|
}
|
|
@@ -1,4 +1,5 @@
|
|
|
1
1
|
import DreamCLI from '../../../../cli/index.js';
|
|
2
|
+
import { testDatabasePoolSize } from '../../../../db/testDatabasePool.js';
|
|
2
3
|
import DreamApp from '../../../../dream-app/index.js';
|
|
3
4
|
import EnvInternal from '../../../../helpers/EnvInternal.js';
|
|
4
5
|
import loadPgClient from './loadPgClient.js';
|
|
@@ -25,7 +26,8 @@ async function maybeDropDuplicateDatabases(client, dbName) {
|
|
|
25
26
|
DreamCLI.logger.logContinueProgress(`dropping fake replica test database ${replicaTestWorkerDatabaseName}`, { logPrefix: ' ├ [db]', logPrefixColor: 'greenBright' });
|
|
26
27
|
await client.query(`DROP DATABASE IF EXISTS ${replicaTestWorkerDatabaseName};`);
|
|
27
28
|
}
|
|
28
|
-
|
|
29
|
+
const poolSize = testDatabasePoolSize(parallelTests);
|
|
30
|
+
for (let i = 2; i <= poolSize; i++) {
|
|
29
31
|
const workerDatabaseName = `${dbName}_${i}`;
|
|
30
32
|
DreamCLI.logger.logContinueProgress(`dropping duplicate test database ${workerDatabaseName}`, {
|
|
31
33
|
logPrefix: ' ├ [db]',
|
|
@@ -1,4 +1,5 @@
|
|
|
1
1
|
import * as util from 'node:util';
|
|
2
|
+
import { claimedTestDatabaseIndexOrNull, claimTestDatabase, testDatabaseNameForIndex, } from '../db/testDatabasePool.js';
|
|
2
3
|
import validateTable from '../db/validators/validateTable.js';
|
|
3
4
|
import Query from '../dream/Query.js';
|
|
4
5
|
import PostgresQueryDriver from '../dream/QueryDriver/Postgres.js';
|
|
@@ -47,8 +48,12 @@ export default class DreamApp {
|
|
|
47
48
|
if (!dreamApp.serializers)
|
|
48
49
|
setCachedSerializers({});
|
|
49
50
|
cacheDreamApp(dreamApp);
|
|
50
|
-
if (!opts.bypassDbConnectionsDuringInit)
|
|
51
|
+
if (!opts.bypassDbConnectionsDuringInit) {
|
|
52
|
+
// Claim this worker's pool database before any connection is built, so
|
|
53
|
+
// the synchronous dbName() always resolves to the claimed database.
|
|
54
|
+
await dreamApp.ensureTestDatabaseClaimed();
|
|
51
55
|
await this.setDatabaseTypeParsers(dreamApp);
|
|
56
|
+
}
|
|
52
57
|
await deferCb?.(dreamApp);
|
|
53
58
|
for (const plugin of dreamApp.plugins) {
|
|
54
59
|
await plugin(dreamApp);
|
|
@@ -289,9 +294,56 @@ A new key can also be generated from the CLI:
|
|
|
289
294
|
get serializers() {
|
|
290
295
|
return getSerializersOrFail();
|
|
291
296
|
}
|
|
297
|
+
/**
|
|
298
|
+
* The database name to connect to for a given connection. In dev/prod this is
|
|
299
|
+
* simply the configured name. Under vitest (test env) it is the per-worker
|
|
300
|
+
* database claimed from the pool — see {@link testDatabaseClaimEnabled} and
|
|
301
|
+
* `src/db/testDatabasePool.ts`.
|
|
302
|
+
*
|
|
303
|
+
* This is synchronous because Kysely builds its `pg.Pool` synchronously. It
|
|
304
|
+
* reads the already-claimed pool index; the claim is awaited earlier in the
|
|
305
|
+
* worker — `DreamApp.init()` calls {@link claimTestDatabase} before the first
|
|
306
|
+
* connection is built, and `dream-spec-helpers`' `truncate` awaits
|
|
307
|
+
* {@link testDatabaseName} — so the index is always populated by the time any
|
|
308
|
+
* connection resolves its name.
|
|
309
|
+
*/
|
|
292
310
|
dbName(connectionName, connection) {
|
|
293
311
|
const conf = this.dbConnectionConfig(connectionName, connection);
|
|
294
|
-
|
|
312
|
+
if (!this.runtimeNeedsTestDatabaseClaim)
|
|
313
|
+
return conf.name;
|
|
314
|
+
this.requireTestDatabaseClaimSupport();
|
|
315
|
+
const index = claimedTestDatabaseIndexOrNull();
|
|
316
|
+
if (index === null)
|
|
317
|
+
throw new Error(`[dream] dbName("${connectionName}", "${connection}") was resolved before this worker claimed ` +
|
|
318
|
+
`a test database. The pool claim must be awaited first (DreamApp.init / truncate do this). ` +
|
|
319
|
+
`This indicates a database connection was opened before app initialization completed.`);
|
|
320
|
+
return testDatabaseNameForIndex(conf.name, index);
|
|
321
|
+
}
|
|
322
|
+
/**
|
|
323
|
+
* Single source of truth for the claimed test-database name, used both by
|
|
324
|
+
* dream's connection resolution and by `dream-spec-helpers`' `truncate`
|
|
325
|
+
* (which opens its own `pg.Client` and so must read — not recompute — the
|
|
326
|
+
* claim). Awaits the pool claim if it has not happened yet in this worker.
|
|
327
|
+
*/
|
|
328
|
+
async testDatabaseName(connectionName, connection) {
|
|
329
|
+
const conf = this.dbConnectionConfig(connectionName, connection);
|
|
330
|
+
if (!this.runtimeNeedsTestDatabaseClaim)
|
|
331
|
+
return conf.name;
|
|
332
|
+
this.requireTestDatabaseClaimSupport();
|
|
333
|
+
const index = await claimTestDatabase(this);
|
|
334
|
+
return testDatabaseNameForIndex(conf.name, index);
|
|
335
|
+
}
|
|
336
|
+
/**
|
|
337
|
+
* Claim this worker's pool database if it hasn't been claimed yet. Idempotent
|
|
338
|
+
* and a no-op outside the test-under-vitest path. Called by `DreamApp.init()`
|
|
339
|
+
* before any connection is built so the synchronous {@link dbName} always
|
|
340
|
+
* sees a populated index.
|
|
341
|
+
*/
|
|
342
|
+
async ensureTestDatabaseClaimed() {
|
|
343
|
+
if (!this.runtimeNeedsTestDatabaseClaim)
|
|
344
|
+
return;
|
|
345
|
+
this.requireTestDatabaseClaimSupport();
|
|
346
|
+
await claimTestDatabase(this);
|
|
295
347
|
}
|
|
296
348
|
dbConnectionConfig(connectionName, connection) {
|
|
297
349
|
const conf = this.dbCredentialsFor(connectionName)?.[connection] || this.dbCredentialsFor(connectionName)?.primary;
|
|
@@ -322,10 +374,57 @@ A new key can also be generated from the CLI:
|
|
|
322
374
|
hasReplicaConfig(connectionName) {
|
|
323
375
|
return !!this.dbCredentials[connectionName]?.replica;
|
|
324
376
|
}
|
|
325
|
-
|
|
326
|
-
|
|
327
|
-
|
|
328
|
-
|
|
377
|
+
/**
|
|
378
|
+
* Whether the runtime *requires* a per-worker test-database claim. Gated on
|
|
379
|
+
* test env *under vitest* (not on `parallelTests > 1`): the slot-reuse
|
|
380
|
+
* collision reproduces even serially, because vitest respawns a process per
|
|
381
|
+
* file and the lingering process overlaps the new one on the same database.
|
|
382
|
+
* CLI db tasks (`db:migrate`, `db:reset`, …) do not run under vitest, so they
|
|
383
|
+
* keep the unsuffixed name and operate on the whole pool explicitly.
|
|
384
|
+
*
|
|
385
|
+
* This is independent of whether the default driver can actually claim — when
|
|
386
|
+
* the need is present but the capability is absent, the claim path fails loud
|
|
387
|
+
* (see {@link requireTestDatabaseClaimSupport}) rather than silently sharing
|
|
388
|
+
* one database.
|
|
389
|
+
*/
|
|
390
|
+
get runtimeNeedsTestDatabaseClaim() {
|
|
391
|
+
return EnvInternal.isTest && runningUnderVitest();
|
|
392
|
+
}
|
|
393
|
+
/**
|
|
394
|
+
* Whether the default connection's query driver implements the per-worker
|
|
395
|
+
* test-database claim seam (`openTestDatabaseLockSession`). The claim is keyed
|
|
396
|
+
* off the **default** connection's capability so a Postgres-default app with a
|
|
397
|
+
* MySQL secondary keeps working (the secondary borrows the Postgres-claimed
|
|
398
|
+
* index for its database name).
|
|
399
|
+
*/
|
|
400
|
+
get defaultConnectionSupportsParallelTestDatabases() {
|
|
401
|
+
return this.dbConnectionQueryDriverClass('default').supportsParallelTestDatabases;
|
|
402
|
+
}
|
|
403
|
+
/**
|
|
404
|
+
* Whether the per-worker test-database pool claim is active: needed by the
|
|
405
|
+
* runtime *and* supported by the default connection's driver. Consumers that
|
|
406
|
+
* key behavior off the claimed index (connection-cache suffixing) use this;
|
|
407
|
+
* the claim entry points additionally fail loud when the need is present but
|
|
408
|
+
* support is absent.
|
|
409
|
+
*/
|
|
410
|
+
get testDatabaseClaimEnabled() {
|
|
411
|
+
return this.runtimeNeedsTestDatabaseClaim && this.defaultConnectionSupportsParallelTestDatabases;
|
|
412
|
+
}
|
|
413
|
+
/**
|
|
414
|
+
* Throw an actionable error when the runtime needs a per-worker test-database
|
|
415
|
+
* claim but the default connection's driver cannot provide one — rather than
|
|
416
|
+
* silently sharing a single database across overlapping vitest workers (which
|
|
417
|
+
* reintroduces the flake this whole mechanism exists to fix).
|
|
418
|
+
*/
|
|
419
|
+
requireTestDatabaseClaimSupport() {
|
|
420
|
+
if (this.defaultConnectionSupportsParallelTestDatabases)
|
|
421
|
+
return;
|
|
422
|
+
const driverName = this.dbConnectionQueryDriverClass('default').name;
|
|
423
|
+
throw new Error(`[dream] running under vitest requires a per-worker test database, but the "default" connection's ` +
|
|
424
|
+
`query driver (${driverName}) does not support parallel test databases. Implement the claim seam ` +
|
|
425
|
+
`on that driver — set \`supportsParallelTestDatabases = true\` and override ` +
|
|
426
|
+
`\`openTestDatabaseLockSession\` (see PostgresQueryDriver / the test-app MysqlQueryDriver for ` +
|
|
427
|
+
`reference) — or do not run this adapter under vitest.`);
|
|
329
428
|
}
|
|
330
429
|
async load(resourceType, resourcePath, importCb) {
|
|
331
430
|
switch (resourceType) {
|
|
@@ -436,6 +535,14 @@ function loggerArgToString(arg) {
|
|
|
436
535
|
}
|
|
437
536
|
return JSON.stringify(util.inspect(arg, { depth: 6 }));
|
|
438
537
|
}
|
|
538
|
+
/**
|
|
539
|
+
* Whether the current process is a vitest worker. vitest sets `VITEST=true` in
|
|
540
|
+
* every worker, and also exposes the per-worker `VITEST_POOL_ID` /
|
|
541
|
+
* `VITEST_WORKER_ID`. CLI db tasks run outside vitest, so this is false there.
|
|
542
|
+
*/
|
|
543
|
+
function runningUnderVitest() {
|
|
544
|
+
return process.env.VITEST === 'true' || process.env.VITEST_POOL_ID != null;
|
|
545
|
+
}
|
|
439
546
|
export const DreamAppAllowedPackageManagersEnumValues = ['pnpm', 'yarn', 'npm', 'bun', 'deno'];
|
|
440
547
|
// GeneratorImportStyles are used by CLI generators to determine how
|
|
441
548
|
// to style import suffixes. When integrating with other apps, this
|
|
@@ -4,7 +4,7 @@ import CalendarDate from '../utils/datetime/CalendarDate.js';
|
|
|
4
4
|
import { DateTime } from '../utils/datetime/DateTime.js';
|
|
5
5
|
/**
|
|
6
6
|
* Performs a deep equality check between two values, supporting primitives, arrays,
|
|
7
|
-
* Dream instances, CalendarDate, DateTime, and objects.
|
|
7
|
+
* Dream instances, CalendarDate, DateTime, ClockTime, ClockTimeTz, and objects.
|
|
8
8
|
*
|
|
9
9
|
* Examples:
|
|
10
10
|
* areEqual(undefined, undefined) // true
|
|
@@ -2,6 +2,8 @@ import DreamApp from '../dream-app/index.js';
|
|
|
2
2
|
import Encrypt from '../encrypt/index.js';
|
|
3
3
|
import ops from '../ops/index.js';
|
|
4
4
|
import CalendarDate from '../utils/datetime/CalendarDate.js';
|
|
5
|
+
import ClockTime from '../utils/datetime/ClockTime.js';
|
|
6
|
+
import ClockTimeTz from '../utils/datetime/ClockTimeTz.js';
|
|
5
7
|
import { DateTime } from '../utils/datetime/DateTime.js';
|
|
6
8
|
import camelize from './camelize.js';
|
|
7
9
|
import capitalize from './capitalize.js';
|
|
@@ -19,6 +21,8 @@ import uniq from './uniq.js';
|
|
|
19
21
|
export default async function loadRepl(context) {
|
|
20
22
|
const dreamApp = DreamApp.getOrFail();
|
|
21
23
|
context.CalendarDate = CalendarDate;
|
|
24
|
+
context.ClockTime = ClockTime;
|
|
25
|
+
context.ClockTimeTz = ClockTimeTz;
|
|
22
26
|
context.DateTime = DateTime;
|
|
23
27
|
context.Encrypt = Encrypt;
|
|
24
28
|
context.camelize = camelize;
|
package/dist/esm/src/Dream.js
CHANGED
|
@@ -2020,10 +2020,10 @@ export default class Dream {
|
|
|
2020
2020
|
* // as: 'ratings',
|
|
2021
2021
|
* // polymorphic: true,
|
|
2022
2022
|
* // source: 'ratings',
|
|
2023
|
-
* //
|
|
2024
|
-
* //
|
|
2025
|
-
* //
|
|
2026
|
-
* //
|
|
2023
|
+
* // and: undefined,
|
|
2024
|
+
* // andNot: undefined,
|
|
2025
|
+
* // selfAnd: undefined,
|
|
2026
|
+
* // selfAndNot: undefined,
|
|
2027
2027
|
* // primaryKeyOverride: null,
|
|
2028
2028
|
* // primaryKey: [Function: primaryKey],
|
|
2029
2029
|
* // primaryKeyValue: [Function: primaryKeyValue],
|
|
@@ -8,6 +8,7 @@
|
|
|
8
8
|
// @ts-ignore
|
|
9
9
|
import { CamelCasePlugin, Kysely } from 'kysely';
|
|
10
10
|
import DreamApp from '../dream-app/index.js';
|
|
11
|
+
import { claimedTestDatabaseIndexOrNull } from './testDatabasePool.js';
|
|
11
12
|
import protectAgainstPollutingAssignment from '../helpers/protectAgainstPollutingAssignment.js';
|
|
12
13
|
import { installDbConnectionLeakDiagnosticsIfEnabled, reportLeakedDbConnections, } from './dbConnectionLeakDiagnostics.js';
|
|
13
14
|
let connections = {};
|
|
@@ -39,9 +40,14 @@ export default class DreamDbConnection {
|
|
|
39
40
|
return dbConn;
|
|
40
41
|
}
|
|
41
42
|
static getConnectionTypeName(connectionType) {
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
43
|
+
// The connection cache is per Node process (per vitest worker), and a worker
|
|
44
|
+
// only ever uses its single claimed pool index, so suffixing the cache key by
|
|
45
|
+
// that index keeps connections cached per claimed database without colliding
|
|
46
|
+
// across the (now process-scoped) registry.
|
|
47
|
+
if (!DreamApp.getOrFail().testDatabaseClaimEnabled)
|
|
48
|
+
return connectionType;
|
|
49
|
+
const index = claimedTestDatabaseIndexOrNull();
|
|
50
|
+
return index === null ? connectionType : `${connectionType}_${index}`;
|
|
45
51
|
}
|
|
46
52
|
}
|
|
47
53
|
export function dreamDbConnections() {
|
|
@@ -41,21 +41,15 @@ function track(client, stack) {
|
|
|
41
41
|
if (!client)
|
|
42
42
|
return;
|
|
43
43
|
checkedOutClients.set(client, { stack: stack ?? '(no stack captured)', since: Date.now() });
|
|
44
|
-
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
45
44
|
const c = client;
|
|
46
|
-
// eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
|
|
47
45
|
if (c.__dreamReleasePatched)
|
|
48
46
|
return;
|
|
49
|
-
// eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
|
|
50
47
|
const originalRelease = c.release;
|
|
51
48
|
if (typeof originalRelease !== 'function')
|
|
52
49
|
return;
|
|
53
|
-
// eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
|
|
54
50
|
c.__dreamReleasePatched = true;
|
|
55
|
-
// eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
|
|
56
51
|
c.release = function patchedRelease(...args) {
|
|
57
52
|
checkedOutClients.delete(client);
|
|
58
|
-
// eslint-disable-next-line @typescript-eslint/no-unsafe-return, @typescript-eslint/no-unsafe-call
|
|
59
53
|
return originalRelease.apply(this, args);
|
|
60
54
|
};
|
|
61
55
|
}
|
|
@@ -67,34 +61,26 @@ export function installDbConnectionLeakDiagnosticsIfEnabled() {
|
|
|
67
61
|
if (!dreamDebugEnabled)
|
|
68
62
|
return;
|
|
69
63
|
enabled = true;
|
|
70
|
-
// eslint-disable-next-line @typescript-eslint/no-unsafe-member-access, @typescript-eslint/no-explicit-any
|
|
71
64
|
const Pool = pg.Pool;
|
|
72
65
|
if (!Pool?.prototype)
|
|
73
66
|
return;
|
|
74
|
-
// eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
|
|
75
67
|
if (Pool.prototype[INSTALLED])
|
|
76
68
|
return;
|
|
77
|
-
// eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
|
|
78
69
|
Pool.prototype[INSTALLED] = true;
|
|
79
|
-
// eslint-disable-next-line @typescript-eslint/no-unsafe-member-access, @typescript-eslint/no-unsafe-assignment
|
|
80
70
|
const originalConnect = Pool.prototype.connect;
|
|
81
|
-
// eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
|
|
82
71
|
Pool.prototype.connect = function patchedConnect(cb) {
|
|
83
72
|
const acquireStack = new Error('db connection acquired here').stack;
|
|
84
73
|
if (typeof cb === 'function') {
|
|
85
|
-
// eslint-disable-next-line @typescript-eslint/no-unsafe-return, @typescript-eslint/no-unsafe-call
|
|
86
74
|
return originalConnect.call(this, (err, client, release) => {
|
|
87
75
|
track(client, acquireStack);
|
|
88
76
|
// `track()` replaces `client.release` with an untracking wrapper.
|
|
89
77
|
// Callback-style users call the 3rd `release` arg, which pg captured
|
|
90
78
|
// *before* that swap — so hand them the wrapped one instead, or a
|
|
91
79
|
// correctly-released client would be reported as a false leak.
|
|
92
|
-
// eslint-disable-next-line @typescript-eslint/no-explicit-any, @typescript-eslint/no-unsafe-member-access
|
|
93
80
|
const wrappedRelease = client ? client.release : release;
|
|
94
81
|
cb(err, client, wrappedRelease);
|
|
95
82
|
});
|
|
96
83
|
}
|
|
97
|
-
// eslint-disable-next-line @typescript-eslint/no-unsafe-call, @typescript-eslint/no-unsafe-member-access
|
|
98
84
|
return originalConnect.call(this).then((client) => {
|
|
99
85
|
track(client, acquireStack);
|
|
100
86
|
return client;
|
|
@@ -1,4 +1,5 @@
|
|
|
1
1
|
import { sql } from 'kysely';
|
|
2
|
+
import InternalEncrypt from '../../encrypt/InternalEncrypt.js';
|
|
2
3
|
export default class DreamMigrationHelpers {
|
|
3
4
|
/**
|
|
4
5
|
* Rename a table and its associated primary key index and sequence.
|
|
@@ -195,6 +196,136 @@ export default class DreamMigrationHelpers {
|
|
|
195
196
|
}
|
|
196
197
|
}
|
|
197
198
|
}
|
|
199
|
+
/**
|
|
200
|
+
* Convert an existing plaintext column into the encrypted-backed form expected by
|
|
201
|
+
* the `@Encrypted` decorator.
|
|
202
|
+
*
|
|
203
|
+
* This renames `column` to `encrypted_<column>`, widens it to `text`, and rewrites
|
|
204
|
+
* every non-null value with the AES-GCM ciphertext produced by the exact same code
|
|
205
|
+
* path the decorator's setter uses (`InternalEncrypt.encryptColumn`). After it runs,
|
|
206
|
+
* the column holds real ciphertext that the decorator's getter can decrypt — which is
|
|
207
|
+
* what a bare column rename does **not** do (renaming a plaintext column to
|
|
208
|
+
* `encrypted_<column>` and decorating the property leaves plaintext in the column, so
|
|
209
|
+
* the getter throws `DecryptionError`).
|
|
210
|
+
*
|
|
211
|
+
* The encryption key and algorithm come from the application's encryption config
|
|
212
|
+
* (`DreamApp` `encryption.columns.current`); if encryption is not configured this
|
|
213
|
+
* throws `MissingColumnEncryptionOpts`. Null values are left null.
|
|
214
|
+
*
|
|
215
|
+
* Intended for text/string columns (the common case is text -> encrypted text). The
|
|
216
|
+
* column is widened to `text` before reading, so values are encrypted as their text
|
|
217
|
+
* form; the inverse `decryptColumn` can restore a non-text type via its `columnType`
|
|
218
|
+
* option.
|
|
219
|
+
*
|
|
220
|
+
* ```ts
|
|
221
|
+
* // plaintext `phone` column -> encrypted `encrypted_phone` text column
|
|
222
|
+
* await DreamMigrationHelpers.encryptColumn(db, { table: 'users', column: 'phone' })
|
|
223
|
+
* ```
|
|
224
|
+
*
|
|
225
|
+
* **Rewrites rows one at a time** (each value needs a fresh random IV computed in Node,
|
|
226
|
+
* so the rewrite cannot be a single SQL `UPDATE`), reading them in keyset batches of
|
|
227
|
+
* `batchSize` to bound memory. It still holds the table for the migration's duration; on
|
|
228
|
+
* very large tables do not use this helper — write your own batched / online migration.
|
|
229
|
+
*
|
|
230
|
+
* **Drop any index on the column first.** Per-row updates pay index-maintenance cost on
|
|
231
|
+
* every write, and an index over ciphertext is useless anyway (encrypted values are not
|
|
232
|
+
* queryable). Remove the index before this migration and do not re-add it.
|
|
233
|
+
*
|
|
234
|
+
* @param db - The Kysely database object passed into the migration up/down function
|
|
235
|
+
* @param options - Configuration options
|
|
236
|
+
* @param options.table - The name of the table
|
|
237
|
+
* @param options.column - The current (plaintext) column name
|
|
238
|
+
* @param options.encryptedColumnName - The target encrypted column name. Defaults to `encrypted_<column>`, matching the `@Encrypted` decorator's default; pass this when the decorator was given a custom encrypted column name.
|
|
239
|
+
* @param options.primaryKey - The primary key column used to keyset-paginate and target each row's update. Defaults to `id`.
|
|
240
|
+
* @param options.batchSize - How many rows to read per batch. Defaults to `1000`.
|
|
241
|
+
*/
|
|
242
|
+
static async encryptColumn(db, { table, column, encryptedColumnName = `encrypted_${column}`, primaryKey = 'id', batchSize = 1000, }) {
|
|
243
|
+
await db.schema.alterTable(table).renameColumn(column, encryptedColumnName).execute();
|
|
244
|
+
await db.schema
|
|
245
|
+
.alterTable(table)
|
|
246
|
+
.alterColumn(encryptedColumnName, col => col.setDataType('text'))
|
|
247
|
+
.execute();
|
|
248
|
+
await this.transformColumnInBatches(db, { table, column: encryptedColumnName, primaryKey, batchSize }, value => InternalEncrypt.encryptColumn(value));
|
|
249
|
+
}
|
|
250
|
+
/**
|
|
251
|
+
* Inverse of {@link DreamMigrationHelpers.encryptColumn}: decrypt an
|
|
252
|
+
* `encrypted_<column>` column back to plaintext and rename it to `column`.
|
|
253
|
+
*
|
|
254
|
+
* Every non-null value is decrypted with the same path the decorator's getter uses
|
|
255
|
+
* (`InternalEncrypt.decryptColumn`, which honors both the `current` and `legacy`
|
|
256
|
+
* encryption keys), then the column is renamed back. Called with the same `table` and
|
|
257
|
+
* `column`, this exactly reverses `encryptColumn`.
|
|
258
|
+
*
|
|
259
|
+
* By default the column is left as `text`, because the original column type cannot be
|
|
260
|
+
* recovered from the encrypted state. Pass `columnType` to restore a specific type
|
|
261
|
+
* (e.g. `'integer'`); the conversion runs `ALTER COLUMN ... TYPE <columnType> USING
|
|
262
|
+
* <column>::<columnType>`.
|
|
263
|
+
*
|
|
264
|
+
* ```ts
|
|
265
|
+
* await DreamMigrationHelpers.decryptColumn(db, { table: 'users', column: 'phone' })
|
|
266
|
+
*
|
|
267
|
+
* // restore the original column type as part of the inverse
|
|
268
|
+
* await DreamMigrationHelpers.decryptColumn(db, { table: 'users', column: 'age', columnType: 'integer' })
|
|
269
|
+
* ```
|
|
270
|
+
*
|
|
271
|
+
* The same per-row, batched, table-locking caveat as `encryptColumn` applies.
|
|
272
|
+
*
|
|
273
|
+
* @param db - The Kysely database object passed into the migration up/down function
|
|
274
|
+
* @param options - Configuration options
|
|
275
|
+
* @param options.table - The name of the table
|
|
276
|
+
* @param options.column - The target (plaintext) column name to rename back to
|
|
277
|
+
* @param options.encryptedColumnName - The current encrypted column name. Defaults to `encrypted_<column>`.
|
|
278
|
+
* @param options.primaryKey - The primary key column used to keyset-paginate and target each row's update. Defaults to `id`.
|
|
279
|
+
* @param options.batchSize - How many rows to read per batch. Defaults to `1000`.
|
|
280
|
+
* @param options.columnType - When provided, the restored column is converted to this type. When omitted, the column is left as `text`.
|
|
281
|
+
*/
|
|
282
|
+
static async decryptColumn(db, { table, column, encryptedColumnName = `encrypted_${column}`, primaryKey = 'id', batchSize = 1000, columnType, }) {
|
|
283
|
+
await this.transformColumnInBatches(db, { table, column: encryptedColumnName, primaryKey, batchSize }, value => InternalEncrypt.decryptColumn(value));
|
|
284
|
+
await db.schema.alterTable(table).renameColumn(encryptedColumnName, column).execute();
|
|
285
|
+
if (columnType !== undefined) {
|
|
286
|
+
await sql `
|
|
287
|
+
ALTER TABLE ${sql.table(table)}
|
|
288
|
+
ALTER COLUMN ${sql.ref(column)}
|
|
289
|
+
TYPE ${sql.raw(columnType)}
|
|
290
|
+
USING ${sql.ref(column)}::${sql.raw(columnType)};
|
|
291
|
+
`.execute(db);
|
|
292
|
+
}
|
|
293
|
+
}
|
|
294
|
+
/**
|
|
295
|
+
* Walk every non-null value of `column` in keyset batches of `batchSize`, applying
|
|
296
|
+
* `transform` in Node and writing the result back one row at a time. Keyset pagination
|
|
297
|
+
* on `primaryKey` (`WHERE pk > last ORDER BY pk`) bounds memory to a single batch and
|
|
298
|
+
* guarantees forward progress, so a row is never read — or transformed — twice (which
|
|
299
|
+
* matters because the transform is not idempotent: re-encrypting ciphertext would
|
|
300
|
+
* double-encrypt, and re-decrypting plaintext would throw).
|
|
301
|
+
*
|
|
302
|
+
* Reads alias the selected columns to fixed keys via `sql.ref(...).as(...)` so a
|
|
303
|
+
* `CamelCasePlugin` on the connection cannot rename the result keys.
|
|
304
|
+
*/
|
|
305
|
+
static async transformColumnInBatches(db, { table, column, primaryKey, batchSize, }, transform) {
|
|
306
|
+
let lastPrimaryKey;
|
|
307
|
+
for (;;) {
|
|
308
|
+
let query = db
|
|
309
|
+
.selectFrom(table)
|
|
310
|
+
.select([sql.ref(primaryKey).as('pk'), sql.ref(column).as('val')])
|
|
311
|
+
.where(column, 'is not', null)
|
|
312
|
+
.orderBy(primaryKey)
|
|
313
|
+
.limit(batchSize);
|
|
314
|
+
if (lastPrimaryKey !== undefined)
|
|
315
|
+
query = query.where(primaryKey, '>', lastPrimaryKey);
|
|
316
|
+
const rows = await query.execute();
|
|
317
|
+
if (rows.length === 0)
|
|
318
|
+
break;
|
|
319
|
+
for (const row of rows) {
|
|
320
|
+
await db
|
|
321
|
+
.updateTable(table)
|
|
322
|
+
.set({ [column]: transform(row.val) })
|
|
323
|
+
.where(primaryKey, '=', row.pk)
|
|
324
|
+
.execute();
|
|
325
|
+
lastPrimaryKey = row.pk;
|
|
326
|
+
}
|
|
327
|
+
}
|
|
328
|
+
}
|
|
198
329
|
}
|
|
199
330
|
async function getEnumValues(db, enumName) {
|
|
200
331
|
const response = await sql `SELECT unnest(enum_range(NULL::${sql.raw(enumName)}))`.execute(db);
|