From 4367fa47da74e450ee67d9a4af76eb36d82470ba Mon Sep 17 00:00:00 2001
From: SunSeekerX <sunseekerx@outlook.com>
Date: Thu, 22 Jan 2026 17:02:00 +0800
Subject: [PATCH] 1

---
 src/services/weeklyClaudeCostInitService.js | 76 +++++++++++++++++++--
 1 file changed, 70 insertions(+), 6 deletions(-)

diff --git a/src/services/weeklyClaudeCostInitService.js b/src/services/weeklyClaudeCostInitService.js
index 32b079bd..1268329f 100644
--- a/src/services/weeklyClaudeCostInitService.js
+++ b/src/services/weeklyClaudeCostInitService.js
@@ -1,6 +1,7 @@
 const redis = require('../models/redis')
 const logger = require('../utils/logger')
 const pricingService = require('./pricingService')
+const serviceRatesService = require('./serviceRatesService')
 const { isClaudeFamilyModel } = require('../utils/modelHelper')
 
 function pad2(n) {
@@ -57,7 +58,7 @@ class WeeklyClaudeCostInitService {
     }
 
     const weekString = redis.getWeekStringInTimezone()
-    const doneKey = `init:weekly_claude_cost:${weekString}:done`
+    const doneKey = `init:weekly_opus_cost:${weekString}:done`
 
     try {
       const alreadyDone = await client.get(doneKey)
@@ -69,7 +70,7 @@ class WeeklyClaudeCostInitService {
       // 尽力而为：读取失败不阻断启动回填流程。
     }
 
-    const lockKey = `lock:init:weekly_claude_cost:${weekString}`
+    const lockKey = `lock:init:weekly_opus_cost:${weekString}`
     const lockValue = `${process.pid}:${Date.now()}`
     const lockTtlMs = 15 * 60 * 1000
 
@@ -86,6 +87,42 @@ class WeeklyClaudeCostInitService {
       const keyIds = await redis.scanApiKeyIds()
       const dates = this._getCurrentWeekDatesInTimezone()
 
+      // 预加载所有 API Key 数据和全局倍率（避免循环内重复查询）
+      const keyDataCache = new Map()
+      const globalRateCache = new Map()
+      const batchSize = 500
+      for (let i = 0; i < keyIds.length; i += batchSize) {
+        const batch = keyIds.slice(i, i + batchSize)
+        const pipeline = client.pipeline()
+        for (const keyId of batch) {
+          pipeline.hgetall(`apikey:${keyId}`)
+        }
+        const results = await pipeline.exec()
+        for (let j = 0; j < batch.length; j++) {
+          const [, data] = results[j] || []
+          if (data && Object.keys(data).length > 0) {
+            keyDataCache.set(batch[j], data)
+          }
+        }
+      }
+      logger.info(`💰 预加载 ${keyDataCache.size} 个 API Key 数据`)
+
+      // 推断账户类型的辅助函数（与运行时 recordOpusCost 一致，只统计 claude-official/claude-console/ccr）
+      const OPUS_ACCOUNT_TYPES = ['claude-official', 'claude-console', 'ccr']
+      const inferAccountType = (keyData) => {
+        if (keyData?.ccrAccountId) {
+          return 'ccr'
+        }
+        if (keyData?.claudeConsoleAccountId) {
+          return 'claude-console'
+        }
+        if (keyData?.claudeAccountId) {
+          return 'claude-official'
+        }
+        // bedrock/azure/gemini 等不计入周费用
+        return null
+      }
+
       const costByKeyId = new Map()
       let scannedKeys = 0
       let matchedClaudeKeys = 0
@@ -165,19 +202,46 @@ class WeeklyClaudeCostInitService {
             }
 
             const costInfo = pricingService.calculateCost(usage, entry.model)
-            const cost = costInfo && costInfo.totalCost ? costInfo.totalCost : 0
-            if (cost <= 0) {
+            const realCost = costInfo && costInfo.totalCost ? costInfo.totalCost : 0
+            if (realCost <= 0) {
               continue
             }
 
-            costByKeyId.set(entry.keyId, (costByKeyId.get(entry.keyId) || 0) + cost)
+            // 应用倍率：全局倍率 × Key 倍率（使用缓存数据）
+            const keyData = keyDataCache.get(entry.keyId)
+            const accountType = inferAccountType(keyData)
+
+            // 与运行时 recordOpusCost 一致：只统计 claude-official/claude-console/ccr 账户
+            if (!accountType || !OPUS_ACCOUNT_TYPES.includes(accountType)) {
+              continue
+            }
+
+            const service = serviceRatesService.getService(accountType, entry.model)
+
+            // 获取全局倍率（带缓存）
+            let globalRate = globalRateCache.get(service)
+            if (globalRate === undefined) {
+              globalRate = await serviceRatesService.getServiceRate(service)
+              globalRateCache.set(service, globalRate)
+            }
+
+            // 获取 Key 倍率
+            let keyRates = {}
+            try {
+              keyRates = JSON.parse(keyData?.serviceRates || '{}')
+            } catch (e) {
+              keyRates = {}
+            }
+            const keyRate = keyRates[service] ?? 1.0
+            const ratedCost = realCost * globalRate * keyRate
+
+            costByKeyId.set(entry.keyId, (costByKeyId.get(entry.keyId) || 0) + ratedCost)
           }
         } while (cursor !== '0')
       }
 
       // 为所有 API Key 写入本周 opus:weekly key
       const ttlSeconds = 14 * 24 * 3600
-      const batchSize = 500
       for (let i = 0; i < keyIds.length; i += batchSize) {
         const batch = keyIds.slice(i, i + batchSize)
         const pipeline = client.pipeline()