请求 Iris 的正文限制中间件-Golang-PHP中文网

首页

后端开发

Golang

请求 Iris 的正文限制中间件

Susan Sarandon

Nov 02, 2024 pm 12:41 PM

Request Body Limit Middleware for Iris

概述

Iris Body Limit 中间件是一个强大的工具，用于控制 Iris Web 应用程序中传入请求正文的大小。通过设置请求正文的大小限制，您可以防止客户端发送过大的负载，否则可能会压垮您的服务器或导致拒绝服务 (DoS) 攻击。该中间件对于处理文件上传、JSON 有效负载或任何其他类型的大小可能变化很大的数据的应用程序特别有用。

为什么要使用 Body Limit 中间件？

安全

使用主体限制中间件的主要原因之一是增强应用程序的安全性。通过限制传入请求正文的大小，您可以降低 DoS 攻击的风险，攻击者会发送大量有效负载以耗尽服务器资源。

表现

限制请求正文的大小还可以提高应用程序的性能。大型有效负载会消耗大量内存和处理能力，从而降低服务器速度并影响用户体验。通过设置合理的限制，您可以确保您的服务器保持响应速度和高效。

资源管理

在处理文件上传或大型 JSON 负载的应用程序中，有效管理资源至关重要。通过设置正文限制，您可以防止客户端上传过大的文件或发送巨大的 JSON 对象，从而导致服务器资源紧张。

安装

要使用 bodylimit 中间件，您需要将其导入到 Iris 应用程序中：

import "github.com/kataras/iris/v12/middleware/bodylimit"

用法

基本设置

要使用身体限制中间件，您需要创建一个 Iris 应用程序并注册中间件。以下是如何设置 2 MB 限制的中间件的示例：

import "github.com/kataras/iris/v12/middleware/bodylimit"

解释

Limit：bodylimit.New 函数采用单个参数，该参数是请求正文的最大大小（以字节为单位）。在上面的示例中，限制设置为 10 个字节。
Handler：处理程序读取请求正文并将其写回响应。如果请求体超出限制，中间件将停止请求并返回 413 Request Entity Too Large 状态。
正文限制中间件使用sync.Pool来管理Reader实例，这些实例用于读取请求正文并强制执行大小限制。这种方法确保了内存的高效使用，并减少了为每个请求创建新 Reader 实例的开销。

使用 BodyLimit 中间件测试处理程序

要测试使用 BodyLimit 中间件的处理程序，您可以使用 Iris 提供的 httptest 包。以下是如何测试处理程序的示例：

package main

import (
    "github.com/kataras/iris/v12"
    "github.com/kataras/iris/v12/middleware/bodylimit"
)

func main() {
    app := iris.New()
    app.Use(bodylimit.New(2 * iris.MB)) // set the limit to 2 MB.

    handler := func(ctx iris.Context) {
        body, err := ctx.Body()
        if err != nil {
            ctx.StopWithPlainError(iris.StatusInternalServerError, err)
            return
        }

        ctx.Write(body) // write the request body back to the client.
    }

    app.Post("/", handler)
    app.Listen(":8080")
}