@aztec/validator-ha-signer 0.0.1-commit.7d4e6cd → 0.0.1-commit.87a0206
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 +42 -37
- package/dest/config.d.ts +71 -17
- package/dest/config.d.ts.map +1 -1
- package/dest/config.js +47 -19
- package/dest/db/postgres.d.ts +34 -5
- package/dest/db/postgres.d.ts.map +1 -1
- package/dest/db/postgres.js +80 -20
- package/dest/db/schema.d.ts +19 -9
- package/dest/db/schema.d.ts.map +1 -1
- package/dest/db/schema.js +46 -18
- package/dest/db/types.d.ts +81 -24
- package/dest/db/types.d.ts.map +1 -1
- package/dest/db/types.js +34 -0
- package/dest/errors.d.ts +9 -5
- package/dest/errors.d.ts.map +1 -1
- package/dest/errors.js +7 -4
- package/dest/factory.d.ts +6 -14
- package/dest/factory.d.ts.map +1 -1
- package/dest/factory.js +6 -11
- package/dest/migrations.d.ts +1 -1
- package/dest/migrations.d.ts.map +1 -1
- package/dest/migrations.js +13 -2
- package/dest/slashing_protection_service.d.ts +16 -6
- package/dest/slashing_protection_service.d.ts.map +1 -1
- package/dest/slashing_protection_service.js +58 -17
- package/dest/test/pglite_pool.d.ts +92 -0
- package/dest/test/pglite_pool.d.ts.map +1 -0
- package/dest/test/pglite_pool.js +210 -0
- package/dest/types.d.ts +91 -14
- package/dest/types.d.ts.map +1 -1
- package/dest/types.js +21 -1
- package/dest/validator_ha_signer.d.ts +10 -13
- package/dest/validator_ha_signer.d.ts.map +1 -1
- package/dest/validator_ha_signer.js +32 -31
- package/package.json +9 -8
- package/src/config.ts +83 -50
- package/src/db/postgres.ts +101 -19
- package/src/db/schema.ts +48 -18
- package/src/db/types.ts +111 -22
- package/src/errors.ts +7 -2
- package/src/factory.ts +8 -13
- package/src/migrations.ts +17 -1
- package/src/slashing_protection_service.ts +94 -23
- package/src/test/pglite_pool.ts +256 -0
- package/src/types.ts +140 -19
- package/src/validator_ha_signer.ts +39 -41
package/dest/factory.js
CHANGED
|
@@ -18,7 +18,7 @@ import { ValidatorHASigner } from './validator_ha_signer.js';
|
|
|
18
18
|
* ```typescript
|
|
19
19
|
* const { signer, db } = await createHASigner({
|
|
20
20
|
* databaseUrl: process.env.DATABASE_URL,
|
|
21
|
-
*
|
|
21
|
+
* haSigningEnabled: true,
|
|
22
22
|
* nodeId: 'validator-node-1',
|
|
23
23
|
* pollingIntervalMs: 100,
|
|
24
24
|
* signingTimeoutMs: 3000,
|
|
@@ -30,22 +30,17 @@ import { ValidatorHASigner } from './validator_ha_signer.js';
|
|
|
30
30
|
* await signer.stop(); // On shutdown
|
|
31
31
|
* ```
|
|
32
32
|
*
|
|
33
|
-
*
|
|
34
|
-
*
|
|
35
|
-
* const { signer, db } = await createHASigner({
|
|
36
|
-
* databaseUrl: process.env.DATABASE_URL,
|
|
37
|
-
* enabled: true,
|
|
38
|
-
* nodeId: 'validator-node-1',
|
|
39
|
-
* runMigrations: true, // Auto-run migrations on startup
|
|
40
|
-
* });
|
|
41
|
-
* signer.start();
|
|
42
|
-
* ```
|
|
33
|
+
* Note: Migrations must be run separately using `aztec migrate-ha-db up` before
|
|
34
|
+
* creating the signer. The factory will verify the schema is initialized via `db.initialize()`.
|
|
43
35
|
*
|
|
44
36
|
* @param config - Configuration for the HA signer
|
|
45
37
|
* @param deps - Optional dependencies (e.g., for testing)
|
|
46
38
|
* @returns An object containing the signer and database instances
|
|
47
39
|
*/ export async function createHASigner(config, deps) {
|
|
48
40
|
const { databaseUrl, poolMaxCount, poolMinCount, poolIdleTimeoutMs, poolConnectionTimeoutMs, ...signerConfig } = config;
|
|
41
|
+
if (!databaseUrl) {
|
|
42
|
+
throw new Error('databaseUrl is required for createHASigner');
|
|
43
|
+
}
|
|
49
44
|
// Create connection pool (or use provided pool)
|
|
50
45
|
let pool;
|
|
51
46
|
if (!deps?.pool) {
|
package/dest/migrations.d.ts
CHANGED
|
@@ -12,4 +12,4 @@ export interface RunMigrationsOptions {
|
|
|
12
12
|
* @returns Array of applied migration names
|
|
13
13
|
*/
|
|
14
14
|
export declare function runMigrations(databaseUrl: string, options?: RunMigrationsOptions): Promise<string[]>;
|
|
15
|
-
//# sourceMappingURL=data:application/json;base64,
|
|
15
|
+
//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoibWlncmF0aW9ucy5kLnRzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vc3JjL21pZ3JhdGlvbnMudHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6IkFBYUEsTUFBTSxXQUFXLG9CQUFvQjtJQUNuQyxpRkFBaUY7SUFDakYsU0FBUyxDQUFDLEVBQUUsSUFBSSxHQUFHLE1BQU0sQ0FBQztJQUMxQixnREFBZ0Q7SUFDaEQsT0FBTyxDQUFDLEVBQUUsT0FBTyxDQUFDO0NBQ25CO0FBRUQ7Ozs7OztHQU1HO0FBQ0gsd0JBQXNCLGFBQWEsQ0FBQyxXQUFXLEVBQUUsTUFBTSxFQUFFLE9BQU8sR0FBRSxvQkFBeUIsR0FBRyxPQUFPLENBQUMsTUFBTSxFQUFFLENBQUMsQ0ErQzlHIn0=
|
package/dest/migrations.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"migrations.d.ts","sourceRoot":"","sources":["../src/migrations.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"migrations.d.ts","sourceRoot":"","sources":["../src/migrations.ts"],"names":[],"mappings":"AAaA,MAAM,WAAW,oBAAoB;IACnC,iFAAiF;IACjF,SAAS,CAAC,EAAE,IAAI,GAAG,MAAM,CAAC;IAC1B,gDAAgD;IAChD,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB;AAED;;;;;;GAMG;AACH,wBAAsB,aAAa,CAAC,WAAW,EAAE,MAAM,EAAE,OAAO,GAAE,oBAAyB,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CA+C9G"}
|
package/dest/migrations.js
CHANGED
|
@@ -1,6 +1,7 @@
|
|
|
1
1
|
/**
|
|
2
2
|
* Programmatic migration runner
|
|
3
3
|
*/ import { createLogger } from '@aztec/foundation/log';
|
|
4
|
+
import { readdirSync } from 'fs';
|
|
4
5
|
import { runner } from 'node-pg-migrate';
|
|
5
6
|
import { dirname, join } from 'path';
|
|
6
7
|
import { fileURLToPath } from 'url';
|
|
@@ -16,16 +17,26 @@ const __dirname = dirname(__filename);
|
|
|
16
17
|
const direction = options.direction ?? 'up';
|
|
17
18
|
const verbose = options.verbose ?? false;
|
|
18
19
|
const log = createLogger('validator-ha-signer:migrations');
|
|
20
|
+
const migrationsDir = join(__dirname, 'db', 'migrations');
|
|
19
21
|
try {
|
|
20
22
|
log.info(`Running migrations ${direction}...`);
|
|
23
|
+
// Filter out .d.ts and .d.ts.map files - node-pg-migrate only needs .js files
|
|
24
|
+
const migrationFiles = readdirSync(migrationsDir);
|
|
25
|
+
const jsMigrationFiles = migrationFiles.filter((file)=>file.endsWith('.js') && !file.endsWith('.d.ts') && !file.endsWith('.d.ts.map'));
|
|
26
|
+
if (jsMigrationFiles.length === 0) {
|
|
27
|
+
log.info('No migration files found');
|
|
28
|
+
return [];
|
|
29
|
+
}
|
|
21
30
|
const appliedMigrations = await runner({
|
|
22
31
|
databaseUrl,
|
|
23
|
-
dir:
|
|
32
|
+
dir: migrationsDir,
|
|
24
33
|
direction,
|
|
25
34
|
migrationsTable: 'pgmigrations',
|
|
26
35
|
count: direction === 'down' ? 1 : Infinity,
|
|
27
36
|
verbose,
|
|
28
|
-
log: (msg)=>verbose ? log.info(msg) : log.debug(msg)
|
|
37
|
+
log: (msg)=>verbose ? log.info(msg) : log.debug(msg),
|
|
38
|
+
// Ignore TypeScript declaration files - node-pg-migrate will try to import them otherwise
|
|
39
|
+
ignorePattern: '.*\\.d\\.(ts|js)$|.*\\.d\\.ts\\.map$'
|
|
29
40
|
});
|
|
30
41
|
if (appliedMigrations.length === 0) {
|
|
31
42
|
log.info('No migrations to apply - schema is up to date');
|
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import { type CheckAndRecordParams, type DeleteDutyParams, type RecordSuccessParams } from './db/types.js';
|
|
2
|
-
import type {
|
|
2
|
+
import type { SlashingProtectionDatabase, ValidatorHASignerConfig } from './types.js';
|
|
3
3
|
/**
|
|
4
4
|
* Slashing Protection Service
|
|
5
5
|
*
|
|
@@ -21,8 +21,10 @@ export declare class SlashingProtectionService {
|
|
|
21
21
|
private readonly log;
|
|
22
22
|
private readonly pollingIntervalMs;
|
|
23
23
|
private readonly signingTimeoutMs;
|
|
24
|
+
private readonly maxStuckDutiesAgeMs;
|
|
24
25
|
private cleanupRunningPromise;
|
|
25
|
-
|
|
26
|
+
private lastOldDutiesCleanupAtMs?;
|
|
27
|
+
constructor(db: SlashingProtectionDatabase, config: ValidatorHASignerConfig);
|
|
26
28
|
/**
|
|
27
29
|
* Check if a duty can be performed and acquire the lock if so.
|
|
28
30
|
*
|
|
@@ -32,7 +34,6 @@ export declare class SlashingProtectionService {
|
|
|
32
34
|
* 2. If insert succeeds, we acquired the lock - return the lockToken
|
|
33
35
|
* 3. If a record exists, handle based on status:
|
|
34
36
|
* - SIGNED: Throw appropriate error (already signed or slashing protection)
|
|
35
|
-
* - FAILED: Delete the failed record
|
|
36
37
|
* - SIGNING: Wait and poll until status changes, then handle result
|
|
37
38
|
*
|
|
38
39
|
* @returns The lockToken that must be used for recordSuccess/deleteDuty
|
|
@@ -64,11 +65,20 @@ export declare class SlashingProtectionService {
|
|
|
64
65
|
* Start running tasks.
|
|
65
66
|
* Cleanup runs immediately on start to recover from any previous crashes.
|
|
66
67
|
*/
|
|
67
|
-
|
|
68
|
+
/**
|
|
69
|
+
* Start the background cleanup task.
|
|
70
|
+
* Also performs one-time cleanup of duties with outdated rollup addresses.
|
|
71
|
+
*/
|
|
72
|
+
start(): Promise<void>;
|
|
68
73
|
/**
|
|
69
74
|
* Stop the background cleanup task.
|
|
70
75
|
*/
|
|
71
76
|
stop(): Promise<void>;
|
|
72
|
-
|
|
77
|
+
/**
|
|
78
|
+
* Close the database connection.
|
|
79
|
+
* Should be called after stop() during graceful shutdown.
|
|
80
|
+
*/
|
|
81
|
+
close(): Promise<void>;
|
|
82
|
+
private cleanup;
|
|
73
83
|
}
|
|
74
|
-
//# sourceMappingURL=data:application/json;base64,
|
|
84
|
+
//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoic2xhc2hpbmdfcHJvdGVjdGlvbl9zZXJ2aWNlLmQudHMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi9zcmMvc2xhc2hpbmdfcHJvdGVjdGlvbl9zZXJ2aWNlLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQVVBLE9BQU8sRUFDTCxLQUFLLG9CQUFvQixFQUN6QixLQUFLLGdCQUFnQixFQUVyQixLQUFLLG1CQUFtQixFQUV6QixNQUFNLGVBQWUsQ0FBQztBQUV2QixPQUFPLEtBQUssRUFBRSwwQkFBMEIsRUFBRSx1QkFBdUIsRUFBRSxNQUFNLFlBQVksQ0FBQztBQUV0Rjs7Ozs7Ozs7Ozs7Ozs7R0FjRztBQUNILHFCQUFhLHlCQUF5QjtJQVVsQyxPQUFPLENBQUMsUUFBUSxDQUFDLEVBQUU7SUFDbkIsT0FBTyxDQUFDLFFBQVEsQ0FBQyxNQUFNO0lBVnpCLE9BQU8sQ0FBQyxRQUFRLENBQUMsR0FBRyxDQUFTO0lBQzdCLE9BQU8sQ0FBQyxRQUFRLENBQUMsaUJBQWlCLENBQVM7SUFDM0MsT0FBTyxDQUFDLFFBQVEsQ0FBQyxnQkFBZ0IsQ0FBUztJQUMxQyxPQUFPLENBQUMsUUFBUSxDQUFDLG1CQUFtQixDQUFTO0lBRTdDLE9BQU8sQ0FBQyxxQkFBcUIsQ0FBaUI7SUFDOUMsT0FBTyxDQUFDLHdCQUF3QixDQUFDLENBQVM7SUFFMUMsWUFDbUIsRUFBRSxFQUFFLDBCQUEwQixFQUM5QixNQUFNLEVBQUUsdUJBQXVCLEVBU2pEO0lBRUQ7Ozs7Ozs7Ozs7Ozs7O09BY0c7SUFDRyxjQUFjLENBQUMsTUFBTSxFQUFFLG9CQUFvQixHQUFHLE9BQU8sQ0FBQyxNQUFNLENBQUMsQ0FpRWxFO0lBRUQ7Ozs7OztPQU1HO0lBQ0csYUFBYSxDQUFDLE1BQU0sRUFBRSxtQkFBbUIsR0FBRyxPQUFPLENBQUMsT0FBTyxDQUFDLENBMkJqRTtJQUVEOzs7Ozs7T0FNRztJQUNHLFVBQVUsQ0FBQyxNQUFNLEVBQUUsZ0JBQWdCLEdBQUcsT0FBTyxDQUFDLE9BQU8sQ0FBQyxDQXdCM0Q7SUFFRDs7T0FFRztJQUNILElBQUksTUFBTSxJQUFJLE1BQU0sQ0FFbkI7SUFFRDs7O09BR0c7SUFDSDs7O09BR0c7SUFDRyxLQUFLLGtCQVdWO0lBRUQ7O09BRUc7SUFDRyxJQUFJLGtCQUdUO0lBRUQ7OztPQUdHO0lBQ0csS0FBSyxrQkFHVjtZQU1hLE9BQU87Q0E2QnRCIn0=
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"slashing_protection_service.d.ts","sourceRoot":"","sources":["../src/slashing_protection_service.ts"],"names":[],"mappings":"AAUA,OAAO,
|
|
1
|
+
{"version":3,"file":"slashing_protection_service.d.ts","sourceRoot":"","sources":["../src/slashing_protection_service.ts"],"names":[],"mappings":"AAUA,OAAO,EACL,KAAK,oBAAoB,EACzB,KAAK,gBAAgB,EAErB,KAAK,mBAAmB,EAEzB,MAAM,eAAe,CAAC;AAEvB,OAAO,KAAK,EAAE,0BAA0B,EAAE,uBAAuB,EAAE,MAAM,YAAY,CAAC;AAEtF;;;;;;;;;;;;;;GAcG;AACH,qBAAa,yBAAyB;IAUlC,OAAO,CAAC,QAAQ,CAAC,EAAE;IACnB,OAAO,CAAC,QAAQ,CAAC,MAAM;IAVzB,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAS;IAC7B,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAS;IAC3C,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAS;IAC1C,OAAO,CAAC,QAAQ,CAAC,mBAAmB,CAAS;IAE7C,OAAO,CAAC,qBAAqB,CAAiB;IAC9C,OAAO,CAAC,wBAAwB,CAAC,CAAS;IAE1C,YACmB,EAAE,EAAE,0BAA0B,EAC9B,MAAM,EAAE,uBAAuB,EASjD;IAED;;;;;;;;;;;;;;OAcG;IACG,cAAc,CAAC,MAAM,EAAE,oBAAoB,GAAG,OAAO,CAAC,MAAM,CAAC,CAiElE;IAED;;;;;;OAMG;IACG,aAAa,CAAC,MAAM,EAAE,mBAAmB,GAAG,OAAO,CAAC,OAAO,CAAC,CA2BjE;IAED;;;;;;OAMG;IACG,UAAU,CAAC,MAAM,EAAE,gBAAgB,GAAG,OAAO,CAAC,OAAO,CAAC,CAwB3D;IAED;;OAEG;IACH,IAAI,MAAM,IAAI,MAAM,CAEnB;IAED;;;OAGG;IACH;;;OAGG;IACG,KAAK,kBAWV;IAED;;OAEG;IACG,IAAI,kBAGT;IAED;;;OAGG;IACG,KAAK,kBAGV;YAMa,OAAO;CA6BtB"}
|
|
@@ -6,7 +6,7 @@
|
|
|
6
6
|
*/ import { createLogger } from '@aztec/foundation/log';
|
|
7
7
|
import { RunningPromise } from '@aztec/foundation/promise';
|
|
8
8
|
import { sleep } from '@aztec/foundation/sleep';
|
|
9
|
-
import { DutyStatus } from './db/types.js';
|
|
9
|
+
import { DutyStatus, getBlockIndexFromDutyIdentifier } from './db/types.js';
|
|
10
10
|
import { DutyAlreadySignedError, SlashingProtectionError } from './errors.js';
|
|
11
11
|
/**
|
|
12
12
|
* Slashing Protection Service
|
|
@@ -28,14 +28,18 @@ import { DutyAlreadySignedError, SlashingProtectionError } from './errors.js';
|
|
|
28
28
|
log;
|
|
29
29
|
pollingIntervalMs;
|
|
30
30
|
signingTimeoutMs;
|
|
31
|
+
maxStuckDutiesAgeMs;
|
|
31
32
|
cleanupRunningPromise;
|
|
33
|
+
lastOldDutiesCleanupAtMs;
|
|
32
34
|
constructor(db, config){
|
|
33
35
|
this.db = db;
|
|
34
36
|
this.config = config;
|
|
35
37
|
this.log = createLogger('slashing-protection');
|
|
36
38
|
this.pollingIntervalMs = config.pollingIntervalMs;
|
|
37
39
|
this.signingTimeoutMs = config.signingTimeoutMs;
|
|
38
|
-
|
|
40
|
+
// Default to 144s (2x 72s Aztec slot duration) if not explicitly configured
|
|
41
|
+
this.maxStuckDutiesAgeMs = config.maxStuckDutiesAgeMs ?? 144_000;
|
|
42
|
+
this.cleanupRunningPromise = new RunningPromise(this.cleanup.bind(this), this.log, this.maxStuckDutiesAgeMs);
|
|
39
43
|
}
|
|
40
44
|
/**
|
|
41
45
|
* Check if a duty can be performed and acquire the lock if so.
|
|
@@ -46,7 +50,6 @@ import { DutyAlreadySignedError, SlashingProtectionError } from './errors.js';
|
|
|
46
50
|
* 2. If insert succeeds, we acquired the lock - return the lockToken
|
|
47
51
|
* 3. If a record exists, handle based on status:
|
|
48
52
|
* - SIGNED: Throw appropriate error (already signed or slashing protection)
|
|
49
|
-
* - FAILED: Delete the failed record
|
|
50
53
|
* - SIGNING: Wait and poll until status changes, then handle result
|
|
51
54
|
*
|
|
52
55
|
* @returns The lockToken that must be used for recordSuccess/deleteDuty
|
|
@@ -81,9 +84,9 @@ import { DutyAlreadySignedError, SlashingProtectionError } from './errors.js';
|
|
|
81
84
|
existingNodeId: record.nodeId,
|
|
82
85
|
attemptingNodeId: nodeId
|
|
83
86
|
});
|
|
84
|
-
throw new SlashingProtectionError(slot, dutyType, record.messageHash, messageHash);
|
|
87
|
+
throw new SlashingProtectionError(slot, dutyType, record.blockIndexWithinCheckpoint, record.messageHash, messageHash, record.nodeId);
|
|
85
88
|
}
|
|
86
|
-
throw new DutyAlreadySignedError(slot, dutyType, record.nodeId);
|
|
89
|
+
throw new DutyAlreadySignedError(slot, dutyType, record.blockIndexWithinCheckpoint, record.nodeId);
|
|
87
90
|
} else if (record.status === DutyStatus.SIGNING) {
|
|
88
91
|
// Another node is currently signing - check for timeout
|
|
89
92
|
if (Date.now() - startTime > this.signingTimeoutMs) {
|
|
@@ -92,7 +95,7 @@ import { DutyAlreadySignedError, SlashingProtectionError } from './errors.js';
|
|
|
92
95
|
timeoutMs: this.signingTimeoutMs,
|
|
93
96
|
signingNodeId: record.nodeId
|
|
94
97
|
});
|
|
95
|
-
throw new DutyAlreadySignedError(slot, dutyType, 'unknown (timeout)');
|
|
98
|
+
throw new DutyAlreadySignedError(slot, dutyType, record.blockIndexWithinCheckpoint, 'unknown (timeout)');
|
|
96
99
|
}
|
|
97
100
|
// Wait and poll
|
|
98
101
|
this.log.debug(`Waiting for signing to complete for duty ${dutyType} at slot ${slot}`, {
|
|
@@ -113,8 +116,9 @@ import { DutyAlreadySignedError, SlashingProtectionError } from './errors.js';
|
|
|
113
116
|
*
|
|
114
117
|
* @returns true if the update succeeded, false if token didn't match
|
|
115
118
|
*/ async recordSuccess(params) {
|
|
116
|
-
const { validatorAddress, slot, dutyType, signature, nodeId, lockToken } = params;
|
|
117
|
-
const
|
|
119
|
+
const { rollupAddress, validatorAddress, slot, dutyType, signature, nodeId, lockToken } = params;
|
|
120
|
+
const blockIndexWithinCheckpoint = getBlockIndexFromDutyIdentifier(params);
|
|
121
|
+
const success = await this.db.updateDutySigned(rollupAddress, validatorAddress, slot, dutyType, signature.toString(), lockToken, blockIndexWithinCheckpoint);
|
|
118
122
|
if (success) {
|
|
119
123
|
this.log.info(`Recorded successful signing for duty ${dutyType} at slot ${slot}`, {
|
|
120
124
|
validatorAddress: validatorAddress.toString(),
|
|
@@ -135,8 +139,9 @@ import { DutyAlreadySignedError, SlashingProtectionError } from './errors.js';
|
|
|
135
139
|
*
|
|
136
140
|
* @returns true if the delete succeeded, false if token didn't match
|
|
137
141
|
*/ async deleteDuty(params) {
|
|
138
|
-
const { validatorAddress, slot, dutyType, lockToken } = params;
|
|
139
|
-
const
|
|
142
|
+
const { rollupAddress, validatorAddress, slot, dutyType, lockToken } = params;
|
|
143
|
+
const blockIndexWithinCheckpoint = getBlockIndexFromDutyIdentifier(params);
|
|
144
|
+
const success = await this.db.deleteDuty(rollupAddress, validatorAddress, slot, dutyType, lockToken, blockIndexWithinCheckpoint);
|
|
140
145
|
if (success) {
|
|
141
146
|
this.log.info(`Deleted duty ${dutyType} at slot ${slot} to allow retry`, {
|
|
142
147
|
validatorAddress: validatorAddress.toString()
|
|
@@ -156,7 +161,17 @@ import { DutyAlreadySignedError, SlashingProtectionError } from './errors.js';
|
|
|
156
161
|
/**
|
|
157
162
|
* Start running tasks.
|
|
158
163
|
* Cleanup runs immediately on start to recover from any previous crashes.
|
|
159
|
-
*/
|
|
164
|
+
*/ /**
|
|
165
|
+
* Start the background cleanup task.
|
|
166
|
+
* Also performs one-time cleanup of duties with outdated rollup addresses.
|
|
167
|
+
*/ async start() {
|
|
168
|
+
// One-time cleanup at startup: remove duties from previous rollup versions
|
|
169
|
+
const numOutdatedRollupDuties = await this.db.cleanupOutdatedRollupDuties(this.config.l1Contracts.rollupAddress);
|
|
170
|
+
if (numOutdatedRollupDuties > 0) {
|
|
171
|
+
this.log.info(`Cleaned up ${numOutdatedRollupDuties} duties with outdated rollup address at startup`, {
|
|
172
|
+
currentRollupAddress: this.config.l1Contracts.rollupAddress.toString()
|
|
173
|
+
});
|
|
174
|
+
}
|
|
160
175
|
this.cleanupRunningPromise.start();
|
|
161
176
|
this.log.info('Slashing protection service started', {
|
|
162
177
|
nodeId: this.config.nodeId
|
|
@@ -171,14 +186,40 @@ import { DutyAlreadySignedError, SlashingProtectionError } from './errors.js';
|
|
|
171
186
|
});
|
|
172
187
|
}
|
|
173
188
|
/**
|
|
174
|
-
*
|
|
175
|
-
|
|
176
|
-
|
|
177
|
-
|
|
178
|
-
|
|
189
|
+
* Close the database connection.
|
|
190
|
+
* Should be called after stop() during graceful shutdown.
|
|
191
|
+
*/ async close() {
|
|
192
|
+
await this.db.close();
|
|
193
|
+
this.log.info('Slashing protection database connection closed');
|
|
194
|
+
}
|
|
195
|
+
/**
|
|
196
|
+
* Periodic cleanup of stuck duties and optionally old signed duties.
|
|
197
|
+
* Runs in the background via RunningPromise.
|
|
198
|
+
*/ async cleanup() {
|
|
199
|
+
// 1. Clean up stuck duties (our own node's duties that got stuck in 'signing' status)
|
|
200
|
+
const numStuckDuties = await this.db.cleanupOwnStuckDuties(this.config.nodeId, this.maxStuckDutiesAgeMs);
|
|
201
|
+
if (numStuckDuties > 0) {
|
|
202
|
+
this.log.verbose(`Cleaned up ${numStuckDuties} stuck duties`, {
|
|
179
203
|
nodeId: this.config.nodeId,
|
|
180
|
-
maxStuckDutiesAgeMs: this.
|
|
204
|
+
maxStuckDutiesAgeMs: this.maxStuckDutiesAgeMs
|
|
181
205
|
});
|
|
182
206
|
}
|
|
207
|
+
// 2. Clean up old signed duties if configured
|
|
208
|
+
// we shouldn't run this as often as stuck duty cleanup.
|
|
209
|
+
if (this.config.cleanupOldDutiesAfterHours !== undefined) {
|
|
210
|
+
const maxAgeMs = this.config.cleanupOldDutiesAfterHours * 60 * 60 * 1000;
|
|
211
|
+
const nowMs = Date.now();
|
|
212
|
+
const shouldRun = this.lastOldDutiesCleanupAtMs === undefined || nowMs - this.lastOldDutiesCleanupAtMs >= maxAgeMs;
|
|
213
|
+
if (shouldRun) {
|
|
214
|
+
const numOldDuties = await this.db.cleanupOldDuties(maxAgeMs);
|
|
215
|
+
this.lastOldDutiesCleanupAtMs = nowMs;
|
|
216
|
+
if (numOldDuties > 0) {
|
|
217
|
+
this.log.verbose(`Cleaned up ${numOldDuties} old signed duties`, {
|
|
218
|
+
cleanupOldDutiesAfterHours: this.config.cleanupOldDutiesAfterHours,
|
|
219
|
+
maxAgeMs
|
|
220
|
+
});
|
|
221
|
+
}
|
|
222
|
+
}
|
|
223
|
+
}
|
|
183
224
|
}
|
|
184
225
|
}
|
|
@@ -0,0 +1,92 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Vendored pg-compatible Pool/Client wrapper for PGlite.
|
|
3
|
+
*
|
|
4
|
+
* Copied from @middle-management/pglite-pg-adapter v0.0.3
|
|
5
|
+
* https://www.npmjs.com/package/@middle-management/pglite-pg-adapter
|
|
6
|
+
*
|
|
7
|
+
* Modifications:
|
|
8
|
+
* - Converted to ESM and TypeScript
|
|
9
|
+
* - Uses PGliteInterface instead of PGlite class to avoid TypeScript
|
|
10
|
+
* type mismatches from ESM/CJS dual package resolution with private fields
|
|
11
|
+
* - Simplified rowCount calculation to handle CTEs properly
|
|
12
|
+
*/
|
|
13
|
+
import type { PGliteInterface } from '@electric-sql/pglite';
|
|
14
|
+
import { EventEmitter } from 'events';
|
|
15
|
+
import type { QueryResult, QueryResultRow } from 'pg';
|
|
16
|
+
import { Readable, Writable } from 'stream';
|
|
17
|
+
export interface PoolConfig {
|
|
18
|
+
pglite: PGliteInterface;
|
|
19
|
+
max?: number;
|
|
20
|
+
min?: number;
|
|
21
|
+
}
|
|
22
|
+
interface ClientConfig {
|
|
23
|
+
pglite: PGliteInterface;
|
|
24
|
+
host?: string;
|
|
25
|
+
port?: number;
|
|
26
|
+
ssl?: boolean;
|
|
27
|
+
}
|
|
28
|
+
export declare class Client extends EventEmitter {
|
|
29
|
+
protected pglite: PGliteInterface;
|
|
30
|
+
protected _connected: boolean;
|
|
31
|
+
readonly host: string;
|
|
32
|
+
readonly port: number;
|
|
33
|
+
readonly ssl: boolean;
|
|
34
|
+
readonly connection: object;
|
|
35
|
+
readonly copyFrom: () => Writable;
|
|
36
|
+
readonly copyTo: () => Readable;
|
|
37
|
+
readonly pauseDrain: () => void;
|
|
38
|
+
readonly resumeDrain: () => void;
|
|
39
|
+
readonly escapeLiteral: (str: string) => string;
|
|
40
|
+
readonly escapeIdentifier: (str: string) => string;
|
|
41
|
+
readonly setTypeParser: () => void;
|
|
42
|
+
readonly getTypeParser: () => (value: string) => unknown;
|
|
43
|
+
constructor(config: ClientConfig);
|
|
44
|
+
connect(): Promise<void>;
|
|
45
|
+
end(): Promise<void>;
|
|
46
|
+
query<R extends QueryResultRow = any>(text: string, values?: any[]): Promise<QueryResult<R>>;
|
|
47
|
+
protected convertPGliteResult<R extends QueryResultRow>(result: {
|
|
48
|
+
rows: R[];
|
|
49
|
+
fields: Array<{
|
|
50
|
+
name: string;
|
|
51
|
+
dataTypeID: number;
|
|
52
|
+
}>;
|
|
53
|
+
affectedRows?: number;
|
|
54
|
+
}): QueryResult<R>;
|
|
55
|
+
get connected(): boolean;
|
|
56
|
+
}
|
|
57
|
+
export declare class Pool extends EventEmitter {
|
|
58
|
+
private clients;
|
|
59
|
+
private availableClients;
|
|
60
|
+
private waitingQueue;
|
|
61
|
+
private _ended;
|
|
62
|
+
private pglite;
|
|
63
|
+
private _config;
|
|
64
|
+
readonly expiredCount = 0;
|
|
65
|
+
readonly options: PoolConfig;
|
|
66
|
+
constructor(config: PoolConfig);
|
|
67
|
+
get totalCount(): number;
|
|
68
|
+
get idleCount(): number;
|
|
69
|
+
get waitingCount(): number;
|
|
70
|
+
get ending(): boolean;
|
|
71
|
+
get ended(): boolean;
|
|
72
|
+
connect(): Promise<PoolClient>;
|
|
73
|
+
query<R extends QueryResultRow = any>(text: string, values?: any[]): Promise<QueryResult<R>>;
|
|
74
|
+
releaseClient(client: PoolClient): void;
|
|
75
|
+
end(): Promise<void>;
|
|
76
|
+
}
|
|
77
|
+
export declare class PoolClient extends Client {
|
|
78
|
+
private pool;
|
|
79
|
+
private _released;
|
|
80
|
+
private _inUse;
|
|
81
|
+
private _userReleased;
|
|
82
|
+
constructor(pglite: PGliteInterface, pool: Pool);
|
|
83
|
+
query<R extends QueryResultRow = any>(text: string, values?: any[]): Promise<QueryResult<R>>;
|
|
84
|
+
release(): void;
|
|
85
|
+
end(): Promise<void>;
|
|
86
|
+
_markInUse(): void;
|
|
87
|
+
_markAvailable(): void;
|
|
88
|
+
_markReleased(): void;
|
|
89
|
+
get connected(): boolean;
|
|
90
|
+
}
|
|
91
|
+
export {};
|
|
92
|
+
//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoicGdsaXRlX3Bvb2wuZC50cyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uL3NyYy90ZXN0L3BnbGl0ZV9wb29sLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUFBOzs7Ozs7Ozs7OztHQVdHO0FBQ0gsT0FBTyxLQUFLLEVBQUUsZUFBZSxFQUFFLE1BQU0sc0JBQXNCLENBQUM7QUFDNUQsT0FBTyxFQUFFLFlBQVksRUFBRSxNQUFNLFFBQVEsQ0FBQztBQUN0QyxPQUFPLEtBQUssRUFBRSxXQUFXLEVBQUUsY0FBYyxFQUFFLE1BQU0sSUFBSSxDQUFDO0FBQ3RELE9BQU8sRUFBRSxRQUFRLEVBQUUsUUFBUSxFQUFFLE1BQU0sUUFBUSxDQUFDO0FBRTVDLE1BQU0sV0FBVyxVQUFVO0lBQ3pCLE1BQU0sRUFBRSxlQUFlLENBQUM7SUFDeEIsR0FBRyxDQUFDLEVBQUUsTUFBTSxDQUFDO0lBQ2IsR0FBRyxDQUFDLEVBQUUsTUFBTSxDQUFDO0NBQ2Q7QUFFRCxVQUFVLFlBQVk7SUFDcEIsTUFBTSxFQUFFLGVBQWUsQ0FBQztJQUN4QixJQUFJLENBQUMsRUFBRSxNQUFNLENBQUM7SUFDZCxJQUFJLENBQUMsRUFBRSxNQUFNLENBQUM7SUFDZCxHQUFHLENBQUMsRUFBRSxPQUFPLENBQUM7Q0FDZjtBQUVELHFCQUFhLE1BQU8sU0FBUSxZQUFZO0lBQ3RDLFNBQVMsQ0FBQyxNQUFNLEVBQUUsZUFBZSxDQUFDO0lBQ2xDLFNBQVMsQ0FBQyxVQUFVLFVBQVM7SUFDN0IsUUFBUSxDQUFDLElBQUksRUFBRSxNQUFNLENBQUM7SUFDdEIsUUFBUSxDQUFDLElBQUksRUFBRSxNQUFNLENBQUM7SUFDdEIsUUFBUSxDQUFDLEdBQUcsRUFBRSxPQUFPLENBQUM7SUFDdEIsUUFBUSxDQUFDLFVBQVUsRUFBRSxNQUFNLENBQUM7SUFHNUIsUUFBUSxDQUFDLFFBQVEsaUJBQWtDO0lBQ25ELFFBQVEsQ0FBQyxNQUFNLGlCQUFrQztJQUNqRCxRQUFRLENBQUMsVUFBVSxhQUFrQjtJQUNyQyxRQUFRLENBQUMsV0FBVyxhQUFrQjtJQUN0QyxRQUFRLENBQUMsYUFBYSwwQkFBMkQ7SUFDakYsUUFBUSxDQUFDLGdCQUFnQiwwQkFBMkQ7SUFDcEYsUUFBUSxDQUFDLGFBQWEsYUFBa0I7SUFDeEMsUUFBUSxDQUFDLGFBQWEsbUNBQWdFO0lBRXRGLFlBQVksTUFBTSxFQUFFLFlBQVksRUFPL0I7SUFFRCxPQUFPLElBQUksT0FBTyxDQUFDLElBQUksQ0FBQyxDQU92QjtJQUVELEdBQUcsSUFBSSxPQUFPLENBQUMsSUFBSSxDQUFDLENBT25CO0lBRUssS0FBSyxDQUFDLENBQUMsU0FBUyxjQUFjLEdBQUcsR0FBRyxFQUFFLElBQUksRUFBRSxNQUFNLEVBQUUsTUFBTSxDQUFDLEVBQUUsR0FBRyxFQUFFLEdBQUcsT0FBTyxDQUFDLFdBQVcsQ0FBQyxDQUFDLENBQUMsQ0FBQyxDQU1qRztJQUVELFNBQVMsQ0FBQyxtQkFBbUIsQ0FBQyxDQUFDLFNBQVMsY0FBYyxFQUFFLE1BQU0sRUFBRTtRQUM5RCxJQUFJLEVBQUUsQ0FBQyxFQUFFLENBQUM7UUFDVixNQUFNLEVBQUUsS0FBSyxDQUFDO1lBQUUsSUFBSSxFQUFFLE1BQU0sQ0FBQztZQUFDLFVBQVUsRUFBRSxNQUFNLENBQUE7U0FBRSxDQUFDLENBQUM7UUFDcEQsWUFBWSxDQUFDLEVBQUUsTUFBTSxDQUFDO0tBQ3ZCLEdBQUcsV0FBVyxDQUFDLENBQUMsQ0FBQyxDQWdCakI7SUFFRCxJQUFJLFNBQVMsSUFBSSxPQUFPLENBRXZCO0NBQ0Y7QUFFRCxxQkFBYSxJQUFLLFNBQVEsWUFBWTtJQUNwQyxPQUFPLENBQUMsT0FBTyxDQUFvQjtJQUNuQyxPQUFPLENBQUMsZ0JBQWdCLENBQW9CO0lBQzVDLE9BQU8sQ0FBQyxZQUFZLENBQTJDO0lBQy9ELE9BQU8sQ0FBQyxNQUFNLENBQVM7SUFDdkIsT0FBTyxDQUFDLE1BQU0sQ0FBa0I7SUFDaEMsT0FBTyxDQUFDLE9BQU8sQ0FBYTtJQUU1QixRQUFRLENBQUMsWUFBWSxLQUFLO0lBQzFCLFFBQVEsQ0FBQyxPQUFPLEVBQUUsVUFBVSxDQUFDO0lBRTdCLFlBQVksTUFBTSxFQUFFLFVBQVUsRUFLN0I7SUFFRCxJQUFJLFVBQVUsSUFBSSxNQUFNLENBRXZCO0lBRUQsSUFBSSxTQUFTLElBQUksTUFBTSxDQUV0QjtJQUVELElBQUksWUFBWSxJQUFJLE1BQU0sQ0FFekI7SUFFRCxJQUFJLE1BQU0sSUFBSSxPQUFPLENBRXBCO0lBRUQsSUFBSSxLQUFLLElBQUksT0FBTyxDQUVuQjtJQUVELE9BQU8sSUFBSSxPQUFPLENBQUMsVUFBVSxDQUFDLENBb0I3QjtJQUVLLEtBQUssQ0FBQyxDQUFDLFNBQVMsY0FBYyxHQUFHLEdBQUcsRUFBRSxJQUFJLEVBQUUsTUFBTSxFQUFFLE1BQU0sQ0FBQyxFQUFFLEdBQUcsRUFBRSxHQUFHLE9BQU8sQ0FBQyxXQUFXLENBQUMsQ0FBQyxDQUFDLENBQUMsQ0FPakc7SUFFRCxhQUFhLENBQUMsTUFBTSxFQUFFLFVBQVUsR0FBRyxJQUFJLENBWXRDO0lBRUQsR0FBRyxJQUFJLE9BQU8sQ0FBQyxJQUFJLENBQUMsQ0FPbkI7Q0FDRjtBQUVELHFCQUFhLFVBQVcsU0FBUSxNQUFNO0lBQ3BDLE9BQU8sQ0FBQyxJQUFJLENBQU87SUFDbkIsT0FBTyxDQUFDLFNBQVMsQ0FBUztJQUMxQixPQUFPLENBQUMsTUFBTSxDQUFRO0lBQ3RCLE9BQU8sQ0FBQyxhQUFhLENBQVM7SUFFOUIsWUFBWSxNQUFNLEVBQUUsZUFBZSxFQUFFLElBQUksRUFBRSxJQUFJLEVBSTlDO0lBRWMsS0FBSyxDQUFDLENBQUMsU0FBUyxjQUFjLEdBQUcsR0FBRyxFQUFFLElBQUksRUFBRSxNQUFNLEVBQUUsTUFBTSxDQUFDLEVBQUUsR0FBRyxFQUFFLEdBQUcsT0FBTyxDQUFDLFdBQVcsQ0FBQyxDQUFDLENBQUMsQ0FBQyxDQU0xRztJQUVELE9BQU8sSUFBSSxJQUFJLENBTWQ7SUFFUSxHQUFHLElBQUksT0FBTyxDQUFDLElBQUksQ0FBQyxDQUc1QjtJQUVELFVBQVUsSUFBSSxJQUFJLENBR2pCO0lBRUQsY0FBYyxJQUFJLElBQUksQ0FHckI7SUFFRCxhQUFhLElBQUksSUFBSSxDQUlwQjtJQUVELElBQWEsU0FBUyxJQUFJLE9BQU8sQ0FFaEM7Q0FDRiJ9
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"pglite_pool.d.ts","sourceRoot":"","sources":["../../src/test/pglite_pool.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AACH,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAC5D,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,KAAK,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,IAAI,CAAC;AACtD,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,QAAQ,CAAC;AAE5C,MAAM,WAAW,UAAU;IACzB,MAAM,EAAE,eAAe,CAAC;IACxB,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,GAAG,CAAC,EAAE,MAAM,CAAC;CACd;AAED,UAAU,YAAY;IACpB,MAAM,EAAE,eAAe,CAAC;IACxB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,GAAG,CAAC,EAAE,OAAO,CAAC;CACf;AAED,qBAAa,MAAO,SAAQ,YAAY;IACtC,SAAS,CAAC,MAAM,EAAE,eAAe,CAAC;IAClC,SAAS,CAAC,UAAU,UAAS;IAC7B,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,GAAG,EAAE,OAAO,CAAC;IACtB,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAG5B,QAAQ,CAAC,QAAQ,iBAAkC;IACnD,QAAQ,CAAC,MAAM,iBAAkC;IACjD,QAAQ,CAAC,UAAU,aAAkB;IACrC,QAAQ,CAAC,WAAW,aAAkB;IACtC,QAAQ,CAAC,aAAa,0BAA2D;IACjF,QAAQ,CAAC,gBAAgB,0BAA2D;IACpF,QAAQ,CAAC,aAAa,aAAkB;IACxC,QAAQ,CAAC,aAAa,mCAAgE;IAEtF,YAAY,MAAM,EAAE,YAAY,EAO/B;IAED,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC,CAOvB;IAED,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC,CAOnB;IAEK,KAAK,CAAC,CAAC,SAAS,cAAc,GAAG,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,GAAG,EAAE,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAMjG;IAED,SAAS,CAAC,mBAAmB,CAAC,CAAC,SAAS,cAAc,EAAE,MAAM,EAAE;QAC9D,IAAI,EAAE,CAAC,EAAE,CAAC;QACV,MAAM,EAAE,KAAK,CAAC;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,UAAU,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;QACpD,YAAY,CAAC,EAAE,MAAM,CAAC;KACvB,GAAG,WAAW,CAAC,CAAC,CAAC,CAgBjB;IAED,IAAI,SAAS,IAAI,OAAO,CAEvB;CACF;AAED,qBAAa,IAAK,SAAQ,YAAY;IACpC,OAAO,CAAC,OAAO,CAAoB;IACnC,OAAO,CAAC,gBAAgB,CAAoB;IAC5C,OAAO,CAAC,YAAY,CAA2C;IAC/D,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,MAAM,CAAkB;IAChC,OAAO,CAAC,OAAO,CAAa;IAE5B,QAAQ,CAAC,YAAY,KAAK;IAC1B,QAAQ,CAAC,OAAO,EAAE,UAAU,CAAC;IAE7B,YAAY,MAAM,EAAE,UAAU,EAK7B;IAED,IAAI,UAAU,IAAI,MAAM,CAEvB;IAED,IAAI,SAAS,IAAI,MAAM,CAEtB;IAED,IAAI,YAAY,IAAI,MAAM,CAEzB;IAED,IAAI,MAAM,IAAI,OAAO,CAEpB;IAED,IAAI,KAAK,IAAI,OAAO,CAEnB;IAED,OAAO,IAAI,OAAO,CAAC,UAAU,CAAC,CAoB7B;IAEK,KAAK,CAAC,CAAC,SAAS,cAAc,GAAG,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,GAAG,EAAE,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAOjG;IAED,aAAa,CAAC,MAAM,EAAE,UAAU,GAAG,IAAI,CAYtC;IAED,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC,CAOnB;CACF;AAED,qBAAa,UAAW,SAAQ,MAAM;IACpC,OAAO,CAAC,IAAI,CAAO;IACnB,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,MAAM,CAAQ;IACtB,OAAO,CAAC,aAAa,CAAS;IAE9B,YAAY,MAAM,EAAE,eAAe,EAAE,IAAI,EAAE,IAAI,EAI9C;IAEc,KAAK,CAAC,CAAC,SAAS,cAAc,GAAG,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,GAAG,EAAE,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAM1G;IAED,OAAO,IAAI,IAAI,CAMd;IAEQ,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC,CAG5B;IAED,UAAU,IAAI,IAAI,CAGjB;IAED,cAAc,IAAI,IAAI,CAGrB;IAED,aAAa,IAAI,IAAI,CAIpB;IAED,IAAa,SAAS,IAAI,OAAO,CAEhC;CACF"}
|
|
@@ -0,0 +1,210 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Vendored pg-compatible Pool/Client wrapper for PGlite.
|
|
3
|
+
*
|
|
4
|
+
* Copied from @middle-management/pglite-pg-adapter v0.0.3
|
|
5
|
+
* https://www.npmjs.com/package/@middle-management/pglite-pg-adapter
|
|
6
|
+
*
|
|
7
|
+
* Modifications:
|
|
8
|
+
* - Converted to ESM and TypeScript
|
|
9
|
+
* - Uses PGliteInterface instead of PGlite class to avoid TypeScript
|
|
10
|
+
* type mismatches from ESM/CJS dual package resolution with private fields
|
|
11
|
+
* - Simplified rowCount calculation to handle CTEs properly
|
|
12
|
+
*/ import { EventEmitter } from 'events';
|
|
13
|
+
import { Readable, Writable } from 'stream';
|
|
14
|
+
export class Client extends EventEmitter {
|
|
15
|
+
pglite;
|
|
16
|
+
_connected = false;
|
|
17
|
+
host;
|
|
18
|
+
port;
|
|
19
|
+
ssl;
|
|
20
|
+
connection;
|
|
21
|
+
// Stub implementations for pg compatibility
|
|
22
|
+
copyFrom = ()=>new Writable();
|
|
23
|
+
copyTo = ()=>new Readable();
|
|
24
|
+
pauseDrain = ()=>{};
|
|
25
|
+
resumeDrain = ()=>{};
|
|
26
|
+
escapeLiteral = (str)=>`'${str.replace(/'/g, "''")}'`;
|
|
27
|
+
escapeIdentifier = (str)=>`"${str.replace(/"/g, '""')}"`;
|
|
28
|
+
setTypeParser = ()=>{};
|
|
29
|
+
getTypeParser = ()=>(value)=>value;
|
|
30
|
+
constructor(config){
|
|
31
|
+
super();
|
|
32
|
+
this.pglite = config.pglite;
|
|
33
|
+
this.host = config.host || 'localhost';
|
|
34
|
+
this.port = config.port || 5432;
|
|
35
|
+
this.ssl = typeof config.ssl === 'boolean' ? config.ssl : !!config.ssl;
|
|
36
|
+
this.connection = {};
|
|
37
|
+
}
|
|
38
|
+
connect() {
|
|
39
|
+
if (this._connected) {
|
|
40
|
+
return Promise.resolve();
|
|
41
|
+
}
|
|
42
|
+
this._connected = true;
|
|
43
|
+
this.emit('connect');
|
|
44
|
+
return Promise.resolve();
|
|
45
|
+
}
|
|
46
|
+
end() {
|
|
47
|
+
if (!this._connected) {
|
|
48
|
+
return Promise.resolve();
|
|
49
|
+
}
|
|
50
|
+
this._connected = false;
|
|
51
|
+
this.emit('end');
|
|
52
|
+
return Promise.resolve();
|
|
53
|
+
}
|
|
54
|
+
async query(text, values) {
|
|
55
|
+
if (!this._connected) {
|
|
56
|
+
throw new Error('Client is not connected');
|
|
57
|
+
}
|
|
58
|
+
const result = await this.pglite.query(text, values);
|
|
59
|
+
return this.convertPGliteResult(result);
|
|
60
|
+
}
|
|
61
|
+
convertPGliteResult(result) {
|
|
62
|
+
return {
|
|
63
|
+
command: '',
|
|
64
|
+
rowCount: 'affectedRows' in result ? result.affectedRows ?? 0 : result.rows.length,
|
|
65
|
+
oid: 0,
|
|
66
|
+
fields: result.fields.map((field)=>({
|
|
67
|
+
name: field.name,
|
|
68
|
+
tableID: 0,
|
|
69
|
+
columnID: 0,
|
|
70
|
+
dataTypeID: field.dataTypeID,
|
|
71
|
+
dataTypeSize: -1,
|
|
72
|
+
dataTypeModifier: -1,
|
|
73
|
+
format: 'text'
|
|
74
|
+
})),
|
|
75
|
+
rows: result.rows
|
|
76
|
+
};
|
|
77
|
+
}
|
|
78
|
+
get connected() {
|
|
79
|
+
return this._connected;
|
|
80
|
+
}
|
|
81
|
+
}
|
|
82
|
+
export class Pool extends EventEmitter {
|
|
83
|
+
clients = [];
|
|
84
|
+
availableClients = [];
|
|
85
|
+
waitingQueue = [];
|
|
86
|
+
_ended = false;
|
|
87
|
+
pglite;
|
|
88
|
+
_config;
|
|
89
|
+
expiredCount = 0;
|
|
90
|
+
options;
|
|
91
|
+
constructor(config){
|
|
92
|
+
super();
|
|
93
|
+
this._config = {
|
|
94
|
+
max: 10,
|
|
95
|
+
min: 0,
|
|
96
|
+
...config
|
|
97
|
+
};
|
|
98
|
+
this.pglite = config.pglite;
|
|
99
|
+
this.options = config;
|
|
100
|
+
}
|
|
101
|
+
get totalCount() {
|
|
102
|
+
return this.clients.length;
|
|
103
|
+
}
|
|
104
|
+
get idleCount() {
|
|
105
|
+
return this.availableClients.length;
|
|
106
|
+
}
|
|
107
|
+
get waitingCount() {
|
|
108
|
+
return this.waitingQueue.length;
|
|
109
|
+
}
|
|
110
|
+
get ending() {
|
|
111
|
+
return this._ended;
|
|
112
|
+
}
|
|
113
|
+
get ended() {
|
|
114
|
+
return this._ended;
|
|
115
|
+
}
|
|
116
|
+
connect() {
|
|
117
|
+
if (this._ended) {
|
|
118
|
+
return Promise.reject(new Error('Pool is ended'));
|
|
119
|
+
}
|
|
120
|
+
if (this.availableClients.length > 0) {
|
|
121
|
+
const client = this.availableClients.pop();
|
|
122
|
+
client._markInUse();
|
|
123
|
+
return Promise.resolve(client);
|
|
124
|
+
}
|
|
125
|
+
if (this.clients.length < (this._config.max || 10)) {
|
|
126
|
+
const client = new PoolClient(this.pglite, this);
|
|
127
|
+
this.clients.push(client);
|
|
128
|
+
return Promise.resolve(client);
|
|
129
|
+
}
|
|
130
|
+
return new Promise((resolve)=>{
|
|
131
|
+
this.waitingQueue.push(resolve);
|
|
132
|
+
});
|
|
133
|
+
}
|
|
134
|
+
async query(text, values) {
|
|
135
|
+
const client = await this.connect();
|
|
136
|
+
try {
|
|
137
|
+
return await client.query(text, values);
|
|
138
|
+
} finally{
|
|
139
|
+
client.release();
|
|
140
|
+
}
|
|
141
|
+
}
|
|
142
|
+
releaseClient(client) {
|
|
143
|
+
const index = this.clients.indexOf(client);
|
|
144
|
+
if (index !== -1) {
|
|
145
|
+
client._markAvailable();
|
|
146
|
+
if (this.waitingQueue.length > 0) {
|
|
147
|
+
const resolve = this.waitingQueue.shift();
|
|
148
|
+
client._markInUse();
|
|
149
|
+
resolve(client);
|
|
150
|
+
} else {
|
|
151
|
+
this.availableClients.push(client);
|
|
152
|
+
}
|
|
153
|
+
}
|
|
154
|
+
}
|
|
155
|
+
end() {
|
|
156
|
+
this._ended = true;
|
|
157
|
+
this.clients.forEach((client)=>client._markReleased());
|
|
158
|
+
this.clients = [];
|
|
159
|
+
this.availableClients = [];
|
|
160
|
+
this.emit('end');
|
|
161
|
+
return Promise.resolve();
|
|
162
|
+
}
|
|
163
|
+
}
|
|
164
|
+
export class PoolClient extends Client {
|
|
165
|
+
pool;
|
|
166
|
+
_released = false;
|
|
167
|
+
_inUse = true;
|
|
168
|
+
_userReleased = false;
|
|
169
|
+
constructor(pglite, pool){
|
|
170
|
+
super({
|
|
171
|
+
pglite
|
|
172
|
+
});
|
|
173
|
+
this.pool = pool;
|
|
174
|
+
this._connected = true;
|
|
175
|
+
}
|
|
176
|
+
async query(text, values) {
|
|
177
|
+
if (this._userReleased && !this._inUse) {
|
|
178
|
+
throw new Error('Client has been released back to the pool');
|
|
179
|
+
}
|
|
180
|
+
const result = await this.pglite.query(text, values);
|
|
181
|
+
return this.convertPGliteResult(result);
|
|
182
|
+
}
|
|
183
|
+
release() {
|
|
184
|
+
if (this._released || this._userReleased) {
|
|
185
|
+
return;
|
|
186
|
+
}
|
|
187
|
+
this._userReleased = true;
|
|
188
|
+
this.pool.releaseClient(this);
|
|
189
|
+
}
|
|
190
|
+
end() {
|
|
191
|
+
this.release();
|
|
192
|
+
return Promise.resolve();
|
|
193
|
+
}
|
|
194
|
+
_markInUse() {
|
|
195
|
+
this._inUse = true;
|
|
196
|
+
this._userReleased = false;
|
|
197
|
+
}
|
|
198
|
+
_markAvailable() {
|
|
199
|
+
this._inUse = false;
|
|
200
|
+
this._userReleased = false;
|
|
201
|
+
}
|
|
202
|
+
_markReleased() {
|
|
203
|
+
this._released = true;
|
|
204
|
+
this._inUse = false;
|
|
205
|
+
this._userReleased = true;
|
|
206
|
+
}
|
|
207
|
+
get connected() {
|
|
208
|
+
return this._connected && !this._released;
|
|
209
|
+
}
|
|
210
|
+
}
|