@render-foundation/utils 0.0.59 → 0.0.61

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.
Files changed (37) hide show
  1. package/lib/cjs/client/{pg.js → pg/v1.js} +18 -214
  2. package/lib/cjs/client/pg/v1.js.map +1 -0
  3. package/lib/cjs/client/pg/v2.js +714 -0
  4. package/lib/cjs/client/pg/v2.js.map +1 -0
  5. package/lib/cjs/client/render.js +26 -4
  6. package/lib/cjs/client/render.js.map +1 -1
  7. package/lib/cjs/dbTypesV2.js +3 -0
  8. package/lib/cjs/dbTypesV2.js.map +1 -0
  9. package/lib/cjs/index.js +15 -1
  10. package/lib/cjs/index.js.map +1 -1
  11. package/lib/esm/src/client/{pg.js → pg/v1.js} +13 -192
  12. package/lib/esm/src/client/pg/v1.js.map +1 -0
  13. package/lib/esm/src/client/pg/v2.js +653 -0
  14. package/lib/esm/src/client/pg/v2.js.map +1 -0
  15. package/lib/esm/src/client/render.js +17 -0
  16. package/lib/esm/src/client/render.js.map +1 -1
  17. package/lib/esm/src/dbTypesV2.js +2 -0
  18. package/lib/esm/src/dbTypesV2.js.map +1 -0
  19. package/lib/esm/src/index.js +2 -1
  20. package/lib/esm/src/index.js.map +1 -1
  21. package/lib/esm/tsconfig.esm.tsbuildinfo +1 -1
  22. package/lib/types/src/client/{pg.d.ts → pg/v1.d.ts} +12 -51
  23. package/lib/types/src/client/pg/v1.d.ts.map +1 -0
  24. package/lib/types/src/client/pg/v2.d.ts +35 -0
  25. package/lib/types/src/client/pg/v2.d.ts.map +1 -0
  26. package/lib/types/src/client/render.d.ts +3 -0
  27. package/lib/types/src/client/render.d.ts.map +1 -1
  28. package/lib/types/src/dbTypes.d.ts +5 -0
  29. package/lib/types/src/dbTypes.d.ts.map +1 -1
  30. package/lib/types/src/dbTypesV2.d.ts +83 -0
  31. package/lib/types/src/dbTypesV2.d.ts.map +1 -0
  32. package/lib/types/src/index.d.ts +2 -1
  33. package/lib/types/src/index.d.ts.map +1 -1
  34. package/package.json +1 -1
  35. package/lib/cjs/client/pg.js.map +0 -1
  36. package/lib/esm/src/client/pg.js.map +0 -1
  37. package/lib/types/src/client/pg.d.ts.map +0 -1
@@ -0,0 +1,714 @@
1
+ "use strict";
2
+ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
3
+ function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
4
+ return new (P || (P = Promise))(function (resolve, reject) {
5
+ function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
6
+ function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
7
+ function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
8
+ step((generator = generator.apply(thisArg, _arguments || [])).next());
9
+ });
10
+ };
11
+ var __classPrivateFieldSet = (this && this.__classPrivateFieldSet) || function (receiver, state, value, kind, f) {
12
+ if (kind === "m") throw new TypeError("Private method is not writable");
13
+ if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a setter");
14
+ if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot write private member to an object whose class did not declare it");
15
+ return (kind === "a" ? f.call(receiver, value) : f ? f.value = value : state.set(receiver, value)), value;
16
+ };
17
+ var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (receiver, state, kind, f) {
18
+ if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
19
+ if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
20
+ return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
21
+ };
22
+ var _Deferred_promise, _Deferred_resolve, _Deferred_reject;
23
+ Object.defineProperty(exports, "__esModule", { value: true });
24
+ exports.Deferred = exports.pgClient = exports.BURN_TABLE = exports.JOB_TABLE = exports.JOB_ID_TABLE = exports.SOL_TRANSFER_INFO_TABLE = exports.SOL_TRANSFER_TABLE = exports.ENTITY_TABLE = exports.MANUAL_BURN_TABLE = exports.EPOCH_TABLE = exports.SOL_TX_TABLE = void 0;
25
+ const pg_1 = require("pg");
26
+ const kysely_1 = require("kysely");
27
+ const logger_1 = require("../../logger");
28
+ const transferTypToChan = (t) => {
29
+ if (t == 'open_compute_rewards') {
30
+ return 'open_compute';
31
+ }
32
+ if (t == 'node_operator_rewards') {
33
+ return 'node_operator';
34
+ }
35
+ return 'upgrade';
36
+ };
37
+ const transferFromDBV2 = (o) => {
38
+ return {
39
+ id: o.id,
40
+ solKey: o.sol_key,
41
+ solTx: o.sig,
42
+ sqTx: o.sq_tx,
43
+ createdAt: o.created_at,
44
+ payedAt: o.executed_at,
45
+ amountDue: BigInt(o.amount_due),
46
+ amountPayed: o.amount_payed ? BigInt(o.amount_payed) : undefined,
47
+ operatorRewards: o.channel == 'node_operator'
48
+ ? {
49
+ epochId: o.ui,
50
+ startTime: o.start_time,
51
+ endTime: o.end_time,
52
+ }
53
+ : undefined,
54
+ computeClientRewards: o.transfer_type == 'open_compute'
55
+ ? {
56
+ epochId: o.ui,
57
+ deviceId: o.device_id,
58
+ }
59
+ : undefined,
60
+ upgradeRewards: o.transfer_type == 'upgrade'
61
+ ? {
62
+ epochId: o.ui,
63
+ epochStartTime: o.start_time,
64
+ points: BigInt(o.points),
65
+ }
66
+ : undefined,
67
+ };
68
+ };
69
+ exports.SOL_TX_TABLE = 'sol_tx';
70
+ exports.EPOCH_TABLE = 'epoch';
71
+ exports.MANUAL_BURN_TABLE = 'manual_burn';
72
+ exports.ENTITY_TABLE = 'entity';
73
+ exports.SOL_TRANSFER_TABLE = 'sol_transfer';
74
+ exports.SOL_TRANSFER_INFO_TABLE = 'sol_transfer_info';
75
+ exports.JOB_ID_TABLE = 'job_id';
76
+ exports.JOB_TABLE = 'job';
77
+ exports.BURN_TABLE = 'burn';
78
+ const pgClient = (config) => {
79
+ var _a, _b;
80
+ const dialect = new kysely_1.PostgresDialect({
81
+ pool: new pg_1.Pool(Object.assign(Object.assign({}, config), { port: (_a = config.port) !== null && _a !== void 0 ? _a : 5432, max: (_b = config.max) !== null && _b !== void 0 ? _b : 10 })),
82
+ });
83
+ // Database interface is passed to Kysely's constructor, and from now on, Kysely
84
+ // knows your database structure.
85
+ // Dialect is passed to Kysely's constructor, and from now on, Kysely knows how
86
+ // to communicate with your database.
87
+ const db = new kysely_1.Kysely({
88
+ dialect,
89
+ });
90
+ const getOrInsertEntity = (db, p, log = logger_1.consoleLogger) => __awaiter(void 0, void 0, void 0, function* () {
91
+ const res = (yield db
92
+ .selectFrom(exports.ENTITY_TABLE)
93
+ .select('id')
94
+ .where('sol_key', '=', p.solKey)
95
+ .executeTakeFirst());
96
+ if (res) {
97
+ log.debug(`already found entity ${res.id}`);
98
+ return res.id;
99
+ }
100
+ const { id } = (yield db
101
+ .insertInto(exports.ENTITY_TABLE)
102
+ .values({
103
+ sol_key: p.solKey,
104
+ })
105
+ .onConflict((oc) => oc.column('sol_key').doNothing())
106
+ .returning((eb) => ['id as id'])
107
+ .executeTakeFirst());
108
+ log.debug(`inserted entity ${id}`);
109
+ return id;
110
+ });
111
+ const getOrInsertTx = (db, p, log = logger_1.consoleLogger) => __awaiter(void 0, void 0, void 0, function* () {
112
+ const res = yield db
113
+ .selectFrom(exports.SOL_TX_TABLE)
114
+ .select('id')
115
+ .where('sig', '=', p.sig)
116
+ .executeTakeFirst();
117
+ if (res) {
118
+ log.debug(`already found tx ${res.id}`);
119
+ return res.id;
120
+ }
121
+ const { id } = (yield db
122
+ .insertInto(exports.SOL_TX_TABLE)
123
+ .values({
124
+ sig: p.sig,
125
+ executed_at: p.executedAt,
126
+ })
127
+ .onConflict((oc) => oc.column('sig').doNothing())
128
+ .returning((eb) => ['id as id'])
129
+ .executeTakeFirst());
130
+ log.debug(`inserted tx ${id}`);
131
+ return id;
132
+ });
133
+ const getOrInsertEpoch = (db, p, log = logger_1.consoleLogger) => __awaiter(void 0, void 0, void 0, function* () {
134
+ const res = yield db
135
+ .selectFrom(exports.EPOCH_TABLE)
136
+ .select('id')
137
+ .where('ui', '=', String(p.uiEpochId))
138
+ .where('channel', '=', p.channel)
139
+ .executeTakeFirst();
140
+ if (res) {
141
+ log.debug(`already found epoch ${res.id}`);
142
+ return res.id;
143
+ }
144
+ const { id } = (yield db
145
+ .insertInto(exports.EPOCH_TABLE)
146
+ .values({
147
+ ui: p.uiEpochId,
148
+ channel: p.channel,
149
+ start_time: p.startTime,
150
+ end_time: p.endTime,
151
+ total_rndr: p.totalRndr,
152
+ })
153
+ .onConflict((oc) => oc.columns(['ui', 'channel']).doNothing())
154
+ .returning((eb) => ['id as id'])
155
+ .executeTakeFirst());
156
+ log.debug(`inserted epoch ${id}`);
157
+ return id;
158
+ });
159
+ const createTransfers = (db, transfers, log = logger_1.consoleLogger, totalRndr = BigInt(0)) => __awaiter(void 0, void 0, void 0, function* () {
160
+ const epochIds = [];
161
+ for (const t of transfers) {
162
+ if (!t.computeClientRewards && !t.upgradeRewards && !t.operatorRewards) {
163
+ log.warn(`malformed transfer doesnt have type ${JSON.stringify(t, (key, value) => typeof value === 'bigint' ? value.toString() : value)}`);
164
+ continue;
165
+ }
166
+ let solTxId;
167
+ if (t.solTx) {
168
+ solTxId = yield getOrInsertTx(db, {
169
+ sig: t.solTx,
170
+ executedAt: t.payedAt,
171
+ }, log);
172
+ }
173
+ let entityId;
174
+ if (t.solKey) {
175
+ entityId = yield getOrInsertEntity(db, { solKey: t.solKey }, log);
176
+ }
177
+ log.info(`inserting transfer ${JSON.stringify(t, (key, value) => typeof value === 'bigint' ? value.toString() : value)}`);
178
+ const { id: transferId } = (yield db
179
+ .insertInto(exports.SOL_TRANSFER_TABLE)
180
+ .values({
181
+ sol_tx_id: solTxId,
182
+ entity_id: entityId,
183
+ channel: transferTypToChan((() => {
184
+ if (t.operatorRewards) {
185
+ return 'node_operator_rewards';
186
+ }
187
+ else if (t.upgradeRewards) {
188
+ return 'upgrade_rewards';
189
+ }
190
+ return 'open_compute_rewards';
191
+ })()),
192
+ created_at: t.createdAt,
193
+ amount_due: BigInt(t.amountDue),
194
+ amount_payed: t.amountPayed ? BigInt(t.amountPayed) : 0,
195
+ })
196
+ .returning((eb) => ['id as id'])
197
+ .executeTakeFirst());
198
+ let epochId;
199
+ let deviceId;
200
+ let points;
201
+ let startTime;
202
+ let endTime;
203
+ let channel;
204
+ if (t.operatorRewards) {
205
+ log.info(`inserting operator rewards ${transferId}`);
206
+ epochId = t.operatorRewards.epochId;
207
+ startTime = t.operatorRewards.startTime;
208
+ endTime = t.operatorRewards.endTime;
209
+ channel = 'node_operator';
210
+ }
211
+ else if (t.computeClientRewards) {
212
+ log.info(`inserting compute client rewards ${transferId}`);
213
+ epochId = t.computeClientRewards.epochId;
214
+ deviceId = t.computeClientRewards.deviceId;
215
+ channel = 'open_compute';
216
+ }
217
+ else if (t.upgradeRewards) {
218
+ log.info(`inserting upgrade rewards ${transferId}`);
219
+ channel = 'upgrade';
220
+ epochId = t.upgradeRewards.epochId;
221
+ points = t.upgradeRewards.points;
222
+ }
223
+ let epochIdDb;
224
+ if (epochId != null) {
225
+ epochIdDb = yield getOrInsertEpoch(db, {
226
+ channel,
227
+ startTime,
228
+ endTime,
229
+ uiEpochId: epochId,
230
+ totalRndr,
231
+ }, log);
232
+ }
233
+ yield db
234
+ .insertInto(exports.SOL_TRANSFER_INFO_TABLE)
235
+ .values({
236
+ transfer_id: transferId,
237
+ device_id: deviceId,
238
+ points,
239
+ epoch_id: epochIdDb ? BigInt(epochIdDb) : null,
240
+ })
241
+ .executeTakeFirst();
242
+ epochIds.push({ epochDbId: epochIdDb });
243
+ }
244
+ return epochIds;
245
+ });
246
+ return {
247
+ db,
248
+ saveNodeOperatorEpoch(epochId, transfers, jobIds, totalRndr, log = logger_1.consoleLogger) {
249
+ return __awaiter(this, void 0, void 0, function* () {
250
+ yield db
251
+ .transaction()
252
+ .setIsolationLevel('repeatable read')
253
+ .execute((trx) => __awaiter(this, void 0, void 0, function* () {
254
+ const epochDbIds = yield createTransfers(trx, transfers, log, totalRndr);
255
+ log.info(`inserted transfers`);
256
+ for (const id of jobIds) {
257
+ yield trx
258
+ .insertInto(exports.JOB_ID_TABLE)
259
+ .values({
260
+ id: BigInt(id),
261
+ epoch_id: epochDbIds[0].epochDbId,
262
+ })
263
+ .executeTakeFirstOrThrow();
264
+ }
265
+ log.debug(`saved epoch`);
266
+ /*const now = new Date(Date.now())
267
+ const res = await trx
268
+ .updateTable(EPOCH_TABLE)
269
+ .set({
270
+ saved_at: now,
271
+ total_rndr: totalRndr,
272
+ })
273
+ .where('channel')
274
+ .where('ui', '=', epochId.toString())
275
+ .where('', '=')
276
+ .executeTakeFirstOrThrow()*/
277
+ }));
278
+ return;
279
+ });
280
+ },
281
+ getLatestNodeOperatorEpoch(log = logger_1.consoleLogger) {
282
+ var _a;
283
+ return __awaiter(this, void 0, void 0, function* () {
284
+ const now = new Date(Date.now());
285
+ try {
286
+ const res = yield db
287
+ .selectFrom(exports.EPOCH_TABLE)
288
+ .selectAll()
289
+ .where('channel', '=', 'node_operator')
290
+ .where('end_time', '<=', now)
291
+ .where('saved_at', 'is', null)
292
+ .orderBy('end_time asc')
293
+ .limit(1)
294
+ .executeTakeFirstOrThrow();
295
+ return {
296
+ epochId: BigInt(res.ui),
297
+ startedAt: res.start_time,
298
+ endedAt: res.end_time,
299
+ savedAt: (_a = res.saved_at) !== null && _a !== void 0 ? _a : undefined,
300
+ };
301
+ }
302
+ catch (e) {
303
+ if (e instanceof kysely_1.NoResultError) {
304
+ log.info(`no node operator epoch found`);
305
+ return undefined;
306
+ }
307
+ throw e;
308
+ }
309
+ });
310
+ },
311
+ getOutstanding(f, log = logger_1.consoleLogger) {
312
+ var _a;
313
+ return __awaiter(this, void 0, void 0, function* () {
314
+ const { typ, nullKey, due } = f;
315
+ //select sol_key, sum(amount_due)-sum(amount_payed) from transfer group by sol_key;
316
+ let q = db
317
+ .selectFrom(exports.SOL_TRANSFER_TABLE)
318
+ .innerJoin(exports.ENTITY_TABLE, 'sol_transfer.entity_id', 'entity.id')
319
+ .select(({ fn, val, ref }) => [
320
+ 'entity.sol_key',
321
+ (0, kysely_1.sql) `${fn.sum('sol_transfer.amount_due')} -
322
+ ${fn.sum('sol_transfer.amount_payed')}`.as('due'),
323
+ ])
324
+ .where('sol_transfer.channel', '=', transferTypToChan(typ))
325
+ .groupBy(['entity.sol_key']);
326
+ if (!nullKey) {
327
+ q = q.where('entity.sol_key', 'is not', null);
328
+ }
329
+ if (due) {
330
+ q = q.having(({ fn, ref }) => (0, kysely_1.sql) `${fn.sum('sol_transfer.amount_due')} -
331
+ ${fn.sum('sol_transfer.amount_payed')}
332
+ >
333
+ 0`);
334
+ }
335
+ const res = yield q.execute();
336
+ log.info(`outstanding query ${q.compile().sql} params ${JSON.stringify(f)} ${res.length} rows`);
337
+ const out = {};
338
+ for (const r of res) {
339
+ out[(_a = r.sol_key) !== null && _a !== void 0 ? _a : ''] = BigInt(r.due);
340
+ }
341
+ return out;
342
+ });
343
+ },
344
+ fetchTransfers(f, log = logger_1.consoleLogger) {
345
+ return __awaiter(this, void 0, void 0, function* () {
346
+ let base = db
347
+ .selectFrom(exports.SOL_TRANSFER_TABLE)
348
+ .selectAll(exports.SOL_TRANSFER_TABLE)
349
+ .innerJoin(exports.SOL_TRANSFER_INFO_TABLE, 'sol_transfer_info.transfer_id', 'sol_transfer.id')
350
+ .select(['device_id', 'points'])
351
+ .leftJoin(exports.SOL_TX_TABLE, 'sol_tx.id', 'sol_transfer.sol_tx_id')
352
+ .select(['sig', 'executed_at'])
353
+ .innerJoin(exports.ENTITY_TABLE, 'entity.id', 'sol_transfer.entity_id')
354
+ .select(['sol_key'])
355
+ .innerJoin(exports.EPOCH_TABLE, 'epoch.id', 'sol_transfer_info.epoch_id')
356
+ .select(['ui', 'start_time', 'end_time'])
357
+ .where('sol_transfer.channel', '=', transferTypToChan(f.typ));
358
+ if (f.solKey) {
359
+ base = base.where('entity.sol_key', '=', f.solKey);
360
+ }
361
+ if (f.payed != undefined) {
362
+ base = base.where('sol_transfer.sol_tx_id', f.payed ? 'is not' : 'is', null);
363
+ }
364
+ if (f.createdBefore != undefined) {
365
+ base = base.where('sol_transfer.created_at', '<=', f.createdBefore);
366
+ }
367
+ if (f.epochId) {
368
+ base = base.where('epoch.ui', '=', String(f.epochId));
369
+ }
370
+ if (f.sort) {
371
+ base = base.orderBy('entity.sol_key asc');
372
+ }
373
+ const res = yield base.execute();
374
+ log.info(`fetched ${base.compile().sql} params ${JSON.stringify(f)} ${res.length} transfers`);
375
+ return res.map((r) => transferFromDBV2(r));
376
+ });
377
+ },
378
+ payoutTransfers(idToAmt, payedAt, solTx, log = logger_1.consoleLogger) {
379
+ return __awaiter(this, void 0, void 0, function* () {
380
+ yield db
381
+ .transaction()
382
+ .setIsolationLevel('repeatable read')
383
+ .execute((tx) => __awaiter(this, void 0, void 0, function* () {
384
+ const alreadyUpdatedIds = (yield tx
385
+ .selectFrom(exports.SOL_TRANSFER_TABLE)
386
+ .innerJoin(exports.SOL_TX_TABLE, 'sol_transfer.sol_tx_id', 'sol_tx.id')
387
+ .select(['sol_transfer.id'])
388
+ .where('sol_transfer.id', 'in', Object.keys(idToAmt))
389
+ .where('sol_tx.sig', 'is not', null)
390
+ .execute()).map(({ id }) => id);
391
+ for (const [id, amt] of Object.entries(idToAmt)) {
392
+ if (alreadyUpdatedIds.includes(id)) {
393
+ log.info(`transfer ${id} already marked payed out. skipping`);
394
+ continue;
395
+ }
396
+ const solTxId = yield getOrInsertTx(tx, { sig: solTx, executedAt: payedAt }, log);
397
+ const res = yield tx
398
+ .updateTable(exports.SOL_TRANSFER_TABLE)
399
+ .set({
400
+ sol_tx_id: solTxId,
401
+ amount_payed: amt,
402
+ })
403
+ .where('id', '=', id.toString())
404
+ .executeTakeFirstOrThrow();
405
+ log.info(`marked ${res.numUpdatedRows} transfer payed out`);
406
+ }
407
+ }));
408
+ });
409
+ },
410
+ payoutTransfer(id, payedAt, solTx, amtPayed, log = logger_1.consoleLogger) {
411
+ return __awaiter(this, void 0, void 0, function* () {
412
+ yield db
413
+ .transaction()
414
+ .setIsolationLevel('repeatable read')
415
+ .execute((tx) => __awaiter(this, void 0, void 0, function* () {
416
+ const solTxId = yield getOrInsertTx(tx, { sig: solTx, executedAt: payedAt }, log);
417
+ const res = yield tx
418
+ .updateTable(exports.SOL_TRANSFER_TABLE)
419
+ .set({
420
+ sol_tx_id: solTxId,
421
+ amount_payed: amtPayed,
422
+ })
423
+ .where('id', '=', id.toString())
424
+ .executeTakeFirst();
425
+ log.info(`marked ${res.numUpdatedRows} transfer payed out`);
426
+ }));
427
+ });
428
+ },
429
+ createTransfers(transfers, log = logger_1.consoleLogger) {
430
+ return __awaiter(this, void 0, void 0, function* () {
431
+ yield db
432
+ .transaction()
433
+ .setIsolationLevel('repeatable read')
434
+ .execute((trx) => __awaiter(this, void 0, void 0, function* () {
435
+ yield createTransfers(trx, transfers, log);
436
+ }));
437
+ });
438
+ },
439
+ fetchInfoForEpoch(epochId, log = logger_1.consoleLogger) {
440
+ var _a;
441
+ return __awaiter(this, void 0, void 0, function* () {
442
+ const q = db
443
+ .selectFrom(exports.JOB_ID_TABLE)
444
+ .innerJoin(exports.EPOCH_TABLE, 'epoch.id', 'job_id.epoch_id')
445
+ .leftJoin(exports.MANUAL_BURN_TABLE, 'manual_burn.epoch_id', 'epoch.id')
446
+ .where('epoch.ui', '=', String(epochId))
447
+ .where('epoch.channel', '=', 'node_operator')
448
+ .select([
449
+ 'job_id.id',
450
+ 'epoch.total_rndr',
451
+ 'epoch.start_time',
452
+ 'epoch.end_time',
453
+ 'manual_burn.burned',
454
+ 'manual_burn.tx_sig',
455
+ ])
456
+ .orderBy('job_id.id asc');
457
+ log.debug(`fetch epoch info q ${q.compile().sql} epochId ${epochId}`);
458
+ const res = yield q.execute();
459
+ if (res.length == 0) {
460
+ throw new Error(`epoch ${epochId} not found`);
461
+ }
462
+ return {
463
+ jobIds: res.map((r) => Number(r.id)),
464
+ totalRndr: BigInt(res[0].total_rndr),
465
+ startTime: res[0].start_time,
466
+ endTime: res[0].end_time,
467
+ burned: BigInt((_a = res[0].burned) !== null && _a !== void 0 ? _a : 0),
468
+ burnTx: res[0].tx_sig,
469
+ };
470
+ });
471
+ },
472
+ updateComputeClientSolKey(deviceId, solKey, log = logger_1.consoleLogger) {
473
+ return __awaiter(this, void 0, void 0, function* () {
474
+ const q = db
475
+ .updateTable(`${exports.ENTITY_TABLE} as e`)
476
+ .set({
477
+ sol_key: solKey,
478
+ })
479
+ .from(`${exports.SOL_TRANSFER_TABLE} as t`)
480
+ .innerJoin(exports.SOL_TRANSFER_INFO_TABLE, 'sol_transfer_info.transfer_id', 't.id')
481
+ //.whereRef('t.id', '=', 'o.transfer_id')
482
+ .where('e.sol_key', 'is', null)
483
+ .where('sol_transfer_info.device_id', '=', deviceId);
484
+ const sql = yield q.compile();
485
+ log.info(`sql ${sql.sql} params ${sql.parameters}`);
486
+ const res = yield q.executeTakeFirstOrThrow();
487
+ log.info(`updated ${res.numUpdatedRows} device-id sol_keys`);
488
+ });
489
+ },
490
+ markJobsBurnt(trx, p, log = logger_1.consoleLogger) {
491
+ return __awaiter(this, void 0, void 0, function* () {
492
+ const solTxId = yield getOrInsertTx(trx, { sig: p.solTx, executedAt: p.executedAt }, log);
493
+ const { id } = yield trx
494
+ .insertInto(exports.BURN_TABLE)
495
+ .values({
496
+ sol_tx_id: solTxId,
497
+ priced_at: p.pricedAt,
498
+ usdc_spent: p.usdcSpent,
499
+ quote_amt: p.quoteAmt,
500
+ burned: p.burned,
501
+ })
502
+ //.onConflict((oc) => oc.column('sol_tx').doNothing())
503
+ .returning((eb) => ['id as id'])
504
+ .executeTakeFirstOrThrow();
505
+ const jobIdsStr = p.jobsIds.map(String);
506
+ const res = yield trx
507
+ .updateTable(exports.JOB_TABLE)
508
+ .where('id', 'in', jobIdsStr)
509
+ .set('burn_id', id)
510
+ .executeTakeFirstOrThrow();
511
+ log.debug(`set burn_id on ${res.numUpdatedRows} jobs`);
512
+ return;
513
+ });
514
+ },
515
+ produceBurnedJobs(jobs, log = logger_1.consoleLogger) {
516
+ return __awaiter(this, void 0, void 0, function* () {
517
+ yield db.transaction().execute((trx) => __awaiter(this, void 0, void 0, function* () {
518
+ var _a;
519
+ let num = BigInt(0);
520
+ for (const { amount, completedAt, jobId } of jobs) {
521
+ const res = yield trx
522
+ .insertInto(exports.JOB_TABLE)
523
+ .values({
524
+ id: jobId,
525
+ render_amt: amount,
526
+ completed_at: completedAt,
527
+ inserted_at: new Date(Date.now()),
528
+ })
529
+ .onConflict((oc) => oc.column('id').doNothing())
530
+ .executeTakeFirstOrThrow();
531
+ num += (_a = res.numInsertedOrUpdatedRows) !== null && _a !== void 0 ? _a : BigInt(0);
532
+ }
533
+ log.debug(`inserted ${num} of ${jobs.length} burn jobs`);
534
+ }));
535
+ });
536
+ },
537
+ markJobsBurned(p, log = logger_1.consoleLogger) {
538
+ return __awaiter(this, void 0, void 0, function* () {
539
+ let { markedAt, solTx } = p;
540
+ let res = yield db
541
+ .updateTable(exports.BURN_TABLE)
542
+ .set({ marked_burned_at: markedAt })
543
+ .from(exports.SOL_TX_TABLE)
544
+ .whereRef('burn.sol_tx_id', '=', 'sol_tx.id')
545
+ .where('sol_tx.sig', '=', solTx)
546
+ .executeTakeFirstOrThrow();
547
+ log.debug(`set ${res.numUpdatedRows} ${solTx} -> marked_burned_at ${markedAt.toISOString()}`);
548
+ });
549
+ },
550
+ consumeAndLockBurnJobs(limit, lowerRndrThreshold, upperRndrThreshold, log = logger_1.consoleLogger) {
551
+ return __awaiter(this, void 0, void 0, function* () {
552
+ const trx = yield begin(db);
553
+ const q = trx.trx
554
+ .selectFrom(trx.trx
555
+ .selectFrom(exports.JOB_TABLE)
556
+ .select(({ fn, val, ref }) => [
557
+ 'id',
558
+ fn
559
+ .sum('render_amt')
560
+ .over((ob) => ob.orderBy(`completed_at`, `asc`))
561
+ .as('total'),
562
+ ])
563
+ .where('burn_id', 'is', null)
564
+ .as('t'))
565
+ .select('id')
566
+ .where('total', '>', lowerRndrThreshold)
567
+ .where('total', '<', upperRndrThreshold);
568
+ log.debug(`outstanding jobs q: ${q.compile().sql} ${JSON.stringify({
569
+ limit,
570
+ lowerRndrThreshold,
571
+ upperRndrThreshold,
572
+ })}`);
573
+ const idRs = yield q.execute();
574
+ const ids = idRs.map((x) => x.id);
575
+ log.debug(`${ids.length} candidate jobs for burn. locking up to ${limit} jobs`);
576
+ if (ids.length == 0) {
577
+ return { trx, jobs: [] };
578
+ }
579
+ const rows = yield trx.trx
580
+ .selectFrom(exports.JOB_TABLE)
581
+ .selectAll()
582
+ .where('id', 'in', ids)
583
+ .limit(limit)
584
+ .forUpdate()
585
+ .skipLocked()
586
+ .execute();
587
+ log.debug(`locked ${rows.length} jobs`);
588
+ const jobs = [];
589
+ for (const r of rows) {
590
+ jobs.push({
591
+ id: Number(r.id),
592
+ completedAt: r.completed_at,
593
+ rndrUsed: BigInt(r.render_amt),
594
+ });
595
+ }
596
+ return { trx, jobs };
597
+ });
598
+ },
599
+ incrBuyBurnMetrics(log = logger_1.consoleLogger) {
600
+ var _a;
601
+ return __awaiter(this, void 0, void 0, function* () {
602
+ const res = yield db
603
+ .selectFrom(exports.JOB_TABLE)
604
+ .select('completed_at')
605
+ .orderBy('completed_at asc')
606
+ .where('burn_id', 'is', null)
607
+ .execute();
608
+ if (res.length == 0) {
609
+ return { outstandingJobs: 0 };
610
+ }
611
+ return {
612
+ outstandingJobs: res.length,
613
+ oldestJobFinishedAt: (_a = res[0].completed_at) !== null && _a !== void 0 ? _a : undefined,
614
+ };
615
+ });
616
+ },
617
+ deleteAll() {
618
+ return __awaiter(this, void 0, void 0, function* () {
619
+ if (config.database.includes('render')) {
620
+ throw new Error('cannot nuke from deployed database. call from tests only');
621
+ }
622
+ yield Promise.all([
623
+ db
624
+ .deleteFrom(exports.SOL_TX_TABLE)
625
+ .where('id', 'is not', null)
626
+ .executeTakeFirst(),
627
+ db
628
+ .deleteFrom(exports.SOL_TRANSFER_TABLE)
629
+ .where('id', 'is not', null)
630
+ .executeTakeFirst(),
631
+ db
632
+ .deleteFrom(exports.SOL_TRANSFER_INFO_TABLE)
633
+ .where('transfer_id', 'is not', null)
634
+ .executeTakeFirst(),
635
+ db
636
+ .deleteFrom(exports.EPOCH_TABLE)
637
+ .where('id', 'is not', null)
638
+ .executeTakeFirst(),
639
+ db
640
+ .deleteFrom(exports.ENTITY_TABLE)
641
+ .where('id', 'is not', null)
642
+ .executeTakeFirst(),
643
+ db
644
+ .deleteFrom(exports.MANUAL_BURN_TABLE)
645
+ .where('epoch_id', 'is not', null)
646
+ .executeTakeFirst(),
647
+ db
648
+ .deleteFrom(exports.JOB_ID_TABLE)
649
+ .where('id', 'is not', null)
650
+ .executeTakeFirst(),
651
+ db.deleteFrom(exports.JOB_TABLE).where('id', 'is not', null).executeTakeFirst(),
652
+ db
653
+ .deleteFrom(exports.BURN_TABLE)
654
+ .where('id', 'is not', null)
655
+ .executeTakeFirst(),
656
+ ]);
657
+ });
658
+ },
659
+ };
660
+ };
661
+ exports.pgClient = pgClient;
662
+ function begin(db) {
663
+ return __awaiter(this, void 0, void 0, function* () {
664
+ const connection = new Deferred();
665
+ const result = new Deferred();
666
+ // Do NOT await this line.
667
+ db.transaction()
668
+ .setIsolationLevel('repeatable read')
669
+ .execute((trx) => {
670
+ connection.resolve(trx);
671
+ return result.promise;
672
+ })
673
+ .catch((err) => {
674
+ // Don't do anything here. Just swallow the exception.
675
+ });
676
+ const trx = yield connection.promise;
677
+ return {
678
+ trx,
679
+ commit() {
680
+ result.resolve(null);
681
+ },
682
+ rollback() {
683
+ result.reject(new Error('rollback'));
684
+ },
685
+ };
686
+ });
687
+ }
688
+ class Deferred {
689
+ constructor() {
690
+ _Deferred_promise.set(this, void 0);
691
+ _Deferred_resolve.set(this, void 0);
692
+ _Deferred_reject.set(this, void 0);
693
+ this.resolve = (value) => {
694
+ if (__classPrivateFieldGet(this, _Deferred_resolve, "f")) {
695
+ __classPrivateFieldGet(this, _Deferred_resolve, "f").call(this, value);
696
+ }
697
+ };
698
+ this.reject = (reason) => {
699
+ if (__classPrivateFieldGet(this, _Deferred_reject, "f")) {
700
+ __classPrivateFieldGet(this, _Deferred_reject, "f").call(this, reason);
701
+ }
702
+ };
703
+ __classPrivateFieldSet(this, _Deferred_promise, new Promise((resolve, reject) => {
704
+ __classPrivateFieldSet(this, _Deferred_reject, reject, "f");
705
+ __classPrivateFieldSet(this, _Deferred_resolve, resolve, "f");
706
+ }), "f");
707
+ }
708
+ get promise() {
709
+ return __classPrivateFieldGet(this, _Deferred_promise, "f");
710
+ }
711
+ }
712
+ exports.Deferred = Deferred;
713
+ _Deferred_promise = new WeakMap(), _Deferred_resolve = new WeakMap(), _Deferred_reject = new WeakMap();
714
+ //# sourceMappingURL=v2.js.map