首頁 > 後端開發 > Golang > 如何使用泛型將介面轉換為指定類型

如何使用泛型將介面轉換為指定類型

WBOY
發布: 2024-02-06 08:25:11
轉載
1037 人瀏覽過

如何使用泛型將介面轉換為指定類型

問題內容

有一個介面宣告和許多實作它的結構

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
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板