フロー制御に Go 言語と Redis を使用する方法-Golang-php.cn

フロー制御に Go 言語と Redis を使用する方法

WBOY

リリース： 2023-10-28 09:48:27

オリジナル

850 人が閲覧しました

フロー制御に Go 言語と Redis を使用する方法

Go 言語と Redis をフロー制御に使用する方法

はじめに
高度な同時実行ネットワークアプリケーションでは、フロー制御は非常に重要なリンクです。システムの安定性と信頼性を確保するには、トラフィックを制限し、管理する必要があります。この記事では、Go 言語と Redis を使用してフロー制御を実装する方法と、具体的なコード例を紹介します。

背景
分散システムでは、フロー制御はシステムの正常な動作を保証する重要な手段の 1 つです。システムが大量の同時リクエストに直面すると、過剰なトラフィックによりシステムがクラッシュしたり、応答が遅くなったりする可能性があります。したがって、システムが過負荷にならないようにトラフィックを制限する必要があります。 Redis は、フロー制御を容易にする豊富なデータ構造とコマンドを提供する高性能インメモリデータベースです。

プロジェクト設計
私たちのソリューション設計は次のとおりです:

Redis のカウンターデータ構造を使用して、各ユーザーのリクエスト数を記録します。
順序付きセット (ソートセット) を使用して、後続のタイムウィンドウ計算のためにユーザーのリクエストタイムスタンプを保存します。
Redis のトランザクション機能を使用して、操作の原子性と一貫性を確保します。
同時処理にはGoroutineを使用します。

具体的な実装
ユーザーは 60 秒間に 100 件のリクエストしか送信できないと想定しています。 Redis のカウンターデータ構造を使用して、この制限を達成できます。以下はサンプルコードです:

package main

import (
    "fmt"
    "strconv"
    "sync"
    "time"

    "github.com/go-redis/redis"
)

var (
    wg sync.WaitGroup
    rdb *redis.Client
)

func main() {
    rdb = redis.NewClient(&redis.Options{
        Addr:     "localhost:6379", // Redis地址
        Password: "",               // Redis密码
        DB:       0,                // Redis数据库
    })

    for i := 0; i < 100; i++ {
        wg.Add(1)
        go sendRequest(i)
    }

    wg.Wait()
}

func sendRequest(userID int) {
    defer wg.Done()

    // 检查用户请求数是否超过限制
    count, err := rdb.Incr(strconv.Itoa(userID)).Result()
    if err != nil {
        fmt.Println("Redis error:", err)
        return
    }

    if count > 100 {
        fmt.Println("Request limit exceeded for user", userID)
        return
    }

    // 获取当前时间戳
    now := time.Now().Unix()

    // 将当前时间戳添加到有序集合中
    _, err = rdb.ZAdd("timestamps", redis.Z{
        Score:  float64(now),
        Member: strconv.Itoa(userID),
    }).Result()
    if err != nil {
        fmt.Println("Redis error:", err)
        return
    }

    // 移除60秒前的时间戳
    _, err = rdb.ZRemRangeByScore("timestamps", "0", strconv.FormatInt(now-60, 10)).Result()
    if err != nil {
        fmt.Println("Redis error:", err)
        return
    }

    fmt.Println("Request sent by user", userID)
}

ログイン後にコピー

説明と呼び出し

まず、Redis クライアントを作成して Redis データベースに接続します。
次に、ループを使用してリクエストを送信する 100 個のコルーチンを作成します。各コルーチンはユーザーを表します。
リクエストを送信する関数 sendRequest では、まず INCR コマンドを使用してユーザーリクエストの数を増やし、制限を超えているかどうかを確認します。
次に、現在のタイムスタンプを取得して、並べ替えられたセットに追加します。
最後に、ZRemRangeByScore コマンドを使用して、期限切れのタイムスタンプを削除します。

結論
この記事では、Go 言語と Redis を使用してフロー制御を実装する方法を紹介します。 Redis のカウンターと順序付きセットのデータ構造を使用することで、ユーザーリクエストの数とタイムスタンプを簡単に記録し、トラフィックを制限できます。このソリューションは、過剰なトラフィックからシステムを効果的に保護し、システムの安定性と信頼性を確保します。

参考資料: