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.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

    }

}

这在返回的对象类型正确时部分起作用，但我无法弄清楚的是如何使用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。