Merge pull request #2 from victorhqc/master

Clean pk builder
mean-expert-official · Apr 1, 2016 · 610e813 · 610e813
2 parents 6838de2 + 24fc1cb
commit 610e813
Showing 1 changed file with 2 additions and 6 deletions.
diff --git 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();
     }
-}
+}