Go + MongoDB：多态查询

Question

我将 Go 和 MongoDB 与 mgo 驱动程序一起使用。我试图在同一个 MongoDB 集合中保留和检索不同的结构（实现通用接口）。我来自 Java 世界（使用 Spring 很容易做到这一点，几乎没有配置），但我很难用 Go 做类似的事情。这是我用于测试的简化设置。假设两个结构S1andS2实现了一个公共接口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")}S1现在我可以保存or的实例并S2轻松使用mgo.Collection.Insert()，但是为了正确填充一个值 usingmgo.Collection.Find().One()例如，我需要传递一个指向S1or的现有值的指针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.Typefunc 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    }}这在返回的对象类型正确时部分起作用，但我无法弄清楚的是如何使用pt从 MongoDB 检索的数据填充值，这些数据存储wrapper.Val为bson.M.所以基本上剩下的问题是：如何从bson.M值填充未知结构？我确定必须有一个简单的解决方案......在此先感谢您的帮助。这是包含所有代码的 Github 要点：https://gist.github.com/ogerardin/5aa272f69563475ba9d7b3194b12ae57

Answer 1

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

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

pt是类型reflect.Value（恰好是一个结构），而不是你应该传递给bson.Unmarshal(). 例如，您应该传递一个指向要解组的结构值的指针（它将被包装在一个interface{}值中）。所以调用Value.Interface()返回的值reflect.New()：

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)

}

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

另请注意，可以将映射直接转换为结构（直接意味着无需编组和解组）。您可以手动实施或使用现成的第 3 方库。

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

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

<type>-<id>

例如：

my.package.S1-123

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

Answer 2

这是一个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

}

从结构到结构的转换由https://github.com/mitchellh/mapstructurebson.M处理，而不是 marshalling-unmarshaling。