/** * Module exports class Model */ module.exports = DataAccessObject; /** * Module dependencies */ var util = require('util'); var jutil = require('./jutil'); var validations = require('./validations.js'); var ValidationError = validations.ValidationError; var List = require('./list.js'); require('./relations.js'); var Inclusion = require('./include.js'); var Relation = require('./relations.js'); var geo = require('./geo'); var Memory = require('./connectors/memory').Memory; var fieldsToArray = require('./utils').fieldsToArray; /** * DAO class - base class for all persist objects * provides **common API** to access any database connector. * This class describes only abstract behavior layer, refer to `lib/connectors/*.js` * to learn more about specific connector implementations * * `DataAccessObject` mixes `Inclusion` classes methods * * @constructor * @param {Object} data - initial object data */ function DataAccessObject() { if(DataAccessObject._mixins) { var self = this; var args = arguments; DataAccessObject._mixins.forEach(function(m) { m.call(self, args); }); } } function idName(m) { return m.dataSource.idName ? m.dataSource.idName(m.modelName) : 'id'; } function getIdValue(m, data) { return data && data[m.dataSource.idName(m.modelName)]; } function setIdValue(m, data, value) { if(data) { data[idName(m)] = value; } } DataAccessObject._forDB = function (data) { if(!(this.dataSource.isRelational && this.dataSource.isRelational())) { return data; } var res = {}; Object.keys(data).forEach(function (propName) { if (this.whatTypeName(propName) === 'JSON' || data[propName] instanceof Array) { res[propName] = JSON.stringify(data[propName]); } else { res[propName] = data[propName]; } }.bind(this)); return res; }; /** * Create new instance of Model class, saved in database * * @param data [optional] * @param callback(err, obj) * callback called with arguments: * * - err (null or Error) * - instance (null or Model) */ DataAccessObject.create = function (data, callback) { if (stillConnecting(this.dataSource, this, arguments)) return; var Model = this; var modelName = Model.modelName; if (typeof data === 'function') { callback = data; data = {}; } if (typeof callback !== 'function') { callback = function () {}; } if (!data) { data = {}; } if (Array.isArray(data)) { var instances = []; var errors = Array(data.length); var gotError = false; var wait = data.length; if (wait === 0) callback(null, []); var instances = []; for (var i = 0; i < data.length; i += 1) { (function(d, i) { instances.push(Model.create(d, function(err, inst) { if (err) { errors[i] = err; gotError = true; } modelCreated(); })); })(data[i], i); } return instances; function modelCreated() { if (--wait === 0) { callback(gotError ? errors : null, instances); } } } var obj; // if we come from save if (data instanceof Model && !getIdValue(this, data)) { obj = data; } else { obj = new Model(data); } data = obj.toObject(true); // validation required obj.isValid(function(valid) { if (valid) { create(); } else { callback(new ValidationError(obj), obj); } }, data); function create() { obj.trigger('create', function(createDone) { obj.trigger('save', function(saveDone) { var _idName = idName(Model); this._adapter().create(modelName, this.constructor._forDB(obj.toObject(true)), function (err, id, rev) { if (id) { obj.__data[_idName] = id; obj.__dataWas[_idName] = id; defineReadonlyProp(obj, _idName, id); } if (rev) { obj._rev = rev; } if (err) { return callback(err, obj); } saveDone.call(obj, function () { createDone.call(obj, function () { callback(err, obj); }); }); }, obj); }, obj); }, obj); } // for chaining return obj; }; DataAccessObject.create.shared = true; DataAccessObject.create.accepts = {arg: 'data', type: 'object', http: {source: 'body'}}; DataAccessObject.create.returns = {arg: 'data', type: 'object', root: true}; DataAccessObject.create.http = {verb: 'post', path: '/'}; function stillConnecting(dataSource, obj, args) { if (dataSource.connected) return false; // Connected var method = args.callee; // Set up a callback after the connection is established to continue the method call dataSource.once('connected', function () { method.apply(obj, [].slice.call(args)); }); if (!dataSource.connecting) { dataSource.connect(); } return true; } /** * Update or insert a model instance * @param {Object} data The model instance data * @param {Function} [callback] The callback function */ DataAccessObject.upsert = DataAccessObject.updateOrCreate = function upsert(data, callback) { if (stillConnecting(this.dataSource, this, arguments)) return; var Model = this; if (!getIdValue(this, data)) return this.create(data, callback); if (this.dataSource.connector.updateOrCreate) { var inst = new Model(data); this.dataSource.connector.updateOrCreate(Model.modelName, inst.toObject(true), function (err, data) { var obj; if (data) { inst._initProperties(data); obj = inst; } else { obj = null; } callback(err, obj); }); } else { this.findById(getIdValue(this, data), function (err, inst) { if (err) return callback(err); if (inst) { inst.updateAttributes(data, callback); } else { var obj = new Model(data); obj.save(data, callback); } }); } }; // upsert ~ remoting attributes DataAccessObject.upsert.shared = true; DataAccessObject.upsert.accepts = {arg: 'data', type: 'object', http: {source: 'body'}}; DataAccessObject.upsert.returns = {arg: 'data', type: 'object', root: true}; DataAccessObject.upsert.http = [ {verb: 'put', path: '/'} ]; /** * Find one record, same as `all`, limited by 1 and return object, not collection, * if not found, create using data provided as second argument * * @param {Object} query - search conditions: {where: {test: 'me'}}. * @param {Object} data - object to create. * @param {Function} cb - callback called with (err, instance) */ DataAccessObject.findOrCreate = function findOrCreate(query, data, callback) { if (typeof query === 'undefined') { query = {where: {}}; } if (typeof data === 'function' || typeof data === 'undefined') { callback = data; data = query && query.where; } if (typeof callback === 'undefined') { callback = function () {}; } var t = this; this.findOne(query, function (err, record) { if (err) return callback(err); if (record) return callback(null, record); t.create(data, callback); }); }; /** * Check whether object exitst in database * * @param {id} id - identifier of object (primary key value) * @param {Function} cb - callbacl called with (err, exists: Bool) */ DataAccessObject.exists = function exists(id, cb) { if (stillConnecting(this.dataSource, this, arguments)) return; if (id) { this.dataSource.connector.exists(this.modelName, id, cb); } else { cb(new Error('Model::exists requires positive id argument')); } }; // exists ~ remoting attributes DataAccessObject.exists.shared = true; DataAccessObject.exists.accepts = {arg: 'id', type: 'any'}; DataAccessObject.exists.returns = {arg: 'exists', type: 'any'}; DataAccessObject.exists.http = [ {verb: 'get', path: '/exists'} ]; /** * Find object by id * * @param {*} id - primary key value * @param {Function} cb - callback called with (err, instance) */ DataAccessObject.findById = function find(id, cb) { if (stillConnecting(this.dataSource, this, arguments)) return; this.dataSource.connector.find(this.modelName, id, function (err, data) { var obj = null; if (data) { if (!getIdValue(this, data)) { setIdValue(this, data, id); } obj = new this(); obj._initProperties(data, false); } cb(err, obj); }.bind(this)); }; // find ~ remoting attributes DataAccessObject.findById.accepts = [ {arg: 'id', type: 'any'} ]; DataAccessObject.findById.returns = [ {arg: 'data', type: 'any', root: true} ]; DataAccessObject.findById.shared = true; DataAccessObject.findById.http = [ {verb: 'get', path: '/:id'} ]; // alias function for backwards compat. DataAccessObject.all = function () { DataAccessObject.find.apply(this, arguments); } /** * Find all instances of Model, matched by query * make sure you have marked as `index: true` fields for filter or sort * * @param {Object} params (optional) * * - where: Object `{ key: val, key2: {gt: 'val2'}}` * - include: String, Object or Array. See DataAccessObject.include documentation. * - order: String * - limit: Number * - skip: Number * * @param {Function} callback (required) called with arguments: * * - err (null or Error) * - Array of instances */ DataAccessObject.find = function find(params, cb) { if (stillConnecting(this.dataSource, this, arguments)) return; if (arguments.length === 1) { cb = params; params = null; } var constr = this; params = params || {}; var fields = params.fields; var near = params && geo.nearFilter(params.where); var supportsGeo = !!this.dataSource.connector.buildNearFilter; // normalize fields as array of included property names if(fields) { params.fields = fieldsToArray(fields, Object.keys(this.properties)); } if(near) { if(supportsGeo) { // convert it this.dataSource.connector.buildNearFilter(params, near); } else if(params.where) { // do in memory query // using all documents this.dataSource.connector.all(this.modelName, {}, function (err, data) { var memory = new Memory(); var modelName = constr.modelName; if(err) { cb(err); } else if(Array.isArray(data)) { memory.define({ properties: constr.dataSource.definitions[constr.modelName].properties, settings: constr.dataSource.definitions[constr.modelName].settings, model: constr }); data.forEach(function (obj) { memory.create(modelName, obj, function () { // noop }); }); memory.all(modelName, params, cb); } else { cb(null, []); } }.bind(this)); // already handled return; } } this.dataSource.connector.all(this.modelName, params, function (err, data) { if (data && data.forEach) { data.forEach(function (d, i) { var obj = new constr; obj._initProperties(d, false, params.fields); if (params && params.include && params.collect) { data[i] = obj.__cachedRelations[params.collect]; } else { data[i] = obj; } }); if (data && data.countBeforeLimit) { data.countBeforeLimit = data.countBeforeLimit; } if(!supportsGeo && near) { data = geo.filter(data, near); } cb(err, data); } else cb(err, []); }); }; // all ~ remoting attributes DataAccessObject.find.accepts = {arg: 'filter', type: 'object'}; DataAccessObject.find.returns = {arg: 'data', type: 'array', root: true}; DataAccessObject.find.shared = true; DataAccessObject.find.http = [ {verb: 'get', path: '/'} ]; /** * Find one record, same as `all`, limited by 1 and return object, not collection * * @param {Object} params - search conditions: {where: {test: 'me'}} * @param {Function} cb - callback called with (err, instance) */ DataAccessObject.findOne = function findOne(params, cb) { if (stillConnecting(this.dataSource, this, arguments)) return; if (typeof params === 'function') { cb = params; params = {}; } params.limit = 1; this.find(params, function (err, collection) { if (err || !collection || !collection.length > 0) return cb(err, null); cb(err, collection[0]); }); }; DataAccessObject.findOne.shared = true; DataAccessObject.findOne.accepts = {arg: 'filter', type: 'object'}; DataAccessObject.findOne.returns = {arg: 'data', type: 'object', root: true}; DataAccessObject.findOne.http = [ {verb: 'get', path: '/findOne'} ]; /** * Destroy all records * @param {Function} cb - callback called with (err) */ DataAccessObject.deleteAll = DataAccessObject.destroyAll = function destroyAll(cb) { if (stillConnecting(this.dataSource, this, arguments)) return; this.dataSource.connector.destroyAll(this.modelName, function (err) { if ('function' === typeof cb) { cb(err); } }.bind(this)); }; /** * Destroy a record by id * @param {*} id The id value * @param {Function} cb - callback called with (err) */ DataAccessObject.deleteById = DataAccessObject.destroyById = function deleteById(id, cb) { if (stillConnecting(this.dataSource, this, arguments)) return; this.dataSource.connector.destroy(this.modelName, id, function (err) { if ('function' === typeof cb) { cb(err); } }.bind(this)); }; // deleteById ~ remoting attributes DataAccessObject.deleteById.accepts = [ {arg: 'id', type: 'any'} ]; DataAccessObject.deleteById.shared = true; DataAccessObject.deleteById.http = [ {verb: 'del', path: '/:id'} ]; /** * Return count of matched records * * @param {Object} where - search conditions (optional) * @param {Function} cb - callback, called with (err, count) */ DataAccessObject.count = function (where, cb) { if (stillConnecting(this.dataSource, this, arguments)) return; if (typeof where === 'function') { cb = where; where = null; } this.dataSource.connector.count(this.modelName, cb, where); }; // count ~ remoting attributes DataAccessObject.count.shared = true; DataAccessObject.count.accepts = [ {arg: 'where', type: 'object'} ]; DataAccessObject.count.returns = [ {arg: 'count', type: 'number'} ]; DataAccessObject.count.http = {verb: 'get', path: '/count'}; /** * Save instance. When instance haven't id, create method called instead. * Triggers: validate, save, update | create * @param options {validate: true, throws: false} [optional] * @param callback(err, obj) */ DataAccessObject.prototype.save = function (options, callback) { if (stillConnecting(this.constructor.dataSource, this, arguments)) return; if (typeof options == 'function') { callback = options; options = {}; } callback = callback || function () {}; options = options || {}; if (!('validate' in options)) { options.validate = true; } if (!('throws' in options)) { options.throws = false; } var inst = this; var data = inst.toObject(true); var Model = this.constructor; var modelName = Model.modelName; if (!getIdValue(Model, this)) { return Model.create(this, callback); } // validate first if (!options.validate) { return save(); } inst.isValid(function (valid) { if (valid) { save(); } else { var err = new ValidationError(inst); // throws option is dangerous for async usage if (options.throws) { throw err; } callback(err, inst); } }); // then save function save() { inst.trigger('save', function (saveDone) { inst.trigger('update', function (updateDone) { inst._adapter().save(modelName, inst.constructor._forDB(data), function (err) { if (err) { return callback(err, inst); } inst._initProperties(data, false); updateDone.call(inst, function () { saveDone.call(inst, function () { callback(err, inst); }); }); }); }, data); }, data); } }; /* // save ~ remoting attributes DataAccessObject.prototype.save.shared = true; DataAccessObject.prototype.save.returns = {arg: 'obj', type: 'object', root: true}; DataAccessObject.prototype.save.http = [ {verb: 'put', path: '/'} ]; */ DataAccessObject.prototype.isNewRecord = function () { return !getIdValue(this.constructor, this); }; /** * Return connector of current record * @private */ DataAccessObject.prototype._adapter = function () { return this.dataSource.connector; }; /** * Delete object from persistence * * @triggers `destroy` hook (async) before and after destroying object */ DataAccessObject.prototype.delete = DataAccessObject.prototype.destroy = function (cb) { if (stillConnecting(this.constructor.dataSource, this, arguments)) return; this.trigger('destroy', function (destroyed) { this._adapter().destroy(this.constructor.modelName, getIdValue(this.constructor, this), function (err) { if (err) { return cb(err); } destroyed(function () { if(cb) cb(); }); }.bind(this)); }); }; /* // destroy ~ remoting attributes DataAccessObject.prototype.destroy.shared = true; DataAccessObject.prototype.destroy.http = [ {verb: 'del', path: '/'} ]; */ /** * Update single attribute * * equals to `updateAttributes({name: value}, cb) * * @param {String} name - name of property * @param {Mixed} value - value of property * @param {Function} callback - callback called with (err, instance) */ DataAccessObject.prototype.updateAttribute = function updateAttribute(name, value, callback) { var data = {}; data[name] = value; this.updateAttributes(data, callback); }; /** * Update set of attributes * * this method performs validation before updating * * @trigger `validation`, `save` and `update` hooks * @param {Object} data - data to update * @param {Function} callback - callback called with (err, instance) */ DataAccessObject.prototype.updateAttributes = function updateAttributes(data, cb) { if (stillConnecting(this.constructor.dataSource, this, arguments)) return; var inst = this; var model = this.constructor.modelName; if (typeof data === 'function') { cb = data; data = null; } if (!data) { data = {}; } // update instance's properties Object.keys(data).forEach(function (key) { inst[key] = data[key]; }); inst.isValid(function (valid) { if (!valid) { if (cb) { cb(new ValidationError(inst), inst); } } else { inst.trigger('save', function (saveDone) { inst.trigger('update', function (done) { Object.keys(data).forEach(function (key) { inst[key] = data[key]; }); inst._adapter().updateAttributes(model, getIdValue(inst.constructor, inst), inst.constructor._forDB(data), function (err) { if (!err) { // update _was attrs Object.keys(data).forEach(function (key) { inst.__dataWas[key] = inst.__data[key]; }); } done.call(inst, function () { saveDone.call(inst, function () { cb(err, inst); }); }); }); }, data); }, data); } }, data); }; // updateAttributes ~ remoting attributes DataAccessObject.prototype.updateAttributes.shared = true; DataAccessObject.prototype.updateAttributes.accepts = {arg: 'data', type: 'object', http: {source: 'body'}}; DataAccessObject.prototype.updateAttributes.returns = {arg: 'data', type: 'object', root: true}; DataAccessObject.prototype.updateAttributes.http = [ {verb: 'put', path: '/'} ]; /** * Reload object from persistence * * @requires `id` member of `object` to be able to call `find` * @param {Function} callback - called with (err, instance) arguments */ DataAccessObject.prototype.reload = function reload(callback) { if (stillConnecting(this.constructor.dataSource, this, arguments)) return; this.constructor.findById(getIdValue(this.constructor, this), callback); }; DataAccessObject.prototype.reload.shared = true; DataAccessObject.prototype.reload.returns = {arg: 'data', type: 'object', root: true}; /** * Define readonly property on object * * @param {Object} obj * @param {String} key * @param {Mixed} value */ function defineReadonlyProp(obj, key, value) { Object.defineProperty(obj, key, { writable: false, enumerable: true, configurable: true, value: value }); } // jutil.mixin(DataAccessObject, validations.Validatable); jutil.mixin(DataAccessObject, Inclusion); jutil.mixin(DataAccessObject, Relation);