Go实现低性能集合Set

最新推荐文章于 2024-03-28 08:22:57 发布

ZZZhonngger

最新推荐文章于 2024-03-28 08:22:57 发布

阅读量1.1k

点赞数

分类专栏： Go 文章标签： golang 后端

本文链接：https://blog.csdn.net/weixin_43395911/article/details/122751389

版权

本文介绍了在Go中如何实现线程不安全和安全的集合Set。线程不安全Set通过Map的Key唯一性实现，而线程安全Set则利用sync.RWMutex读写锁解决并发问题，确保多线程环境下的正确性。

摘要由CSDN通过智能技术生成

前言

从去年校招入职至今，Java转Go已经有大半年了，在开发过程中发现Go还是有一些痛点的，它的基本数据结构主要是数组、切片、Map，像我们常用到的集合Set其实是没有的，需要自己实现一套。Set表示一个集合，Set里的元素不能重复，实现方式有：

作为一名Bytedancer，当然要享受造轮子的过程，那么现在我们来自己动手写实现一个Set

我目前使用的是Go 1.16版本，还不支持范型，由于开发中常用string这一基本类型，那么我们就以string为例造一个存string的集合～

package hashset

type StringSet map[string]struct{
   }

func NewStringSet() StringSet {
   
   return make(map[string]struct{
   })
}

func (set StringSet) Add(element string) string {
   
   set[element] = struct{
   }{
   }
   return element
}

把元素塞到Map里的Key中，Value是什么我们不关心

func (set StringSet) Remove(element string) string {
   
   delete(set, element)
   return element
}

调用Go语言提供的一个内置函数 delete()，用于删除Map内键值对

func (set StringSet) Contain(element string) bool {
   
   if _, ok := set[element]; ok {
   
      return true
   }
   return false
}

这个没啥好说的，利用Map的属性，查看Map中是否存在以element为Key的元素

func (set StringSet) Length() int {
   
   return len(set)
}

这个也没啥好说的，用内置函数len()就可以解决

package hashset

import (
   "fmt"
   "testing"
)

func TestAdd(t *testing.T) {
   
   set := NewStringSet()
   addElem

关注

专栏目录