import redis
import time
def rate_limit_check(r, key, limits):
period_lengths = [_[0] for _ in sorted(limits.items())]
period_limits = [_[1] for _ in sorted(limits.items())]
pipe = r.pipeline()
for period_length in period_lengths:
current_period = int(time.time() / period_length)
redis_key = 'rate_limit:{key}:{period_length}:{current_period}'.format(key=key, period_length=period_length, current_period=current_period)
pipe.incr(redis_key).expire(redis_key, period_length*3)
return not any(hits > period_limit for period_limit, hits in zip(period_limits, pipe.execute()[::2]))
if __name__ == '__main__':
r = redis.Redis()
print rate_limit_check(r, '127.0.0.1', {1: 3, 60: 20})
return rate_limit_check(r, '127.0.0.1', {1: 3, 60: 20}) and rate_limit_check(r, '127.0.0.1+/login/', {1: 2, 60: 5})
机械节能产品生产企业官网模板...
大气智能家居家具装修装饰类企业通用网站模板...
礼品公司网站模板
宽屏简约大气婚纱摄影影楼模板...
蓝白WAP手机综合医院类整站源码(独立后台)...苏ICP备2024110244号-2 苏公网安备32050702011978号 增值电信业务经营许可证编号:苏B2-20251499 | Copyright 2018 - 2025 源码网商城 (www.ymwmall.com) 版权所有