01 目標(biāo)

假設(shè)我們要實(shí)現(xiàn)一個(gè)blog系統(tǒng)，在該系統(tǒng)中有以下兩個(gè)結(jié)構(gòu)體：

type Category struct {
    ID int32
    Name string
    Slug string
}
type Post struct {
    ID int32
    Categories []Category
    Title string
    Text string
    Slug string
}

為了提高系統(tǒng)的性能，我們需要實(shí)現(xiàn)一個(gè)緩存系統(tǒng)，該緩存可以用于緩存各種類型，在該示例中我們限定為只能緩存Category和Post類型。

02 實(shí)現(xiàn)

根據(jù)Go泛型使用的三步曲提到的：類型參數(shù)化、定義類型約束、類型實(shí)例化我們一步步來定義我們的緩存結(jié)構(gòu)體。

第一步：定義類型約束

這里我們先定義類型約束。因?yàn)樵诜盒椭袑?duì)類型參數(shù)進(jìn)行約束是必要條件。所以要先定義類型約束。

因?yàn)橐獙?duì)分類Category類型和文章Post類型進(jìn)行緩存，所以我們這里的緩存類型約束限制在了這兩個(gè)類型上。約束接口定義如下：

type cacheable interface {
    Category | Post
}

第二步：對(duì)類型進(jìn)行參數(shù)化

現(xiàn)在我們創(chuàng)建一個(gè)名為cache的泛型結(jié)構(gòu)體，并使用cacheable對(duì)其進(jìn)行約束。

type cache[T cacheable] struct {
    data map[string]T
}

我們看到cache的底層實(shí)際上是用map來進(jìn)行存儲(chǔ)數(shù)據(jù)的，map的key是具體的類型字符串，而map的值是參數(shù)化的類型T，即要在具體使用時(shí)根據(jù)需要對(duì)該參數(shù)T進(jìn)行實(shí)例化。

為了能夠在cache結(jié)構(gòu)體中存儲(chǔ)和獲取數(shù)據(jù)，我們?cè)俣x兩個(gè)方法如下：

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
}

這里需要大家注意的是在泛型結(jié)構(gòu)體類型中，定義方法的時(shí)候，也需要將類型參數(shù)T帶上的。因?yàn)橹挥性谡{(diào)用時(shí)對(duì)類型參數(shù)實(shí)例化后結(jié)構(gòu)體中的類型才是明確的。

第三步：類型實(shí)例化

為了實(shí)例化cache結(jié)構(gòu)體，我們創(chuàng)建了一個(gè)New函數(shù)來專門構(gòu)造cache的實(shí)例。

func New[T cacheable]() *cache[T]{
	c := cache[T]{}
	c.data = make(map[string]T)

	return &c
}

這里大家需要注意的是因?yàn)槲覀兪褂昧朔盒徒Y(jié)構(gòu)體類型cache，所以函數(shù)New也必須是泛型函數(shù)，只有這樣才能將泛型類型T的具體值傳遞到泛型結(jié)構(gòu)體類型中。

當(dāng)然，這里還有另外一種實(shí)例化的cache的方法就是直接使用，這樣就不需要使用泛型函數(shù)New了。如下：

c := &cache[Category]{
    data: make(map[string]T)
}

好了，下面我們給出具體的main函數(shù)使用示例：

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))
}

好了，以上就是今天跟大家分享的內(nèi)容。

到此這篇關(guān)于Go泛型實(shí)戰(zhàn)教程之如何在結(jié)構(gòu)體中使用泛型的文章就介紹到這了,更多相關(guān)go結(jié)構(gòu)體泛型內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論