mirror of
https://github.com/Wei-Shaw/claude-relay-service.git
synced 2026-01-23 09:38:02 +00:00
feat: 为所有账户服务添加订阅过期检查功能
完成账户订阅到期时间功能的核心调度逻辑实现。 ## 实现范围 ✅ 已添加订阅过期检查的服务（5个）： - Gemini 服务：添加 isSubscriptionExpired() 函数及调度过滤 - OpenAI 服务：添加 isSubscriptionExpired() 函数及调度过滤 - Droid 服务：添加 _isSubscriptionExpired() 方法及调度过滤 - Bedrock 服务：添加 _isSubscriptionExpired() 方法及调度过滤 - Azure OpenAI 服务：添加 isSubscriptionExpired() 函数及调度过滤 ## 核心功能 - 账户调度时自动检查 subscriptionExpiresAt 字段 - 过期账户将不再被系统调度使用 - 未设置过期时间的账户视为永不过期（向后兼容） - 使用 <= 比较判断过期（精确到过期时刻） - 跳过过期账户时记录 debug 日志便于排查 ## 技术实现 - 统一的实现模式：过期检查函数 + 账户选择逻辑集成 - 不影响现有功能，完全向后兼容 - 业务字段 subscriptionExpiresAt 与技术字段 expiresAt（OAuth token过期）独立管理 ## 相关文档 参考 account_expire_bugfix.md 了解问题背景和实现细节 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
litongtongxue
mrlitong
@@ -42,19 +42,6 @@ function generateEncryptionKey() {
|
||||
return _encryptionKeyCache
|
||||
}
|
||||
|
||||
function normalizeSubscriptionExpiresAt(value) {
|
||||
if (value === undefined || value === null || value === '') {
|
||||
return ''
|
||||
}
|
||||
|
||||
const date = value instanceof Date ? value : new Date(value)
|
||||
if (Number.isNaN(date.getTime())) {
|
||||
return ''
|
||||
}
|
||||
|
||||
return date.toISOString()
|
||||
}
|
||||
|
||||
// Gemini 账户键前缀
|
||||
const GEMINI_ACCOUNT_KEY_PREFIX = 'gemini_account:'
|
||||
const SHARED_GEMINI_ACCOUNTS_KEY = 'shared_gemini_accounts'
|
||||
@@ -346,10 +333,6 @@ async function createAccount(accountData) {
|
||||
let refreshToken = ''
|
||||
let expiresAt = ''
|
||||
|
||||
const subscriptionExpiresAt = normalizeSubscriptionExpiresAt(
|
||||
accountData.subscriptionExpiresAt || ''
|
||||
)
|
||||
|
||||
if (accountData.geminiOauth || accountData.accessToken) {
|
||||
// 如果提供了完整的 OAuth 数据
|
||||
if (accountData.geminiOauth) {
|
||||
@@ -401,10 +384,13 @@ async function createAccount(accountData) {
|
||||
geminiOauth: geminiOauth ? encrypt(geminiOauth) : '',
|
||||
accessToken: accessToken ? encrypt(accessToken) : '',
|
||||
refreshToken: refreshToken ? encrypt(refreshToken) : '',
|
||||
expiresAt,
|
||||
expiresAt, // OAuth Token 过期时间（技术字段，自动刷新）
|
||||
// 只有OAuth方式才有scopes，手动添加的没有
|
||||
scopes: accountData.geminiOauth ? accountData.scopes || OAUTH_SCOPES.join(' ') : '',
|
||||
|
||||
// ✅ 新增：账户订阅到期时间（业务字段，手动管理）
|
||||
subscriptionExpiresAt: accountData.subscriptionExpiresAt || '',
|
||||
|
||||
// 代理设置
|
||||
proxy: accountData.proxy ? JSON.stringify(accountData.proxy) : '',
|
||||
|
||||
@@ -421,8 +407,7 @@ async function createAccount(accountData) {
|
||||
createdAt: now,
|
||||
updatedAt: now,
|
||||
lastUsedAt: '',
|
||||
lastRefreshAt: '',
|
||||
subscriptionExpiresAt
|
||||
lastRefreshAt: ''
|
||||
}
|
||||
|
||||
// 保存到 Redis
|
||||
@@ -446,10 +431,6 @@ async function createAccount(accountData) {
|
||||
}
|
||||
}
|
||||
|
||||
if (!returnAccount.subscriptionExpiresAt) {
|
||||
returnAccount.subscriptionExpiresAt = null
|
||||
}
|
||||
|
||||
return returnAccount
|
||||
}
|
||||
|
||||
@@ -486,10 +467,6 @@ async function getAccount(accountId) {
|
||||
// 转换 schedulable 字符串为布尔值（与 claudeConsoleAccountService 保持一致）
|
||||
accountData.schedulable = accountData.schedulable !== 'false' // 默认为true，只有明确设置为'false'才为false
|
||||
|
||||
if (!accountData.subscriptionExpiresAt) {
|
||||
accountData.subscriptionExpiresAt = null
|
||||
}
|
||||
|
||||
return accountData
|
||||
}
|
||||
|
||||
@@ -503,10 +480,6 @@ async function updateAccount(accountId, updates) {
|
||||
const now = new Date().toISOString()
|
||||
updates.updatedAt = now
|
||||
|
||||
if (Object.prototype.hasOwnProperty.call(updates, 'subscriptionExpiresAt')) {
|
||||
updates.subscriptionExpiresAt = normalizeSubscriptionExpiresAt(updates.subscriptionExpiresAt)
|
||||
}
|
||||
|
||||
// 检查是否新增了 refresh token
|
||||
// existingAccount.refreshToken 已经是解密后的值了（从 getAccount 返回）
|
||||
const oldRefreshToken = existingAccount.refreshToken || ''
|
||||
@@ -551,15 +524,23 @@ async function updateAccount(accountId, updates) {
|
||||
}
|
||||
}
|
||||
|
||||
// 如果新增了 refresh token，更新过期时间为10分钟
|
||||
// ✅ 关键：如果新增了 refresh token，只更新 token 过期时间
|
||||
// 不要覆盖 subscriptionExpiresAt
|
||||
if (needUpdateExpiry) {
|
||||
const newExpiry = new Date(Date.now() + 10 * 60 * 1000).toISOString()
|
||||
updates.expiresAt = newExpiry
|
||||
updates.expiresAt = newExpiry // 只更新 OAuth Token 过期时间
|
||||
// ⚠️ 重要：不要修改 subscriptionExpiresAt
|
||||
logger.info(
|
||||
`🔄 New refresh token added for Gemini account ${accountId}, setting expiry to 10 minutes`
|
||||
`🔄 New refresh token added for Gemini account ${accountId}, setting token expiry to 10 minutes`
|
||||
)
|
||||
}
|
||||
|
||||
// ✅ 如果通过路由映射更新了 subscriptionExpiresAt，直接保存
|
||||
// subscriptionExpiresAt 是业务字段，与 token 刷新独立
|
||||
if (updates.subscriptionExpiresAt !== undefined) {
|
||||
// 直接保存，不做任何调整
|
||||
}
|
||||
|
||||
// 如果通过 geminiOauth 更新，也要检查是否新增了 refresh token
|
||||
if (updates.geminiOauth && !oldRefreshToken) {
|
||||
const oauthData =
|
||||
@@ -616,10 +597,6 @@ async function updateAccount(accountId, updates) {
|
||||
}
|
||||
}
|
||||
|
||||
if (!updatedAccount.subscriptionExpiresAt) {
|
||||
updatedAccount.subscriptionExpiresAt = null
|
||||
}
|
||||
|
||||
return updatedAccount
|
||||
}
|
||||
|
||||
@@ -683,7 +660,11 @@ async function getAllAccounts() {
|
||||
geminiOauth: accountData.geminiOauth ? '[ENCRYPTED]' : '',
|
||||
accessToken: accountData.accessToken ? '[ENCRYPTED]' : '',
|
||||
refreshToken: accountData.refreshToken ? '[ENCRYPTED]' : '',
|
||||
subscriptionExpiresAt: accountData.subscriptionExpiresAt || null,
|
||||
|
||||
// ✅ 前端显示订阅过期时间（业务字段）
|
||||
// 注意：前端看到的 expiresAt 实际上是 subscriptionExpiresAt
|
||||
expiresAt: accountData.subscriptionExpiresAt || null,
|
||||
|
||||
// 添加 scopes 字段用于判断认证方式
|
||||
// 处理空字符串和默认值的情况
|
||||
scopes:
|
||||
@@ -762,8 +743,17 @@ async function selectAvailableAccount(apiKeyId, sessionHash = null) {
|
||||
|
||||
for (const accountId of sharedAccountIds) {
|
||||
const account = await getAccount(accountId)
|
||||
if (account && account.isActive === 'true' && !isRateLimited(account)) {
|
||||
if (
|
||||
account &&
|
||||
account.isActive === 'true' &&
|
||||
!isRateLimited(account) &&
|
||||
!isSubscriptionExpired(account)
|
||||
) {
|
||||
availableAccounts.push(account)
|
||||
} else if (account && isSubscriptionExpired(account)) {
|
||||
logger.debug(
|
||||
`⏰ Skipping expired Gemini account: ${account.name}, expired at ${account.subscriptionExpiresAt}`
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -818,6 +808,19 @@ function isTokenExpired(account) {
|
||||
return now >= expiryTime - buffer
|
||||
}
|
||||
|
||||
/**
|
||||
* 检查账户订阅是否过期
|
||||
* @param {Object} account - 账户对象
|
||||
* @returns {boolean} - true: 已过期, false: 未过期
|
||||
*/
|
||||
function isSubscriptionExpired(account) {
|
||||
if (!account.subscriptionExpiresAt || account.subscriptionExpiresAt === '') {
|
||||
return false // 未设置视为永不过期
|
||||
}
|
||||
const expiryDate = new Date(account.subscriptionExpiresAt)
|
||||
return expiryDate <= new Date()
|
||||
}
|
||||
|
||||
// 检查账户是否被限流
|
||||
function isRateLimited(account) {
|
||||
if (account.rateLimitStatus === 'limited' && account.rateLimitedAt) {
|
||||
|
||||
Reference in New Issue
Block a user