Djangorestframework之频率组件

Djangoframework的频率组件

频率组件的使用：

我们不妨先来看看频率组件的使用效果

views.py文件里面：

class Throttle(SimpleRateThrottle):
    scope = 'xxx'
    def get_cache_key(self, request, view):
        # return request.META.get('REMOTE_ADDR')
        #返回什么值，就以什么做过滤，返回用户id，就以用户id做过滤
        return self.get_ident(request)


class Books(APIView):
    throttle_classes=[Throttle,]
    def get(self,request):
        response ={'code':100,'msg':'获取成功','data':'所有图书'}
        return Response(response)

在settings里面配置一下：

REST_FRAMEWORK = {

    'DEFAULT_THROTTLE_RATES':{
        'xxx':'3/minute'
    }
}

这时候我们去发请求，看看会有什么效果

前三次都是可以拿到数据的，第四次的时候就会出现一些小意外：

就这样，就被服务端无情的拒绝了，让我在门外等待39s，让我不禁想彻底搞明白这频率组件，要看源码

 频率组件的源码分析

首先还是要来到APIView的diapatch方法，走APIView的innitial方法

走的就是APIView方法的check_throttles方法，就是这个就是throttle的核心源代码了，进入throttle类的实例化对象里面，看看allow_request的返回值是True，还是False

这其实就是在实例化一个个throttle类的对象

我们来到throttle类中看看allow_request方法，不难看出这个get_cache 是我们throttle自定义的一个方法

 走到get_ident时自己和SimpleRateThrottle都没有该方法，就会走父类(BaseThrottle)的方法，不难看出返回了ip值作为频率限制的对象。

返回到allow_request方法，对这个ip的访问历史进行判断，并存储。循环判断当ip的时间列表，有值，并且当前时间减去列表的最后一个时间大于60s，

把这种数据删除掉，这样的话，列表里面就只有60s以内的时间，当大于等于，说明一分钟访问超过三次，返回false验证失败，当小于三次时

列表里面插入值，缓存信息更新，返回true，验证成功

这时候又走到了核心源代码当返回true时就结束了，敲黑板，这是核心源代码，返回false时就会走wait方法

先看自己没有，就走SimpleRateThrottle的wait方法

 这样就形成了闭环，提高了代码的健壮性。

那么我们其实也是按照这个思路写一个MyThrottle类的频率组件验证

class MyThrottle(BaseThrottle):
    VISIT_RECORD = {}
    def __init__(self):
        self.history=None
    def allow_request(self,request,view):
        #自定义控制每分钟访问多少次，运行访问返回true，不允许访问返回false
        # （1）取出访问者ip{ip1:[第二次访问时间，第一次访问时间],ip2:[]}
        # （2）判断当前ip不在访问字典里，如果不在添加进去，并且直接返回True,表示第一次访问，在字典里，继续往下走
        # （3）循环判断当前ip的列表，有值，并且当前时间减去列表的最后一个时间大于60s，把这种数据pop掉，这样列表中只有60s以内的访问时间，
        # （4）判断，当列表小于3，说明一分钟以内访问不足三次，把当前时间插入到列表第一个位置，返回True，顺利通过
        # （5）当大于等于3，说明一分钟内访问超过三次，返回False验证失败
        # （1）取出访问者ip
        # print(request.META)
        #取出访问者ip
        ip = request.META.get('REMOTE_ADDR')
        import time
        #拿到当前时间
        ctime = time.time()
        # （2）判断当前ip不在访问字典里，添加进去，并且直接返回True,表示第一次访问
        if ip not in self.VISIT_RECORD:
            self.VISIT_RECORD[ip] = [ctime, ]
            return True
        #是个当前访问者ip对应的时间列表 [第一次访问的时间，]
        self.history = self.VISIT_RECORD.get(ip)
        # （3）循环判断当前ip的列表，有值，并且当前时间减去列表的最后一个时间大于60s，把这种数据pop掉，这样列表中只有60s以内的访问时间，
        while self.history and ctime - self.history[-1] > 60:
            self.history.pop()
        # （4）判断，当列表小于3，说明一分钟以内访问不足三次，把当前时间插入到列表第一个位置，返回True，顺利通过
        # （5）当大于等于3，说明一分钟内访问超过三次，返回False验证失败
        if len(self.history) < 3:
            self.history.insert(0, ctime)
            return True
        else:
            return False

    def wait(self):
        import time
        ctime = time.time()
        return 60 - (ctime - self.history[-1])

class Books(APIView):

    throttle_classes=[MyThrottle,]
    def get(self,request):
        response ={'code':100,'msg':'获取成功','data':'所有图书'}
        return Response(response)