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