add transaction & wallet model

Author: khaihkd

models/mongodb/index.js

@@ -0,0 +1,23 @@
+'use strict'
+
+const fs = require('fs')
+const path = require('path')
+const mongoose = require('mongoose')
+const config = require('config')
+const db = {}
+
+mongoose.Promise = global.Promise
+mongoose.connect(config.get('db.uri'), { useNewUrlParser: true })
+
+// import all file in this dir, except index.js
+fs.readdirSync(__dirname)
+    .filter(function (file) {
+        return (file.indexOf('.') !== 0) && (file !== 'index.js')
+    })
+    .forEach(function (file) {
+        var model = require(path.join(__dirname, file))
+        db[model.modelName] = model
+    })
+
+db.mongoose = mongoose
+module.exports = db

models/mongodb/transaction.js

@@ -0,0 +1,32 @@
+'use strict'
+
+const mongoose = require('mongoose')
+const Schema = mongoose.Schema
+
+const Transaction = new Schema({
+    hash: {
+        type: String,
+        index: true,
+        unique: true
+    },
+    block: {
+        type: Number,
+        index: true
+    },
+    fromAccount: {
+        type: String,
+        index: true
+    },
+    toAccount: {
+        type: String,
+        index: true
+    },
+    tokenAmount: {
+        type: Number
+    },
+    isProcess: {
+        type: Boolean
+    }
+}, { timestamps: true })
+
+module.exports = mongoose.model('Transaction', Transaction)

models/mongodb/wallet.js

@@ -0,0 +1,22 @@
+'use strict'
+
+const mongoose = require('mongoose')
+const Schema = mongoose.Schema
+
+const Wallet = new Schema({
+    address: {
+        type: String,
+        index: true,
+        unique: true
+    },
+    balance: {
+        type: Number,
+        index: true
+    },
+    transactionCount: {
+        type: Number,
+        index: true
+    }
+}, { timestamps: true })
+
+module.exports = mongoose.model('Wallet', Wallet)