From 0f31173291efafd4cb5594f56d54fb68903179a6 Mon Sep 17 00:00:00 2001
From: kercylan98 <kercylan@gmail.com>
Date: Mon, 14 Aug 2023 12:24:20 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20sole=20=E5=8C=85=E6=96=B0=E5=A2=9E=20On?=
 =?UTF-8?q?ce=20=E7=BB=93=E6=9E=84=E4=BD=93=EF=BC=8C=E7=94=A8=E4=BA=8E?=
 =?UTF-8?q?=E6=95=B0=E6=8D=AE=E5=8F=96=E5=80=BC=E5=8E=BB=E9=87=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 utils/sole/once.go | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)
 create mode 100644 utils/sole/once.go

diff --git a/utils/sole/once.go b/utils/sole/once.go
new file mode 100644
index 0000000..9452d05
--- /dev/null
+++ b/utils/sole/once.go
@@ -0,0 +1,35 @@
+package sole
+
+// NewOnce 创建一个用于数据取值去重的结构实例
+func NewOnce[V any]() *Once[V] {
+	once := &Once[V]{
+		r: map[any]struct{}{},
+	}
+	return once
+}
+
+// Once 用于数据取值去重的结构体
+type Once[V any] struct {
+	r map[any]struct{}
+}
+
+// Get 获取一个值，当该值已经被获取过的时候，返回 defaultValue，否则返回 value
+func (slf *Once[V]) Get(key any, value V, defaultValue V) V {
+	_, exist := slf.r[key]
+	if exist {
+		return defaultValue
+	}
+	slf.r[key] = struct{}{}
+	return value
+}
+
+// Reset 当 key 数量大于 0 时，将会重置对应 key 的记录，否则重置所有记录
+func (slf *Once[V]) Reset(key ...any) {
+	if len(key) > 0 {
+		for _, k := range key {
+			delete(slf.r, k)
+		}
+		return
+	}
+	clear(slf.r)
+}