소스 검색

fix: 更新用户请求次数缓存失效时间为第二天零点,修复max_request配置达不到预期的问题 (#192)

Finly 2 년 전
부모
커밋
0adbece285
1개의 변경된 파일4개의 추가작업 그리고 1개의 파일을 삭제
  1. 4 1
      pkg/cache/user_requese.go

+ 4 - 1
pkg/cache/user_requese.go

@@ -6,7 +6,10 @@ import (
 
 // SetUseRequestCount 设置用户请求次数
 func (s *UserService) SetUseRequestCount(userId string, current int) {
-	s.cache.Set(userId+"_request", current, time.Hour*24)
+	expiration := time.Now().Add(time.Hour * 24).Truncate(time.Hour * 24)
+	duration := expiration.Sub(time.Now())
+	// 设置缓存失效时间为第二天零点
+	s.cache.Set(userId+"_request", current, duration)
 }
 
 // GetUseRequestCount 获取当前用户已请求次数