Go + MongoDB：多态查询

学习Golang要努力，但是不要急！今天的这篇文章《Go + MongoDB：多态查询》将会介绍到等等知识点，如果你想深入学习Golang，可以关注我！我会持续更新相关文章的，希望对大家都能有所帮助！

（抱歉这个问题比我想象的要长......）

我正在使用 go 和 mongodb 以及 mgo 驱动程序。我试图在同一个 mongodb 集合中保留并检索不同的结构（实现通用接口）。我来自 java 世界（用 spring 很容易完成，几乎不需要任何配置），但我很难用 go 做类似的事情。 我已经阅读了我能找到的所有相关文章或帖子或 stackexchange 问题，但仍然没有找到完整的解决方案。这包括：

• 使用 mgo 的非结构化 mongodb 集合
• 如何在 go 中运行时根据结构体的类型创建结构体的新实例？
• golang 反映：从名称获取类型表示？

这是我用于测试的简化设置。假设有两个结构体 s1 和 s2，实现一个公共接口 i。 s2 具有 s1 类型的隐式字段，这意味着结构方面的 s2 嵌入了 s1 值，并且类型方面的 s2 实现了 i。

type i interface {
    f1()
}

type s1 struct {
    x int
}

type s2 struct {
    s1
    y int
}

func (*s1) f1() {
    fmt.println("f1")
}

现在我可以使用 mgo.collection.insert() 轻松保存 s1 或 s2 的实例，但要使用 mgo.collection.find().one() 正确填充值，例如，我需要传递指向 s1 或 s2 现有值的指针，这意味着我已经知道我要读取的对象的类型！ 我希望能够从 mongodb 集合中检索文档，而不知道它是 s1 还是 s2，或者实际上是任何实现 i 的对象。

到目前为止，我的情况是：我没有直接保存想要持久化的对象，而是保存了一个 wrapper 结构，该结构保存 mongodb id、类型标识符和实际值。类型标识符是 packagename + “.” 的串联。 + typename，它用于在类型注册表中查找类型，因为 go 中没有从类型名称映射到 type 对象的本机机制。这意味着我需要注册我希望能够保留和检索的类型，但我可以接受这一点。事情是这样的：

typeregistry.register(reflect.typeof((*s1)(nil)).elem())
typeregistry.register(reflect.typeof((*s2)(nil)).elem())

这是类型注册表的代码：

var types map[string]reflect.type

func init() {
    types = make(map[string]reflect.type)
}

func register(t reflect.type) {
    key := getkey(t)
    types[key] = t
}

func getkey(t reflect.type) string {
    key := t.pkgpath() + "." + t.name()
    return key
}

func gettype(key string) reflect.type {
    t := types[key]
    return t
}

保存对象的代码非常简单：

func save(coll *mgo.collection, s i) (bson.objectid, error) {
    t := reflect.typeof(s)
    wrapper := wrapper{
        id:      bson.newobjectid(),
        typekey: typeregistry.getkey(t),
        val:     s,
    }
    return wrapper.id, coll.insert(wrapper)
}

检索对象的代码有点棘手：

func getbyid(coll *mgo.collection, id interface{}) (*i, error) {
    // read wrapper
    wrapper := wrapper{}
    err := coll.find(bson.m{"_id": id}).one(&wrapper)
    if err != nil {
        return nil, err
    }

    // obtain type from registry
    t := typeregistry.gettype(wrapper.typekey)

    // get a pointer to a new value of this type
    pt := reflect.new(t)

    // fixme populate value using wrapper.val (type bson.m)
    // how ???

    // return the value as *i
    i := pt.elem().interface().(i)
    return &i, nil
}

这部分有效，因为返回的对象输入正确，但我不知道如何使用从 mongodb 检索到的数据填充值 pt，该数据存储在 wrapper.val 中作为 bson.m。 p>

我已尝试以下方法，但不起作用：

m := wrapper.Val.(bson.M)
bsonBytes, _ := bson.Marshal(m)
bson.Unmarshal(bsonBytes, pt)

所以基本上剩下的问题是：如何从 bson.m 值填充未知结构？我确信必须有一个简单的解决方案...... 预先感谢您的任何帮助。

这是包含所有代码的 github 要点：https://gist.github.com/ogerardin/5aa272f69563475ba9d7b3194b12ae57

解决方案

首先，您应该始终检查返回的错误。 bson.Marshal() 和 bson.Unmarshal() 返回您未检查的错误。这样做揭示了为什么它不起作用：

unmarshal 无法处理结构值。使用指针

pt 的类型为 reflect.Value（恰好是一个结构体），而不是您应该传递给 bson.unmarshal() 的内容。你应该通过例如指向要解组到的结构体值的指针（将包装在 interface{} 值中）。因此，对 reflect.New() 返回的值调用 Value.Interface()：

pt := reflect.new(t).interface()

您可以将其传递给 bson.unmarshal()：

bsonbytes, err := bson.marshal(m)
if err != nil {
    panic(err)
}
if err = bson.unmarshal(bsonbytes, pt); err != nil {
    panic(err)
}

（在你的真实代码中，你想做一些除了恐慌之外的事情，这只是为了表明你应该始终检查错误！）

另请注意，可以直接将映射转换为结构（直接意味着无需编组和解组）。您可以手动实现它或使用现成的第三方库。详情见Converting map to struct

另请注意，还有更聪明的方法来解决您想做的事情。您可以将类型存储在 id 本身中，因此如果您有 id，则可以构造该类型的值以将其解组到查询结果中，这样您就可以跳过整个过程。这会更简单、更高效。

例如，您可以使用以下 id 结构：

<type>-<id>

例如：

my.package.s1-123

获取/加载此文档时，您可以使用反射创建 my.package.s1 的值，并直接解组到该值（将其传递给 query.one()）。

as根据 @icza 的评论，这是实际有效的 getbyid() 的修改版本：

func getById(coll *mgo.Collection, id interface{}) (*I, error) {
    // read wrapper
    wrapper := Wrapper{}
    err := coll.Find(bson.M{"_id": id}).One(&wrapper)
    if err != nil {
        return nil, err
    }

    // obtain Type from registry
    t := typeregistry.GetType(wrapper.TypeKey)

    // get a pointer to a new value of this type
    pt := reflect.New(t)

    // populate value using wrapper.Val
    err = mapstructure.Decode(wrapper.V, pt.Interface())
    if err != nil {
        return nil, err
    }

    // return the value as *I
    i := pt.Elem().Interface().(I)
    return &i, nil
}

从 bson.m 到结构的转换由 https://github.com/mitchellh/mapstructure 处理，而不是编组-解编组。

好了，本文到此结束，带大家了解了《Go + MongoDB：多态查询》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！