Write cache.json on startup if it doesn't exist.
Write mempool and block data to cache on every new block fixes #146
This commit is contained in:
parent
6a8deff706
commit
d260a1ed73
@ -3,6 +3,7 @@ import logger from '../logger';
|
|||||||
import memPool from './mempool';
|
import memPool from './mempool';
|
||||||
import { Block, TransactionExtended, TransactionMinerInfo } from '../interfaces';
|
import { Block, TransactionExtended, TransactionMinerInfo } from '../interfaces';
|
||||||
import { Common } from './common';
|
import { Common } from './common';
|
||||||
|
import diskCache from './disk-cache';
|
||||||
|
|
||||||
class Blocks {
|
class Blocks {
|
||||||
private static INITIAL_BLOCK_AMOUNT = 8;
|
private static INITIAL_BLOCK_AMOUNT = 8;
|
||||||
@ -99,6 +100,7 @@ class Blocks {
|
|||||||
if (this.newBlockCallbacks.length) {
|
if (this.newBlockCallbacks.length) {
|
||||||
this.newBlockCallbacks.forEach((cb) => cb(block, txIds, transactions));
|
this.newBlockCallbacks.forEach((cb) => cb(block, txIds, transactions));
|
||||||
}
|
}
|
||||||
|
diskCache.$saveCacheToDiskAsync();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -6,10 +6,16 @@ import blocks from './blocks';
|
|||||||
import logger from '../logger';
|
import logger from '../logger';
|
||||||
|
|
||||||
class DiskCache {
|
class DiskCache {
|
||||||
static FILE_NAME = './cache.json';
|
private static FILE_NAME = './cache.json';
|
||||||
|
private enabled = true;
|
||||||
|
|
||||||
constructor() {
|
constructor() {
|
||||||
if (process.env.workerId === '0' || !config.MEMPOOL.SPAWN_CLUSTER_PROCS) {
|
if (process.env.workerId === '0' || !config.MEMPOOL.SPAWN_CLUSTER_PROCS) {
|
||||||
|
if (!fs.existsSync(DiskCache.FILE_NAME)) {
|
||||||
|
fs.closeSync(fs.openSync(DiskCache.FILE_NAME, 'w'));
|
||||||
|
logger.info('Disk cache file created');
|
||||||
|
}
|
||||||
|
|
||||||
process.on('SIGINT', () => {
|
process.on('SIGINT', () => {
|
||||||
this.saveCacheToDisk();
|
this.saveCacheToDisk();
|
||||||
process.exit(2);
|
process.exit(2);
|
||||||
@ -19,19 +25,28 @@ class DiskCache {
|
|||||||
this.saveCacheToDisk();
|
this.saveCacheToDisk();
|
||||||
process.exit(2);
|
process.exit(2);
|
||||||
});
|
});
|
||||||
|
} else {
|
||||||
|
this.enabled = false;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
saveCacheToDisk() {
|
async $saveCacheToDiskAsync(): Promise<void> {
|
||||||
this.saveData(JSON.stringify({
|
if (!this.enabled) {
|
||||||
mempool: memPool.getMempool(),
|
return;
|
||||||
blocks: blocks.getBlocks(),
|
}
|
||||||
}));
|
try {
|
||||||
logger.info('Mempool and blocks data saved to disk cache');
|
await this.$saveDataAsync(JSON.stringify({
|
||||||
|
mempool: memPool.getMempool(),
|
||||||
|
blocks: blocks.getBlocks(),
|
||||||
|
}));
|
||||||
|
logger.info('Mempool and blocks data saved to disk cache');
|
||||||
|
} catch (e) {
|
||||||
|
logger.warn('Error writing to cache file asynchronously');
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
loadMempoolCache() {
|
loadMempoolCache() {
|
||||||
const cacheData = this.loadData();
|
const cacheData = this.loadDataSync();
|
||||||
if (cacheData) {
|
if (cacheData) {
|
||||||
logger.info('Restoring mempool and blocks data from disk cache');
|
logger.info('Restoring mempool and blocks data from disk cache');
|
||||||
const data = JSON.parse(cacheData);
|
const data = JSON.parse(cacheData);
|
||||||
@ -40,11 +55,30 @@ class DiskCache {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private saveData(dataBlob: string) {
|
private saveCacheToDisk() {
|
||||||
|
this.saveDataSync(JSON.stringify({
|
||||||
|
mempool: memPool.getMempool(),
|
||||||
|
blocks: blocks.getBlocks(),
|
||||||
|
}));
|
||||||
|
logger.info('Mempool and blocks data saved to disk cache');
|
||||||
|
}
|
||||||
|
|
||||||
|
private $saveDataAsync(dataBlob: string): Promise<void> {
|
||||||
|
return new Promise((resolve, reject) => {
|
||||||
|
fs.writeFile(DiskCache.FILE_NAME, dataBlob, (err) => {
|
||||||
|
if (err) {
|
||||||
|
return reject(err);
|
||||||
|
}
|
||||||
|
resolve();
|
||||||
|
});
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
private saveDataSync(dataBlob: string) {
|
||||||
fs.writeFileSync(DiskCache.FILE_NAME, dataBlob, 'utf8');
|
fs.writeFileSync(DiskCache.FILE_NAME, dataBlob, 'utf8');
|
||||||
}
|
}
|
||||||
|
|
||||||
private loadData(): string {
|
private loadDataSync(): string {
|
||||||
return fs.readFileSync(DiskCache.FILE_NAME, 'utf8');
|
return fs.readFileSync(DiskCache.FILE_NAME, 'utf8');
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user