Django实现文件上传、文件列表查看、修改、限流和日志记录7

Django实现文件上传、文件列表查看、修改、限流和日志记录7

不管调用的接口在内网，还是外网都需要做好限制保证接口的访问和限流降级处理，本章节新增限流功能。

限流功能主要针对两个方面： IP和用户

针对IP限流

文件列表接口进行限流处理，限制每分钟每个 IP 访问 10 次，你可以使用 Django 的内置限流功能结合缓存来实现

使用了 Django 的缓存系统来存储每个 IP 地址的访问次数。对于每个请求，我们首先获取客户端的 IP 地址，然后检查该 IP 地址的访问次数。如果访问次数超过限制（10 次），我们返回 403 Forbidden 响应。否则，我们增加访问次数并将缓存时间设置为 1 分钟。

from django.core.cache import cache
from django.http import HttpResponseForbidden

def file_list(request):
    # 获取客户端 IP 地址
    client_ip = request.META.get('REMOTE_ADDR')

    # 检查该 IP 地址的访问次数
    cache_key = f'file_list:{client_ip}'
    access_count = cache.get(cache_key, 0)

    # 如果访问次数超过限制，返回 403 Forbidden
    if access_count >= 10:
        logging.warning('用户 {} 在 {} 访问用户列表超过限制'.format(request.user.username, datetime.now()))
        return HttpResponseForbidden('访问次数超过限制')

    # 增加访问次数并设置缓存时间为 1 分钟
    cache.set(cache_key, access_count + 1, 60)

    # 继续处理文件列表逻辑
    # ...

    return render(request, 'file_list.html', {'file_list': file_list})

 

访问十次提示次数超限

 

针对用户限流

限制每个用户每分钟访问文件列表的次数，你可以使用 Django 的内置限流功能结合缓存来实现。

以下是一个示例代码，演示如何对每个用户进行限流处理，限制每分钟访问文件列表的次数为 10 次：

from django.core.cache import cache
from django.http import HttpResponseForbidden

def file_list(request):
    # 获取当前用户
    user = request.user

    # 检查该用户的访问次数
    cache_key = f'file_list:{user.username}'
    access_count = cache.get(cache_key, 0)

    # 如果访问次数超过限制，返回 403 Forbidden
    if access_count >= 10:
        logging.warning('用户 {} 在 {} 访问用户列表超过限制'.format(request.user.username, datetime.now()))
        return HttpResponseForbidden('访问次数超过限制')

    # 增加访问次数并设置缓存时间为 1 分钟
    cache.set(cache_key, access_count + 1, 60)

    # 继续处理文件列表逻辑
    # ...

    return render(request, 'file_list.html', {'file_list': file_list})

 

访问十次依然提示次数超限

装饰器限流

将限流定义为一个函数，并在其他接口中引用并设置不同的访问次数限制，你可以将限流逻辑封装在一个装饰器函数中，并在需要进行限流的视图函数上应用该装饰器。

#定义限流装饰器
from django.core.cache import cache
from django.http import HttpResponseForbidden

def rate_limit(limit_count):
    def decorator(view_func):
        def wrapper(request, *args, **kwargs):
            # 获取当前用户
            user = request.user

            # 检查该用户的访问次数
            cache_key = f'rate_limit:{user.username}'
            access_count = cache.get(cache_key, 0)

            # 如果访问次数超过限制，返回 403 Forbidden
            if access_count >= limit_count:
                logging.error('用户 {} 在 {} 查看上传文件列表失败'.format(request.user.username, datetime.now()))
                return HttpResponseForbidden('访问次数超过限制')

            # 增加访问次数并设置缓存时间为 1 分钟
            cache.set(cache_key, access_count + 1, 60)

            # 继续执行视图函数
            return view_func(request, *args, **kwargs)

        return wrapper

    return decorator

定义了一个名为 rate_limit 的限流函数，它接受一个访问次数限制作为参数。该函数返回一个装饰器函数，用于包装需要进行限流的视图函数。

要在其他接口上应用限流，你可以使用 @rate_limit(limit_count) 这样的语法，将装饰器应用到视图函数上。

###   调用示例一
#导入装饰器的限流函数
from .views import rate_limit

@rate_limit(10) #设置访问次数限制为10
@login_required
def file_list(request):

###   调用示例二
#导入装饰器的限流函数，若找不到rate_limit模块检查导入文件路径，将限流装饰器放在调用视图前面
from .views import rate_limit

@rate_limit(10) #设置访问次数限制为10
@login_required
def view_file(request, file_name):

访问上述两个路由地址超过十次报错，超过次数限制

 

django-ratelimit库进行限流

RatelimitMiddleware是django-ratelimit库提供的中间件，用于在请求到达视图之前进行请求限流，本次不做使用演示可自行了解使用，以下是一个演示示例：

#在Django的settings.py配置文件中添加以下配置：
# settings.py

INSTALLED_APPS = [
  # ...
  'ratelimit',
  # ...
]

MIDDLEWARE = [
  # ...
  'ratelimit.middleware.RatelimitMiddleware',
  # ...
]

# 在应用限流的视图函数或者类上使用ratelimit装饰器来进行限流设置
# myapp/views.py

from ratelimit.decorators import ratelimit

@ratelimit(key='ip', rate='5/m')  # 设置每分钟每个用户最多5个请求
def my_view(request):
  if request.limited:
        # 请求被限流的处理逻辑
        return HttpResponse('请求过于频繁，请稍后再试', status=429)

  # 处理请求的代码
  pass

 

至此，接口限流器的配置已经完成。