首页 > 后端开发 > Golang > 如何使用泛型将接口转换为指定类型

如何使用泛型将接口转换为指定类型

WBOY
发布: 2024-02-06 08:25:11
转载
1043 人浏览过

如何使用泛型将接口转换为指定类型

问题内容

有一个接口声明和许多实现它的结构

type datainterface interface {
    get(string) string
}

type dataa struct {
    d map[string]string
}

func (d *dataa) get(key string) string {
    return d.d[key]
}

func (d *dataa) getid() string {
    return d.get("id")
}

type datab struct {
    d map[string]string
}

func (d *datab) get(key string) string {
    return d.d[key]
}

func (d *datab) getfile() string {
    return d.get("file")
}


type datac...
登录后复制

还包括 datac,d,e...

我会将这些 datax 结构实例存储到 type dataslice []datainterface

现在,如果我想获取 datax ,我可以这样做:

type dataslice []datainterface
func (d dataslice) geta() []*dataa {
    var ret []*dataa
    for _, di := range d {
        if v, ok := di.(*dataa); ok {
            ret = append(ret, v)
        }
    }
    return ret
}

func (d dataslice) getb() []*datab {
    var ret []*datab
    for _, di := range d {
        if v, ok := di.(*datab); ok {
            ret = append(ret, v)
        }
    }
    return ret
}

func (d dataslice) getc() .....
登录后复制

显然这里有很多重复的代码:

var ret []*datax
for _, di := range d {
    if v, ok := di.(*datax); ok {
        ret = append(ret, v)
    }
}
登录后复制

所以我想我可以使用泛型来解决这个问题,然后我定义了这个函数:

func GetDataX[T any] (d DataInterface) *T {
    return d.(*T)
}
登录后复制

但出现错误:impossible 类型断言:'*t' 未实现 'datainterface

所以我想知道这样真的不可能吗?或者可以通过其他方式完成?


正确答案


您应该能够使用以下代码来满足您的需求:

package main

import "fmt"

// interface
type DataInterface interface {
    Get(string) string
}

// struct implementing the interface
type DataA struct {
    d map[string]string
}

func (d DataA) Get(key string) string {
    return d.d[key]
}

type DataB struct {
    d map[string]string
}

func (d DataB) Get(key string) string {
    return d.d[key]
}

type DataSlice []DataInterface

func GetDataX[T any](d DataInterface) T {
    return d.(T)
}

func main() {
    a := DataA{map[string]string{"a": "1"}}
    b := DataB{map[string]string{"b": "2"}}

    ds := DataSlice{a, b}

    for _, v := range ds {
        if value, ok := v.(DataA); ok {
            fmt.Printf("A\t%q\n", GetDataX[DataA](value))
            continue
        }

        if value, ok := v.(DataB); ok {
            fmt.Printf("B\t%q\n", GetDataX[DataB](value))
            continue
        }

        // add unknown type handling logic here
    }
}
登录后复制

首先,我简化了代码,仅考虑 dataadatab 结构。然后,我将指针接收器更改为值接收器,因为您不会更改传递给方法的实际实例的状态。由于此更改,getdatax 可以成功工作,并且您可以获得所有类似结构的信息。

如果这能解决您的问题或者您需要其他东西,请告诉我,谢谢!

以上是如何使用泛型将接口转换为指定类型的详细内容。更多信息请关注PHP中文网其他相关文章!

来源:stackoverflow.com
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板