reorg code, integration publish/subscribe working, still some TODOs

2023-02-01 13:01:51 +11:00 · 2023-02-01 13:01:51 +11:00 · 1af6d56e2d
commit 1af6d56e2d
parent 043a95d9ef
35 changed files with 3052 additions and 1401 deletions
--- a/miner/miner.js
+++ b/miner/miner.js
@ -0,0 +1,169 @@
+const Block = require('../blockchain/block');
+
+const Payment = require('../blockchain/payment');
+const Integration = require('../blockchain/integration');
+const SensorRegistration = require('../blockchain/sensor-registration');
+const BrokerRegistration = require('../blockchain/broker-registration');
+
+const ITERATIONS = 1;
+
+const STATE_RUNNING = 0;
+const STATE_INTERRUPTED = 1;
+
+function mine(miner) {
+  if (miner.state !== STATE_RUNNING) {
+    this.startMine();
+    return;
+  }
+  const timestamp = Date.now();
+  const difficulty = Block.adjustDifficulty(miner.lastBlock, timestamp);
+
+  for (let i = 0; i < ITERATIONS; ++i) {
+    const hash = Block.hash(
+      timestamp,
+      miner.lastBlock.hash,
+      miner.reward,
+      miner.txs.payments.mining,
+      miner.txs.sensorRegistrations.mining,
+      miner.txs.brokerRegistrations.mining,
+      miner.txs.integrations.mining,
+      miner.nonce,
+      difficulty);
+
+    if (hash.substring(0, difficulty) === '0'.repeat(difficulty)) {
+      //success
+      const endTime = process.hrtime.bigint();
+      console.log(`Mined a block of difficulty ${difficulty} in ${Number(endTime - miner.minedStartTime) / 1000000}ms`);
+      miner.onMined(new Block(
+        timestamp,
+        miner.lastBlock.hash,
+        hash,
+        miner.reward,
+        miner.txs.payments.mining,
+        miner.txs.sensorRegistrations.mining,
+        miner.txs.brokerRegistrations.mining,
+        miner.txs.integrations.mining,
+        miner.nonce,
+        difficulty));
+      miner.state = STATE_INTERRUPTED;
+      setImmediate(() => { startMine(miner) });
+      return;
+    } else {
+      //failure
+      miner.nonce++;
+    }
+  }
+  setImmediate(() => { mine(miner) });
+}
+
+function startMine(miner) {
+  //only continue if state is waiting or restarting
+  if (miner.state !== STATE_INTERRUPTED) {
+    return;
+  }
+
+  miner.minedStartTime = process.hrtime.bigint();
+
+  //TODO make sure these transactions actually work as a collective instead of individually
+  miner.txs.payments.mining = [...miner.txs.payments.pool];
+  miner.txs.integrations.mining = [...miner.txs.integrations.pool];
+  miner.txs.sensorRegistrations.mining = [...miner.txs.sensorRegistrations.pool];
+  miner.txs.brokerRegistrations.mining = [...miner.txs.brokerRegistrations.pool];
+
+  miner.lastBlock = miner.blockchain.chain[miner.blockchain.chain.length - 1];
+
+  miner.nonce = 0;
+  miner.state = STATE_RUNNING;
+
+  mine(miner);
+}
+
+function findTx(tx) {
+  return t => t.input === tx.input && t.counter === tx.counter;
+}
+
+function clearFromBlock(miner, txs, blockTxs) {
+  for (const tx of blockTxs) {
+    const foundIndex = txs.pool.findIndex(findTx(tx));
+
+    if (foundIndex !== -1) {
+      txs.pool.splice(foundIndex, 1);
+    }
+
+    if (txs.mining.some(findTx(tx))) {
+      miner.state = STATE_INTERRUPTED;
+    }
+  }
+}
+
+class Miner {
+  constructor(blockchain, reward, onMined) {
+    this.blockchain = blockchain;
+    this.onMined = onMined;
+    this.state = STATE_INTERRUPTED;
+    this.lastBlock = null;
+    this.reward = reward;
+
+    this.minedStartTime = null;
+
+    this.txs = {
+      payments: {
+        pool: [],
+        mining: []
+      },
+      integrations: {
+        pool: [],
+        mining: []
+      },
+      sensorRegistrations: {
+        pool: [],
+        mining: []
+      },
+      brokerRegistrations: {
+        pool: [],
+        mining: []
+      }
+    };
+
+    startMine(this);
+  }
+
+  addTransaction(tx) {
+    const verifyRes = tx.type.verify(tx.transaction);
+    if (!verifyRes.result) {
+      console.log("Couldn't add tx to miner, tx couldn't be verified: " + verifyRes.reason);
+      return;
+    }
+
+    let txs = null;
+
+    switch (tx.type) {
+      case Payment: txs = this.txs.payments; break;
+      case Integration: txs = this.txs.integrations; break;
+      case SensorRegistration: txs = this.txs.sensorRegistrations; break;
+      case BrokerRegistration: txs = this.txs.brokerRegistrations; break;
+      default: throw new Error(`unknown tx type: ${tx.type.name()}`);
+    }
+
+    const foundIndex = txs.pool.findIndex(findTx(tx.transaction));
+
+    if (foundIndex !== -1) {
+      txs.pool[foundIndex] = tx.transaction;
+      if (txs.mining.some(findTx(tx.transaction))) {
+        this.state = STATE_INTERRUPTED;
+      }
+    } else {
+      txs.pool.push(tx.transaction);
+    }
+  }
+
+  onNewBlock(block) {
+    clearFromBlock(this, this.txs.payments, Block.getPayments(block));
+    clearFromBlock(this, this.txs.integrations, Block.getIntegrations(block));
+    clearFromBlock(this, this.txs.sensorRegistrations, Block.getSensorRegistrations(block));
+    clearFromBlock(this, this.txs.brokerRegistrations, Block.getBrokerRegistrations(block));
+  }
+}
+
+module.exports = Miner;
+