loopback/lib/persisted-model.js

1073 lines
31 KiB
JavaScript
Raw Normal View History

2014-02-20 01:09:36 +00:00
/*!
* Module Dependencies.
*/
2014-02-20 01:09:36 +00:00
var Model = require('./model');
var registry = require('./registry');
var runtime = require('./runtime');
var assert = require('assert');
var async = require('async');
2014-02-20 01:09:36 +00:00
/**
* Extends Model with basic query and CRUD support.
*
2014-04-14 21:49:29 +00:00
* **Change Event**
*
* Listen for model changes using the `change` event.
*
* ```js
2014-06-05 07:45:09 +00:00
* MyPersistedModel.on('changed', function(obj) {
2014-04-14 21:49:29 +00:00
* console.log(obj) // => the changed model
* });
* ```
*
2014-06-05 07:45:09 +00:00
* @class PersistedModel
2014-02-20 01:09:36 +00:00
*/
2014-06-05 07:45:09 +00:00
var PersistedModel = module.exports = Model.extend('PersistedModel');
2014-02-20 01:09:36 +00:00
/*!
2014-06-05 07:45:09 +00:00
* Setup the `PersistedModel` constructor.
*/
2014-06-05 07:45:09 +00:00
PersistedModel.setup = function setupPersistedModel() {
2014-05-03 03:04:06 +00:00
// call Model.setup first
Model.setup.call(this);
2014-06-05 07:45:09 +00:00
var PersistedModel = this;
// enable change tracking (usually for replication)
if (this.settings.trackChanges) {
2014-06-05 07:45:09 +00:00
PersistedModel._defineChangeModel();
PersistedModel.once('dataSourceAttached', function() {
PersistedModel.enableChangeTracking();
});
}
2014-10-16 22:54:40 +00:00
2014-06-05 07:45:09 +00:00
PersistedModel.setupRemoting();
2014-10-16 22:54:40 +00:00
};
/*!
* Throw an error telling the user that the method is not available and why.
*/
function throwNotAttached(modelName, methodName) {
throw new Error(
'Cannot call ' + modelName + '.' + methodName + '().' +
' The ' + methodName + ' method has not been setup.' +
' The PersistedModel has not been correctly attached to a DataSource!'
);
2014-02-20 01:09:36 +00:00
}
/*!
* Convert null callbacks to 404 error objects.
* @param {HttpContext} ctx
* @param {Function} cb
*/
function convertNullToNotFoundError(ctx, cb) {
if (ctx.result !== null) return cb();
var modelName = ctx.method.sharedClass.name;
var id = ctx.getArgByName('id');
2014-05-16 03:32:12 +00:00
var msg = 'Unknown "' + modelName + '" id "' + id + '".';
2014-02-20 01:09:36 +00:00
var error = new Error(msg);
error.statusCode = error.status = 404;
cb(error);
}
/**
* Create new instance of Model class, saved in database
*
* @param {Object}|[{Object}] data Optional data object. Can be either a single model instance or an array of instances.
* @param {Function} cb Callback function with `cb(err, obj)` signature,
* where `err` is error object and `obj` is null or Model instance.
2014-02-20 01:09:36 +00:00
*/
PersistedModel.create = function(data, callback) {
throwNotAttached(this.modelName, 'create');
2014-02-20 01:09:36 +00:00
};
/**
* Update or insert a model instance
* @param {Object} data The model instance data
* @param {Function} [callback] The callback function
*/
2014-06-05 07:45:09 +00:00
PersistedModel.upsert = PersistedModel.updateOrCreate = function upsert(data, callback) {
throwNotAttached(this.modelName, 'upsert');
2014-02-20 01:09:36 +00:00
};
/**
* Find one record, same as `find`, but limited to one object. Returns an object, not collection.
* If not found, create the object using data provided as second argument.
2014-02-20 01:09:36 +00:00
*
* @param {Object} query Search conditions: {where: {test: 'me'}}.
* @param {Object} data Object to create.
* @param {Function} cb Callback called with `cb(err, instance)` signature.
2014-02-20 01:09:36 +00:00
*/
2014-06-05 07:45:09 +00:00
PersistedModel.findOrCreate = function findOrCreate(query, data, callback) {
throwNotAttached(this.modelName, 'findOrCreate');
2014-02-20 01:09:36 +00:00
};
2014-07-16 16:09:07 +00:00
PersistedModel.findOrCreate._delegate = true;
2014-02-20 01:09:36 +00:00
/**
* Check whether a model instance exists in database.
2014-02-20 01:09:36 +00:00
*
* @param {id} id Identifier of object (primary key value)
* @param {Function} cb Callback function called with (err, exists: Bool)
2014-02-20 01:09:36 +00:00
*/
2014-06-05 07:45:09 +00:00
PersistedModel.exists = function exists(id, cb) {
throwNotAttached(this.modelName, 'exists');
2014-02-20 01:09:36 +00:00
};
/**
* Find object by ID.
2014-02-20 01:09:36 +00:00
*
* @param {*} id - primary key value
2014-10-15 20:45:15 +00:00
* @param {Function} cb Callback function called with `(err, instances)`. Required.
2014-02-20 01:09:36 +00:00
*/
2014-06-05 07:45:09 +00:00
PersistedModel.findById = function find(id, cb) {
throwNotAttached(this.modelName, 'findById');
2014-02-20 01:09:36 +00:00
};
/**
* Find all model instances that match `filter` specification.
* See [Querying models](http://docs.strongloop.com/display/LB/Querying+models).
2014-02-20 01:09:36 +00:00
*
* @options {Object} [filter] Optional Filter JSON object; see below.
* @property {String|Object|Array} fields Identify fields to include in return result.
* <br/>See [Fields filter](http://docs.strongloop.com/display/LB/Fields+filter).
* @property {String|Object|Array} include See PersistedModel.include documentation.
* <br/>See [Include filter](http://docs.strongloop.com/display/LB/Include+filter).
* @property {Number} limit Maximum number of instances to return.
* <br/>See [Limit filter](http://docs.strongloop.com/display/LB/Limit+filter).
* @property {String} order Sort order: either "ASC" for ascending or "DESC" for descending.
* <br/>See [Order filter](http://docs.strongloop.com/display/LB/Order+filter).
* @property {Number} skip Number of results to skip.
* <br/>See [Skip filter](http://docs.strongloop.com/display/LB/Skip+filter).
* @property {Object} where Where clause, like
* ```
* { key: val, key2: {gt: 'val2'}, ...}
* ```
* <br/>See [Where filter](http://docs.strongloop.com/display/LB/Where+filter).
2014-02-20 01:09:36 +00:00
*
* @param {Function} Callback function called with `(err, returned-instances)`.
* @returns {Object} Array of model instances that match the filter; or Error.
2014-02-20 01:09:36 +00:00
*/
2014-06-05 07:45:09 +00:00
PersistedModel.find = function find(params, cb) {
throwNotAttached(this.modelName, 'find');
2014-02-20 01:09:36 +00:00
};
/**
* Find one model instance that matches `filter` specification.
* Same as `find`, but limited to one result;
* Returns object, not collection.
2014-02-20 01:09:36 +00:00
*
* @options {Object} [filter] Optional Filter JSON object; see below.
* @property {String|Object|Array} fields Identify fields to include in return result.
* <br/>See [Fields filter](http://docs.strongloop.com/display/LB/Fields+filter).
* @property {String|Object|Array} include See PersistedModel.include documentation.
* <br/>See [Include filter](http://docs.strongloop.com/display/LB/Include+filter).
* @property {Number} limit Maximum number of instances to return.
* <br/>See [Limit filter](http://docs.strongloop.com/display/LB/Limit+filter).
* @property {String} order Sort order: either "ASC" for ascending or "DESC" for descending.
* <br/>See [Order filter](http://docs.strongloop.com/display/LB/Order+filter).
* @property {Number} skip Number of results to skip.
* <br/>See [Skip filter](http://docs.strongloop.com/display/LB/Skip+filter).
* @property {Object} where Where clause, like
* ```
* { key: val, key2: {gt: 'val2'}, ...}
* ```
* <br/>See [Where filter](http://docs.strongloop.com/display/LB/Where+filter).
*
* @param {Function} Callback function called with `(err, returned-instance)`. Required.
* @returns {Object} First model instance that matches the filter; or Error.
2014-02-20 01:09:36 +00:00
*/
2014-06-05 07:45:09 +00:00
PersistedModel.findOne = function findOne(params, cb) {
throwNotAttached(this.modelName, 'findOne');
2014-02-20 01:09:36 +00:00
};
/**
* Destroy all model instances that match the optional `filter` specification.
*
* @options {Object} [where] Optional where filter JSON object; see below.
* @property {Object} where Where clause, like
* ```
* { key: val, key2: {gt: 'val2'}, ...}
* ```
*
* @param {Function} [cb] - callback called with `(err)`.
2014-02-20 01:09:36 +00:00
*/
2014-06-05 07:45:09 +00:00
PersistedModel.destroyAll = function destroyAll(where, cb) {
throwNotAttached(this.modelName, 'destroyAll');
2014-02-20 01:09:36 +00:00
};
/**
* Alias for `destroyAll`
*/
PersistedModel.remove = PersistedModel.destroyAll;
/**
* Alias for `destroyAll`
*/
PersistedModel.deleteAll = PersistedModel.destroyAll;
2014-06-21 06:08:35 +00:00
/**
* Update multiple instances that match the where clause
*
* Example:
*
*```js
* Employee.updateAll({managerId: 'x001'}, {managerId: 'x002'}, function(err, count) {
2014-06-21 06:08:35 +00:00
* ...
* });
* ```
*
* @options {Object} [where] Optional where filter JSON object; see below.
* @property {Object} where Where clause, like
* ```
* { key: val, key2: {gt: 'val2'}, ...}
* ```
2014-06-21 06:08:35 +00:00
* @param {Object} data Changes to be made
* @param {Function} cb Callback function called with (err, count).
2014-06-21 06:08:35 +00:00
*/
PersistedModel.updateAll = function updateAll(where, data, cb) {
throwNotAttached(this.modelName, 'updateAll');
};
/**
* Alias for updateAll.
*/
PersistedModel.update = PersistedModel.updateAll;
2014-06-21 06:08:35 +00:00
2014-02-20 01:09:36 +00:00
/**
* Destroy model instance with the specified ID.
* @param {*} id The ID value of model instance to delete.
* @param {Function} cb Callback function called with (err).
2014-02-20 01:09:36 +00:00
*/
2014-06-05 07:45:09 +00:00
PersistedModel.destroyById = function deleteById(id, cb) {
throwNotAttached(this.modelName, 'deleteById');
2014-02-20 01:09:36 +00:00
};
/**
* Alias for destroyById.
*/
2014-10-16 22:54:40 +00:00
PersistedModel.removeById = PersistedModel.destroyById;
/**
* Alias for destroyById.
*/
PersistedModel.deleteById = PersistedModel.destroyById;
/**
* Return the number of records that match the optional filter.
* @options {Object} [filter] Optional where filter JSON object; see below.
* @property {Object} where Where clause, like
* ```
* { key: val, key2: {gt: 'val2'}, ...}
* ```
* @param {Function} cb Callback function called with (err, count).
2014-02-20 01:09:36 +00:00
*/
PersistedModel.count = function(where, cb) {
throwNotAttached(this.modelName, 'count');
2014-02-20 01:09:36 +00:00
};
/**
* Save model instance. If the instance doesn't have an ID, then the [create](#persistedmodelcreatedata-cb) method is called instead.
* Triggers: validate, save, update, or create.
* @options {Object} [options] See below.
* @property {Boolean} validate
* @property {Boolean} throws
* @param {Function} [callback] Callback function called with (err, obj).
2014-02-20 01:09:36 +00:00
*/
PersistedModel.prototype.save = function(options, callback) {
var Model = this.constructor;
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 id = this.getId();
if (!id) {
return Model.create(this, callback);
}
// validate first
if (!options.validate) {
return save();
}
inst.isValid(function(valid) {
if (valid) {
save();
} else {
var err = new Model.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) {
Model.upsert(inst, function(err) {
inst._initProperties(data);
updateDone.call(inst, function() {
saveDone.call(inst, function() {
callback(err, inst);
});
});
});
}, data);
}, data);
}
2014-02-20 01:09:36 +00:00
};
/**
* Determine if the data model is new.
* @returns {Boolean} Returns true if the data model is new; false otherwise.
2014-02-20 01:09:36 +00:00
*/
PersistedModel.prototype.isNewRecord = function() {
throwNotAttached(this.constructor.modelName, 'isNewRecord');
2014-02-20 01:09:36 +00:00
};
/**
* Deletes the model from persistence.
* Triggers `destroy` hook (async) before and after destroying object.
* @param {Function} callback Callback function.
2014-02-20 01:09:36 +00:00
*/
PersistedModel.prototype.destroy = function(cb) {
throwNotAttached(this.constructor.modelName, 'destroy');
2014-02-20 01:09:36 +00:00
};
/**
* Alias for destroy.
* @header PersistedModel.remove
*/
PersistedModel.prototype.remove = PersistedModel.prototype.destroy;
/**
* Alias for destroy.
* @header PersistedModel.delete
*/
PersistedModel.prototype.delete = PersistedModel.prototype.destroy;
2014-06-05 07:45:09 +00:00
PersistedModel.prototype.destroy._delegate = true;
2014-02-20 01:09:36 +00:00
/**
* Update a single attribute.
* Equivalent to `updateAttributes({name: 'value'}, cb)`
2014-02-20 01:09:36 +00:00
*
* @param {String} name Name of property
* @param {Mixed} value Value of property
* @param {Function} callback Callback function called with (err, instance).
2014-02-20 01:09:36 +00:00
*/
2014-06-05 07:45:09 +00:00
PersistedModel.prototype.updateAttribute = function updateAttribute(name, value, callback) {
throwNotAttached(this.constructor.modelName, 'updateAttribute');
2014-02-20 01:09:36 +00:00
};
/**
* Update set of attributes. Performs validation before updating.
2014-02-20 01:09:36 +00:00
*
* Trigger: `validation`, `save` and `update` hooks
* @param {Object} data Dta to update.
* @param {Function} callback Callback function called with (err, instance).
2014-02-20 01:09:36 +00:00
*/
2014-06-05 07:45:09 +00:00
PersistedModel.prototype.updateAttributes = function updateAttributes(data, cb) {
throwNotAttached(this.modelName, 'updateAttributes');
2014-02-20 01:09:36 +00:00
};
/**
* Reload object from persistence. Requires `id` member of `object` to be able to call `find`.
* @param {Function} callback Callback function called with (err, instance) arguments.
2014-02-20 01:09:36 +00:00
*/
2014-06-05 07:45:09 +00:00
PersistedModel.prototype.reload = function reload(callback) {
throwNotAttached(this.constructor.modelName, 'reload');
2014-02-20 01:09:36 +00:00
};
/**
2014-06-05 07:45:09 +00:00
* Set the correct `id` property for the `PersistedModel`. If a `Connector` defines
* a `setId` method it will be used. Otherwise the default lookup is used.
* Override this method to handle complex IDs.
*
* @param {*} val The `id` value. Will be converted to the type that the `id` property specifies.
*/
2014-06-05 07:45:09 +00:00
PersistedModel.prototype.setId = function(val) {
var ds = this.getDataSource();
this[this.getIdName()] = val;
2014-10-16 22:54:40 +00:00
};
/**
2014-06-05 07:45:09 +00:00
* Get the `id` value for the `PersistedModel`.
*
* @returns {*} The `id` value
*/
2014-06-05 07:45:09 +00:00
PersistedModel.prototype.getId = function() {
var data = this.toObject();
if (!data) return;
return data[this.getIdName()];
2014-10-16 22:54:40 +00:00
};
/**
* Get the id property name of the constructor.
*
* @returns {String} The `id` property name
*/
2014-06-05 07:45:09 +00:00
PersistedModel.prototype.getIdName = function() {
return this.constructor.getIdName();
2014-10-16 22:54:40 +00:00
};
/**
* Get the id property name of the constructor.
*
* @returns {String} The `id` property name
*/
2014-06-05 07:45:09 +00:00
PersistedModel.getIdName = function() {
var Model = this;
var ds = Model.getDataSource();
if (ds.idName) {
return ds.idName(Model.modelName);
} else {
return 'id';
}
2014-10-16 22:54:40 +00:00
};
2014-06-05 07:45:09 +00:00
PersistedModel.setupRemoting = function() {
var PersistedModel = this;
var typeName = PersistedModel.modelName;
var options = PersistedModel.settings;
function setRemoting(scope, name, options) {
var fn = scope[name];
fn._delegate = true;
2014-06-05 07:45:09 +00:00
options.isStatic = scope === PersistedModel;
PersistedModel.remoteMethod(name, options);
2014-05-16 19:32:54 +00:00
}
2014-06-05 07:45:09 +00:00
setRemoting(PersistedModel, 'create', {
description: 'Create a new instance of the model and persist it into the data source',
accepts: {arg: 'data', type: 'object', description: 'Model instance data', http: {source: 'body'}},
returns: {arg: 'data', type: typeName, root: true},
http: {verb: 'post', path: '/'}
});
2014-06-05 07:45:09 +00:00
setRemoting(PersistedModel, 'upsert', {
aliases: ['updateOrCreate'],
description: 'Update an existing model instance or insert a new one into the data source',
accepts: {arg: 'data', type: 'object', description: 'Model instance data', http: {source: 'body'}},
returns: {arg: 'data', type: typeName, root: true},
http: {verb: 'put', path: '/'}
});
2014-06-05 07:45:09 +00:00
setRemoting(PersistedModel, 'exists', {
description: 'Check whether a model instance exists in the data source',
accepts: {arg: 'id', type: 'any', description: 'Model id', required: true},
returns: {arg: 'exists', type: 'boolean'},
2014-07-31 04:57:45 +00:00
http: [
{verb: 'get', path: '/:id/exists'},
{verb: 'head', path: '/:id'}
],
rest: {
// After hook to map exists to 200/404 for HEAD
after: function(ctx, cb) {
if (ctx.req.method === 'GET') {
// For GET, return {exists: true|false} as is
return cb();
}
if (!ctx.result.exists) {
2014-07-31 04:57:45 +00:00
var modelName = ctx.method.sharedClass.name;
var id = ctx.getArgByName('id');
var msg = 'Unknown "' + modelName + '" id "' + id + '".';
var error = new Error(msg);
error.statusCode = error.status = 404;
cb(error);
} else {
cb();
}
}
}
});
2014-06-05 07:45:09 +00:00
setRemoting(PersistedModel, 'findById', {
description: 'Find a model instance by id from the data source',
accepts: {
arg: 'id', type: 'any', description: 'Model id', required: true,
http: {source: 'path'}
},
returns: {arg: 'data', type: typeName, root: true},
http: {verb: 'get', path: '/:id'},
rest: {after: convertNullToNotFoundError}
});
2014-06-05 07:45:09 +00:00
setRemoting(PersistedModel, 'find', {
description: 'Find all instances of the model matched by filter from the data source',
accepts: {arg: 'filter', type: 'object', description: 'Filter defining fields, where, orderBy, offset, and limit'},
returns: {arg: 'data', type: [typeName], root: true},
http: {verb: 'get', path: '/'}
});
2014-06-05 07:45:09 +00:00
setRemoting(PersistedModel, 'findOne', {
description: 'Find first instance of the model matched by filter from the data source',
accepts: {arg: 'filter', type: 'object', description: 'Filter defining fields, where, orderBy, offset, and limit'},
returns: {arg: 'data', type: typeName, root: true},
2014-06-21 06:48:46 +00:00
http: {verb: 'get', path: '/findOne'},
rest: {after: convertNullToNotFoundError}
});
2014-06-05 07:45:09 +00:00
setRemoting(PersistedModel, 'destroyAll', {
description: 'Delete all matching records',
accepts: {arg: 'where', type: 'object', description: 'filter.where object'},
2014-05-16 00:27:02 +00:00
http: {verb: 'del', path: '/'},
shared: false
});
2014-06-21 06:08:35 +00:00
setRemoting(PersistedModel, 'updateAll', {
aliases: ['update'],
2014-06-21 06:08:35 +00:00
description: 'Update instances of the model matched by where from the data source',
accepts: [
{arg: 'where', type: 'object', http: {source: 'query'},
description: 'Criteria to match model instances'},
{arg: 'data', type: 'object', http: {source: 'body'},
description: 'An object of model property name/value pairs'},
],
http: {verb: 'post', path: '/update'}
});
2014-07-16 16:09:07 +00:00
setRemoting(PersistedModel, 'deleteById', {
aliases: ['destroyById', 'removeById'],
description: 'Delete a model instance by id from the data source',
2014-05-03 03:04:06 +00:00
accepts: {arg: 'id', type: 'any', description: 'Model id', required: true,
http: {source: 'path'}},
http: {verb: 'del', path: '/:id'}
});
2014-06-05 07:45:09 +00:00
setRemoting(PersistedModel, 'count', {
description: 'Count instances of the model matched by where from the data source',
accepts: {arg: 'where', type: 'object', description: 'Criteria to match model instances'},
returns: {arg: 'count', type: 'number'},
http: {verb: 'get', path: '/count'}
});
2014-06-05 07:45:09 +00:00
setRemoting(PersistedModel.prototype, 'updateAttributes', {
description: 'Update attributes for a model instance and persist it into the data source',
accepts: {arg: 'data', type: 'object', http: {source: 'body'}, description: 'An object of model property name/value pairs'},
returns: {arg: 'data', type: typeName, root: true},
http: {verb: 'put', path: '/'}
});
if (options.trackChanges) {
2014-06-05 07:45:09 +00:00
setRemoting(PersistedModel, 'diff', {
description: 'Get a set of deltas and conflicts since the given checkpoint',
accepts: [
{arg: 'since', type: 'number', description: 'Find deltas since this checkpoint'},
{arg: 'remoteChanges', type: 'array', description: 'an array of change objects',
http: {source: 'body'}}
2014-10-16 22:54:40 +00:00
],
returns: {arg: 'result', type: 'object', root: true},
http: {verb: 'post', path: '/diff'}
});
2014-06-05 07:45:09 +00:00
setRemoting(PersistedModel, 'changes', {
description: 'Get the changes to a model since a given checkpoint.' +
'Provide a filter object to reduce the number of results returned.',
accepts: [
{arg: 'since', type: 'number', description: 'Only return changes since this checkpoint'},
{arg: 'filter', type: 'object', description: 'Only include changes that match this filter'}
],
returns: {arg: 'changes', type: 'array', root: true},
http: {verb: 'get', path: '/changes'}
});
2014-10-16 22:54:40 +00:00
2014-06-05 07:45:09 +00:00
setRemoting(PersistedModel, 'checkpoint', {
description: 'Create a checkpoint.',
returns: {arg: 'checkpoint', type: 'object', root: true},
http: {verb: 'post', path: '/checkpoint'}
});
2014-06-05 07:45:09 +00:00
setRemoting(PersistedModel, 'currentCheckpoint', {
description: 'Get the current checkpoint.',
returns: {arg: 'checkpoint', type: 'object', root: true},
http: {verb: 'get', path: '/checkpoint'}
});
2014-06-05 07:45:09 +00:00
setRemoting(PersistedModel, 'createUpdates', {
description: 'Create an update list from a delta list',
accepts: {arg: 'deltas', type: 'array', http: {source: 'body'}},
returns: {arg: 'updates', type: 'array', root: true},
http: {verb: 'post', path: '/create-updates'}
});
2014-06-05 07:45:09 +00:00
setRemoting(PersistedModel, 'bulkUpdate', {
description: 'Run multiple updates at once. Note: this is not atomic.',
accepts: {arg: 'updates', type: 'array'},
http: {verb: 'post', path: '/bulk-update'}
});
2014-06-05 07:45:09 +00:00
setRemoting(PersistedModel, 'rectifyAllChanges', {
description: 'Rectify all Model changes.',
http: {verb: 'post', path: '/rectify-all'}
});
2014-06-05 07:45:09 +00:00
setRemoting(PersistedModel, 'rectifyChange', {
description: 'Tell loopback that a change to the model with the given id has occurred.',
accepts: {arg: 'id', type: 'any', http: {source: 'path'}},
http: {verb: 'post', path: '/:id/rectify-change'}
});
}
2014-10-16 22:54:40 +00:00
};
/**
* Get a set of deltas and conflicts since the given checkpoint.
*
* See `Change.diff()` for details.
2014-10-16 22:54:40 +00:00
*
* @param {Number} since Find deltas since this checkpoint
* @param {Array} remoteChanges An array of change objects
2014-10-16 22:54:40 +00:00
* @param {Function} callback
*/
2014-06-05 07:45:09 +00:00
PersistedModel.diff = function(since, remoteChanges, callback) {
var Change = this.getChangeModel();
Change.diff(this.modelName, since, remoteChanges, callback);
2014-10-16 22:54:40 +00:00
};
/**
* Get the changes to a model since a given checkpoint. Provide a filter object
* to reduce the number of results returned.
* @param {Number} since Only return changes since this checkpoint
* @param {Object} filter Only include changes that match this filter
* (same as `Model.find(filter, ...)`)
* @callback {Function} callback
* @param {Error} err
* @param {Array} changes An array of `Change` objects
* @end
*/
2014-06-05 07:45:09 +00:00
PersistedModel.changes = function(since, filter, callback) {
if (typeof since === 'function') {
filter = {};
callback = since;
since = -1;
}
if (typeof filter === 'function') {
callback = filter;
since = -1;
filter = {};
}
var idName = this.dataSource.idName(this.modelName);
var Change = this.getChangeModel();
var model = this;
filter = filter || {};
filter.fields = {};
filter.where = filter.where || {};
filter.fields[idName] = true;
// TODO(ritch) this whole thing could be optimized a bit more
Change.find({
checkpoint: {gt: since},
modelName: this.modelName
}, function(err, changes) {
if (err) return callback(err);
var ids = changes.map(function(change) {
return change.getModelId();
});
filter.where[idName] = {inq: ids};
model.find(filter, function(err, models) {
if (err) return callback(err);
var modelIds = models.map(function(m) {
return m[idName].toString();
});
callback(null, changes.filter(function(ch) {
if (ch.type() === Change.DELETE) return true;
return modelIds.indexOf(ch.modelId) > -1;
}));
});
});
2014-10-16 22:54:40 +00:00
};
/**
* Create a checkpoint.
2014-10-16 22:54:40 +00:00
*
* @param {Function} callback
*/
2014-06-05 07:45:09 +00:00
PersistedModel.checkpoint = function(cb) {
var Checkpoint = this.getChangeModel().getCheckpointModel();
this.getSourceId(function(err, sourceId) {
if (err) return cb(err);
Checkpoint.create({
sourceId: sourceId
}, cb);
});
2014-10-16 22:54:40 +00:00
};
/**
* Get the current checkpoint id.
2014-10-16 22:54:40 +00:00
*
* @callback {Function} callback
* @param {Error} err
* @param {Number} currentCheckpointId
* @end
*/
2014-06-05 07:45:09 +00:00
PersistedModel.currentCheckpoint = function(cb) {
var Checkpoint = this.getChangeModel().getCheckpointModel();
Checkpoint.current(cb);
2014-10-16 22:54:40 +00:00
};
/**
* Replicate changes since the given checkpoint to the given target model.
*
* @param {Number} [since] Since this checkpoint
* @param {Model} targetModel Target this model class
* @param {Object} [options]
* @param {Object} [options.filter] Replicate models that match this filter
* @callback {Function} [callback]
* @param {Error} err
* @param {Conflict[]} conflicts A list of changes that could not be replicated
* due to conflicts.
*/
2014-06-05 07:45:09 +00:00
PersistedModel.replicate = function(since, targetModel, options, callback) {
var lastArg = arguments[arguments.length - 1];
if (typeof lastArg === 'function' && arguments.length > 1) {
callback = lastArg;
}
if (typeof since === 'function' && since.modelName) {
targetModel = since;
since = -1;
}
options = options || {};
var sourceModel = this;
var diff;
var updates;
var Change = this.getChangeModel();
var TargetChange = targetModel.getChangeModel();
var changeTrackingEnabled = Change && TargetChange;
assert(
changeTrackingEnabled,
'You must enable change tracking before replicating'
);
callback = callback || function defaultReplicationCallback(err) {
if (err) throw err;
2014-10-16 22:54:40 +00:00
};
var tasks = [
getSourceChanges,
getDiffFromTarget,
createSourceUpdates,
bulkUpdate,
checkpoint
];
async.waterfall(tasks, done);
function getSourceChanges(cb) {
sourceModel.changes(since, options.filter, cb);
}
function getDiffFromTarget(sourceChanges, cb) {
targetModel.diff(since, sourceChanges, cb);
}
function createSourceUpdates(_diff, cb) {
diff = _diff;
diff.conflicts = diff.conflicts || [];
if (diff && diff.deltas && diff.deltas.length) {
sourceModel.createUpdates(diff.deltas, cb);
} else {
// nothing to replicate
done();
}
}
function bulkUpdate(updates, cb) {
targetModel.bulkUpdate(updates, cb);
}
function checkpoint() {
var cb = arguments[arguments.length - 1];
sourceModel.checkpoint(cb);
}
function done(err) {
if (err) return callback(err);
var conflicts = diff.conflicts.map(function(change) {
return new Change.Conflict(
change.modelId, sourceModel, targetModel
);
});
if (conflicts.length) {
sourceModel.emit('conflicts', conflicts);
}
if (callback) callback(null, conflicts);
}
2014-10-16 22:54:40 +00:00
};
/**
* Create an update list (for `Model.bulkUpdate()`) from a delta list
* (result of `Change.diff()`).
2014-10-16 22:54:40 +00:00
*
* @param {Array} deltas
* @param {Function} callback
*/
2014-06-05 07:45:09 +00:00
PersistedModel.createUpdates = function(deltas, cb) {
var Change = this.getChangeModel();
var updates = [];
var Model = this;
var tasks = [];
deltas.forEach(function(change) {
change = new Change(change);
var type = change.type();
var update = {type: type, change: change};
switch (type) {
case Change.CREATE:
case Change.UPDATE:
tasks.push(function(cb) {
Model.findById(change.modelId, function(err, inst) {
if (err) return cb(err);
if (!inst) {
2014-05-12 17:36:10 +00:00
console.error('missing data for change:', change);
return cb &&
cb(new Error('missing data for change: ' + change.modelId));
2014-05-12 17:36:10 +00:00
}
if (inst.toObject) {
update.data = inst.toObject();
} else {
update.data = inst;
}
updates.push(update);
cb();
});
});
break;
case Change.DELETE:
updates.push(update);
break;
}
});
async.parallel(tasks, function(err) {
if (err) return cb(err);
cb(null, updates);
});
2014-10-16 22:54:40 +00:00
};
/**
* Apply an update list.
*
* **Note: this is not atomic**
2014-10-16 22:54:40 +00:00
*
* @param {Array} updates An updates list (usually from Model.createUpdates())
* @param {Function} callback
*/
2014-06-05 07:45:09 +00:00
PersistedModel.bulkUpdate = function(updates, callback) {
var tasks = [];
var Model = this;
var idName = this.dataSource.idName(this.modelName);
var Change = this.getChangeModel();
updates.forEach(function(update) {
switch (update.type) {
case Change.UPDATE:
case Change.CREATE:
// var model = new Model(update.data);
// tasks.push(model.save.bind(model));
tasks.push(function(cb) {
var model = new Model(update.data);
model.save(cb);
});
break;
case Change.DELETE:
var data = {};
data[idName] = update.change.modelId;
var model = new Model(data);
tasks.push(model.destroy.bind(model));
break;
}
});
async.parallel(tasks, callback);
2014-10-16 22:54:40 +00:00
};
/**
* Get the `Change` model.
2014-10-16 22:54:40 +00:00
*
* @throws {Error} Throws an error if the change model is not correctly setup.
* @return {Change}
*/
2014-06-05 07:45:09 +00:00
PersistedModel.getChangeModel = function() {
var changeModel = this.Change;
var isSetup = changeModel && changeModel.dataSource;
assert(isSetup, 'Cannot get a setup Change model');
return changeModel;
2014-10-16 22:54:40 +00:00
};
/**
* Get the source identifier for this model / dataSource.
2014-10-16 22:54:40 +00:00
*
* @callback {Function} callback
* @param {Error} err
* @param {String} sourceId
*/
2014-06-05 07:45:09 +00:00
PersistedModel.getSourceId = function(cb) {
var dataSource = this.dataSource;
if (!dataSource) {
this.once('dataSourceAttached', this.getSourceId.bind(this, cb));
}
assert(
2014-10-16 22:54:40 +00:00
dataSource.connector.name,
'Model.getSourceId: cannot get id without dataSource.connector.name'
);
var id = [dataSource.connector.name, this.modelName].join('-');
cb(null, id);
2014-10-16 22:54:40 +00:00
};
/**
* Enable the tracking of changes made to the model. Usually for replication.
*/
2014-06-05 07:45:09 +00:00
PersistedModel.enableChangeTracking = function() {
var Model = this;
var Change = this.Change || this._defineChangeModel();
var cleanupInterval = Model.settings.changeCleanupInterval || 30000;
assert(this.dataSource, 'Cannot enableChangeTracking(): ' + this.modelName
+ ' is not attached to a dataSource');
Change.attachTo(this.dataSource);
Change.getCheckpointModel().attachTo(this.dataSource);
2014-05-20 19:46:43 +00:00
Model.afterSave = function afterSave(next) {
Model.rectifyChange(this.getId(), next);
2014-10-16 22:54:40 +00:00
};
2014-05-20 19:46:43 +00:00
Model.afterDestroy = function afterDestroy(next) {
Model.rectifyChange(this.getId(), next);
2014-10-16 22:54:40 +00:00
};
Model.on('deletedAll', cleanup);
if (runtime.isServer) {
// initial cleanup
cleanup();
// cleanup
setInterval(cleanup, cleanupInterval);
}
function cleanup() {
Model.rectifyAllChanges(function(err) {
if (err) {
console.error(Model.modelName + ' Change Cleanup Error:');
console.error(err);
}
});
}
2014-10-16 22:54:40 +00:00
};
2014-06-05 07:45:09 +00:00
PersistedModel._defineChangeModel = function() {
var BaseChangeModel = registry.getModel('Change');
assert(BaseChangeModel,
'Change model must be defined before enabling change replication');
this.Change = BaseChangeModel.extend(this.modelName + '-change',
{},
{
trackModel: this
}
);
return this.Change;
2014-10-16 22:54:40 +00:00
};
2014-06-05 07:45:09 +00:00
PersistedModel.rectifyAllChanges = function(callback) {
this.getChangeModel().rectifyAll(callback);
2014-10-16 22:54:40 +00:00
};
/**
* Handle a change error. Override this method in a subclassing model to customize
* change error handling.
*
* @param {Error} err
*/
2014-06-05 07:45:09 +00:00
PersistedModel.handleChangeError = function(err) {
if (err) {
console.error(Model.modelName + ' Change Tracking Error:');
console.error(err);
}
2014-10-16 22:54:40 +00:00
};
/**
* Tell loopback that a change to the model with the given id has occurred.
*
* @param {*} id The id of the model that has changed
* @callback {Function} callback
* @param {Error} err
*/
2014-06-05 07:45:09 +00:00
PersistedModel.rectifyChange = function(id, callback) {
var Change = this.getChangeModel();
Change.rectifyModelChanges(this.modelName, [id], callback);
2014-10-16 22:54:40 +00:00
};
2014-06-05 07:45:09 +00:00
PersistedModel.setup();