单个模式数组中的多个模式引用 - mongoose

时间:2016-02-17 22:15:39

标签: javascript mongodb mongoose mongoose-populate

您可以使用对几个不同架构选项的引用来填充mongoose架构中的数组吗?

为了澄清这个问题,请说我有以下架构:

var scenarioSchema = Schema({
  _id     : Number,
  name    : String,
  guns : []
});

var ak47 = Schema({
  _id     : Number
  //Bunch of AK specific parameters
});

var m16 = Schema({
  _id     : Number
  //Bunch of M16 specific parameters
});

我可以用一堆ak47 OR m16填充枪阵吗?我可以将 BOTH 放在同一个枪阵列中吗?或者它是否需要在资产数组中使用填充引用,这样会将其限制为单个特定类型?

guns: [{ type: Schema.Types.ObjectId, ref: 'm16' }]

我知道我可以为不同的枪类型提供单独的数组,但是随着项目的扩展,这将在模式中创建大量的额外字段,其中大部分将根据加载的场景保留为空。

var scenarioSchema = Schema({
  _id     : Number,
  name    : String,
  ak47s : [{ type: Schema.Types.ObjectId, ref: 'ak47' }],
  m16s: [{ type: Schema.Types.ObjectId, ref: 'm16' }]
});

回到这个问题,我可以在单个数组中粘贴多个模式引用吗?

1 个答案:

答案 0 :(得分:19)

您在这里寻找的是猫鼬.discriminator()方法。这基本上允许您将不同类型的对象存储在同一个集合中,但将它们作为可区分的第一类对象。

请注意"相同的集合"这里的原则对于.populate()如何工作以及包含模型中引用的定义很重要。因为你真的只能指向"一个"无论如何,参考模型,但还有一些其他魔法可以使一个模型出现多少。

示例清单:

var util = require('util'),
    async = require('async'),
    mongoose = require('mongoose'),
    Schema = mongoose.Schema;

mongoose.connect('mongodb://localhost/gunshow');

//mongoose.set("debug",true);

var scenarioSchema = new Schema({
  "name": String,
  "guns": [{ "type": Schema.Types.ObjectId, "ref": "Gun" }]
});

function BaseSchema() {
  Schema.apply(this, arguments);

  // Common Gun stuff
  this.add({
    "createdAt": { "type": Date, "default": Date.now }
  });
}

util.inherits(BaseSchema, Schema);

var gunSchema = new BaseSchema();

var ak47Schema = new BaseSchema({
  // Ak74 stuff
});

ak47Schema.methods.shoot = function() {
  return "Crack!Crack";
};

var m16Schema = new BaseSchema({
  // M16 Stuff
});

m16Schema.methods.shoot = function() {
  return "Blam!!"
};


var Scenario = mongoose.model("Scenario", scenarioSchema);

var Gun = mongoose.model("Gun", gunSchema );
var Ak47 = Gun.discriminator("Ak47", ak47Schema );
var M16 = Gun.discriminator("M16", m16Schema );


async.series(
  [
    // Cleanup
    function(callback) {
      async.each([Scenario,Gun],function(model,callback) {
        model.remove({},callback);
      },callback);
    },

    // Add some guns and add to scenario
    function(callback) {
      async.waterfall(
        [
          function(callback) {
            async.map([Ak47,M16],function(gun,callback) {
              gun.create({},callback);
            },callback);
          },
          function(guns,callback) {
            Scenario.create({
              "name": "Test",
              "guns": guns
            },callback);
          }
        ],
        callback
      );
    },

    // Get populated scenario
    function(callback) {
      Scenario.findOne().populate("guns").exec(function(err,data) {

        console.log("Populated:\n%s",JSON.stringify(data,undefined,2));

        // Shoot each gun for fun!
        data.guns.forEach(function(gun) {
          console.log("%s says %s",gun.__t,gun.shoot());
        });

        callback(err);
      });
    },

    // Show the Guns collection
    function(callback) {
      Gun.find().exec(function(err,guns) {
        console.log("Guns:\n%s", JSON.stringify(guns,undefined,2));
        callback(err);
      });
    },

    // Show magic filtering
    function(callback) {
      Ak47.find().exec(function(err,ak47) {
        console.log("Magic!:\n%s", JSON.stringify(ak47,undefined,2));
        callback(err);
      });
    }
  ],
  function(err) {
    if (err) throw err;
    mongoose.disconnect();
  }
);

输出

Populated:
{
  "_id": "56c508069d16fab84ead921d",
  "name": "Test",
  "__v": 0,
  "guns": [
    {
      "_id": "56c508069d16fab84ead921b",
      "__v": 0,
      "__t": "Ak47",
      "createdAt": "2016-02-17T23:53:42.853Z"
    },
    {
      "_id": "56c508069d16fab84ead921c",
      "__v": 0,
      "__t": "M16",
      "createdAt": "2016-02-17T23:53:42.862Z"
    }
  ]
}
Ak47 says Crack!Crack
M16 says Blam!!
Guns:
[
  {
    "_id": "56c508069d16fab84ead921b",
    "__v": 0,
    "__t": "Ak47",
    "createdAt": "2016-02-17T23:53:42.853Z"
  },
  {
    "_id": "56c508069d16fab84ead921c",
    "__v": 0,
    "__t": "M16",
    "createdAt": "2016-02-17T23:53:42.862Z"
  }
]
Magic!:
[
  {
    "_id": "56c508069d16fab84ead921b",
    "__v": 0,
    "__t": "Ak47",
    "createdAt": "2016-02-17T23:53:42.853Z"
  }
]

您还可以取消注释列表中的mongoose.set("debug",true)行,以了解mongoose实际构建调用的方式。

所以这证明了你可以将不同的模式应用于不同的第一类对象,甚至使用不同的方法,就像真实对象一样。猫鼬将这些全部存放在枪支中。使用附加模型的集合,它将包含所有"类型"由鉴别者引用:

var Gun = mongoose.model("Gun", gunSchema );
var Ak47 = Gun.discriminator("Ak47", ak47Schema );
var M16 = Gun.discriminator("M16", m16Schema );

但是每种不同的"类型"以特殊的方式引用它自己的模型。所以你看到当mongoose存储并读取对象时,有一个特殊的__t字段告诉它哪个"模型"申请,因此附加架构。

作为一个示例,我们调用.shoot()方法,该方法针对每个模型/模式进行了不同的定义。此外,您还可以将每个模型单独用作查询或其他操作的模型,因为Ak47会自动在所有查询/更新中应用__t值。

因此,尽管存储在一个集合中,但它看起来可能是许多集合,但也有将它们保存在一起以进行其他有用操作的好处。这就是你如何应用"多态性"你在寻找。