From 9fcc75e0d7545fcbdb65f87ec9e1a12b03b7bce0 Mon Sep 17 00:00:00 2001
From: kercylan98 <kercylan@gmail.com>
Date: Wed, 26 Jul 2023 17:28:48 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20hash=20=E5=8C=85=E6=96=B0=E5=A2=9E=20Se?=
 =?UTF-8?q?t=20=E6=95=B0=E6=8D=AE=E7=BB=93=E6=9E=84?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 utils/hash/set.go | 90 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 90 insertions(+)
 create mode 100644 utils/hash/set.go

diff --git a/utils/hash/set.go b/utils/hash/set.go
new file mode 100644
index 0000000..76477cf
--- /dev/null
+++ b/utils/hash/set.go
@@ -0,0 +1,90 @@
+package hash
+
+// NewSet 创建一个 Set 集合
+func NewSet[K comparable](ks ...K) Set[K] {
+	s := make(Set[K])
+	s.AddAll(ks...)
+	return s
+}
+
+// Set 集合
+type Set[K comparable] map[K]struct{}
+
+// Exist 检查特定 key 是否存在
+func (s Set[K]) Exist(key K) bool {
+	_, exist := s[key]
+	return exist
+}
+
+// AllExist 检查多个 key 是否存在
+func (s Set[K]) AllExist(keys ...K) bool {
+	for _, key := range keys {
+		if _, exist := s[key]; !exist {
+			return false
+		}
+	}
+	return true
+}
+
+// Add 添加元素
+func (s Set[K]) Add(key K) {
+	s[key] = struct{}{}
+}
+
+// AddAll 添加多个元素
+func (s Set[K]) AddAll(keys ...K) {
+	for _, key := range keys {
+		s[key] = struct{}{}
+	}
+}
+
+// Remove 移除元素
+func (s Set[K]) Remove(key K) {
+	delete(s, key)
+}
+
+// RemoveAll 移除多个元素
+func (s Set[K]) RemoveAll(keys ...K) {
+	for _, key := range keys {
+		delete(s, key)
+	}
+}
+
+// Clear 清空集合
+func (s Set[K]) Clear() {
+	for key := range s {
+		delete(s, key)
+	}
+}
+
+// Size 集合长度
+func (s Set[K]) Size() int {
+	return len(s)
+}
+
+// ToSlice 转换为切片
+func (s Set[K]) ToSlice() []K {
+	keys := make([]K, 0, len(s))
+	for key := range s {
+		keys = append(keys, key)
+	}
+	return keys
+}
+
+// ToMap 转换为 map
+func (s Set[K]) ToMap() map[K]struct{} {
+	return s
+}
+
+// ToJson 转换为 json 字符串
+func (s Set[K]) ToJson() string {
+	return ToJson(s)
+}
+
+// RandomGet 随机获取一个元素
+func (s Set[K]) RandomGet() (k K) {
+	for k = range s {
+		return k
+	}
+	return
+}