diff --git a/builders/pk-builder.js b/builders/pk-builder.js index 562ac80..d3cef63 100644 --- a/builders/pk-builder.js +++ b/builders/pk-builder.js @@ -5,10 +5,6 @@ module.exports = class PrimaryKeyBuilder { constructor(Model) { this.Model = Model; } build() { - let pk = 'id'; - if (!this.Model.settings.idInjection) - for (let key in this.Model.rawProperties) - if (this.Model.rawProperties[key].id) pk = key; - return pk; + return this.Model.getIdName(); } -} \ No newline at end of file +}