如何在Redis中實(shí)現(xiàn)分布式限流

在Redis中實(shí)現(xiàn)分布式限流可以通過使用Redis的計(jì)數(shù)器和過期時(shí)間來實(shí)現(xiàn)。以下是一種基于計(jì)數(shù)器和過期時(shí)間的簡單實(shí)現(xiàn)方式：

1. 使用Redis的INCR命令來遞增一個(gè)計(jì)數(shù)器，表示當(dāng)前已經(jīng)被請求的次數(shù)。
2. 使用Redis的EXPIRE命令來設(shè)置一個(gè)過期時(shí)間，保證計(jì)數(shù)器在一定時(shí)間內(nèi)自動(dòng)清零。
3. 在每次請求到來時(shí)，先判斷計(jì)數(shù)器是否已經(jīng)達(dá)到限流閾值，如果未達(dá)到則遞增計(jì)數(shù)器并處理請求，否則拒絕請求。

以下是一個(gè)Python示例代碼：

import redis

class RateLimiter:
    def __init__(self, redis_host, redis_port, limit, expire_time):
        self.redis_conn = redis.Redis(host=redis_host, port=redis_port)
        self.limit = limit
        self.expire_time = expire_time

    def is_allowed(self, key):
        current_count = self.redis_conn.incr(key)
        if current_count == 1:
            self.redis_conn.expire(key, self.expire_time)
        if current_count > self.limit:
            return False
        return True

# 使用示例
limiter = RateLimiter('localhost', 6379, 10, 60) # 限流10次/分鐘
if limiter.is_allowed('user1'):
    print("Allow request")
else:
    print("Reject request")

在上面的示例中，RateLimiter類初始化時(shí)傳入了Redis的主機(jī)和端口、限流閾值和過期時(shí)間。is_allowed方法用于判斷當(dāng)前請求是否允許，通過遞增計(jì)數(shù)器并判斷是否超過限流閾值來實(shí)現(xiàn)限流功能。