Menggunakan Redis untuk melaksanakan pengehadan arus teragih

Tajuk: Menggunakan Redis untuk melaksanakan pengehadan semasa yang diedarkan

Teks:

Dengan perkembangan pesat Internet, bilangan lawatan serentak ke laman web dan perkhidmatan terus meningkat Untuk melindungi kestabilan belakang-. sistem akhir, mengehadkan bilangan lawatan serentak telah menjadi satu tugas yang penting. Dalam sistem teragih, untuk memastikan status dikongsi antara berbilang contoh perkhidmatan, kami boleh menggunakan Redis sebagai alat pengehad semasa yang diedarkan.

Redis ialah sistem storan nilai kunci berprestasi tinggi dengan kelajuan baca dan tulis pantas serta sokongan struktur data yang kaya, dan digunakan secara meluas dalam sistem teragih. Di bawah ini kami akan memperkenalkan cara menggunakan Redis untuk melaksanakan pengehadan semasa yang diedarkan dan memberikan contoh kod khusus.

Pertama, kita perlu menentukan strategi pengehadan semasa. Algoritma pengehad semasa biasa termasuk algoritma baldi bocor dan algoritma baldi token. Dalam artikel ini, kami menggunakan algoritma baldi token sebagai contoh.

Prinsip algoritma baldi token adalah untuk mengedarkan token kepada setiap permintaan, dan apabila bilangan token dalam baldi token tidak mencukupi, permintaan baharu akan ditolak. Kita boleh menggunakan pembilang Redis dan set diisih untuk melaksanakan algoritma baldi token.

Berikut ialah kod sampel (ditulis dalam bahasa Python) yang menggunakan Redis untuk melaksanakan pengehadan kadar teragih:

import redis
import time


class DistributedRateLimiter:
    def __init__(self, host, port, password, limit, interval):
        self.r = redis.Redis(host=host, port=port, password=password)
        self.limit = limit
        self.interval = interval

    def limit_request(self, key):
        current_time = int(time.time() * 1000)
        self.r.zremrangebyscore(key, 0, current_time - self.interval)
        requests_count = self.r.zcard(key)
        if requests_count < self.limit:
            self.r.zadd(key, {current_time: current_time})
            return True
        return False


if __name__ == '__main__':
    limiter = DistributedRateLimiter('localhost', 6379, 'password', 100, 1000)
    for _ in range(10):
        if limiter.limit_request('api:rate_limit'):
            print('Allow request')
        else:
            print('Limit exceeded')

Dalam kod di atas, kami mencipta kelas bernama DistributedRateLimiter, yang mengandungi Logik yang berkaitan algoritma had semasa dijelaskan. Kaedah pembinaan menerima parameter sambungan Redis, ambang had semasa dan selang had semasa. DistributedRateLimiter的类，其中包含了限流算法的相关逻辑。构造方法接受Redis的连接参数、限流的阈值和限流的时间间隔。

limit_request方法用于进行限流判断，它首先清理过期的令牌，然后获取当前令牌桶中的请求数量，如果请求数量小于限制，则将当前时间添加到有序集合中，并返回允许请求的标志位。

在示例代码的主函数中，我们创建了一个DistributedRateLimiter

Atas ialah kandungan terperinci Menggunakan Redis untuk melaksanakan pengehadan arus teragih. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!