django利用django-ratelimit设置接口请求频率限制

2020-12-15 12:00:12 浏览数 (1)

 一、概述

平台中需要编写接口供第三方调用,需要控制调用频率,需求为5s内调用一次后不得再次调用。

Django官方插件库中有个django-ratelimit插件可以满足要求, django-ratelimit文档地址,很灵活很强大。

官方文档:https://django-ratelimit.readthedocs.io/en/stable/

安装插件

代码语言:javascript复制
pip3 install django-ratelimit

二、演示效果

要对某个url做限制,django-ratelimit有2种方法,一个是在路由里面配置,一个是在视图函数中添加装饰圈。

当然,还有其他方式,比如中间件,这里不做演示,具体请参考官方文档。

新建项目test1,app名为web

修改web/views.py,增加视图函数index

代码语言:javascript复制
from django.shortcuts import render

# Create your views here.
def index(request):
    return render(request,"index.html")

在templates目录下,创建index.html

代码语言:javascript复制
<!DOCTYPE html>
<html>
<head>
    <meta charset="UTF-8">
    <title>Title</title>
</head>
<body>
Hello world
</body>
</html>

修改test1/urls.py,增加路由

代码语言:javascript复制
from django.contrib import admin
from django.urls import path
from web import views

urlpatterns = [
    path('admin/', admin.site.urls),
    path('',views.index),
]

启动项目,访问默认页面

代码语言:javascript复制
http://127.0.0.1:8000/

效果如下:

准备工作已经做好了,下面开始正式演示频率限制。

路由

修改test1/urls.py,增加限制

代码语言:javascript复制
from django.contrib import admin
from django.urls import path
from ratelimit.decorators import ratelimit
from web import views

urlpatterns = [
    path('admin/', admin.site.urls),
    path('', ratelimit(key='ip', method='GET', rate='1/5s',block=True)(views.index)),
]

参数说明:

ip 使用请求ip(获取request.META['REMOTE_ADDR']中的ip地址)

rate 频率参数,列表如下:

代码语言:javascript复制
s - second
m - minute
h - hour
d - day

上面写的1/5s,表示5秒内,只能调用一次。

block=True 注意:必须设置为true,才能生效。

重启项目,测试访问

视图函数

先注释掉路由,还原为之前的

修改test1/urls.py

代码语言:javascript复制
from django.contrib import admin
from django.urls import path
from web import views

urlpatterns = [
    path('admin/', admin.site.urls),
    path('',views.index),
]

修改web/views.py,在视图函数index前,增加装饰器

代码语言:javascript复制
from django.shortcuts import render
from ratelimit.decorators import ratelimit

# Create your views here.
@ratelimit(key='ip', rate='1/m',block=True)
def index(request):
    return render(request,"index.html")

重启项目,测试访问。效果同上!

总结:还是用url里面定义是最方便的,不需要操作视图函数。

注意:针对uwsgi nginx环境中,如果uwsgi运行了多个进程,这个限制效果会有影响。因为uwsgi多进程之间,内存是不共享的。

如果一个请求,刚好分配到一个新的uwsgi中,则是不受限制的。如果刚好分配到刚刚访问到的uswgi,才会有频率限制效果。

本文参考链接:

https://blog.csdn.net/qq_36387683/article/details/95452061

0 人点赞