首页 > 编程语言 >drf之频率类源码

drf之频率类源码

时间:2023-06-01 20:23:23浏览次数:56  
标签:return get self request rate 源码 频率 history drf

1 频率类

写一个类,继承SimpleRateThrottle,重写get_cache_key,返回[ip,用户id]什么,就以什么做限制,编写类属性  scope = 字符串,在配置文件中配置
      'DEFAULT_THROTTLE_RATES': {
        '字符串': '3/m',
      }
      配置在视图类,全局使用(配置在配置文件中)

2 自定义频率类

    源码中找---》自定义频率类如何写---》写个类,重写allow_request,如果能访问就返回True,不能访问返回False

    APIView---》dispatch---》走三大认证的时候
    def check_throttles(self, request):
        throttle_durations = []
        # 在视图类上配置的频率类的列表中一个个频率类的对象
        for throttle in self.get_throttles():
            if not throttle.allow_request(request, self):
            throttle_durations.append(throttle.wait())

        if throttle_durations:
            durations = [
               duration for duration in throttle_durations
               if duration is not None
            ]
            duration = max(durations, default=None)
            self.throttled(request, duration)

# 3 自定义频率类 这是我们自己写的
class MyThrottles():
    VISIT_RECORD = {} # 访问者 ip 字典,格式是{ip1:[时间4,时间3,时间2,时间1],ip2:[时间,时间],ip3:[当前时间]}
    def __init__(self):
        self.history = None
    def allow_request(self, request, view):
        # (1)取出访问者ip
        # print(request.META)
        ip = request.META.get('REMOTE_ADDR')
        import time
        ctime = time.time()
        # (2)判断当前ip不在访问字典里,添加进去,并且直接返回True,表示第一次访问
        if ip not in self.VISIT_RECORD:
            # {ip地址作为key:[当前时间,时间1,时间2,时间3]}
            self.VISIT_RECORD[ip] = [ctime, ]
            return True
        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])

SimpleRateThrottle源码分析

    """
    -allow_request:必须有的,频率类,必须重写它
    -get_cache_key:没有具体实现,直接抛了异常,需要子类重写
    -wait:必须有的,返回一个数字,告诉前端,还有多长时间能访问
    -------都是为了给allow_request辅助的-----
    get_rate
    parse_rate
    throttle_failure
    throttle_success
    """

def allow_request(self, request, view):
        if self.rate is None: # 要取init中看,self.rate=3/m,如果自己在频率类中写rate=5/m,我们就不需要写scope和配置文件了
            return True
        # get_cache_key返回了ip:192.168.1.19
        self.key = self.get_cache_key(request, view)
        if self.key is None: # 如果get_cache_key返回None,也不会做频率限制
            return True
        # self.history = self.VISIT_RECORD.get(self.key, [])
        # self.cache 是缓存,相当于咱们的self.VISIT_RECORD
        self.history = self.cache.get(self.key, [])
        self.now = self.timer()# 取出当前时间
        # self.duration:配置的1分钟访问3次,self.duration就是60
        while self.history and self.history[-1] <= self.now - self.duration:
            self.history.pop()
        if len(self.history) >= self.num_requests: # '5/m',num_requests就是5
            return self.throttle_failure()
        return self.throttle_success()

# __init__    
def __init__(self):
  if not getattr(self, 'rate', None): # 如果没有,执行下面的代码
      self.rate = self.get_rate()
  self.num_requests, self.duration = self.parse_rate(self.rate)

# self.get_rate()
    def get_rate(self):
        # self.scope  我们自己写的字符串
        # self.THROTTLE_RATES配置文件中的那个字典
        return self.THROTTLE_RATES[self.scope]  # 咱们配置的 3/m

# self.num_requests, self.duration = self.parse_rate(self.rate)---》self.rate是3/s
    def parse_rate(self, rate):
        if rate is None:
            return (None, None)
        # 3/m---->num=3,period=m
        num, period = rate.split('/')
        # num_requests=数字3
        num_requests = int(num)
        # 
        d={'s': 1, 'm': 60, 'h': 3600, 'd': 86400}
        period[0]='m'
        d[m]
        duration = {'s': 1, 'm': 60, 'h': 3600, 'd': 86400}[period[0]]
        return (num_requests, duration)

标签:return,get,self,request,rate,源码,频率,history,drf
From: https://www.cnblogs.com/XxMa/p/17449080.html

相关文章

  • springboot整合shiro实现认证授权源码
    shiro-admin介绍springboot整合shiro实现前后端分离架构(swagger文档协调前端开发)源码地址:https://gitee.com/liujinxin_ark/shiro-admin软件架构架构说明springboot+shiro+mysql+swagger使用说明运行项目后访问http://localhost:8080/doc.html即可进入swagger接口文档界......
  • 源码、二进制可执行文件、jar包
    源码(Sourcecode):源码是开发人员编写的人类可读的程序代码,它以文本文件的形式存在,并使用特定的编程语言编写。源码包含了程序的逻辑、算法和操作步骤,以及相关的注释和文档说明。源码通常存储在版本控制系统中,并且用于开发、调试和维护软件。二进制文件(Binaryfile):二进制文件是......
  • 源码和二进制文件
    源码(Sourcecode)是开发人员编写的人类可读的程序代码。它通常以文本文件的形式存在,使用特定的编程语言编写。源码包含了程序的逻辑、算法和操作步骤,以及相关的注释和文档说明。二进制文件(Binaryfile)是计算机能够直接执行的机器码形式的文件。它是由源代码经过编译或汇编生成的,以......
  • 怎么编译源码,得到可执行二进制文件
    源码(Sourcecode)是开发人员编写的人类可读的程序代码。它通常以文本文件的形式存在,使用特定的编程语言编写。源码包含了程序的逻辑、算法和操作步骤,以及相关的注释和文档说明。二进制文件(Binaryfile)是计算机能够直接执行的机器码形式的文件。它是由源代码经过编译或汇编生成的,以......
  • 【虚幻引擎】UE4源码解析FWorldContent、UWorld、ULevel、UGameInstance、UEngine
    一、UEngineEngine,因为也是很基础的类,再加上开发过程中会经常访问到该类型,因此UE4引擎也在代码全局范围内定义了一个该类型的全局变量:UEngine*GEngine供开发者直接调用。该最基础的类型分化成了两个子类:UGameEngine和UEditorEngine。UGameEngine保存了唯一的一个UGameInstan......
  • 直播小程序源码,flutter TextField 限制输入长度,限制输入数字文字
    直播小程序源码,flutterTextField限制输入长度,限制输入数字文字//限制长度inputFormatters:[LengthLimitingTextInputFormatter(11)], //限制输入数字文字等类型inputFormatters:[WhitelistingTextInputFormatter.digitsOnly], //键盘类型keyboardType:TextInputType.tex......
  • C# WINFORM 打砖块游戏,可以进行两队PK 程序源码
    C#WINFORM打砖块游戏红色绿色阵营,球可以自定义添加,图片可以设置为网络地址的头像,可以进行与评论和弹幕进行建设。 支持球增加支持球加速支持一键初始化游戏支持pk支持积分累计程序代码还有完全开发完毕,有些小细节还需要再进行优化程序使用纯原生代码,没有使用任何第三......
  • 成品直播源码推荐,Android 自定义颜色样式
    成品直播源码推荐,Android自定义颜色样式<?xmlversion="1.0"encoding="utf-8"?><resources>  <colorname="colorPrimary">#7bb736</color>  <colorname="colorPrimaryDark">#16c24b</color>  <c......
  • 直播app源码技术开发知识:横竖屏功能的实现
    在快节奏时代的今天,直播短视频日益火爆,许多人选择去进入直播短视频源码搭建平台的行业,去开发制作自己的直播短视频app平台。当然,要开发制作自己的直播短视频app平台就要去顺应市场的需要,将用户们所需要的和市场各大app好的功能都要添加到自己的app中,不知道大家有没有发现这么一个功......
  • 还在用BeanUtils拷贝对象,MapStruct才是yyds | 附源码
    前几天,远在北京的小伙伴在群里抛出了“MapStruct”的概念。对于只闻其名,未见其人的我来说,决定对其研究一番。本文我们就从MapStruct的概念出发,通过具体的代码示例来研究它的使用情况,最后与“市面上”的其它工具来做个对比!官方介绍首先我们打开MapStruct的官网地址,映入眼帘的就......