Ruby on Rails中如何追踪第三方API调用以控制每日限额？

方案对比与选型 #

1. 内存存储：直接排除 #

内存存储完全不靠谱——进程重启、多实例部署时计数直接清零，没法跨进程共享数据，根本满足不了生产环境的可靠性要求，直接pass。

2. 数据库：适合低调用量场景 #

• 优势：无需额外依赖，数据持久化，用事务可以保证计数准确。
• 劣势：每次调用都要写库，高并发下会拖慢响应速度，增加数据库负载。
• 实现思路：
  可以建两张表：一张api_call_records存每一次调用的明细（可选，用于对账），一张daily_api_limits存每日聚合计数。每次调用前执行：

  # 服务对象中的核心逻辑
  def can_call?
    today = Date.today
    limit_record = DailyApiLimit.find_or_create_by(service_name: 'third_party', date: today)
    limit_record.call_count < 100 # 假设每日限额100
  end
  
  def call_api
    ActiveRecord::Base.transaction do
      limit_record = DailyApiLimit.find_or_create_by(service_name: 'third_party', date: Date.today)
      if limit_record.call_count < 100
        limit_record.increment!(:call_count)
        # 发起第三方API请求
      else
        raise "API调用已超出每日限额"
      end
    end
  end
  

  注意用事务防止并发下的计数超量问题。

3. Redis：最优推荐 #

Redis是这个场景下的完美选择，理由如下：

• 高性能：内存操作，比数据库快几个数量级，高并发下毫无压力。

• 原子操作：INCR、Lua脚本可以保证计数的准确性，不会出现并发竞争导致的超限额。

• 自动过期：可以给每日计数的key设置过期时间，自动清理旧数据，不用手动维护。

• 实现思路：
  在服务对象里封装Redis计数逻辑，推荐用Lua脚本保证原子性（避免先查后改的竞态问题）：

  class ThirdPartyApiService
    LIMIT = 100
    REDIS = Redis.current
  
    def call_api
      today_key = "api:limit:third_party:#{Date.today}"
      # 执行Lua脚本，原子完成计数检查、递增、过期设置
      result = REDIS.eval(<<~LUA, keys: [today_key], argv: [LIMIT])
        local key = KEYS[1]
        local limit = tonumber(ARGV[1])
        local current = tonumber(redis.call('GET', key) or 0)
        if current < limit then
          redis.call('INCR', key)
          -- 设置过期时间为次日凌晨，自动清理当日数据
          local ttl = 86400 - (tonumber(os.time()) % 86400)
          redis.call('EXPIRE', key, ttl)
          return current + 1
        else
          return -1
        end
      LUA
  
      if result == -1
        raise "已超出每日API调用限额"
      end
  
      # 这里发起第三方API请求
    end
  end
  

4. 补充优化方案 #

• 双数据源备份：用Redis做实时计数，每日凌晨定时把Redis里的计数同步到数据库，做持久化备份，防止Redis数据丢失。
• 对接API响应头：如果第三方API返回剩余限额（比如X-RateLimit-Remaining），可以用这个值来校准本地计数，避免本地和API侧计数不一致的情况。

落地注意事项 #

1. 所有第三方API调用必须统一通过服务对象处理，把计数逻辑封装在基础方法里，不要散在各处。
2. 给Redis加降级逻辑：如果Redis不可用，临时切换到数据库计数，同时触发告警通知运维。
3. 加监控：记录每日调用量、剩余限额，当剩余量低于10%时触发告警，提前处理。

内容的提问来源于stack exchange，提问作者michaelosmith