This repository has been archived by the owner on May 19, 2023. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 3
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
8 changed files
with
130 additions
and
168 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,115 @@ | ||
| import fs from 'fs' | ||
| import config from 'config' | ||
| import path from 'path' | ||
| import BigNumber from 'bignumber.js' | ||
| import { BlockHeader, Eth } from 'web3-eth' | ||
|
|
||
| import { AutoStartStopEventEmitter, NEW_BLOCK_EVENT_NAME } from './blockchain/new-block-emitters' | ||
| import { DbBackUpConfig } from './definitions' | ||
| import { getNewBlockEmitter } from './blockchain/utils' | ||
|
|
||
| export type BackUpEntry = { name: string, block: { hash: string, number: BigNumber } } | ||
|
|
||
| export function parseBackUps (backUpName: string): BackUpEntry { | ||
| const [block] = backUpName.split('.')[0].split('-') | ||
| const [hash, blockNumber] = block.split(':') | ||
|
|
||
| return { | ||
| name: backUpName, | ||
| block: { number: new BigNumber(blockNumber), hash } | ||
| } | ||
| } | ||
|
|
||
| export function getBackUps (): BackUpEntry[] { | ||
| const backupConfig = config.get<DbBackUpConfig>('dbBackUp') | ||
|
|
||
| const backups = fs.readdirSync(path.resolve(process.cwd(), `./${backupConfig.path}`)) | ||
|
|
||
| if (backups.length) { | ||
| return backups | ||
| .map(parseBackUps) | ||
| .sort( | ||
| (a: Record<string, any>, b: Record<string, any>) => | ||
| a.block.number.gt(b.block.number) ? -1 : 1 | ||
| ) | ||
| } | ||
|
|
||
| return [] | ||
| } | ||
|
|
||
| export class DbBackUpJob { | ||
| readonly newBlockEmitter: AutoStartStopEventEmitter | ||
| readonly db: string | ||
| readonly eth: Eth | ||
| readonly backUpConfig: DbBackUpConfig | ||
|
|
||
| constructor (eth: Eth) { | ||
| if (!config.has('dbBackUp')) { | ||
| throw new Error('DB Backup config not exist') | ||
| } | ||
| this.backUpConfig = config.get<DbBackUpConfig>('dbBackUp') | ||
| this.db = config.get<string>('db') | ||
|
|
||
| if (!fs.existsSync(this.backUpConfig.path)) { | ||
| fs.mkdirSync(this.backUpConfig.path) | ||
| } | ||
|
|
||
| this.eth = eth | ||
| this.newBlockEmitter = getNewBlockEmitter(eth) | ||
| } | ||
|
|
||
| /** | ||
| * Back-up database if blocks condition met | ||
| * @return {Promise<void>} | ||
| * @param block | ||
| */ | ||
| private async backupDb (block: BlockHeader): Promise<void> { | ||
| const [lastBackUp, previousBackUp] = getBackUps() | ||
|
|
||
| if (!lastBackUp || new BigNumber(block.number).minus(this.backUpConfig.blocks).gte(lastBackUp.block.number)) { | ||
| // copy and rename current db | ||
| await fs.promises.copyFile(this.db, path.resolve(this.backUpConfig.path, `${block.hash}:${block.number}-${this.db}`)) | ||
|
|
||
| // remove the oldest version | ||
| if (previousBackUp) { | ||
| await fs.promises.unlink(path.resolve(this.backUpConfig.path, previousBackUp.name)) | ||
| } | ||
| } | ||
| } | ||
|
|
||
| /** | ||
| * Restore database backup | ||
| * @return {Promise<void>} | ||
| */ | ||
| public async restoreDb (): Promise<void> { | ||
| const backUps = getBackUps() | ||
| const [_, oldest] = backUps | ||
|
|
||
| if (backUps.length < 2) { | ||
| throw new Error('Should be two backups to be able to restore') | ||
| } | ||
|
|
||
| // Check if back up block hash exist after reorg | ||
| const block = await this.eth.getBlock(oldest.block.hash).catch(() => false) | ||
|
|
||
| if (!block) { | ||
| throw new Error('Invalid backup. Block Hash is not valid!') | ||
| } | ||
|
|
||
| // remove current db | ||
| await fs.promises.unlink(this.db) | ||
|
|
||
| // restore backup | ||
| await fs.promises.copyFile(path.resolve(this.backUpConfig.path, oldest.name), path.resolve(process.cwd(), this.db)) | ||
| } | ||
|
|
||
| public run (): void { | ||
| this.newBlockEmitter.on(NEW_BLOCK_EVENT_NAME, this.backupDb.bind(this)) | ||
| } | ||
|
|
||
| public stop (): void { | ||
| this.newBlockEmitter.stop() | ||
| } | ||
| } | ||
|
|
||
| export default DbBackUpJob |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,5 +1,5 @@ | ||
| import { appFactory } from './app' | ||
| import { startApp } from './app' | ||
|
|
||
| (async function (): Promise<void> { | ||
| await appFactory({ appResetCallBack: () => { throw new Error('Reset callback not implemented') } }) | ||
| await startApp({ appResetCallBack: () => { throw new Error('Reset callback not implemented') } }) | ||
| })() |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.