Go + MongoDB：多态查询

Question

（对不起，这个问题的结果比我想象的要长......）

我正在使用Go和MongoDB和mgo驱动程序。我试图在同一个MongoDB集合中持久化并检索不同的结构（实现一个通用接口）。我来自Java世界（使用Spring非常容易完成，没有配置），而且我很难用Go做类似的事情。我已经阅读了我能找到的最后一篇相关文章或帖子或StackExchange问题，但我还没有找到完整的解决方案。这包括：

这是我用于测试的简化设置。假设两个结构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")
}

现在我可以使用S1轻松保存S2或mgo.Collection.Insert()的实例，但是为了使用mgo.Collection.Find().One()正确填充值，我需要传递指向S1或S2的现有值的指针，这意味着我已经知道了对象的类型我想看!!我希望能够从MongoDB集合中检索文档，而不知道它是S1，还是S2，或者实际上是任何实现I的对象。

这是我到目前为止的地方：我保存了一个包含MongoDB id，类型标识符和实际值的Wrapper结构，而不是直接保存我想要保留的对象。类型标识符是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。

我尝试了以下但它不起作用：

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

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

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

Answer 1

另一答案