db.js

var mongoose = require('mongoose');
var Schema = mongoose.Schema;
var DBName = 'blockDB';

var { waitFor } = require('./lib/utils');

var Block = new Schema(
  {
    "number": { type: Number, index: { unique: true } },
    "hash": String,
    "parentHash": String,
    "nonce": String,
    "sha3Uncles": String,
    "logsBloom": String,
    "transactionsRoot": String,
    "stateRoot": String,
    "receiptRoot": String,
    "proposer": String,
    "difficulty": String,
    "totalDifficulty": String,
    "size": Number,
    "extraData": String,
    "gasLimit": Number,
    "gasUsed": Number,
    "timestamp": Number,
    "blockTime": Number,
    "validators": [String]
  });

var Account = new Schema(
  {
    "address": { type: String, index: { unique: true } },
    "balance": Number,
    "blockNumber": Number,
    "type": { type: Number, default: 0 } // address: 0x0, contract: 0x1
  });

var Contract = new Schema(
  {
    "address": { type: String, index: { unique: true } },
    "creationTransaction": String,
    "contractName": String,
    "compilerVersion": String,
    "optimization": Boolean,
    "sourceCode": String,
    "abi": String,
    "byteCode": String
  }, { collection: "Contract" });

var Transaction = new Schema(
  {
    "hash": { type: String, index: { unique: true } },
    "nonce": Number,
    "blockHash": String,
    "blockNumber": Number,
    "transactionIndex": Number,
    "from": String,
    "to": String,
    "value": String,
    "gas": Number,
    "gasPrice": String,
    "timestamp": Number,
    "input": String
  }, { collection: "Transaction" });

var BlockStat = new Schema(
  {
    "number": { type: Number, index: { unique: true } },
    "timestamp": Number,
    "difficulty": String,
    "hashrate": String,
    "txCount": Number,
    "gasUsed": Number,
    "gasLimit": Number,
    "miner": String,
    "blockTime": Number,
    "uncleCount": Number
  });

// create indices
Transaction.index({ blockNumber: -1 });
Transaction.index({ from: 1, blockNumber: -1 });
Transaction.index({ to: 1, blockNumber: -1 });
Account.index({ balance: -1 });
Account.index({ balance: -1, blockNumber: -1 });
Account.index({ type: -1, balance: -1 });
Block.index({ miner: 1 });
Block.index({ miner: 1, blockNumber: -1 });

mongoose.model('BlockStat', BlockStat);
mongoose.model('Block', Block);
mongoose.model('Account', Account);
mongoose.model('Contract', Contract);
mongoose.model('Transaction', Transaction);
module.exports.BlockStat = mongoose.model('BlockStat');
module.exports.Block = mongoose.model('Block');
module.exports.Contract = mongoose.model('Contract');
module.exports.Transaction = mongoose.model('Transaction');
module.exports.Account = mongoose.model('Account');

mongoose.Promise = global.Promise;

mongoose.set('useNewUrlParser', true);

module.exports.Connect = () => {
  return new Promise((resolve) => {
    mongoUri = process.env.MONGO_URI || `mongodb://localhost/${DBName}`;
    mongoose.connect(mongoUri, async (err) => {
      if (err) {
        console.error(`Cannot connect to mongodb ${mongoUri}. Retrying in 1s...`);
        await waitFor(1);
        await this.Connect();
      }

      console.log(`Connected correctly to mongodb ${mongoUri}`);
      resolve();
    });

  });
};

// mongoose.set('debug', true);