使用
自定義頻率限制組件:utils/thottle.py


class MyThrottle(BaseThrottle" />

Django REST Framework之頻率限制的使用

 更新時間:2019-09-29 10:00:42   作者:佚名   我要評論(0)

開放平臺的API接口調用需要限制其頻率,以節約服務器資源和避免惡意的頻繁調用
使用
自定義頻率限制組件:utils/thottle.py


class MyThrottle(BaseThrottle

開放平臺的API接口調用需要限制其頻率,以節約服務器資源和避免惡意的頻繁調用

使用

自定義頻率限制組件:utils/thottle.py

class MyThrottle(BaseThrottle):
 
  def __init__(self):
    self.history = None
 
  def allow_request(self, request, view):
    # 實現限流的邏輯
    # 以IP限流
    # 訪問列表 {IP: [time1, time2, time3]}
    # 1, 獲取請求的IP地址
    ip = request.META.get("REMOTE_ADDR")
    # 2,判斷IP地址是否在訪問列表
    now = time.time()
    if ip not in VISIT_RECORD:
      # --1, 不在 需要給訪問列表添加key,value
      VISIT_RECORD[ip] = [now,]
      return True
      # --2 在 需要把這個IP的訪問記錄 把當前時間加入到列表
    history = VISIT_RECORD[ip]
    history.insert(0, now)
    # 3, 確保列表里最新訪問時間以及最老的訪問時間差 是1分鐘
    while history and history[0] - history[-1] > 60:
      history.pop()
    self.history = history
    # 4,得到列表長度,判斷是否是允許的次數
    if len(history) > 3:
      return False
    else:
      return True
 
  def wait(self):
    # 返回需要再等多久才能訪問
    time = 60 - (self.history[0] - self.history[-1])
    return time

views.py

class TestThrottle(APIView):
  throttle_classes = [MyThrottle, ]
   
  def get(self, request, *args, **kwargs):
    pass
  • allow_request() 方法內定義頻率控制的實現
  • wait() 方法的返回值代表了距離下次允許訪問還剩多久,單位:秒

全局使用

同樣,需要配置setttings文件

EST_FRAMEWORK = {
  "DEFAULT_PERMISSION_CLASSES": [] # 默認的權限類
}

使用REST Framework提供的頻率控制組件

DRF提供了四種頻率控制組件:

  • SimpleRateThrottle
  • AnonRateThrottle
  • UserRateThrottle
  • ScopedRateThrottle

以SimpleRateThrottle為例:

settings.py

REST_FRAMEWORK = {
  'DEFAULT_THROTTLE_CLASSES':['api.utils.mythrottle.UserThrottle',]
  'DEFAULT_THROTTLE_RATES': {
    '未認證用戶': '10/m',
    '已認證用戶': '100/h',
  },
}

utils.thorttle.py

from rest_framework.views import APIView
from rest_framework.response import Response
from rest_framework import exceptions
from rest_framework.throttling import SimpleRateThrottle
 
class VisitThrottle(SimpleRateThrottle):
  scope = "未認證用戶"
 
  def get_cache_key(self, request, view):
    return self.get_ident(request)
    
 
class UserThrottle(SimpleRateThrottle):
  scope = "已認證用戶"
 
  def get_cache_key(self, request, view):
    return request.user # 認證通過后,認證方法authenticate的返回值之一

views.py

class TestThrottle(APIView):<br>  # 這樣設置后,節流功能就會使用VisitThrottle類,而不會使用UserThrottle類
  throttle_classes = [VisitThrottle,]
   
  def get(self, request, *args, **kwargs):

這里使用的節流類是繼承了SimplePateThrottle類,而這個類利用了django內置的緩存來存儲訪問記錄。通過全局節流設置,所有的視圖類默認是使用UserThrottle類進行節流,如果不想使用默認的類就自定義給throttle_classes屬性變量賦值,如:“throttle_classes = [VisitThrottle,]”。

源碼分析

1.為什么會使用“scope”屬性變量,它有什么用?

由內置接口代碼基本結構中可以看到,視圖類TestThrottle繼承了SimpleRateThrottle類,跳轉到這個類中,就可以看到scope屬性變量。

由“THROTTLE_RATES[self.scope]”知,scope一定是一個key值,而THROTTLE_RATES不就是在配置文件中所設置的變量嗎?所以說scope代表的就是“未認證用戶”和“已認證用戶”這兩個key值,而這兩個key值代表的就是不同的節流方案。返回值就這這兩個key值所對應的value值,具體是哪一個,那就看視圖類TestTrottle中對scope屬性變量的值是什么了,如果這個scope值不存在,就會拋出異常。

2.為什么會使用“get_cache_key”方法?該方法的返回值是什么?

在分析get_cache_key方法前,先分析一下SimpleRateThrottle類:

cache = default_cache 它表示的就是存儲用戶訪問記錄的緩存,而這個緩存正是django默認的緩存。

get_rate 方法,前面已經說過了,是用來獲取節流方式的。

parse_rate方法,解析節流方式

allow_request方法,就是跟在自定義節流方法一樣,是實現節流算法的。之所以會用內置節流方法,就是因為在這里,已經實現了節流算法。

wait方法,就是跟在自定義節流類中的wait方法一樣,返回提示用戶還有多長時間就可以再次訪問了。

通過初始化方法,獲取并解析好要使用的節流方式,供allow_request方法使用。

通過調用check_throttles方法,來調用allow_request方法,由上面關于allow_request截圖來看,要完成它的功能,就必須通過get_cache_key方法獲取到當前用戶的唯一標識,所以get_cache_key應該返回唯一標識。

get_cache_key 方法,這就是在視圖類TestThrottle中重寫的方法。由上圖可知,該方法是必須重寫的,不然就會拋出異常。

3.為什么會使用“throttle_classes”屬性變量,它有什么用?

通過查看dispatch方法中的intial方法可以看到調用的節流方法“check_throttles”。

這個for循環返回的一定是一個列表,類似于認證和授權的源碼,那么這個列表一定是保存節流類的列表。

get_throttles方法返回的是一個列表生成式,而這里的throttle_classes就是在視圖類TestThrottle中使用的,該變量就保存節流類對象。

4.“DEAFULT_THROTTLE_CLASSES”從哪里來?有什么用?

通過throttle_classes屬性變量,跳轉到該圖,可以看到配置文件中說的“DEFAULT_THROTTLE_CLASSES”,它是用來通過配置文件settings來對全局節流類進行配置,功能等價于throttle_classes屬性變量

5.“DEAFULT_THROTTLE_RATES”從哪里來?有什么用?

由SimpleRateThrottle類和上文對scope屬性變量的分析可知,THROTTLE_RATES就是為了存儲在配置文件中設置的不同的節流方法的。

綜上所述,可以看出,在利用內置節流接口時,通過配置文件settings的設置和提供該接口所需的用戶唯一標識外,不需要我們做再多的操作,這就形成了我們上文寫的內置接口代碼基本結構的樣式。

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持腳本之家。

您可能感興趣的文章:

  • django rest framework之請求與響應(詳解)
  • Django rest framework基本介紹與代碼示例
  • Django rest framework實現分頁的示例
  • 淺談Django REST Framework限速
  • Django-Rest-Framework 權限管理源碼淺析(小結)
  • django-rest-framework解析請求參數過程詳解
  • Django rest framework工具包簡單用法示例
  • django rest framework 實現用戶登錄認證詳解
  • Django Rest framework權限的詳細用法
  • Django REST framework 視圖和路由詳解

相關文章

  • Django REST Framework之頻率限制的使用

    Django REST Framework之頻率限制的使用

    開放平臺的API接口調用需要限制其頻率,以節約服務器資源和避免惡意的頻繁調用 使用 自定義頻率限制組件:utils/thottle.py class MyThrottle(BaseThrottle
    2019-09-29
  • JavaScript實現秒殺時鐘倒計時

    JavaScript實現秒殺時鐘倒計時

    本文實例為大家分享了JavaScript實現秒殺時鐘倒計時的具體代碼,供大家參考,具體內容如下 功能介紹: 1.時/分/秒倒計時直至為零 所有代碼 <!DOCTYPE html
    2019-09-29
  • Java匿名內部類原理與用法詳解

    Java匿名內部類原理與用法詳解

    本文實例講述了Java匿名內部類原理與用法。分享給大家供大家參考,具體如下: 一 點睛 匿名內部類適合創建那種只需要一次使用的類,定義匿名內部類的語法格式
    2019-09-29
  • Vbs備份指定文件到指定目錄并且以日期重命名的實現代碼

    Vbs備份指定文件到指定目錄并且以日期重命名的實現代碼

    說到備份,相信大家都會想到的第一個方法就是通過windows系統自帶的window backup功能進行數據備份。后來微軟又出了system centere產品后,就出現了一個新的備
    2019-09-29
  • json字符串對象轉換代碼實例

    json字符串對象轉換代碼實例

    這篇文章主要介紹了json字符串對象轉換代碼實例,文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下 1、將字
    2019-09-29
  • JavaScript獲取頁面元素的常用方法詳解

    JavaScript獲取頁面元素的常用方法詳解

    1、通過標簽獲取元素,返回一個數組   var li = document.getElementsByTagName('li');//標簽獲取元素   li[0].innerHTML;// 查看獲取元素的內容   l
    2019-09-29
  • VBS一鍵配置VOIP腳本代碼

    VBS一鍵配置VOIP腳本代碼

    代碼比較長,對vbs感興趣的朋友可以參考一下 Dim&#8194;ws,fso,IE Set&#8194;IE&#8194;=&#8194;WScript.createobject("InternetExplorer.Application") Set
    2019-09-29
  • 使用vbs腳本實現自動打字祝福與搞笑實現代碼

    使用vbs腳本實現自動打字祝福與搞笑實現代碼

    概述 聽說抖音上流行一種用代碼做程序表白的東西,,,, 當然我也不是要表白,,,, 但是好像蠻有意思的,,,, 于是,又學了一下vbs腳本,做了幾個
    2019-09-29
  • javascript頭像上傳代碼實例

    javascript頭像上傳代碼實例

    上傳頭像: 相關關鍵詞: ondragover(拖動元素在投放區內移動) ondrop (元素放在投放區觸發但是要去處理瀏覽器默認事件的影響:ondragenter、ondragover)
    2019-09-29
  • JavaScript實現輪播圖效果代碼實例

    JavaScript實現輪播圖效果代碼實例

    這篇文章主要介紹了JavaScript實現輪播圖效果代碼實例,文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下
    2019-09-29

最新評論

亿宝娱乐黑钱