ChatGPT API充值全流程技术解析：从支付接口到配额管理

1次阅读

没有评论

共计 2468 个字符，预计需要花费 7 分钟才能阅读完成。

在自建 AI 服务时，开发者常常会遇到账户充值系统的复杂性，这主要体现在以下几个方面：

多币种结算：用户可能来自不同国家，支付时需要支持多种货币。这不仅涉及汇率转换，还需要考虑不同支付网关的手续费差异。
配额实时生效：用户充值后，配额需要立即生效，这对系统的实时性要求非常高。
支付安全：支付过程中需要防范各种安全风险，如 CSRF 攻击、中间人攻击等。
高并发处理：在促销活动期间，系统可能面临大量并发支付请求，如何保证系统稳定性和数据一致性是一个挑战。

不同的支付方案在 API 设计上有显著差异：

Stripe：
支持全球多种支付方式（信用卡、Apple Pay 等）。
提供了丰富的 Webhook 事件，如 payment_intent.succeeded、charge.refunded 等。
支持复杂的订阅计费模式。
Alipay：
主要面向中国市场，支持支付宝余额、花呗等本地化支付方式。
使用 RSA 签名验证回调请求。
提供了简洁的即时到账接口。
OpenAI 官方接口：
设计上更注重开发者友好性，提供了清晰的配额管理接口。
使用 HMAC-SHA256 签名验证 Webhook 请求。
支持实时查询账户余额和使用情况。

以下是一个使用 Python 实现的支付回调验证逻辑，包含 HMAC 签名校验：

from typing import Optional
import hmac
import hashlib

def verify_webhook_signature(
    payload: bytes,
    signature_header: Optional[str],
    secret_key: str
) -> bool:
    """
    验证 Webhook 请求的 HMAC 签名
    :param payload: 原始请求体
    :param signature_header: 请求头中的签名
    :param secret_key: 用于签名的密钥
    :return: 签名是否有效
    """
    if not signature_header:
        return False

    try:
        expected_signature = hmac.new(secret_key.encode('utf-8'),
            payload,
            hashlib.sha256
        ).hexdigest()

        return hmac.compare_digest(expected_signature, signature_header)
    except Exception as e:
        print(f"签名验证失败: {e}")
        return False

配额管理需要保证原子性操作，以下是使用 Redis+Lua 实现的示例：

-- quota_management.lua
local user_key = KEYS[1]
local quota_to_add = tonumber(ARGV[1])
local max_quota = tonumber(ARGV[2])

-- 获取当前配额
local current_quota = tonumber(redis.call('GET', user_key)) or 0

-- 检查是否超过最大配额
if current_quota + quota_to_add > max_quota then
    return {err = "quota limit exceeded"}
end

-- 更新配额
redis.call('INCRBY', user_key, quota_to_add)
return {ok = "quota updated successfully"}

为了防止重复处理支付结果，可以使用以下数据库表设计：

CREATE TABLE payment_notifications (
    id BIGSERIAL PRIMARY KEY,
    notification_id VARCHAR(255) UNIQUE NOT NULL,
    payment_id VARCHAR(255) NOT NULL,
    status VARCHAR(50) NOT NULL,
    amount DECIMAL(15, 2) NOT NULL,
    currency VARCHAR(3) NOT NULL,
    processed_at TIMESTAMP WITH TIME ZONE,
    created_at TIMESTAMP WITH TIME ZONE DEFAULT NOW());

在处理跨国交易时，汇率转换需要注意以下几点：