Iris 请求速率限制中间件-Golang-PHP中文网

Request Rate Limiting Middleware for Iris

概述

速率中间件为 Iris Web 框架提供速率限制功能。它允许开发人员控制对其应用程序的请求速率，确保公平使用并防止滥用。中间件基于令牌桶算法，这是一种流行的限速方法。

安装

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

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

登录后复制

用法

基本设置

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

package main

import (
    "time"

    "github.com/kataras/iris/v12"
    "github.com/kataras/iris/v12/middleware/rate"
)

func main() {
    app := iris.New()
    app.Logger().SetLevel("debug")

    limit := rate.Limit(1, 5, rate.PurgeEvery(time.Minute, 5*time.Minute))
    app.Use(limit)

    app.Get("/", index)
    app.Get("/other", other)

    app.Listen(":8080")
}

func index(ctx iris.Context) {
    ctx.HTML("<h1>Index Page</h1>")
}

func other(ctx iris.Context) {
    ctx.HTML("<h1>Other Page</h1>")
}

登录后复制

此示例允许每秒 1 个请求，最大突发大小为 5。如果旧条目 5 分钟内没有出现，它还会每分钟清除一次。

使用率.各帮手

这个例子演示了如何使用速率。每个助手设置速率限制器：

package main

import (
    "time"

    "github.com/kataras/iris/v12"
    "github.com/kataras/iris/v12/middleware/rate"
)

func main() {
    app := iris.New()
    app.Logger().SetLevel("debug")

    // Use rate.Every helper to set up the rate limiter.
    limit := rate.Limit(rate.Every(time.Minute), 5)
    app.Use(limit)

    app.Get("/", index)
    app.Get("/other", other)

    app.Listen(":8080")
}

func index(ctx iris.Context) {
    ctx.HTML("<h1>Index Page</h1>")
}

func other(ctx iris.Context) {
    ctx.HTML("<h1>Other Page</h1>")
}

登录后复制

使用API密钥进行速率限制

此示例演示如何设置使用 API 密钥而不是客户端远程 IP 地址的速率限制器：

package main

import (
    "time"

    "github.com/kataras/iris/v12"
    "github.com/kataras/iris/v12/middleware/rate"
)

func main() {
    app := iris.New()
    app.Logger().SetLevel("debug")

    // Use API key for rate limiting.
    app.Use(useAPIKey)

    limit := rate.Limit(rate.Every(time.Minute), 300, rate.PurgeEvery(5*time.Minute, 15*time.Minute))
    app.Use(limit)

    app.Get("/list", list)

    app.Listen(":8080")
}

func useAPIKey(ctx iris.Context) {
    apiKey := ctx.Header("X-API-Key")
    if apiKey == "" {
        ctx.StopWithStatus(iris.StatusForbidden)
        return
    }

    rate.SetIdentifier(ctx, apiKey)
    ctx.Next()
}

func list(ctx iris.Context) {
    ctx.JSON(iris.Map{"key": "value"})
}

登录后复制

自定义超出处理程序

此示例演示如何设置超出速率限制时执行的自定义处理程序：

package main

import (
    "time"

    "github.com/kataras/iris/v12"
    "github.com/kataras/iris/v12/middleware/rate"
)

func main() {
    app := iris.New()
    app.Logger().SetLevel("debug")

    // Set a custom exceed handler.
    limit := rate.Limit(1, 5, rate.ExceedHandler(func(ctx iris.Context) {
        ctx.StopWithStatus(429)
    }))
    app.Use(limit)

    app.Get("/", index)
    app.Get("/other", other)

    app.Listen(":8080")
}

func index(ctx iris.Context) {
    ctx.HTML("<h1>Index Page</h1>")
}

func other(ctx iris.Context) {
    ctx.HTML("<h1>Other Page</h1>")
}

登录后复制

自定义客户数据

此示例演示如何为每个客户端存储自定义数据：

package main

import (
    "time"

    "github.com/kataras/iris/v12"
    "github.com/kataras/iris/v12/middleware/rate"
)

func main() {
    app := iris.New()
    app.Logger().SetLevel("debug")

    // Store custom data for each client.
    limit := rate.Limit(1, 5, rate.ClientData(func(ctx iris.Context) any {
        return ctx.RemoteAddr()
    }))
    app.Use(limit)

    app.Get("/", index)
    app.Get("/other", other)

    app.Listen(":8080")
}

func index(ctx iris.Context) {
    ctx.HTML("<h1>Index Page</h1>")
}

func other(ctx iris.Context) {
    ctx.HTML("<h1>Other Page</h1>")
}

登录后复制

解释

速率限制：rate.Limit 函数用于创建新的速率限制器。它需要三个参数：
- limit：每秒允许的最大请求数。
- 突发：最大突发大小。
- options：附加选项，例如用于清理旧条目的 PurgeEvery。
令牌桶算法：该算法通过维护令牌桶来控制请求速率。每个请求都会消耗一个令牌，并且令牌会以固定的速率添加到桶中。如果存储桶为空，则请求被拒绝。