refactor: move init_data to standalone script (not cron endpoint)

- scripts/init_data.py: standalone CLI tool, connects directly to sage DB
- Remove HTTP endpoint (cron/init_data.dspy) - init is one-time operation
- Remove init_datamart from init.py - no longer exposed as web service
- Remove RBAC permission for init_data.dspy

Usage: cd /path/to/sage && python /path/to/sage_datamart/scripts/init_data.py
This commit is contained in:
yumoqing 2026-07-20 13:44:09 +08:00
parent 7f5c1ed372
commit 2f9a890128
4 changed files with 167 additions and 15 deletions

View File

@ -3,7 +3,7 @@ from ahserver.serverenv import ServerEnv
from appPublic.log import debug
from sqlor.dbpools import get_sor_context
from .etl import sync_call_fact, aggregate_daily_perf, aggregate_provider_cost, ensure_tables
from .init_data import run_init as _run_init
from .dashboards import (
get_today_usage, get_today_amount, get_success_rate, get_fail_count,
get_active_users, get_top_models, get_top_providers, get_top_users,
@ -186,15 +186,7 @@ async def cron_etl_provider_cost(request):
return {'status': 'ok', 'aggregated': count}
async def init_datamart(request):
"""初始化数据集市:从 llmusage + llmusage_history 导入全量历史"""
ip = request['client_ip']
if ip not in ['127.0.0.1']:
return {'error': f'IP {ip} not allowed'}
env = request._run_ns
async with get_sor_context(env, MODULE_NAME) as sor:
result = await _run_init(sor)
return result
def load_sage_datamart():
@ -217,5 +209,5 @@ def load_sage_datamart():
env.cron_etl_sync = cron_etl_sync
env.cron_etl_aggregate = cron_etl_aggregate
env.cron_etl_provider_cost = cron_etl_provider_cost
env.init_datamart = init_datamart
debug("[sage_datamart] registered all API endpoints and cron functions")

163
scripts/init_data.py Normal file
View File

@ -0,0 +1,163 @@
#!/usr/bin/env python3
"""
数据集市初始化脚本 llmusage + llmusage_history 导入全量历史数据到 dm_*
用法: cd /path/to/sage && python /path/to/sage_datamart/scripts/init_data.py
"""
import asyncio
import json
import sys
from datetime import datetime
from sqlor.dbpools import DBPools
from appPublic.jsonConfig import getConfig
from appPublic.uniqueID import getID
from appPublic.dictObject import DictObject
DB_NAME = 'sage'
def parse_usages(usages_str):
"""从 llmusage.usages JSON 中提取 token 数"""
try:
u = json.loads(usages_str) if isinstance(usages_str, str) else usages_str
if isinstance(u, dict):
return (
u.get('prompt_tokens', 0) or 0,
u.get('completion_tokens', 0) or 0,
u.get('total_tokens', 0) or 0
)
except:
pass
return 0, 0, 0
async def batch_sync_call_fact(sor, source_table, batch_size=1000):
"""
批量从 llmusage / llmusage_history 导入 dm_model_call_fact
自动去重已存在 luid 跳过
"""
offset = 0
total_inserted = 0
total_skipped = 0
while True:
sql = (
'SELECT lu.*, l.model as llm_model, l.catelogid, l.providerid, l.ownerid '
'FROM ' + source_table + ' lu '
'JOIN llm l ON l.id = lu.llmid '
'ORDER BY lu.use_time '
'LIMIT ${limit}$ OFFSET ${offset}$'
)
rows = await sor.sqlExe(sql, {'limit': batch_size, 'offset': offset})
if not rows:
break
print(f' [{source_table}] batch offset={offset} rows={len(rows)}')
for r in rows:
r = DictObject(r)
existing = await sor.sqlExe(
'SELECT id FROM dm_model_call_fact WHERE luid=${luid}$',
{'luid': r.id}
)
if existing:
total_skipped += 1
continue
prompt_tokens, completion_tokens, _ = parse_usages(r.usages)
try:
call_time = datetime.strptime(str(r.use_time)[:19], '%Y-%m-%d %H:%M:%S')
except Exception:
total_skipped += 1
continue
ns = {
'id': getID(),
'luid': r.id,
'llmid': r.llmid,
'model': r.llm_model or r.model,
'catelogid': r.catelogid,
'userid': r.userid,
'userorgid': r.userorgid,
'ownerid': r.ownerid,
'providerid': r.providerid,
'call_date': r.use_date or call_time.strftime('%Y-%m-%d'),
'call_hour': call_time.hour,
'call_time': r.use_time,
'ttft_ms': int(r.responsed_seconds * 1000) if r.responsed_seconds else None,
'ttot_ms': int(r.finish_seconds * 1000) if r.finish_seconds else None,
'prompt_tokens': prompt_tokens,
'completion_tokens': completion_tokens,
'status': r.status or 'SUCCEEDED',
'amount': r.amount or 0,
'currency': getattr(r, 'currency', 'CNY') or 'CNY',
}
await sor.C('dm_model_call_fact', ns)
total_inserted += 1
offset += batch_size
print(f' [{source_table}] done: inserted={total_inserted}, skipped={total_skipped}')
return total_inserted, total_skipped
async def batch_aggregate_history(sor):
"""按 dm_model_call_fact 中所有历史日期重算聚合表"""
from sage_datamart.etl import aggregate_daily_perf, aggregate_provider_cost
rows = await sor.sqlExe(
'SELECT DISTINCT call_date FROM dm_model_call_fact ORDER BY call_date',
{}
)
dates = [r['call_date'].strftime('%Y-%m-%d') if hasattr(r['call_date'], 'strftime') else str(r['call_date']) for r in rows]
print(f' aggregating {len(dates)} distinct dates...')
for d in dates:
await aggregate_daily_perf(sor, stat_date=d)
await aggregate_provider_cost(sor, stat_date=d)
return len(dates)
async def run_init(sor):
"""
初始化流程:
1. 建表如不存在
2. llmusage_history 批量导入历史
3. llmusage 导入当前数据
4. 重算所有日期的聚合表
"""
from sage_datamart.etl import ensure_tables
print('[1/4] Creating tables...')
await ensure_tables(sor)
print('[2/4] Importing llmusage_history...')
ins_hist, skip_hist = await batch_sync_call_fact(sor, 'llmusage_history')
print('[3/4] Importing llmusage...')
ins_curr, skip_curr = await batch_sync_call_fact(sor, 'llmusage')
print('[4/4] Aggregating historical dates...')
dates = await batch_aggregate_history(sor)
return {
'llmusage_history': {'inserted': ins_hist, 'skipped': skip_hist},
'llmusage': {'inserted': ins_curr, 'skipped': skip_curr},
'aggregated_dates': dates,
}
async def main():
config = getConfig('.')
db = DBPools(config.databases)
async with db.sqlorContext(DB_NAME) as sor:
result = await run_init(sor)
print('\n=== Initialization Complete ===')
print(json.dumps(result, indent=2))
if __name__ == '__main__':
asyncio.run(main())

View File

@ -12,7 +12,7 @@ PATHS_ANY = [
f"/{MOD}/cron/etl_sync.dspy",
f"/{MOD}/cron/etl_aggregate.dspy",
f"/{MOD}/cron/etl_provider_cost.dspy",
f"/{MOD}/cron/init_data.dspy",
]
PATHS_LOGINED = [

View File

@ -1,3 +0,0 @@
# 数据集市初始化:从 llmusage + llmusage_history 导入全量历史数据
result = await request._run_ns.init_datamart(request)
return str(result)