-
Notifications
You must be signed in to change notification settings - Fork 0
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
29 changed files
with
706 additions
and
832 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
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,260 @@ | ||
| import {Wallet} from "./Wallet"; | ||
| import {BlockchainExplorer, RawDaemon_Transaction} from "./blockchain/BlockchainExplorer"; | ||
| import {Transaction} from "./Transaction"; | ||
| import {TransactionsExplorer} from "./TransactionsExplorer"; | ||
|
|
||
| export class WalletWatchdog{ | ||
|
|
||
| wallet : Wallet; | ||
| explorer : BlockchainExplorer; | ||
|
|
||
| constructor(wallet: Wallet, explorer : BlockchainExplorer) { | ||
| this.wallet = wallet; | ||
| this.explorer = explorer; | ||
|
|
||
| this.initWorker(); | ||
| this.initMempool(); | ||
| } | ||
|
|
||
| initWorker(){ | ||
| let self = this; | ||
| this.workerProcessing = new Worker('./workers/TransferProcessingEntrypoint.js'); | ||
| this.workerProcessing.onmessage = function(data : MessageEvent){ | ||
| let message : string|any = data.data; | ||
| if(message === 'ready'){ | ||
| console.info('worker ready'); | ||
| self.signalWalletUpdate(); | ||
| }else if(message === 'readyWallet'){ | ||
| self.workerProcessingReady = true; | ||
| }else if(message.type){ | ||
| if(message.type === 'processed'){ | ||
| let transactions = message.transactions; | ||
| if(transactions.length > 0) { | ||
| for(let tx of transactions) | ||
| self.wallet.addNew(Transaction.fromRaw(tx)); | ||
| self.signalWalletUpdate(); | ||
| } | ||
| if(self.workerCurrentProcessing.length>0) { | ||
| let transactionHeight = self.workerCurrentProcessing[self.workerCurrentProcessing.length-1].height; | ||
| if(typeof transactionHeight !== 'undefined') | ||
| self.wallet.lastHeight = transactionHeight; | ||
| } | ||
|
|
||
| self.workerProcessingWorking = false; | ||
| } | ||
| } | ||
| }; | ||
| } | ||
|
|
||
| signalWalletUpdate(){ | ||
| let self = this; | ||
| console.log('wallet update'); | ||
| this.lastBlockLoading = -1;//reset scanning | ||
| this.workerProcessing.postMessage({ | ||
| type:'initWallet', | ||
| wallet:this.wallet.exportToRaw() | ||
| }); | ||
| clearInterval(this.intervalTransactionsProcess); | ||
| this.intervalTransactionsProcess = setInterval(function(){ | ||
| self.checkTransactionsInterval(); | ||
| }, this.wallet.options.readSpeed); | ||
|
|
||
| //force mempool update after a wallet update (new tx, ...) | ||
| self.checkMempool(); | ||
| } | ||
|
|
||
| intervalMempool = 0; | ||
| initMempool(force : boolean = false){ | ||
| let self = this; | ||
| if(this.intervalMempool === 0 || force){ | ||
| if(force && this.intervalMempool !== 0){ | ||
| clearInterval(this.intervalMempool); | ||
| } | ||
|
|
||
| this.intervalMempool = setInterval(function(){ | ||
| self.checkMempool(); | ||
| }, config.avgBlockTime/2*1000); | ||
| } | ||
| self.checkMempool(); | ||
| } | ||
|
|
||
| stopped : boolean = false; | ||
|
|
||
| stop(){ | ||
| clearInterval(this.intervalTransactionsProcess); | ||
| this.transactionsToProcess = []; | ||
| clearInterval(this.intervalMempool); | ||
| this.stopped = true; | ||
| } | ||
|
|
||
| checkMempool() : boolean{ | ||
| let self = this; | ||
| if(this.lastMaximumHeight - this.lastBlockLoading > 1){//only check memory pool if the user is up to date to ensure outs & ins will be found in the wallet | ||
| return false; | ||
| } | ||
|
|
||
| this.wallet.txsMem = []; | ||
| this.explorer.getTransactionPool().then(function(data : any){ | ||
| if(typeof data.transactions !== 'undefined') | ||
| for(let rawTx of data.transactions){ | ||
| let tx = TransactionsExplorer.parse(rawTx.tx_json,self.wallet); | ||
| if(tx !== null){ | ||
| tx.hash = rawTx.id_hash; | ||
| tx.fees = rawTx.fee; | ||
| self.wallet.txsMem.push(tx); | ||
| } | ||
| } | ||
| }).catch(function(){}); | ||
| return true; | ||
| } | ||
|
|
||
| terminateWorker(){ | ||
| this.workerProcessing.terminate(); | ||
| this.workerProcessingReady = false; | ||
| this.workerCurrentProcessing = []; | ||
| this.workerProcessingWorking = false; | ||
| this.workerCountProcessed = 0; | ||
| } | ||
|
|
||
| transactionsToProcess : RawDaemon_Transaction[] = []; | ||
| intervalTransactionsProcess = 0; | ||
|
|
||
| workerProcessing !: Worker; | ||
| workerProcessingReady = false; | ||
| workerProcessingWorking = false; | ||
| workerCurrentProcessing : RawDaemon_Transaction[] = []; | ||
| workerCountProcessed = 0; | ||
|
|
||
| checkTransactions(rawTransactions : RawDaemon_Transaction[]){ | ||
| for(let rawTransaction of rawTransactions){ | ||
| let height = rawTransaction.height; | ||
| if(typeof height !== 'undefined') { | ||
| let transaction = TransactionsExplorer.parse(rawTransaction, this.wallet); | ||
| if (transaction !== null) { | ||
| this.wallet.addNew(transaction); | ||
| } | ||
| if (height - this.wallet.lastHeight >= 2) { | ||
| this.wallet.lastHeight = height - 1; | ||
| } | ||
| } | ||
| } | ||
| if(this.transactionsToProcess.length == 0){ | ||
| this.wallet.lastHeight = this.lastBlockLoading; | ||
| } | ||
| } | ||
|
|
||
| checkTransactionsInterval(){ | ||
| if(this.workerProcessingWorking || !this.workerProcessingReady) { | ||
| return; | ||
| } | ||
|
|
||
| //we destroy the worker in charge of decoding the transactions every 5k transactions to ensure the memory is not corrupted | ||
| //cnUtil bug, see https://github.com/mymonero/mymonero-core-js/issues/8 | ||
| if(this.workerCountProcessed >= 5*1000){ | ||
| console.log('Recreate worker..'); | ||
| this.terminateWorker(); | ||
| this.initWorker(); | ||
| return; | ||
| } | ||
|
|
||
| let transactionsToProcess : RawDaemon_Transaction[] = this.transactionsToProcess.splice(0, 30); | ||
| if(transactionsToProcess.length > 0) { | ||
| this.workerCurrentProcessing = transactionsToProcess; | ||
| this.workerProcessing.postMessage({ | ||
| type:'process', | ||
| transactions:transactionsToProcess | ||
| }); | ||
| this.workerCountProcessed+=this.transactionsToProcess.length; | ||
| this.workerProcessingWorking = true; | ||
| }else{ | ||
| clearInterval(this.intervalTransactionsProcess); | ||
| this.intervalTransactionsProcess = 0; | ||
| } | ||
| } | ||
|
|
||
| processTransactions(transactions : RawDaemon_Transaction[]){ | ||
| let transactionsToAdd = []; | ||
| for(let tr of transactions){ | ||
| if(typeof tr.height !== 'undefined') | ||
| if(tr.height > this.wallet.lastHeight){ | ||
| transactionsToAdd.push(tr); | ||
| } | ||
| } | ||
|
|
||
| this.transactionsToProcess.push.apply(this.transactionsToProcess, transactionsToAdd); | ||
| if(this.intervalTransactionsProcess === 0){ | ||
| let self = this; | ||
| this.intervalTransactionsProcess = setInterval(function(){ | ||
| self.checkTransactionsInterval(); | ||
| }, this.wallet.options.readSpeed); | ||
| } | ||
| } | ||
|
|
||
|
|
||
| lastBlockLoading = -1; | ||
| lastMaximumHeight = 0; | ||
|
|
||
| loadHistory(){ | ||
| if(this.stopped)return; | ||
|
|
||
| let self = this; | ||
|
|
||
| if(this.transactionsToProcess.length > 500){ | ||
| //to ensure no pile explosion | ||
| setTimeout(function () { | ||
| self.loadHistory(); | ||
| }, 2*1000); | ||
| return; | ||
| } | ||
|
|
||
| // console.log('checking'); | ||
| this.explorer.getHeight().then(function(height){ | ||
| if(height > self.lastMaximumHeight)self.lastMaximumHeight = height; | ||
|
|
||
| if(self.lastBlockLoading === -1)self.lastBlockLoading = self.wallet.lastHeight; | ||
|
|
||
| if(self.lastBlockLoading !== height){ | ||
| let previousStartBlock = self.lastBlockLoading; | ||
| let endBlock = previousStartBlock+99; | ||
|
|
||
| if(previousStartBlock >= self.lastMaximumHeight)previousStartBlock = self.lastMaximumHeight; | ||
| if(endBlock >= self.lastMaximumHeight)endBlock = self.lastMaximumHeight; | ||
|
|
||
| self.explorer.getTransactionsForBlocks(previousStartBlock, endBlock, false).then(function(transactions : RawDaemon_Transaction[]){ | ||
| //to ensure no pile explosion | ||
| if(transactions.length > 0){ | ||
| let lastTx = transactions[transactions.length-1]; | ||
| if(typeof lastTx.height !== 'undefined') { | ||
| self.lastBlockLoading = lastTx.height + 1; | ||
| } | ||
| self.processTransactions(transactions); | ||
| setTimeout(function () { | ||
| self.loadHistory(); | ||
| }, 1); | ||
| }else{ | ||
| self.lastBlockLoading = endBlock+1; | ||
| if(self.lastBlockLoading >= self.lastMaximumHeight)self.lastBlockLoading = self.lastMaximumHeight; | ||
|
|
||
| setTimeout(function () { | ||
| self.loadHistory(); | ||
| }, 30*1000); | ||
| } | ||
| }).catch(function(){ | ||
| setTimeout(function () { | ||
| self.loadHistory(); | ||
| }, 30*1000);//retry 30s later if an error occurred | ||
| }); | ||
| }else{ | ||
| setTimeout(function () { | ||
| self.loadHistory(); | ||
| }, 30*1000); | ||
| } | ||
| }).catch(function(){ | ||
| setTimeout(function () { | ||
| self.loadHistory(); | ||
| }, 30*1000);//retry 30s later if an error occurred | ||
| }); | ||
| } | ||
|
|
||
|
|
||
| } |
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.