RedisとHaskellを使ってリソース制限機能を実装する方法-Redis-php.cn

RedisとHaskellを使ってリソース制限機能を実装する方法

Redis と Haskell を使用してリソース制限機能を実装する方法

現代のネットワークアプリケーションでは、リソースの管理と制限が非常に重要です。リソース制限によりサーバーの安定性が確保され、悪用や悪意のある動作が防止されます。この記事では、Redis と Haskell を使用してリソース制限機能を実装する方法と、具体的なコード例を紹介します。

Redis の概要

Redis は、さまざまなデータ構造をサポートする高性能のキー/値ストレージデータベースです。保存、カウント、有効期限などを含む豊富な機能を提供します。リソース制限機能では、Redis のカウント機能と有効期限機能を使用します。

Haskell の概要

Haskell は、強力な型システムと豊富な関数の組み合わせ機能を備えた純粋な関数型プログラミング言語です。 Haskell を使用して、Redis と対話するサーバー側ロジックを作成します。

リソース制限機能の実装

リソース制限機能を実装するにはさまざまな方法がありますが、ここではカウンタとタイムウィンドウに基づいた方法を紹介します。

3.1 Counter

Counter は、リソース制限を実装するための基本ツールです。 Redis の INCR コマンドを使用して、キーに対してカウンター操作を実行できます。以下はサンプルコードです。

import Database.Redis incrementCounter :: Connection -> ByteString -> IO Integer incrementCounter conn key = runRedis conn $ do res <- incr key case res of Right n -> return n Left _ -> return 0

ログイン後にコピー

このコードは、まず Redis サーバーに接続し、次に incr コマンドを使用して指定されたキーをインクリメントします。キーが存在しない場合は、自動的に作成され、値は 0 に初期化されます。

3.2 タイムウィンドウ

タイムウィンドウは、リソースの使用量を制限する時間範囲です。たとえば、時間枠を 1 分に設定できます。これは、ユーザーが 1 分間に特定の回数だけ特定のリソースにアクセスできることを意味します。以下はサンプルコードです。

import Data.Time.Clock.POSIX import Database.Redis withinTimeWindow :: Connection -> ByteString -> Integer -> Integer -> IO Bool withinTimeWindow conn key limit window = do timestamp <- round . (* 1000) <$> getPOSIXTime runRedis conn $ do res <- zadd key [(fromIntegral timestamp, "")] -- Add a timestamp to the sorted set case res of Right _ -> do _ <- zremrangebyscore key 0 (fromIntegral (timestamp - window * 1000)) -- Remove old timestamps x <- zcount key (fromIntegral timestamp) (fromIntegral (timestamp + 1)) -- Count the number of timestamps within the window return (x <= limit) Left _ -> return False

ログイン後にコピー

このコードは、まず現在のタイムスタンプを取得し、それをミリ秒レベルに変換します。次に、zadd コマンドを使用してソートセットにタイムスタンプを追加し、zremrangebyscore コマンドを使用して古いタイムスタンプを削除し、zcount コマンドを使用してウィンドウ内のタイムスタンプの数をカウントします。最後に、タイムスタンプの数が制限数以下であるかどうかを確認します。

サンプルアプリケーション

上記の 2 つの関数を使用して、単純なリソース制限アプリケーションを実装できます。

ユーザーが 1 分間に 100 メッセージしか送信できないように制限したいとします。次のコードを使用できます。

import Control.Monad import Data.ByteString.Char8 (pack) import Database.Redis main :: IO () main = do conn <- connect defaultConnectInfo replicateM_ 200 $ do count <- incrementCounter conn "user1" within <- withinTimeWindow conn "user1" 100 60 if count <= 100 && within then putStrLn "Allow" else putStrLn "Reject"

ログイン後にコピー

このコードは、最初に Redis サーバーに接続し、次に、replicateM_ 関数を使用して、ユーザーが 200 個のメッセージを送信することをシミュレートします。メッセージが送信されるたびに、まずカウンタがインクリメントされ、制限時間内か制限値を超えていないかが判断され、制限値を超えていなければメッセージの送信が許可され、そうでない場合は送信が拒否されます。。