(对不起,这个问题的结果比我想象的要长......)
我正在使用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
首先,您应该始终检查返回的错误。 bson.Marshal()
和bson.Unmarshal()
返回您不检查的错误。这样做揭示了它为什么不起作用:
unmarshal无法处理struct值。使用指针
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)
}
(在您的真实代码中,除了恐慌之外,您还要做其他事情,这只是为了表明您应该始终检查错误!)
还要注意,可以直接将地图转换为结构(直接意味着没有编组和解编)。您可以手动实现它或使用准备好的第三方库。有关详细信息,请参阅Converting map to struct
另请注意,有更聪明的方法可以解决您想要做的事情。您可以将类型存储在ID本身中,因此如果您有ID,则可以构造要在查询结果中解组的类型值,这样您就可以跳过整个过程。它会更简单,效率更高。
例如,您可以使用以下ID结构:
<type>-<id>
例如:
my.package.S1-123
在获取/加载此文档时,您可以使用反射创建my.package.S1
的值,并直接解组(将其传递给Query.One()
)。
根据@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处理而不是编组 - 解组。