用Python定义支持用户速率限制检查的数据结构
假设我们想要开发一个数据结构,可以在到期时间之内建立,支持一个函数,该函数采用用户ID和时间戳。这将检查特定用户ID在给定时间戳时请求是否失败。只有当用户的最后一次请求在给定的过期时间内时,请求才会失败。
因此,如果输入的参数是expire = 6,则构建一个对象obj,并调用函数obj.limit(0,10),obj.limit(0,16),obj.limit(0,17)和obj.limit(1,20),那么输出将分别为False,False,True和False,因为对于用户0,最初没有请求,所以是false,然后在时间16时,它不大于上次请求的过期时间6,但在17时为true,最后一次请求是用于用户1,因此最初的请求是false。
为了解决这个问题,我们将采取以下步骤:
定义构造函数。这将采用expire。
- lastCall :创建默认值为-1的字典
- 定义一个函数limit()。这将需要uid,timestamp
- last : lastCall[uid]
- 如果last与-1相同或(last + expire) <= timestamp,那么
- lastCall[uid] := timestamp
- return False
- return True
示例
让我们看下面的实现以便更好地理解。