diff --git a/lib/cmds/blockchain/blockchain.js b/lib/cmds/blockchain/blockchain.js index a8d78c49e..617a7f6e3 100644 --- a/lib/cmds/blockchain/blockchain.js +++ b/lib/cmds/blockchain/blockchain.js @@ -196,29 +196,8 @@ Blockchain.prototype.readyCallback = function() { this.onReadyCallback(); } - const NetcatClient = require('netcat/client'); - const isWin = process.platform === "win32"; - - let ipcPath; - if (isWin) { - ipcPath = '\\\\.\\pipe\\geth.ipc'; - } else { - ipcPath = self.datadir + '/geth.ipc'; - } - - const nc2 = new NetcatClient(); - nc2.unixSocket(ipcPath) - .enc('utf8') - .on('data', function(res){ - console.log(res); - }) - .connect() - .send(JSON.stringify({"jsonrpc":"2.0","method":"rpc_modules","params":[],"id":1})); - - setTimeout(() => { - console.log('Stop miner'); - nc2.send(JSON.stringify({"jsonrpc":"2.0","method":"miner_stop","params":[],"id":1})); - }, 5000); + const GethMiner = require('./miner'); + this.miner = new GethMiner(); }; Blockchain.prototype.kill = function() { diff --git a/lib/cmds/blockchain/miner.js b/lib/cmds/blockchain/miner.js new file mode 100644 index 000000000..6f26ff14d --- /dev/null +++ b/lib/cmds/blockchain/miner.js @@ -0,0 +1,165 @@ +const NetcatClient = require('netcat/client'); + +//Constants +const minerStart = 'miner_start'; +const minerStop = 'miner_stop'; +const getHashRate = 'miner_getHashrate'; + +class GethMiner { + constructor() { + const self = this; + // TODO: Find a way to load mining config from YML. + // In the meantime, just set an empty config object + this.config = {}; + + const defaults = { + interval_ms: 15000, + initial_ether: 15000000000000000000, + mine_pending_txns: true, + mine_periodically: false, + mine_normally: false, + threads: 1 + }; + + for (let key in defaults) { + if (this.config[key] === undefined) { + this.config[key] = defaults[key]; + } + } + + const isWin = process.platform === "win32"; + + let ipcPath; + if (isWin) { + ipcPath = '\\\\.\\pipe\\geth.ipc'; + } else { + ipcPath = this.datadir + '/geth.ipc'; + } + + this.client = new NetcatClient(); + this.client.unixSocket(ipcPath) + .enc('utf8') + .connect() + .on('data', function(res){ + if (self.callback) { + self.callback(null, res); + self.callback = null; + } + }); + + if (this.config.mine_normally) { + return this.sendCommand(minerStart); + } + + self.sendCommand(minerStop, () => { + self.fundAccount(function () { + if (this.config.mine_periodically) self.start_periodic_mining(); + if (this.config.mine_pending_txns) self.start_transaction_mining(); + }); + }); + + } + + sendCommand(method, callback) { + if (callback) { + this.callback = callback; + } + this.client.send(JSON.stringify({"jsonrpc": "2.0", "method": method, "params": [], "id": 1})); + } + + fundAccount(cb) { + const self = this; + const accountFunded = function () { + // TODO check https://github.com/ethereum/wiki/wiki/JSON-RPC for APIs + return (eth.getBalance(eth.coinbase) >= self.config.initial_ether); + }; + + if (accountFunded()) { + return cb(); + } + + console.log("== Funding account"); + this.sendCommand(minerStart); + + const blockWatcher = web3.eth.filter("latest").watch(function () { + if (accountFunded()) { + console.log("== Account funded"); + + blockWatcher.stopWatching(); + self.sendCommand(minerStop, cb); + } + }); + } + + pendingTransactions() { + if (web3.eth.pendingTransactions === undefined || web3.eth.pendingTransactions === null) { + return txpool.status.pending || txpool.status.queued; + } + else if (typeof web3.eth.pendingTransactions === "function") { + return web3.eth.pendingTransactions().length > 0; + } + else { + return web3.eth.pendingTransactions.length > 0 || web3.eth.getBlock('pending').transactions.length > 0; + } + } + + start_periodic_mining() { + const self = this; + let last_mined_ms = Date.now(); + let timeout_set = false; + + self.sendCommand(minerStart); + web3.eth.filter("latest").watch(function () { + if ((self.config.mine_pending_txns && self.pendingTransactions()) || timeout_set) { + return; + } + + timeout_set = true; + + const now = Date.now(); + const ms_since_block = now - last_mined_ms; + last_mined_ms = now; + + let next_block_in_ms; + + if (ms_since_block > self.config.interval_ms) { + next_block_in_ms = 0; + } else { + next_block_in_ms = (self.config.interval_ms - ms_since_block); + } + + self.sendCommand(minerStop); + console.log("== Looking for next block in " + next_block_in_ms + "ms"); + + setTimeout(function () { + console.log("== Looking for next block"); + timeout_set = false; + //miner_obj.start(config.threads); + self.sendCommand(minerStart); + }, next_block_in_ms); + }); + } + + start_transaction_mining() { + const self = this; + web3.eth.filter("pending").watch(function () { + self.sendCommand(getHashRate, (err, response) => { + if (response.result > 0) return; + + console.log("== Pending transactions! Looking for next block..."); + self.sendCommand(minerStart); + }); + }); + + if (self.config.mine_periodically) return; + + web3.eth.filter("latest").watch(function () { + if (!self.pendingTransactions()) { + console.log("== No transactions left. Stopping miner..."); + self.sendCommand(minerStop); + } + }); + } +} + +module.exports = GethMiner;