利用中间件判断连接在一段时间只能连接3次，在这段时间内超过3次连接则直接返回...-优快云博客

本文介绍了一种基于时间窗口的限流策略实现，利用Django中间件限制IP在60秒内的请求次数不超过三次，详细解释了算法流程及其实现细节。

利用中间件判断连接在一段时间只能连接3次，在这段时间内超过3次连接则直接返回

from django.utils.deprecation import MiddlewareMixin
from django.shortcuts import HttpResponse,render,redirect



limit_dic = {
    "127.0.0.1": [1542252426.783346, 1542252426.23423]
}  #存放ip地址与最新连接时间的对应关系
class Limit(MiddlewareMixin):
    def process_request(self, request):
        ip = request.META["REMOTE_ADDR"]  #获取连接的ip
        if not ip in limit_dic:
            limit_dic[ip] = []  #新连接过来时，给他重新建立一个对应的列表
        history = limit_dic[ip] #得到时间列表
        now = time.time()
        while history and now - history[-1] > 60:
          # 首先判断时间列表里面是否存在数据，如果不存在数据，说明连接已经存在很久，数据被删除了，或者是一个新来的连接；存在数据的话判断最先连接的时间，间隔如果大于60s则会被清空，循环直至时间列表里面的数据都是60s内的连接时间时方止。
            history.pop()
        history.insert(0, now) #将最新的连接时间插入到时间列表的第一位
        print(history)
        if len(history) > 3: #判断时间列表的长度，大于3的话肯定是在60s内进来太多的连接
            return HttpResponse("滚")

转载于:https://www.cnblogs.com/xiayuhao/p/10034082.html