From 2a0c5b84a83e8bf6db6263dcd329cf225ed8d79f Mon Sep 17 00:00:00 2001
From: kercylan98 <kercylan@gmail.com>
Date: Fri, 28 Jul 2023 18:11:04 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20times=20=E5=8C=85=E6=96=B0=E5=A2=9E=20S?=
 =?UTF-8?q?ystemNewDay=20=E5=92=8C=20OffsetTimeNewDay=20=E4=BA=8B=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 utils/times/newday.go | 53 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 53 insertions(+)
 create mode 100644 utils/times/newday.go

diff --git a/utils/times/newday.go b/utils/times/newday.go
new file mode 100644
index 0000000..9be9061
--- /dev/null
+++ b/utils/times/newday.go
@@ -0,0 +1,53 @@
+package times
+
+import (
+	"github.com/kercylan98/minotaur/utils/offset"
+	"github.com/kercylan98/minotaur/utils/timer"
+	"time"
+)
+
+type (
+	SystemNewDayEventHandle     func()
+	OffsetTimeNewDayEventHandle func()
+)
+
+var (
+	systemNewDayEventHandles     = make(map[string][]SystemNewDayEventHandle)
+	offsetTimeNewDayEventHandles = make(map[string][]OffsetTimeNewDayEventHandle)
+)
+
+// RegSystemNewDayEvent 注册系统新的一天事件
+//   - 建议全局注册一个事件后再另行拓展
+//   - 将特定 name 的定时任务注册到 ticker 中，在系统时间到达每天的 00:00:00 时触发，如果 trigger 为 true，则立即触发一次
+func RegSystemNewDayEvent(ticker *timer.Ticker, name string, trigger bool, handle SystemNewDayEventHandle) {
+	ticker.Loop(name, GetNextDayInterval(time.Now()), Day, timer.Forever, OnSystemNewDayEvent, name)
+	systemNewDayEventHandles[name] = append(systemNewDayEventHandles[name], handle)
+	if trigger {
+		OnSystemNewDayEvent(name)
+	}
+}
+
+// OnSystemNewDayEvent 系统新的一天事件
+func OnSystemNewDayEvent(name string) {
+	for _, handle := range systemNewDayEventHandles[name] {
+		handle()
+	}
+}
+
+// RegOffsetTimeNewDayEvent 注册偏移时间新的一天事件
+//   - 建议全局注册一个事件后再另行拓展
+//   - 与 RegSystemNewDayEvent 类似，但是触发时间为 offset 时间到达每天的 00:00:00
+func RegOffsetTimeNewDayEvent(ticker *timer.Ticker, name string, offset *offset.Time, trigger bool, handle OffsetTimeNewDayEventHandle) {
+	ticker.Loop(name, GetNextDayInterval(offset.Now()), Day, timer.Forever, OnOffsetTimeNewDayEvent, name)
+	offsetTimeNewDayEventHandles[name] = append(offsetTimeNewDayEventHandles[name], handle)
+	if trigger {
+		OnOffsetTimeNewDayEvent(name)
+	}
+}
+
+// OnOffsetTimeNewDayEvent 偏移时间新的一天事件
+func OnOffsetTimeNewDayEvent(name string) {
+	for _, handle := range offsetTimeNewDayEventHandles[name] {
+		handle()
+	}
+}