什么是更好的 mongodb 文档结构？

Question

我正在尝试创建一个良好的文档结构，该结构具有良好的阅读性能，而不是那么慢的写入性能。 我需要存储有关 UserConnection 文档的信息，该文档表示我的数据库中两个用户之间的链接。每个链接都有一个取决于参数列表的权重。

以下是表示数据、链接和权重组件的架构：

UserConnection      Services          Components
  UserA             Name              Name
  UserB             Weight            Weight
  Weight            Components
  Services

UserConnection.Weight = sum(UserConnection.Services.Weight)
UserConnection.Services.Weight = sum(UserConnection.Services.Weight.Components)

我目前在 Django 中使用 Mongoengine。我尝试使用 EmbeddedDocument 和 EmbeddedDocumentListField 定义一些结构，但我对此并不满意。

为了简洁起见，我没有添加测试。

谁能帮我找到一个适合 MongDB 的结构？

谢谢。

Answer 1

由于 mongoengine 在 python 类中定义了文档结构，您可以使用普通的 python 类@property 装饰方法返回计算值。

考虑以下示例：

import mongoengine as mdb

mdb.connect("so-37396173")


class User(mdb.Document):
    name = mdb.StringField()


class UserConnection(mdb.Document):
    user_a = mdb.ReferenceField('User')
    user_b = mdb.ReferenceField('User')
    services = mdb.ListField(mdb.ReferenceField('Service'))

    @property
    def weight(self):
        return sum([s.weight for s in self.services])


class Component(mdb.EmbeddedDocument):
    name = mdb.StringField()
    weight = mdb.FloatField()


class Service(mdb.Document):
    name = mdb.StringField()
    components = mdb.EmbeddedDocumentListField('Component')

    @property
    def weight(self):
        return sum([c.weight for c in self.components])

当您拥有UserConnection 对象时，您可以访问weight 属性：

>>> uc = UserConnection.objects.first()
>>> uc.weight
0.8544546532

这样做的缺点是weight 永远不会在UserConnection 的上下文中存储在数据库中，因此您无法在该级别对其进行聚合或排序，但aggregation framework 可能会提供一些不错的选择.如果您需要保存权重，那么您可以定义一些 signals 以在保存文档之前包含它：

import mongoengine as mdb

mdb.connect("so-37396173")

class User(mdb.Document):
    name = mdb.StringField()


class UserConnection(mdb.Document):
    user_a = mdb.ReferenceField('User')
    user_b = mdb.ReferenceField('User')
    services = mdb.ListField(mdb.ReferenceField('Service'))
    weight = mdb.FloatField()


    @classmethod
    def calc_weight(cls, sender, document, **kwargs):
        document.weight =  sum([s.weight for s in document.services])

mdb.signals.pre_save.connect(UserConnection.calc_weight, sender=UserConnection)

class Component(mdb.EmbeddedDocument):
    name = mdb.StringField()
    weight = mdb.FloatField()


class Service(mdb.Document):
    name = mdb.StringField()
    components = mdb.EmbeddedDocumentListField('Component')
    weight = mdb.FloatField()

    @classmethod
    def calc_weight(cls, sender, document, **kwargs):
        document.weight =  sum([s.weight for s in document.components])

mdb.signals.pre_save.connect(Service.calc_weight, sender=Service)

这样做的一个缺点是您必须调用 save 方法，这意味着使用 upsert=True 执行 update 不会创建权重。

您是使用嵌入式还是参考取决于what else you want to do with the data。