cita
283c052fa5
- Implemented `normalizeMigrationInput` to standardize database migration input for sqlite, mysql, and postgres. - Added `maskConnectionString` to obscure sensitive credentials in connection strings. - Created comprehensive tests for migration input normalization and connection string masking. - Established database schema creation logic for multiple dialects (sqlite, mysql, postgres). - Developed functions for managing database transactions, including begin, commit, and rollback. - Implemented data insertion logic from backup snapshots into the target database.
60 lines
1.8 KiB
TypeScript
60 lines
1.8 KiB
TypeScript
import { db, schema } from '../db/index.js';
|
|
import { eq } from 'drizzle-orm';
|
|
import { sendNotification } from './notifyService.js';
|
|
import { setAccountRuntimeHealth } from './accountHealthService.js';
|
|
import { appendSessionTokenRebindHint } from './alertRules.js';
|
|
|
|
export async function reportTokenExpired(params: {
|
|
accountId: number;
|
|
username?: string | null;
|
|
siteName?: string | null;
|
|
detail?: string;
|
|
}) {
|
|
const accountLabel = params.username || `ID:${params.accountId}`;
|
|
const siteLabel = params.siteName || 'unknown-site';
|
|
const detailText = params.detail ? appendSessionTokenRebindHint(params.detail) : '';
|
|
const detail = detailText ? ` (${detailText})` : '';
|
|
|
|
await db.insert(schema.events).values({
|
|
type: 'token',
|
|
title: 'Token 已失效',
|
|
message: `${accountLabel} @ ${siteLabel} 的 Token 无效或已过期${detail}`,
|
|
level: 'error',
|
|
relatedId: params.accountId,
|
|
relatedType: 'account',
|
|
}).run();
|
|
|
|
await db.update(schema.accounts).set({
|
|
status: 'expired',
|
|
updatedAt: new Date().toISOString(),
|
|
}).where(eq(schema.accounts.id, params.accountId)).run();
|
|
|
|
setAccountRuntimeHealth(params.accountId, {
|
|
state: 'unhealthy',
|
|
reason: detailText ? `访问令牌失效：${detailText}` : '访问令牌失效',
|
|
source: 'auth',
|
|
});
|
|
|
|
await sendNotification(
|
|
'Token 已失效',
|
|
`${accountLabel} @ ${siteLabel} 的 Token 无效或已过期${detail}`,
|
|
'error',
|
|
);
|
|
}
|
|
|
|
export async function reportProxyAllFailed(params: { model: string; reason: string }) {
|
|
await db.insert(schema.events).values({
|
|
type: 'proxy',
|
|
title: '代理全部失败',
|
|
message: `模型=${params.model}, 原因=${params.reason}`,
|
|
level: 'error',
|
|
relatedType: 'route',
|
|
}).run();
|
|
|
|
await sendNotification(
|
|
'代理全部失败',
|
|
`模型=${params.model}, 原因=${params.reason}`,
|
|
'error',
|
|
);
|
|
}
|