Golang實(shí)現(xiàn)支持多種類(lèi)型的set

更新時(shí)間：2023年05月25日 08:50:52 作者：nil

在項(xiàng)目開(kāi)發(fā)中，常常會(huì)用到set去重，為什么不寫(xiě)一個(gè)set呢，而且go現(xiàn)在支持了泛型，所以本文就來(lái)用Golang實(shí)現(xiàn)一個(gè)支持多種類(lèi)型的set呢

寫(xiě)在前面

今天項(xiàng)目中需要用到string類(lèi)型的set，想起來(lái)項(xiàng)目中不少地方都需要，而且都是用map[string]bool實(shí)現(xiàn)的，既然這么多地方用到set去重，為什么不寫(xiě)一個(gè)set呢？而且go現(xiàn)在支持了泛型，為啥不寫(xiě)一個(gè)支持多種類(lèi)型的set呢？說(shuō)干就干

代碼

    package set
    var (
    	ElemValue = struct{}{}
    )
    type SetType interface {
    	int | int32 | int64 | string | float32 | float64
    }
    // Set[T SetType]. 集合，線程不安全
    type Set[T SetType] interface {
    	Add(T)
    	Remove(T)
    	Contains(T) bool
    	Empty() bool
    	Values() []T
    }
    type set[T SetType] struct {
    	m map[T]struct{}
    }
    func NewSet[T SetType](ss ...T) Set[T] {
    	newSet := set[T]{
    		m: make(map[T]struct{}, len(ss)),
    	}
    	for _, s := range ss {
    		newSet.Add(s)
    	}
    	return &newSet
    }
    func (s *set[T]) Add(elem T) {
    	s.m[elem] = ElemValue
    }
    func (s *set[T]) Remove(elem T) {
    	delete(s.m, elem)
    }
    func (s *set[T]) Contains(elem T) bool {
    	_, ok := s.m[elem]
    	return ok
    }
    func (s *set[T]) Empty() bool {
    	empty := true
    	for _, _ = range s.m {
    		empty = false
    		break
    	}
    	return empty
    }
    func (s *set[T]) Values() []T {
    	ss := make([]T, 0)
    	for k, _ := range s.m {
    		ss = append(ss, k)
    	}
    	return ss
    }

寫(xiě)完之后發(fā)現(xiàn)才60多行代碼，非常簡(jiǎn)單，而且value使用的是struct{}{}，不占用任何內(nèi)存

單測(cè)

    package set_test
    import (
    	"set"
    	"testing"
    	"github.com/stretchr/testify/assert"
    )
    func TestStringSet(t *testing.T) {
    	newSet := set.NewSet[string]("a", "b")
    	assert.False(t, newSet.Empty())
    	newSet.Add("c")
    	newSet.Remove("a")
    	assert.False(t, newSet.Contains("a"))
    	assert.True(t, newSet.Contains("c"))
    	newSet.Remove("c")
    	result := newSet.Values()
    	assert.Equal(t, "b", result[0])
    }
    func TestInt64Set(t *testing.T) {
    	newSet := set.NewSet[int64](1, 2)
    	assert.False(t, newSet.Empty())
    	newSet.Add(3)
    	newSet.Remove(1)
    	assert.False(t, newSet.Contains(1))
    	assert.True(t, newSet.Contains(3))
    	newSet.Remove(3)
    	result := newSet.Values()
    	assert.Equal(t, int64(2), result[0])
    }

造輪子感想

之前沒(méi)有用過(guò)go 泛型，今天寫(xiě)這個(gè)set體驗(yàn)了一下泛型，確實(shí)挺好用的。本人喜歡造輪子、支持造輪子。只有這樣才能提升自己，一起學(xué)習(xí)吧

到此這篇關(guān)于Golang實(shí)現(xiàn)支持多種類(lèi)型的set的文章就介紹到這了,更多相關(guān)Golang實(shí)現(xiàn)set內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: