forked from Boris-code/feapder
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathexpirefilter.py
More file actions
70 lines (54 loc) · 1.89 KB
/
Copy pathexpirefilter.py
File metadata and controls
70 lines (54 loc) · 1.89 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
# -*- coding: utf-8 -*-
"""
Created on 2018/12/13 9:44 PM
---------
@summary: 带有有效期的去重集合
---------
@author: Boris
@email: [email protected]
"""
import time
from feapder.db.redisdb import RedisDB
class ExpireFilter:
redis_db = None
def __init__(
self, name: str, expire_time: int, expire_time_record_key=None, redis_url=None
):
if not name:
raise ValueError("name cant't be None")
if not expire_time:
raise ValueError("please set expire time, units is seconds")
if not self.__class__.redis_db:
self.__class__.redis_db = RedisDB(url=redis_url)
self.name = name
self.expire_time = expire_time
self.expire_time_record_key = expire_time_record_key
self.del_expire_key_time = None
self.record_expire_time()
self.del_expire_key()
def __repr__(self):
return "<ExpireSet: {}>".format(self.name)
@property
def current_timestamp(self):
return int(time.time())
def add(self, keys, *args, **kwargs):
"""
@param keys: 检查关键词在zset中是否存在，支持列表批量
@return: list / 单个值
"""
if self.current_timestamp - self.del_expire_key_time > self.expire_time:
self.del_expire_key()
is_added = self.redis_db.zadd(self.name, keys, self.current_timestamp)
return is_added
def get(self, keys):
return self.redis_db.zexists(self.name, keys)
def del_expire_key(self):
self.redis_db.zremrangebyscore(
self.name, "-inf", self.current_timestamp - self.expire_time
)
self.del_expire_key_time = self.current_timestamp
def record_expire_time(self):
if self.expire_time_record_key:
self.redis_db.hset(
self.expire_time_record_key, key=self.name, value=self.expire_time
)