cerc-io/watcher-ts
10
1
Fork 0
mirror of https://github.com/cerc-io/watcher-ts synced 2025-08-01 12:12:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Implement transaction for uni-info-watcher blocks and events update. (#203)

Browse Source 
Co-authored-by: nabarun <nabarun@deepstacksoft.com>
This commit is contained in:
Ashwin Phatak 2021-08-11 15:23:44 +05:30 committed by GitHub
parent 1372fc74ef
commit cbc8dce88b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 88 additions and 58 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
packages/uni-info-watcher/src/database.ts
View File

@ -684,7 +684,7 @@ export class Database {
.getMany(); .getMany();
} }
async saveEvents (block: Block, events: DeepPartial<Event>[]): Promise<void> { async saveEvents (queryRunner: QueryRunner, block: Block, events: DeepPartial<Event>[]): Promise<void> {
const { const {
hash: blockHash, hash: blockHash,
number: blockNumber, number: blockNumber,
@ -700,9 +700,8 @@ export class Database {
// In a transaction: // In a transaction:
// (1) Save all the events in the database. // (1) Save all the events in the database.
// (2) Add an entry to the block progress table. // (2) Add an entry to the block progress table.
await this._conn.transaction(async (tx) => {
const numEvents = events.length; const numEvents = events.length;
const blockProgressRepo = tx.getRepository(BlockProgress); const blockProgressRepo = queryRunner.manager.getRepository(BlockProgress);
let blockProgress = await blockProgressRepo.findOne({ where: { blockHash } }); let blockProgress = await blockProgressRepo.findOne({ where: { blockHash } });
if (!blockProgress) { if (!blockProgress) {
@ -721,18 +720,16 @@ export class Database {
// Bulk insert events. // Bulk insert events.
events.forEach(event => { event.block = blockProgress; }); events.forEach(event => { event.block = blockProgress; });
await tx.createQueryBuilder().insert().into(Event).values(events).execute(); await queryRunner.manager.createQueryBuilder().insert().into(Event).values(events).execute();
} }
});
} }
async getEvent (id: string): Promise<Event | undefined> { async getEvent (id: string): Promise<Event | undefined> {
return this._conn.getRepository(Event).findOne(id, { relations: ['block'] }); return this._conn.getRepository(Event).findOne(id, { relations: ['block'] });
} }
async updateSyncStatus (blockHash: string, blockNumber: number): Promise<SyncStatus> { async updateSyncStatus (queryRunner: QueryRunner, blockHash: string, blockNumber: number): Promise<SyncStatus> {
return await this._conn.transaction(async (tx) => { const repo = queryRunner.manager.getRepository(SyncStatus);
const repo = tx.getRepository(SyncStatus);
let entity = await repo.findOne(); let entity = await repo.findOne();
if (!entity) { if (!entity) {
@ -748,7 +745,6 @@ export class Database {
} }
return await repo.save(entity); return await repo.save(entity);
});
} }
async getSyncStatus (queryRunner: QueryRunner): Promise<SyncStatus | undefined> { async getSyncStatus (queryRunner: QueryRunner): Promise<SyncStatus | undefined> {
@ -761,9 +757,8 @@ export class Database {
return repo.findOne({ where: { blockHash } }); return repo.findOne({ where: { blockHash } });
} }
async updateBlockProgress (blockHash: string, lastProcessedEventIndex: number): Promise<void> { async updateBlockProgress (queryRunner: QueryRunner, blockHash: string, lastProcessedEventIndex: number): Promise<void> {
await this._conn.transaction(async (tx) => { const repo = queryRunner.manager.getRepository(BlockProgress);
const repo = tx.getRepository(BlockProgress);
const entity = await repo.findOne({ where: { blockHash } }); const entity = await repo.findOne({ where: { blockHash } });
if (entity && !entity.isComplete) { if (entity && !entity.isComplete) {
if (lastProcessedEventIndex <= entity.lastProcessedEventIndex) { if (lastProcessedEventIndex <= entity.lastProcessedEventIndex) {
@ -778,7 +773,6 @@ export class Database {
await repo.save(entity); await repo.save(entity);
} }
});
} }
async _getPrevEntityVersion<Entity> (queryRunner: QueryRunner, repo: Repository<Entity>, findOptions: { [key: string]: any }): Promise<Entity | undefined> { async _getPrevEntityVersion<Entity> (queryRunner: QueryRunner, repo: Repository<Entity>, findOptions: { [key: string]: any }): Promise<Entity | undefined> {

42
packages/uni-info-watcher/src/indexer.ts
View File

@ -164,7 +164,20 @@ export class Indexer {
} }
async updateSyncStatus (blockHash: string, blockNumber: number): Promise<SyncStatus> { async updateSyncStatus (blockHash: string, blockNumber: number): Promise<SyncStatus> {
return this._db.updateSyncStatus(blockHash, blockNumber); const dbTx = await this._db.createTransactionRunner();
let res;
try {
res = await this._db.updateSyncStatus(dbTx, blockHash, blockNumber);
await dbTx.commitTransaction();
} catch (error) {
await dbTx.rollbackTransaction();
throw error;
} finally {
await dbTx.release();
}
return res;
} }
async getSyncStatus (): Promise<SyncStatus | undefined> { async getSyncStatus (): Promise<SyncStatus | undefined> {
@ -198,7 +211,20 @@ export class Indexer {
} }
async updateBlockProgress (blockHash: string, lastProcessedEventIndex: number): Promise<void> { async updateBlockProgress (blockHash: string, lastProcessedEventIndex: number): Promise<void> {
return this._db.updateBlockProgress(blockHash, lastProcessedEventIndex); const dbTx = await this._db.createTransactionRunner();
let res;
try {
res = await this._db.updateBlockProgress(dbTx, blockHash, lastProcessedEventIndex);
await dbTx.commitTransaction();
} catch (error) {
await dbTx.rollbackTransaction();
throw error;
} finally {
await dbTx.release();
}
return res;
} }
async getBundle (id: string, block: BlockHeight): Promise<Bundle | undefined> { async getBundle (id: string, block: BlockHeight): Promise<Bundle | undefined> {
@ -319,7 +345,17 @@ export class Indexer {
}); });
} }
await this._db.saveEvents(block, dbEvents); const dbTx = await this._db.createTransactionRunner();
try {
await this._db.saveEvents(dbTx, block, dbEvents);
await dbTx.commitTransaction();
} catch (error) {
await dbTx.rollbackTransaction();
throw error;
} finally {
await dbTx.release();
}
} }
async _handlePoolCreated (block: Block, contractAddress: string, tx: Transaction, poolCreatedEvent: PoolCreatedEvent): Promise<void> { async _handlePoolCreated (block: Block, contractAddress: string, tx: Transaction, poolCreatedEvent: PoolCreatedEvent): Promise<void> {