Go + MongoDB:多态查询

问题描述 投票:2回答:2

(对不起,这个问题的结果比我想象的要长......)

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

这是我用于测试的简化设置。假设两个结构S1S2,实现一个共同的接口IS2有一个类型为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轻松保存S2mgo.Collection.Insert()的实例,但是为了使用mgo.Collection.Find().One()正确填充值,我需要传递指向S1S2的现有值的指针,这意味着我已经知道了对象的类型我想看!!我希望能够从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

mongodb go polymorphism mgo
2个回答
1
投票

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

unmarshal无法处理struct值。使用指针

ptreflect.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)
}

(在您的真实代码中,除了恐慌之外,您还要做其他事情,这只是为了表明您应该始终检查错误!)

还要注意,可以直接将地图转换为结构(直接意味着没有编组和解编)。您可以手动实现它或使用准备好的第三方库。有关详细信息,请参阅Converting map to struct

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

例如,您可以使用以下ID结构:

<type>-<id>

例如:

my.package.S1-123

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


0
投票

根据@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处理而不是编组 - 解组。

© www.soinside.com 2019 - 2024. All rights reserved.