/** * Module dependencies. */ var Collection = require('../../collection') , NativeCollection = require('mongodb').Collection , utils = require('../../utils') /** * Native collection * * @api private */ function MongooseCollection () { Collection.apply(this, arguments); }; /** * Inherit from abstract Collection. */ MongooseCollection.prototype.__proto__ = Collection.prototype; /** * Called when the connection opens * * @api private */ MongooseCollection.prototype.onOpen = function () { var self = this; this.conn.db.collection(this.name, function (err, collection) { if (err) { // likely a strict mode error self.conn.emit('error', err); } else { self.collection = collection; Collection.prototype.onOpen.call(self); } }); }; /** * Called when the connection closes * * @api private */ MongooseCollection.prototype.onClose = function () { Collection.prototype.onClose.call(this); }; /** * Copy the collection methods and make them subject to queues */ for (var i in NativeCollection.prototype) (function(i){ MongooseCollection.prototype[i] = function () { // BENCHMARKME: is it worth caching the prototype methods? probably if (!this.buffer) { var collection = this.collection , args = arguments , self = this; process.nextTick(function(){ var debug = self.conn.base.options.debug; if (debug) { if ('function' === typeof debug) { debug.apply(debug , [self.name, i].concat(utils.args(args, 0, args.length-1))); } else { console.error('\x1B[0;36mMongoose:\x1B[0m %s.%s(%s) %s %s %s' , self.name , i , print(args[0]) , print(args[1]) , print(args[2]) , print(args[3])) } } collection[i].apply(collection, args); }); } else { this.addQueue(i, arguments); } }; })(i); /** * Debug print helper * @private */ function print (arg) { var type = typeof arg; if ('function' === type || 'undefined' === type) return ''; return format(arg); } /** * Debug print helper * @private */ function format (obj, sub) { var x = utils.clone(obj); if (x) { if ('Binary' === x.constructor.name) { x = '[object Buffer]'; } else if ('Object' === x.constructor.name) { var keys = Object.keys(x) , i = keys.length , key while (i--) { key = keys[i]; if (x[key]) { if ('Binary' === x[key].constructor.name) { x[key] = '[object Buffer]'; } else if ('Object' === x[key].constructor.name) { x[key] = format(x[key], true); } else if (Array.isArray(x[key])) { x[key] = x[key].map(function (o) { return format(o, true) }); } } } } if (sub) return x; } return require('util') .inspect(x, false, 10, true) .replace(/\n/g, '') .replace(/\s{2,}/g, ' ') } /** * Implement getIndexes * * @param {Function} callback * @api private */ MongooseCollection.prototype.getIndexes = MongooseCollection.prototype.indexInformation; /** * Override signature of ensureIndex. -native one is not standard. * * @param {Object} spec * @param {Object} options * @param {Function} callback * @api public */ var oldEnsureIndex = NativeCollection.prototype.ensureIndex; function noop () {}; NativeCollection.prototype.ensureIndex = function(fields, options, fn){ if (!this.buffer) { return oldEnsureIndex.apply(this, arguments); } }; /** * Module exports. */ module.exports = MongooseCollection;