Allow filtering of attributes for eager loading

lib/dao-factory.js

@@ -169,7 +169,7 @@ module.exports = (function() {
 
 
     this.DAO.prototype.__factory        = this
     this.DAO.prototype.__factory        = this
     this.DAO.prototype.hasDefaultValues = !Utils._.isEmpty(this.DAO.prototype.defaultValues)
     this.DAO.prototype.hasDefaultValues = !Utils._.isEmpty(this.DAO.prototype.defaultValues)
-
+    
     return this
     return this
   }
   }
 
 
@@ -334,8 +334,8 @@ module.exports = (function() {
         options = {where: parseInt(Number(options) || 0, 0)}
         options = {where: parseInt(Number(options) || 0, 0)}
       }
       }
     }
     }
-
-    options.limit = 1
+options
+    .limit = 1
 
 
     return this.QueryInterface.select(this, this.getTableName(), options, Utils._.defaults({
     return this.QueryInterface.select(this, this.getTableName(), options, Utils._.defaults({
       plain: true,
       plain: true,
@@ -651,6 +651,18 @@ module.exports = (function() {
         delete include.model
         delete include.model
       }
       }
 
 
+      if (include.hasOwnProperty('attributes')) {
+        var primaryKeys;
+        if (include.daoFactory.hasPrimaryKeys) {
+          primaryKeys = include.daoFactory.primaryKeys
+        } else {
+          primaryKeys = ['id']
+        }
+        include.attributes = include.attributes.concat(primaryKeys)
+      } else {
+        include.attributes = Object.keys(include.daoFactory.attributes)
+      }
+
       if (include.hasOwnProperty('daoFactory') && (include.hasOwnProperty('as'))) {
       if (include.hasOwnProperty('daoFactory') && (include.hasOwnProperty('as'))) {
         var usesAlias   = (include.as !== include.daoFactory.tableName)
         var usesAlias   = (include.as !== include.daoFactory.tableName)
           , association = (usesAlias ? this.getAssociationByAlias(include.as) : this.getAssociation(include.daoFactory))
           , association = (usesAlias ? this.getAssociationByAlias(include.as) : this.getAssociation(include.daoFactory))

lib/dialects/mysql/query-generator.js

@@ -181,7 +181,7 @@ module.exports = (function() {
         var optAttributes = options.attributes === '*' ? [options.table + '.*'] : [options.attributes]
         var optAttributes = options.attributes === '*' ? [options.table + '.*'] : [options.attributes]
 
 
         options.include.forEach(function(include) {
         options.include.forEach(function(include) {
-          var attributes = Object.keys(include.daoFactory.attributes).map(function(attr) {
+          var attributes = include.attributes.map(function(attr) {
             return this.quoteIdentifier(include.as) + "." + this.quoteIdentifier(attr) + " AS " + this.quoteIdentifier(include.as + "." + attr)
             return this.quoteIdentifier(include.as) + "." + this.quoteIdentifier(attr) + " AS " + this.quoteIdentifier(include.as + "." + attr)
           }.bind(this))
           }.bind(this))
 
 

lib/dialects/postgres/query-generator.js

@@ -252,7 +252,7 @@ module.exports = (function() {
         var optAttributes = options.attributes === '*' ? [options.table + '.*'] : [options.attributes]
         var optAttributes = options.attributes === '*' ? [options.table + '.*'] : [options.attributes]
 
 
         options.include.forEach(function(include) {
         options.include.forEach(function(include) {
-          var attributes = Object.keys(include.daoFactory.attributes).map(function(attr) {
+          var attributes = include.attributes.map(function(attr) {
             return this.quoteIdentifier(include.as) + "." + this.quoteIdentifier(attr) + " AS " + this.quoteIdentifier(include.as + "." + attr, true)
             return this.quoteIdentifier(include.as) + "." + this.quoteIdentifier(attr) + " AS " + this.quoteIdentifier(include.as + "." + attr, true)
           }.bind(this))
           }.bind(this))
 
 

lib/dialects/sqlite/query-generator.js

@@ -163,7 +163,7 @@ module.exports = (function() {
         var optAttributes = options.attributes === '*' ? [options.table + '.*'] : [options.attributes]
         var optAttributes = options.attributes === '*' ? [options.table + '.*'] : [options.attributes]
 
 
         options.include.forEach(function(include) {
         options.include.forEach(function(include) {
-          var attributes = Object.keys(include.daoFactory.attributes).map(function(attr) {
+          var attributes = include.attributes.map(function(attr) {
             return this.quoteIdentifier(include.as) + "." + this.quoteIdentifier(attr) + " AS " + this.quoteIdentifier(include.as + "." + attr)
             return this.quoteIdentifier(include.as) + "." + this.quoteIdentifier(attr) + " AS " + this.quoteIdentifier(include.as + "." + attr)
           }.bind(this))
           }.bind(this))
 
 

test/dao-factory.test.js

@@ -1404,6 +1404,36 @@ describe(Support.getTestDialectTeaser("DAOFactory"), function () {
         })
         })
       })
       })
 
 
+      it('returns the selected fields for both the base table and the included table as instance.selectedValues', function(done) {
+        var self = this
+        self.Mission = self.sequelize.define('Mission', {
+          title:  {type: Sequelize.STRING, defaultValue: 'another mission!!'},
+          foo:    {type: Sequelize.INTEGER, defaultValue: 4},
+        })
+
+        self.Mission.belongsTo(self.User)
+        self.User.hasMany(self.Mission)
+
+        self.Mission.sync({ force: true }).success(function() {
+          self.Mission.create().success(function(mission) {
+            self.User.create({username: 'Brain Picker'}).success(function(user) {
+              mission.setUser(user).success(function() {
+                self.User.find({
+                  where: { username: 'Brain Picker' },
+                  attributes: ['username'],
+                  include: [{model: self.Mission, as: self.Mission.tableName, attributes: ['title']}]
+                }).success(function(user) {
+                  expect(user.selectedValues).to.deep.equal({ username: 'Brain Picker' })
+                  expect(user.missions[0].selectedValues).to.deep.equal({ id: 1, title: 'another mission!!'})
+                  expect(user.missions[0].foo).not.to.exist
+                  done()
+                })
+              })
+            })
+          })
+        })
+      })
+
       it('always honors ZERO as primary key', function(_done) {
       it('always honors ZERO as primary key', function(_done) {
         var self = this
         var self = this
           , permutations = [
           , permutations = [