Apply custom timestamps to graphql

This commit is contained in:
DESKTOP-9GMDKNB\NeOMakinG 2019-01-25 14:37:22 +01:00
parent 897bd447e4
commit e6e6515d6e

View File

@ -59,7 +59,9 @@ module.exports = {
}
// Add timestamps attributes.
if (_.isArray(_.get(model, 'options.timestamps')) || _.get(model, 'options.timestamps') == true) {
const isArrayTimestamps = _.isArray(_.get(model, 'options.timestamps'));
if (isArrayTimestamps || _.get(model, 'options.timestamps') == true) {
Object.assign(initialState, {
createdAt: 'DateTime!',
updatedAt: 'DateTime!',
@ -68,11 +70,11 @@ module.exports = {
Object.assign(acc.resolver[globalId], {
createdAt: (obj) => {
// eslint-disable-line no-unused-vars
return obj.createdAt || obj.created_at;
return isArrayTimestamps ? obj[_.get(model, 'options.timestamps[0]')] : obj.createdAt || obj.created_at;
},
updatedAt: (obj) => {
// eslint-disable-line no-unused-vars
return obj.updatedAt || obj.updated_at;
return isArrayTimestamps ? obj[_.get(model, 'options.timestamps[1]')] : obj.updatedAt || obj.updated_at;
},
});
}