'use strict'; require('classtool'); function spec() { var mongoose = require('mongoose'); var config = require('../config/config'); var Block = require('../app/models/Block'); var Transaction = require('../app/models/Transaction'); var sockets = require('../app/controllers/socket.js'); function Sync() { this.tx_count = 0; } Sync.prototype.storeBlock = function(block, cb) { var that = this; Block.customCreate(block, function(err, block, inserted_txs){ if (err) return cb(err); if (block && that.opts.broadcast_blocks) { sockets.broadcast_block(block); } if (inserted_txs && that.opts.broadcast_txs) { inserted_txs.forEach(function(tx) { sockets.broadcast_tx(tx); }); } if (inserted_txs) that.tx_count += inserted_txs.length; return cb(); }); }; Sync.prototype.storeTxs = function(txs, inTime, cb) { var that = this; var time = inTime ? inTime : Math.round(new Date().getTime() / 1000); Transaction.createFromArray(txs, time, function(err, inserted_txs) { if (!err && inserted_txs && that.opts.broadcast_txs) { inserted_txs.forEach(function(tx) { sockets.broadcast_tx(tx); }); } return cb(err); }); }; Sync.prototype.init = function(opts, cb) { var that = this; that.opts = opts; if (!(opts && opts.skip_db_connection)) { if (!mongoose.connection) { mongoose.connect(config.db, {server: {auto_reconnect: true}} ); } this.db = mongoose.connection; this.db.on('error', function(err) { console.log('connection error:' + err); mongoose.disconnect(); }); this.db.on('disconnect', function(err) { console.log('disconnect:' + err); mongoose.connect(config.db, {server: {auto_reconnect: true}} ); }); return that.db.once('open', cb); } else return cb(); }; Sync.prototype.close = function() { if (!(this.opts && this.opts.skip_db_connection)) { this.db.close(); } }; return Sync; } module.defineClass(spec);