@render-foundation/utils 0.0.78 → 0.0.79

package/lib/esm/src/client/pg/v2Client.js

@@ -0,0 +1,710 @@
+import { Kysely, NoResultError, PostgresDialect, sql, } from 'kysely';
+import { consoleLogger } from '../../logger';
+import { Pool } from 'pg';
+export const SOL_TX_TABLE = 'sol_tx';
+export const EPOCH_TABLE = 'epoch';
+export const MANUAL_BURN_TABLE = 'manual_burn';
+export const ENTITY_TABLE = 'entity';
+export const SOL_TRANSFER_TABLE = 'sol_transfer1';
+export const LIABILITY_TABLE = 'liability';
+export const JOB_ID_TABLE = 'job_id';
+export const JOB_TABLE = 'job';
+export const BURN_TABLE = 'burn';
+BigInt.prototype.toJSON = function () {
+    return this.toString();
+};
+export const pgClient = (config) => {
+    const dialect = new PostgresDialect({
+        pool: new Pool({
+            ...config,
+            port: config.port ?? 5432,
+            max: config.max ?? 10,
+        }),
+    });
+    // Database interface is passed to Kysely's constructor, and from now on, Kysely
+    // knows your database structure.
+    // Dialect is passed to Kysely's constructor, and from now on, Kysely knows how
+    // to communicate with your database.
+    const db = new Kysely({
+        dialect,
+    });
+    const getOrInsertEntity = async (db, p, log = consoleLogger) => {
+        const res = (await db
+            .selectFrom(ENTITY_TABLE)
+            .select('id')
+            .where('sol_key', '=', p.solKey)
+            .executeTakeFirst());
+        if (res) {
+            log.debug(`already found entity ${res.id}`);
+            return res.id;
+        }
+        const { id } = (await db
+            .insertInto(ENTITY_TABLE)
+            .values({
+            sol_key: p.solKey,
+        })
+            .onConflict((oc) => oc.column('sol_key').doNothing())
+            .returning((eb) => ['id as id'])
+            .executeTakeFirst());
+        log.debug(`inserted entity ${id}`);
+        return id;
+    };
+    const getOrInsertTx = async (db, p, log = consoleLogger) => {
+        const res = await db
+            .selectFrom(SOL_TX_TABLE)
+            .select('id')
+            .where('sig', '=', p.sig)
+            .executeTakeFirst();
+        if (res) {
+            log.debug(`already found tx ${res.id}`);
+            return res.id;
+        }
+        const { id } = (await db
+            .insertInto(SOL_TX_TABLE)
+            .values({
+            sig: p.sig,
+            executed_at: p.executedAt,
+        })
+            .onConflict((oc) => oc.column('sig').doNothing())
+            .returning((eb) => ['id as id'])
+            .executeTakeFirst());
+        log.debug(`inserted tx ${id}`);
+        return id;
+    };
+    const getOrInsertEpoch = async (db, p, log = consoleLogger) => {
+        const res = await db
+            .selectFrom(EPOCH_TABLE)
+            .select('id')
+            .where('ui', '=', String(p.uiEpochId))
+            .where('channel', '=', p.channel)
+            .executeTakeFirst();
+        if (res) {
+            log.debug(`already found epoch ${res.id}`);
+            return res.id;
+        }
+        const { id } = (await db
+            .insertInto(EPOCH_TABLE)
+            .values({
+            ui: p.uiEpochId,
+            channel: p.channel,
+            start_time: p.startTime,
+            end_time: p.endTime,
+            saved_at: p.channel == 'node_operator' ? new Date() : undefined,
+        })
+            .onConflict((oc) => oc.columns(['ui', 'channel']).doNothing())
+            .returning((eb) => ['id as id'])
+            .executeTakeFirst());
+        log.debug(`inserted epoch ${id}`);
+        return id;
+    };
+    const updateTotalRndr = async (db, p, totalRndr, savedAt, log) => {
+        const res = (await db
+            .updateTable(EPOCH_TABLE)
+            .set({
+            total_rndr: totalRndr,
+            saved_at: savedAt,
+        })
+            .where('channel', '=', p.channel)
+            .where('ui', '=', String(p.uiEpochId))
+            .executeTakeFirst());
+        log.info(`set ${res.numUpdatedRows} rows. channel: ${p.channel}, ui: ${p.uiEpochId} -> totalRndr ${totalRndr} `);
+    };
+    const createPayments = async (db, ps, log = consoleLogger) => {
+        const ids = [];
+        const epochDbIds = [];
+        for (const p of ps) {
+            let liabilityId;
+            let transferId;
+            if (p.liability) {
+                let { id: dbId, solKey, channel, epochId, createdAt, amountDue, deviceId, points, } = p.liability;
+                try {
+                    let entityId;
+                    if (solKey && solKey != '') {
+                        entityId = await getOrInsertEntity(db, { solKey });
+                    }
+                    const dbEpochId = await getOrInsertEpoch(db, {
+                        channel,
+                        uiEpochId: epochId,
+                    }, log);
+                    epochDbIds.push(BigInt(dbEpochId));
+                    log.debug(`inserting liability ${JSON.stringify(p.liability)}`);
+                    const { id } = await db
+                        .insertInto(LIABILITY_TABLE)
+                        .values({
+                        id: dbId,
+                        entity_id: entityId,
+                        epoch_id: dbEpochId,
+                        channel,
+                        created_at: createdAt ?? new Date(),
+                        amount_due: amountDue,
+                        device_id: deviceId,
+                        points,
+                    })
+                        .returning((eb) => ['id as id'])
+                        .executeTakeFirstOrThrow();
+                    log.debug(`inserted liability ${id}`);
+                    liabilityId = BigInt(id);
+                }
+                catch (e) {
+                    throw new Error(`failed inserting solKey ${solKey} epochId ${epochId} ${JSON.stringify(p.liability)}: ${e}`);
+                }
+            }
+            if (p.transfer) {
+                let { executedAt, symbol, amountPayed, solTx, channel, fromSolKey, toSolKey, id: dbId, } = p.transfer;
+                if (!fromSolKey || !toSolKey || !solTx) {
+                    log.debug(`skipping transfer. one of undefined: fromKey ${fromSolKey} toKey ${toSolKey} tx ${solTx}`);
+                    continue;
+                }
+                log.debug(`inserting transfer ${JSON.stringify(p.transfer)}`);
+                const fromEntityId = await getOrInsertEntity(db, { solKey: fromSolKey });
+                const toEntityId = await getOrInsertEntity(db, { solKey: toSolKey });
+                const solTxId = await getOrInsertTx(db, {
+                    sig: solTx,
+                    executedAt,
+                }, log);
+                const { id } = await db
+                    .insertInto(SOL_TRANSFER_TABLE)
+                    .values({
+                    id: dbId,
+                    sol_tx_id: solTxId,
+                    from_entity_id: fromEntityId,
+                    to_entity_id: toEntityId,
+                    channel,
+                    symbol,
+                    amount_payed: amountPayed,
+                })
+                    .returning((eb) => ['id as id'])
+                    .executeTakeFirstOrThrow();
+                transferId = BigInt(id);
+            }
+            ids.push({ transferId, liabilityId });
+        }
+        return { epochDbIds, inserted: ids };
+    };
+    return {
+        db: {
+            commit: () => { },
+            rollback: () => { },
+            trx: db,
+        },
+        async startTransaction() {
+            return begin(db);
+        },
+        async createPayments(db, ps, log = consoleLogger) {
+            return createPayments(db.trx, ps, log);
+        },
+        async fetchTransfers(db, f) {
+            let offset = 0;
+            if (f.cursor) {
+                offset = Number(atob(f.cursor));
+            }
+            let q = db.trx
+                .selectFrom(SOL_TRANSFER_TABLE)
+                .selectAll([SOL_TRANSFER_TABLE])
+                .innerJoin('sol_tx', 'sol_tx.id', 'sol_transfer1.sol_tx_id')
+                .select(['sol_tx.sig', 'sol_tx.executed_at'])
+                .innerJoin('entity as f', 'f.id', 'sol_transfer1.from_entity_id')
+                .select(['f.sol_key as from_sol_key'])
+                .innerJoin('entity as t', 't.id', 'sol_transfer1.to_entity_id')
+                .select(['t.sol_key as to_sol_key']);
+            if (f.executedAfter) {
+                q = q.where('sol_tx.executed_at', '>=', f.executedAfter);
+            }
+            if (f.executedBefore) {
+                q = q.where('sol_tx.executed_at', '<=', f.executedBefore);
+            }
+            if (f.channel) {
+                q = q.where('sol_transfer1.channel', '=', f.channel);
+            }
+            let limit = 10000;
+            if (f.limit) {
+                limit = f.limit;
+            }
+            q = q.limit(limit).orderBy(`${SOL_TRANSFER_TABLE}.id`);
+            if (offset) {
+                q = q.offset(offset);
+            }
+            console.log(`fetching transactions ${q.compile().sql} ps ${JSON.stringify(f)}`);
+            const rows = await q.execute();
+            const trs = [];
+            for (const r of rows) {
+                trs.push({
+                    id: BigInt(r.id),
+                    fromSolKey: r.from_sol_key,
+                    toSolKey: r.to_sol_key,
+                    symbol: r.symbol,
+                    channel: r.channel,
+                    solTx: r.sig,
+                    executedAt: r.executed_at,
+                    amountPayed: BigInt(r.amount_payed),
+                });
+            }
+            return {
+                transfers: trs,
+                cursor: trs.length == limit ? btoa(`${trs.length + offset}`) : undefined,
+            };
+        },
+        async getLiabilitiesAndTransfers(f, log = consoleLogger) {
+            let lq = db
+                .selectFrom(LIABILITY_TABLE)
+                .selectAll()
+                .innerJoin(ENTITY_TABLE, `${LIABILITY_TABLE}.entity_id`, 'entity.id')
+                .select(['entity.sol_key'])
+                .innerJoin(EPOCH_TABLE, `${EPOCH_TABLE}.id`, 'liability.epoch_id')
+                .select(['epoch.ui', 'epoch.start_time', 'epoch.end_time'])
+                .where('entity.sol_key', '=', f.user);
+            let tq = db
+                .selectFrom(SOL_TRANSFER_TABLE)
+                .selectAll()
+                .innerJoin(SOL_TX_TABLE, 'sol_tx.id', 'sol_transfer1.sol_tx_id')
+                .select(['sol_tx.sig', 'sol_tx.executed_at'])
+                .innerJoin(`${ENTITY_TABLE} as t`, `${SOL_TRANSFER_TABLE}.to_entity_id`, 't.id')
+                .select(['t.sol_key as to_sol_key'])
+                .innerJoin(`${ENTITY_TABLE} as f`, `${SOL_TRANSFER_TABLE}.from_entity_id`, 'f.id')
+                .select(['f.sol_key as from_sol_key'])
+                .where('t.sol_key', '=', f.user);
+            log.info(`fetch liabilities ${lq.compile().sql} ps ${JSON.stringify(f)}`);
+            log.info(`fetch transfers ${tq.compile().sql} ps ${JSON.stringify(f)}`);
+            const rows = await Promise.all([lq.execute(), tq.execute()]);
+            const epochs = {};
+            const liabilities = [];
+            for (const l of rows[0]) {
+                epochs[`${l.ui}-${l.channel}`] = {
+                    startTime: l.start_time,
+                    endTime: l.end_time,
+                    channel: l.channel,
+                    ui: Number(l.ui),
+                };
+                liabilities.push({
+                    id: BigInt(l.id),
+                    solKey: l.sol_key,
+                    epochId: Number(l.ui),
+                    channel: l.channel,
+                    createdAt: l.created_at,
+                    amountDue: BigInt(l.amount_due),
+                    deviceId: l.device_id,
+                    points: l.points ? BigInt(l.points) : undefined,
+                });
+            }
+            const transfers = rows[1].map((r) => ({
+                id: BigInt(r.id),
+                fromSolKey: r.from_sol_key,
+                toSolKey: r.to_sol_key,
+                symbol: r.symbol,
+                channel: r.channel,
+                solTx: r.sig,
+                executedAt: r.executed_at,
+                amountPayed: BigInt(r.amount_payed),
+            }));
+            return { transfers, liabilities, epochs: Object.values(epochs) };
+        },
+        async getOutstanding(f, log = consoleLogger) {
+            let q = db
+                .selectFrom((eb) => eb
+                .selectFrom((eb) => {
+                let b = eb
+                    .selectFrom(SOL_TRANSFER_TABLE)
+                    .select(({ fn }) => fn.sum(`amount_payed`).as('payed'))
+                    .innerJoin(ENTITY_TABLE, `${SOL_TRANSFER_TABLE}.to_entity_id`, 'entity.id')
+                    .select(['entity.sol_key'])
+                    .groupBy('entity.sol_key');
+                if (f.banned) {
+                    b = b.where('entity.banned', '=', true);
+                }
+                else {
+                    b = b.where('entity.banned', '=', false);
+                }
+                if (f.channels && f.channels.length > 0) {
+                    b = b.where((eb) => eb.or(f.channels.map((c) => eb('sol_transfer1.channel', '=', c))));
+                }
+                else {
+                    b = b.where((eb) => eb.or([
+                        eb('sol_transfer1.channel', '=', 'node_operator'),
+                        eb('sol_transfer1.channel', '=', 'availability'),
+                    ]));
+                }
+                return b.as('y');
+            })
+                .fullJoin((eb) => {
+                let b = eb
+                    .selectFrom(LIABILITY_TABLE)
+                    .select(({ fn }) => fn.sum(`amount_due`).as('due'))
+                    .innerJoin(ENTITY_TABLE, `${LIABILITY_TABLE}.entity_id`, 'entity.id')
+                    .select(['entity.sol_key'])
+                    .groupBy('entity.sol_key');
+                if (f.banned) {
+                    b = b.where('entity.banned', '=', true);
+                }
+                else {
+                    b = b.where('entity.banned', '=', false);
+                }
+                if (f.channels && f.channels.length > 0) {
+                    b = b.where((eb) => eb.or(f.channels.map((c) => eb('liability.channel', '=', c))));
+                }
+                else {
+                    b = b.where((eb) => eb.or([
+                        eb('liability.channel', '=', 'node_operator'),
+                        eb('liability.channel', '=', 'availability'),
+                    ]));
+                }
+                return b.as('x');
+            }, (join) => join.onRef('y.sol_key', '=', 'x.sol_key'))
+                .select(({ fn }) => [
+                fn.coalesce('y.sol_key', 'x.sol_key').as('sol_key'),
+                sql `${fn.coalesce(`x.due`, sql `0`)} - ${fn.coalesce(`y.payed`, sql `0`)}`.as('out'),
+            ])
+                .as('z'))
+                .selectAll();
+            if (f.due == 'overpayed') {
+                q = q.where('out', '<', '0');
+            }
+            else if (f.due == 'underpayed') {
+                q = q.where('out', '>', '0');
+            }
+            log.info(`${q.compile().sql} ps ${JSON.stringify(f)}`);
+            const res = await q.execute();
+            const out = {};
+            for (const r of res) {
+                out[r.sol_key] = BigInt(r.out);
+            }
+            return out;
+        },
+        async fetchNodeOperatorEpoch(epochId, log = consoleLogger) {
+            const q = db
+                .selectFrom(JOB_ID_TABLE)
+                .innerJoin(EPOCH_TABLE, 'epoch.id', 'job_id.epoch_id')
+                .leftJoin(MANUAL_BURN_TABLE, 'manual_burn.epoch_id', 'epoch.id')
+                .where('epoch.ui', '=', epochId.toString())
+                .where('epoch.channel', '=', 'node_operator')
+                .select([
+                'job_id.id',
+                'epoch.total_rndr',
+                'epoch.start_time',
+                'epoch.end_time',
+                'manual_burn.burned',
+                'manual_burn.tx_sig',
+            ])
+                .orderBy('job_id.id asc');
+            log.debug(`fetch epoch info q ${q.compile().sql} epochId ${epochId}`);
+            const res = await q.execute();
+            if (res.length == 0) {
+                throw new Error(`epoch ${epochId} not found`);
+            }
+            log.debug(`fetchNodeOperatorEpoch -> fetching liablities`);
+            const ks = await db
+                .selectFrom(LIABILITY_TABLE)
+                .innerJoin(ENTITY_TABLE, 'liability.entity_id', 'entity.id')
+                .select(({ fn }) => [sql `distinct(entity.sol_key)`.as('sol_key')])
+                .innerJoin(EPOCH_TABLE, 'liability.epoch_id', 'epoch.id')
+                .where('epoch.ui', '=', epochId.toString())
+                .where('epoch.channel', '=', 'node_operator')
+                .where('liability.channel', '=', 'node_operator')
+                .execute();
+            log.debug(`fetchNodeOperatorEpoch -> fetched liablities`);
+            const solKeys = ks.reduce((r, k) => {
+                r[k.sol_key] = {};
+                return r;
+            }, {});
+            return {
+                jobIds: res.map((r) => Number(r.id)),
+                totalRndr: BigInt(res[0].total_rndr),
+                startTime: res[0].start_time,
+                endTime: res[0].end_time,
+                burned: BigInt(res[0].burned ?? 0),
+                burnTx: res[0].tx_sig,
+                solKeys,
+            };
+        },
+        async getLatestNodeOperatorEpoch(log = consoleLogger) {
+            const now = new Date(Date.now());
+            try {
+                const res = await db
+                    .selectFrom(EPOCH_TABLE)
+                    .selectAll()
+                    .where('channel', '=', 'node_operator')
+                    .where('end_time', '<=', now)
+                    .where('saved_at', 'is', null)
+                    .orderBy('end_time asc')
+                    .limit(1)
+                    .executeTakeFirstOrThrow();
+                return {
+                    epochId: BigInt(res.ui),
+                    startedAt: res.start_time,
+                    endedAt: res.end_time,
+                    savedAt: res.saved_at ?? undefined,
+                };
+            }
+            catch (e) {
+                if (e instanceof NoResultError) {
+                    log.info(`no node operator epoch found`);
+                    return undefined;
+                }
+                throw e;
+            }
+        },
+        async saveNodeOperatorEpoch(epochId, liabilities, jobIds, totalRndr, log = consoleLogger) {
+            await db
+                .transaction()
+                .setIsolationLevel('repeatable read')
+                .execute(async (trx) => {
+                const { epochDbIds } = await createPayments(trx, liabilities.map((l) => ({
+                    liability: l,
+                })), log);
+                log.info(`inserted liabilities`);
+                for (const id of jobIds) {
+                    await trx
+                        .insertInto(JOB_ID_TABLE)
+                        .values({
+                        id: BigInt(id),
+                        epoch_id: epochDbIds[0],
+                    })
+                        .executeTakeFirstOrThrow();
+                }
+                const now = new Date();
+                await updateTotalRndr(db, { uiEpochId: epochId, channel: 'node_operator' }, totalRndr, now, log);
+                log.debug(`saved epoch`);
+                /*const now = new Date(Date.now())
+                const res = await trx
+                  .updateTable(EPOCH_TABLE)
+                  .set({
+                    saved_at: now,
+                    total_rndr: totalRndr,
+                  })
+                  .where('channel')
+                  .where('ui', '=', epochId.toString())
+                  .where('', '=')
+                  .executeTakeFirstOrThrow()*/
+            });
+            return;
+        },
+        async produceBurnedJobs(jobs, log = consoleLogger) {
+            await db.transaction().execute(async (trx) => {
+                let num = BigInt(0);
+                for (const { amount, completedAt, jobId } of jobs) {
+                    const res = await trx
+                        .insertInto(JOB_TABLE)
+                        .values({
+                        id: jobId,
+                        render_amt: amount,
+                        completed_at: completedAt,
+                        inserted_at: new Date(Date.now()),
+                    })
+                        .onConflict((oc) => oc.column('id').doNothing())
+                        .executeTakeFirstOrThrow();
+                    num += res.numInsertedOrUpdatedRows ?? BigInt(0);
+                }
+                log.debug(`inserted ${num} of ${jobs.length} burn jobs`);
+            });
+        },
+        async markJobsBurnt(trx, p, log = consoleLogger) {
+            const solTxId = await getOrInsertTx(trx, { sig: p.solTx, executedAt: p.executedAt }, log);
+            const { id } = await trx
+                .insertInto(BURN_TABLE)
+                .values({
+                sol_tx_id: solTxId,
+                priced_at: p.pricedAt,
+                usdc_spent: p.usdcSpent,
+                quote_amt: p.quoteAmt,
+                burned: p.burned,
+            })
+                //.onConflict((oc) => oc.column('sol_tx').doNothing())
+                .returning((eb) => ['id as id'])
+                .executeTakeFirstOrThrow();
+            const jobIdsStr = p.jobsIds.map(String);
+            const res = await trx
+                .updateTable(JOB_TABLE)
+                .where('id', 'in', jobIdsStr)
+                .set('burn_id', id)
+                .executeTakeFirstOrThrow();
+            log.debug(`set burn_id on ${res.numUpdatedRows} jobs`);
+            return;
+        },
+        async markJobsBurned(p, log = consoleLogger) {
+            let { markedAt, solTx } = p;
+            let res = await db
+                .updateTable(BURN_TABLE)
+                .set({ marked_burned_at: markedAt })
+                .from(SOL_TX_TABLE)
+                .whereRef('burn.sol_tx_id', '=', 'sol_tx.id')
+                .where('sol_tx.sig', '=', solTx)
+                .executeTakeFirstOrThrow();
+            log.debug(`set ${res.numUpdatedRows} ${solTx} -> marked_burned_at ${markedAt.toISOString()}`);
+        },
+        async consumeAndLockBurnJobs(limit, lowerRndrThreshold, upperRndrThreshold, log = consoleLogger) {
+            const trx = await begin(db);
+            const q = trx.trx
+                .selectFrom(trx.trx
+                .selectFrom(JOB_TABLE)
+                .select(({ fn, val, ref }) => [
+                'id',
+                fn
+                    .sum('render_amt')
+                    .over((ob) => ob.orderBy(`completed_at`, `asc`))
+                    .as('total'),
+            ])
+                .where('burn_id', 'is', null)
+                .as('t'))
+                .select('id')
+                .where('total', '>', lowerRndrThreshold)
+                .where('total', '<', upperRndrThreshold);
+            log.debug(`outstanding jobs q: ${q.compile().sql} ${JSON.stringify({
+                limit,
+                lowerRndrThreshold,
+                upperRndrThreshold,
+            })}`);
+            const idRs = await q.execute();
+            const ids = idRs.map((x) => x.id);
+            log.debug(`${ids.length} candidate jobs for burn. locking up to ${limit} jobs`);
+            if (ids.length == 0) {
+                return { trx, jobs: [] };
+            }
+            const rows = await trx.trx
+                .selectFrom(JOB_TABLE)
+                .selectAll()
+                .where('id', 'in', ids)
+                .limit(limit)
+                .forUpdate()
+                .skipLocked()
+                .execute();
+            log.debug(`locked ${rows.length} jobs`);
+            const jobs = [];
+            for (const r of rows) {
+                jobs.push({
+                    id: Number(r.id),
+                    completedAt: r.completed_at,
+                    rndrUsed: BigInt(r.render_amt),
+                });
+            }
+            return { trx, jobs };
+        },
+        async incrBuyBurnMetrics(log = consoleLogger) {
+            const res = await db
+                .selectFrom(JOB_TABLE)
+                .select('completed_at')
+                .orderBy('completed_at asc')
+                .where('burn_id', 'is', null)
+                .execute();
+            if (res.length == 0) {
+                return { outstandingJobs: 0 };
+            }
+            return {
+                outstandingJobs: res.length,
+                oldestJobFinishedAt: res[0].completed_at ?? undefined,
+            };
+        },
+        async getLiabilitySolKeys(f) {
+            let q = db
+                .selectFrom(LIABILITY_TABLE)
+                .innerJoin(ENTITY_TABLE, `${LIABILITY_TABLE}.entity_id`, 'entity.id')
+                .select(({ fn }) => [sql `distinct(entity.sol_key)`.as('sol_key')])
+                .where(`${LIABILITY_TABLE}.channel`, '=', f.channel);
+            if (f.epoch) {
+                q = q.where('liability.epoch_id', '=', db
+                    .selectFrom('epoch')
+                    .select('id')
+                    .where('channel', '=', f.channel)
+                    .where('ui', '=', String(f.epoch)));
+            }
+            const ks = await q.execute();
+            const solKeys = {};
+            for (const k of ks) {
+                solKeys[k.sol_key] = {};
+            }
+            return solKeys;
+        },
+        async deleteAll(log = consoleLogger) {
+            if (config.database.includes('render')) {
+                throw new Error('cannot nuke from deployed database. call from tests only');
+            }
+            const tx = await begin(db);
+            try {
+                const res = await Promise.all([
+                    tx.trx
+                        .deleteFrom(SOL_TX_TABLE)
+                        .where('id', 'is not', null)
+                        .executeTakeFirst(),
+                    tx.trx.deleteFrom(LIABILITY_TABLE).executeTakeFirstOrThrow(),
+                    tx.trx.deleteFrom(SOL_TRANSFER_TABLE).executeTakeFirstOrThrow(),
+                    tx.trx
+                        .deleteFrom(EPOCH_TABLE)
+                        .where('id', 'is not', null)
+                        .executeTakeFirst(),
+                    tx.trx
+                        .deleteFrom(ENTITY_TABLE)
+                        .where('id', 'is not', null)
+                        .executeTakeFirst(),
+                    tx.trx
+                        .deleteFrom(MANUAL_BURN_TABLE)
+                        .where('epoch_id', 'is not', null)
+                        .executeTakeFirst(),
+                    tx.trx
+                        .deleteFrom(JOB_ID_TABLE)
+                        .where('id', 'is not', null)
+                        .executeTakeFirst(),
+                    tx.trx
+                        .deleteFrom(JOB_TABLE)
+                        .where('id', 'is not', null)
+                        .executeTakeFirst(),
+                    tx.trx
+                        .deleteFrom(BURN_TABLE)
+                        .where('id', 'is not', null)
+                        .executeTakeFirst(),
+                ]);
+                tx.commit();
+            }
+            catch (e) {
+                log.error(`caught ${e}`);
+                tx.rollback();
+            }
+        },
+    };
+};
+async function begin(db) {
+    const connection = new Deferred();
+    const result = new Deferred();
+    // Do NOT await this line.
+    db.transaction()
+        .setIsolationLevel('repeatable read')
+        .execute((trx) => {
+        connection.resolve(trx);
+        return result.promise;
+    })
+        .catch((err) => {
+        // Don't do anything here. Just swallow the exception.
+    });
+    const trx = await connection.promise;
+    return {
+        trx,
+        commit() {
+            result.resolve(null);
+        },
+        rollback() {
+            result.reject(new Error('rollback'));
+        },
+    };
+}
+export class Deferred {
+    #promise;
+    #resolve;
+    #reject;
+    constructor() {
+        this.#promise = new Promise((resolve, reject) => {
+            this.#reject = reject;
+            this.#resolve = resolve;
+        });
+    }
+    get promise() {
+        return this.#promise;
+    }
+    resolve = (value) => {
+        if (this.#resolve) {
+            this.#resolve(value);
+        }
+    };
+    reject = (reason) => {
+        if (this.#reject) {
+            this.#reject(reason);
+        }
+    };
+}
+//# sourceMappingURL=v2Client.js.map