登录
首页 >  Golang >  Go问答

使用动态反射生成的结构体接口,如何输入断言?

来源:stackoverflow

时间:2024-03-22 08:30:50 390浏览 收藏

在 Go 中使用动态生成的结构体作为 JSON 解组的目标时,由于 Go 的接口使用空接口,导致无法正确解组数据。本文介绍了一种通过修改 `makeinstance` 函数,获取元素地址并添加目标结构来解决该问题的解决方案。这种方法通过公开目标结构的字段,允许 JSON 解组器正确地将数据解组到相应的字段中。

问题内容

我是 go 新手,所以如果这是一个小问题,请耐心等待。我正在使用自制的“类型注册表”将类型名称映射到其类型,以便根据指向各种类型名称的用例动态生成它们(我基本上是在尝试为多态聚合 json 响应寻找一个简单的解决方案elasticsearch 中的结构,但当然这也适用于许多其他动态/多态情况)。 我在这个问题中使用 dolmen 提供的解决方案:有没有办法从字符串创建结构体的实例? :

var typeregistry = make(map[string]reflect.type)

func registertype(typednil interface{}) {
    t := reflect.typeof(typednil).elem()
    typeregistry[t.name()] = t
}

func init() {
    registertype((*playlistidaggregation)(nil))
    registertype((*srcidaggregation)(nil))
    registertype((*assetidaggregation)(nil))
}

func makeinstance(name string) interface{} {
    return reflect.new(typeregistry[name]).elem().interface()
}

然后,我想使用动态生成的结构作为 es 响应中聚合节点的 json 解组的目标:

playlistidagg := makeinstance("playlistidaggregation")
err = json.unmarshal(esresponse.aggregations, &playlistidagg)

这并不像我想要的那样工作,因为 unmarshal 正在尝试将其解组为空接口而不是底层结构类型。它将数据放在 playlistidagg 变量中的“数据”节点下,这些数据字段当然是 map[string]interface{}。我只是缺少输入断言 playlistidagg 接口的方法还是有更好的方法来执行此操作?

编辑--- 评论中的问题让我意识到早就应该对这个问题进行编辑了。 在我的特定情况下,我定义的用于绑定到 elasticsearch 返回的 bucket 聚合的结构具有类似的结构,仅因根 json 标签而有所不同,es 使用根 json 标签来命名聚合并对其进行强类型化。例如

type Aggregation struct {
    Agg BucketAggregationWithCamIDCardinality `json:""`
}

因此,我的特定问题可以通过根据特定用例在结构上动态设置 json 标记来解决,而不是类型注册表。

此外,一个更重但更强大的选择是利用 oliver eilhard 的 elasticsearch go 客户端库(称为 elastic),它内置了对所有 es 聚合响应结构的支持: https://github.com/olivere/elastic/


解决方案


我通过获取元素地址并添加带有公开字段的目标结构来更改 makeinstance 函数。

func makeinstance(name string) interface{} {
    return reflect.new(typeregistry[name]).elem().addr().interface()
}

这是工作代码

package main

import (
    "encoding/json"
    "fmt"
    "reflect"
)

type playlistIDAggregation struct {
    PlaylistID string
}

type srcIDAggregation struct {
    SrcID string
}

type assetIDAggregation struct {
    AssetID string
}

var typeRegistry = make(map[string]reflect.Type)

func registerType(typedNil interface{}) {
    t := reflect.TypeOf(typedNil).Elem()
    typeRegistry[t.Name()] = t
}

func init() {
    registerType((*playlistIDAggregation)(nil))
    registerType((*srcIDAggregation)(nil))
    registerType((*assetIDAggregation)(nil))
}

func makeInstance(name string) interface{} {
    return reflect.New(typeRegistry[name]).Elem().Addr().Interface()
}

func main() {
    playlistIDAgg := makeInstance("playlistIDAggregation")
    fmt.Printf("Type = %[1]T => %#[1]v\n", playlistIDAgg)
    err := json.Unmarshal([]byte(`{"PlayListID": "dummy-id"}`), &playlistIDAgg)
    if err != nil {
        panic(err)
    }
    fmt.Printf("Type = %[1]T => %#[1]v\n", playlistIDAgg)
}

https://play.golang.org/p/dn19_iG5Xjz

到这里,我们也就讲完了《使用动态反射生成的结构体接口,如何输入断言?》的内容了。个人认为,基础知识的学习和巩固,是为了更好的将其运用到项目中,欢迎关注golang学习网公众号,带你了解更多关于的知识点!

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>