如何利用 Redis 實(shí)現(xiàn)接口頻次限制

介紹：

我們可以利用 redis 過期Key來實(shí)現(xiàn)接口的頻次限制?？梢宰远x一些訪問的(速度)限制條件來把那些觸發(fā)限制的請(qǐng)求拒之門外.一般常用來進(jìn)行對(duì)爬蟲的限制.

下面就利用 redis 來實(shí)現(xiàn)了一個(gè)簡單的案例：

裝飾器實(shí)現(xiàn)

def frequency_limit(f):
  @wraps(f)
  def frequency_function(*args, **kwargs):
    if 'csrf_token' in session:
      token = session.get("csrf_token")
      url_ = request.url_rule
      redis_key = token + str(url_)
      conn = redis.StrictRedis(host="127.0.0.1", port="6379", password="123456", db=0)
      clicks = conn.get(redis_key)
      if not clicks:
        conn.set(redis_key, 1)
        conn.expire(redis_key, 60)
      else:
        if int(clicks) >= 5:
          return jsonify({'code': 500, 'status': 0, 'message': "您的訪問頻率太快，請(qǐng)稍后再試", 'data': [],
                  'token': token})
        overdue = 1 if conn.ttl(redis_key) = 0 else conn.ttl(redis_key)
        conn.set(redis_key, int(clicks) + 1)
        conn.expire(redis_key, overdue)
    return f(*args, **kwargs)

  return frequency_function

注：在使用 redis Key過期的時(shí)候需要注意，在設(shè)置了過期時(shí)間后，再次改變 Key 的 Value 值時(shí)，之前設(shè)置的過期時(shí)間會(huì)失效。

解決辦法：

1）在修改 Value 值的時(shí)候，查一下過期時(shí)間還有多少 ttl 在修改值的時(shí)候把過期時(shí)間重新賦值回去（本文用的就是此方法）

2）redis 中設(shè)置了過期時(shí)間，如果 list 結(jié)構(gòu)中添加一個(gè)數(shù)據(jù)或者改變 hset 數(shù)據(jù)的一個(gè)字段是不會(huì)清除超時(shí)時(shí)間的；

官方網(wǎng)站看了一下expire的說明：
這樣解釋的：

The timeout will only be cleared by commands that delete or overwrite the contents of the key, including DEL, SET, GETSET and all the *STORE commands. This means that all the operations that conceptually alter the value stored at the key without replacing it with a new one will leave the timeout untouched. For instance, incrementing the value of a key with INCR, pushing a new value into a list with LPUSH, or altering the field value of a hash with HSET are all operations that will leave the timeout untouched.

如果用DEL, SET, GETSET會(huì)將key對(duì)應(yīng)存儲(chǔ)的值替換成新的，命令也會(huì)清除掉超時(shí)時(shí)間；如果list結(jié)構(gòu)中添加一個(gè)數(shù)據(jù)或者改變hset數(shù)據(jù)的一個(gè)字段是不會(huì)清除超時(shí)時(shí)間的；如果想要通過set去覆蓋值那就必須重新設(shè)置expire。

到此這篇關(guān)于如何利用 Redis 實(shí)現(xiàn)接口頻次限制的文章就介紹到這了,更多相關(guān)Redis 實(shí)現(xiàn)接口頻次限制內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！