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

ゞ浴缸里的玫瑰 2022-12-27 12:42 196阅读 0赞

#  一、概述 #

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

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

官方文档：[https://django-ratelimit.readthedocs.io/en/stable/][https_django-ratelimit.readthedocs.io_en_stable]

安装插件

pip3 install django-ratelimit

# 二、演示效果 #

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

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

新建项目test1，app名为web

![6446c42be7dd8c9246707d818bf029b0.png][]

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

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

在templates目录下，创建index.html

<!DOCTYPE html>
    <html lang="en">
    <head>
        <meta charset="UTF-8">
        <title>Title</title>
    </head>
    <body>
    Hello world
    </body>
    </html>

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

from django.contrib import admin
    from django.urls import path
    from web import views
    
    urlpatterns = [
        path('admin/', admin.site.urls),
        path('',views.index),
    ]

启动项目，访问默认页面

http://127.0.0.1:8000/

效果如下：

![12858d08b9fde9f93c0999af2d6ecd31.png][]

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

## 路由 ##

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

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 频率参数，列表如下：

s - second
    m - minute
    h - hour
    d - day

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

block=True **注意：必须设置为true，才能生效。**

重启项目，测试访问

![43dfb316a58548d4aba9945c7fabec62.gif][]

## 视图函数 ##

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

修改test1/urls.py

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前，增加装饰器

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][https_blog.csdn.net_qq_36387683_article_details_95452061]

[https_django-ratelimit.readthedocs.io_en_stable]: https://django-ratelimit.readthedocs.io/en/stable/
[6446c42be7dd8c9246707d818bf029b0.png]: /images/20221120/9bae2955774b4625ad5faa96dfd546d6.png
[12858d08b9fde9f93c0999af2d6ecd31.png]: /images/20221120/6d877fb01a0a4aafad38176b90964f0b.png
[43dfb316a58548d4aba9945c7fabec62.gif]: https://img-blog.csdnimg.cn/img_convert/43dfb316a58548d4aba9945c7fabec62.gif
[https_blog.csdn.net_qq_36387683_article_details_95452061]: https://blog.csdn.net/qq_36387683/article/details/95452061