Pythonでのユーザーのレート制限チェックをサポートするデータ構造を定義するプログラム

有効期限で構築でき、ユーザーIDとタイムスタンプを受け取る関数をサポートするデータ構造を開発するとします。これにより、指定されたタイムスタンプで指定されたuser_idを持つユーザーが、要求が失敗するかどうかがチェックされます。ユーザーが指定された有効期限より前にリクエストが成功した場合にのみ失敗します。

したがって、入力がexpire =6のような場合は、オブジェクトobjを作成し、関数obj.limit（0,10）、obj.limit（0,16）、obj.limit（0,17）、およびobj.limit（ 1,20）の場合、出力はそれぞれFalse、False、True、Falseになります。これは、ユーザー0の場合、最初はリクエストがなかったためFalseであり、時間16で最後のリクエスト10よりも有効期限6を超えないためです。ただし、17の場合はtrueであり、最後のリクエストの場合はユーザー1の場合であるため、最初のリクエストはfalseです。

これを解決するには、次の手順に従います-

コンストラクターを定義します。これには有効期限がかかります

• lastCall：=デフォルト値が-1の辞書を作成します
• 関数limit（）を定義します。これにはuid、タイムスタンプが必要です
• last：=lastCall [uid]
• lastが-1または（last + Expire）<=タイムスタンプと同じ場合、
  • lastCall [uid]：=タイムスタンプ
  • Falseを返す
• Trueを返す

例

理解を深めるために、次の実装を見てみましょう-

from collections import defaultdict
class RateLimit:
   def __init__(self, expire):
      self.expire = expire
      self.lastCall = defaultdict(lambda: -1)
   def limit(self, uid, timestamp):
      last = self.lastCall[uid]
      if last == -1 or last + self.expire <= timestamp:
         self.lastCall[uid] = timestamp
         return False
      return True

expire = 6
obj = RateLimit(expire)
print(obj.limit(0,10))
print(obj.limit(0,16))
print(obj.limit(0,17))
print(obj.limit(1,20))

入力

RateLimit(6)
obj.limit(0,10)
obj.limit(0,16)
obj.limit(0,17)
obj.limit(1,20)

出力

False
False
True
False