Go泛型實戰教程之如何在結構體中使用泛型
01 目標
假設我們要實現一個blog系統,在該系統中有以下兩個結構體:
type Category struct { ID int32 Name string Slug string } type Post struct { ID int32 Categories []Category Title string Text string Slug string }
為瞭提高系統的性能,我們需要實現一個緩存系統,該緩存可以用於緩存各種類型,在該示例中我們限定為隻能緩存Category和Post類型。
02 實現
根據Go泛型使用的三步曲提到的:類型參數化、定義類型約束、類型實例化我們一步步來定義我們的緩存結構體。
第一步:定義類型約束
這裡我們先定義類型約束。因為在泛型中對類型參數進行約束是必要條件。所以要先定義類型約束。
因為要對分類Category類型和文章Post類型進行緩存,所以我們這裡的緩存類型約束限制在瞭這兩個類型上。約束接口定義如下:
type cacheable interface { Category | Post }
第二步:對類型進行參數化
現在我們創建一個名為cache的泛型結構體,並使用cacheable對其進行約束。
type cache[T cacheable] struct { data map[string]T }
我們看到cache的底層實際上是用map來進行存儲數據的,map的key是具體的類型字符串,而map的值是參數化的類型T,即要在具體使用時根據需要對該參數T進行實例化。
為瞭能夠在cache結構體中存儲和獲取數據,我們再定義兩個方法如下:
func (c *cache[T]) Set(key string, value T) { c.data[key] = value } func (c *cache[T]) Get(key string) (v T) { if v, ok := c.data[key]; ok { return v } return }
這裡需要大傢註意的是在泛型結構體類型中,定義方法的時候,也需要將類型參數T帶上的。因為隻有在調用時對類型參數實例化後結構體中的類型才是明確的。
第三步:類型實例化
為瞭實例化cache結構體,我們創建瞭一個New函數來專門構造cache的實例。
func New[T cacheable]() *cache[T]{ c := cache[T]{} c.data = make(map[string]T) return &c }
這裡大傢需要註意的是因為我們使用瞭泛型結構體類型cache,所以函數New也必須是泛型函數,隻有這樣才能將泛型類型T的具體值傳遞到泛型結構體類型中。
當然,這裡還有另外一種實例化的cache的方法就是直接使用,這樣就不需要使用泛型函數New瞭。如下:
c := &cache[Category]{ data: make(map[string]T) }
好瞭,下面我們給出具體的main函數使用示例:
package main import ( "fmt" ) func main() { // create a new category category := Category{ ID: 1, Name: "Go Generics", Slug: "go-generics", } // create cache for blog.Category struct cc := New[Category]() // add category to cache cc.Set(category.Slug, category) fmt.Printf("cp get:%+v\n", cc.Get(category.Slug)) // create a new post post := Post{ ID: 1, Categories: []Category{ {ID: 1, Name: "Go Generics", Slug: "go-generics"}, }, Title: "Generics in Golang structs", Text: "Here go's the text", Slug: "generics-in-golang-structs", } // create cache for blog.Post struct cp := New[Post]() // add post to cache cp.Set(post.Slug, post) fmt.Printf("cp get:%+v\n", cp.Get(post.Slug)) }
好瞭,以上就是今天跟大傢分享的內容。
到此這篇關於Go泛型實戰教程之如何在結構體中使用泛型的文章就介紹到這瞭,更多相關go結構體泛型內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!
推薦閱讀:
- Go1.18新特性使用Generics泛型進行流式處理
- 淺談Go1.18中的泛型編程
- Go 修改map slice array元素值操作
- Swift泛型Generics淺析講解
- Go語言參數傳遞是傳值還是傳引用