| 1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279 |
3×
3×
3×
3×
3×
3×
3×
3×
3×
3×
426×
426×
426×
120×
306×
306×
306×
306×
306×
306×
306×
306×
306×
2×
306×
306×
306×
306×
306×
306×
306×
2×
306×
306×
45×
12×
12×
12×
12×
12×
12×
12×
12×
12×
12×
12×
12×
12×
4×
8×
12×
12×
12×
12×
12×
14×
12×
2×
306×
306×
306×
306×
2×
306×
306×
306×
306×
306×
306×
306×
306×
12×
12×
294×
294×
324×
294×
294×
294×
324×
294×
294×
324×
294×
2×
294×
294×
294×
33×
33×
294×
258×
36×
36×
36×
36×
| /**
* Module Dependencies
*/
var async = require('async');
var _ = require('lodash');
var usageError = require('../../utils/usageError');
var utils = require('../../utils/helpers');
var normalize = require('../../utils/normalize');
var Deferred = require('../deferred');
var callbacks = require('../../utils/callbacksRunner');
var nestedOperations = require('../../utils/nestedOperations');
var hop = utils.object.hasOwnProperty;
/**
* Update all records matching criteria
*
* @param {Object} criteria
* @param {Object} values
* @param {Function} cb
* @return Deferred object if no callback
*/
module.exports = function(criteria, values, cb, metaContainer) {
var self = this;
Iif (typeof criteria === 'function') {
cb = criteria;
criteria = null;
}
// Return Deferred or pass to adapter
if (typeof cb !== 'function') {
return new Deferred(this, this.update, criteria, values);
}
// If there was something defined in the criteria that would return no results, don't even
// run the query and just return an empty result set.
Iif (criteria === false) {
return cb(null, []);
}
// Ensure proper function signature
var usage = utils.capitalize(this.identity) + '.update(criteria, values, callback)';
Iif (!values) return usageError('No updated values specified!', usage, cb);
// Format Criteria and Values
var valuesObject = prepareArguments.call(this, criteria, values);
// Create any of the belongsTo associations and set the foreign key values
createBelongsTo.call(this, valuesObject, function(err) {
Iif (err) return cb(err);
beforeCallbacks.call(self, valuesObject.values, function(err) {
Iif (err) return cb(err);
updateRecords.call(self, valuesObject, cb, metaContainer);
});
}, metaContainer);
};
/**
* Prepare Arguments
*
* @param {Object} criteria
* @param {Object} values
* @return {Object}
*/
function prepareArguments(criteria, values) {
// Check if options is an integer or string and normalize criteria
// to object, using the specified primary key field.
criteria = normalize.expandPK(this, criteria);
// Normalize criteria
criteria = normalize.criteria(criteria);
// Pull out any associations in the values
var _values = _.cloneDeep(values);
var associations = nestedOperations.valuesParser.call(this, this.identity, this.waterline.schema, values);
// Replace associated models with their foreign key values if available.
// Unless the association has a custom primary key (we want to create the object)
values = nestedOperations.reduceAssociations.call(this, this.identity, this.waterline.schema, values, 'update');
// Cast values to proper types (handle numbers as strings)
values = this._cast.run(values);
return {
criteria: criteria,
values: values,
originalValues: _values,
associations: associations
};
}
/**
* Create BelongsTo Records
*
*/
function createBelongsTo(valuesObject, cb, metaContainer) {
var self = this;
async.each(valuesObject.associations.models.slice(0), function(item, next) {
// Check if value is an object. If not don't try and create it.
if (!_.isPlainObject(valuesObject.values[item])) return next();
// Check for any transformations
var attrName = hop(self._transformer._transformations, item) ? self._transformer._transformations[item] : item;
var attribute = self._schema.schema[attrName];
var modelName;
Iif (hop(attribute, 'collection')) modelName = attribute.collection;
Eif (hop(attribute, 'model')) modelName = attribute.model;
Iif (!modelName) return next();
var model = self.waterline.collections[modelName];
var pkValue = valuesObject.originalValues[item][model.primaryKey];
var criteria = {};
var pkField = hop(model._transformer._transformations, model.primaryKey) ? model._transformer._transformations[model.primaryKey] : model.primaryKey;
criteria[pkField] = pkValue;
// If a pkValue if found, do a findOrCreate and look for a record matching the pk.
var query;
if (pkValue) {
query = model.findOrCreate(criteria, valuesObject.values[item]);
} else {
query = model.create(valuesObject.values[item]);
}
Iif(metaContainer) {
query.meta(metaContainer);
}
query.exec(function(err, val) {
Iif (err) return next(err);
// attach the new model's pk value to the original value's key
var pk = val[model.primaryKey];
valuesObject.values[item] = pk;
// now we have pk value attached, remove it from models
_.remove(valuesObject.associations.models, function(_item) { return _item == item; });
next();
});
}, cb);
}
/**
* Run Before* Lifecycle Callbacks
*
* @param {Object} values
* @param {Function} cb
*/
function beforeCallbacks(values, cb) {
var self = this;
async.series([
// Run Validation with Validation LifeCycle Callbacks
function(cb) {
callbacks.validate(self, values, true, cb);
},
// Before Update Lifecycle Callback
function(cb) {
callbacks.beforeUpdate(self, values, cb);
}
], cb);
}
/**
* Update Records
*
* @param {Object} valuesObjecy
* @param {Function} cb
*/
function updateRecords(valuesObject, cb, metaContainer) {
var self = this;
// Automatically change updatedAt (if enabled)
Eif (this.autoUpdatedAt) {
valuesObject.values.updatedAt = new Date();
}
// Transform Values
valuesObject.values = this._transformer.serialize(valuesObject.values);
// Clean attributes
valuesObject.values = this._schema.cleanValues(valuesObject.values);
// Transform Search Criteria
valuesObject.criteria = self._transformer.serialize(valuesObject.criteria);
// Pass to adapter
self.adapter.update(valuesObject.criteria, valuesObject.values, function(err, values) {
if (err) {
Eif (typeof err === 'object') { err.model = self._model.globalId; }
return cb(err);
}
// If values is not an array, return an array
if (!Array.isArray(values)) values = [values];
// Unserialize each value
var transformedValues = values.map(function(value) {
return self._transformer.unserialize(value);
});
// Update any nested associations and run afterUpdate lifecycle callbacks for each parent
updatedNestedAssociations.call(self, valuesObject, transformedValues, function(err) {
Iif (err) return cb(err);
async.each(transformedValues, function(record, callback) {
callbacks.afterUpdate(self, record, callback);
}, function(err) {
Iif (err) return cb(err);
var models = transformedValues.map(function(value) {
return new self._model(value);
});
cb(null, models);
});
});
}, metaContainer);
}
/**
* Update Nested Associations
*
* @param {Object} valuesObject
* @param {Object} values
* @param {Function} cb
*/
function updatedNestedAssociations(valuesObject, values, cb) {
var self = this;
var associations = valuesObject.associations || {};
// Only attempt nested updates if values are an object or an array
associations.models = _.filter(associations.models, function(model) {
var vals = valuesObject.originalValues[model];
return _.isPlainObject(vals) || Array.isArray(vals);
});
// If no associations were used, return callback
if (associations.collections.length === 0 && associations.models.length === 0) {
return cb();
}
// Create an array of model instances for each parent
var parents = values.map(function(val) {
return new self._model(val);
});
// Update any nested associations found in the values object
var args = [parents, valuesObject.originalValues, valuesObject.associations, cb];
nestedOperations.update.apply(self, args);
}
|