优化代码

internal/utils/time.go

@@ -23,6 +23,21 @@ func UnixTime() int64 {
 	return unixTime
 }
 
+// FloorUnixTime 取整
+func FloorUnixTime(seconds int) int64 {
+	return UnixTime() / int64(seconds) * int64(seconds)
+}
+
+// CeilUnixTime 取整并加1
+func CeilUnixTime(seconds int) int64 {
+	return UnixTime()/int64(seconds)*int64(seconds) + int64(seconds)
+}
+
+// NextMinuteUnixTime 获取下一分钟开始的时间戳
+func NextMinuteUnixTime() int64 {
+	return CeilUnixTime(60)
+}
+
 // UnixTimeMilli 获取时间戳，精确到毫秒
 func UnixTimeMilli() int64 {
 	return unixTimeMilli

internal/utils/time_test.go

@@ -1,6 +1,7 @@
 package utils
 
 import (
+	timeutil "github.com/iwind/TeaGo/utils/time"
 	"testing"
 	"time"
 )
@@ -19,3 +20,11 @@ func TestGMTUnixTime(t *testing.T) {
 func TestGMTTime(t *testing.T) {
 	t.Log(GMTTime(time.Now()))
 }
+
+func TestFloorUnixTime(t *testing.T) {
+	var timestamp = time.Now().Unix()
+	t.Log("floor 60:", timestamp, FloorUnixTime(60), timeutil.FormatTime("Y-m-d H:i:s", FloorUnixTime(60)))
+	t.Log("ceil 60:", timestamp, CeilUnixTime(60), timeutil.FormatTime("Y-m-d H:i:s", CeilUnixTime(60)))
+	t.Log("floor 300:", timestamp, FloorUnixTime(300), timeutil.FormatTime("Y-m-d H:i:s", FloorUnixTime(300)))
+	t.Log("next minute:", NextMinuteUnixTime())
+}