fix

utils/basic/basic.go

@@ -52,10 +52,11 @@ var (
 	CodeS3PutSizeLimitErr     = &StatusResponse{5061, "s3 over limit size error"}     // s3 超过文件大小限制 错误
 	CodeS3CategoryErr         = &StatusResponse{5062, "s3 category not exists error"} // s3 类别不存在  错误
 
-	CodeEmailNotFoundErr  = &StatusResponse{5050, "email not found"}  // 未找到email
-	CodeUserIdNotFoundErr = &StatusResponse{5051, "user not found"}   // 未找到用户
-	CodePasswordErr       = &StatusResponse{5052, "invalid password"} // 无效密码
-	CodeEmailExistsErr    = &StatusResponse{5053, "email exists"}     // email存在
+	CodeEmailNotFoundErr  = &StatusResponse{5050, "email not found"}                            // 未找到email
+	CodeUserIdNotFoundErr = &StatusResponse{5051, "user not found"}                             // 未找到用户
+	CodePasswordErr       = &StatusResponse{5052, "invalid password"}                           // 无效密码
+	CodeEmailExistsErr    = &StatusResponse{5053, "email exists"}                               // email存在
+	CodeEmailTimeShortErr = &StatusResponse{5053, "email with the time of resend is too short"} // email存在
 
 	CodeSafeValueRangeErr = &StatusResponse{5040, "value not in range"}    // 值不在范围内
 	CodeTemplateErr       = &StatusResponse{5040, "template parsed error"} // 模板解析错误

utils/check/limit.go

@@ -1,7 +1,53 @@
+// Timeup是一个限制某个值的重复频率的工具
+// 通过内部计时器实现限制同一个值的访问频率不超过指定时间间隔
 package check
 
-import "sync"
+import (
+	"sync"
+	"time"
+)
 
-type LimitAfter struct {
-	sync.Map
+// TimeLimit结构体包含需要限频的映射表,以及时间间隔
+// 使用sync.Mutex保证对字典的线程安全访问
+type TimeLimit[T comparable] struct {
+	mu   sync.Mutex
+	dict map[T]struct{}
+	dur  time.Duration
+}
+
+// NewTimelimit构造函数,接收限频的时间间隔
+// 并初始化内部字典和间隔字段
+func NewTimelimit[T comparable](dur time.Duration) *TimeLimit[T] {
+	return &TimeLimit[T]{
+		dict: make(map[T]struct{}),
+		dur:  dur,
+	}
+}
+
+// WithTime方法用于更新限频的时间间隔
+func (tup *TimeLimit[T]) WithTime(dur time.Duration) *TimeLimit[T] {
+	tup.mu.Lock()
+	defer tup.mu.Unlock()
+	tup.dur = dur
+	return tup
+}
+
+// Is方法检查传入值是否是一个新的值
+// 首先会查询字典,如果存在则表示在间隔内已经访问过
+// 否则将其添加到字典,并启动一个定时器在间隔后删除
+// 返回true表示新值,false表示重复值
+func (tup *TimeLimit[T]) Is(v T) bool {
+	tup.mu.Lock()
+	defer tup.mu.Unlock()
+
+	if _, ok := tup.dict[v]; ok {
+		return false
+	}
+	tup.dict[v] = struct{}{}
+	time.AfterFunc(tup.dur, func() {
+		tup.mu.Lock()
+		defer tup.mu.Unlock()
+		delete(tup.dict, v)
+	})
+	return true
 }