72 lines
2.2 KiB
TypeScript
72 lines
2.2 KiB
TypeScript
|
import { prisma } from '@/lib/db';
|
|||
|
|
import { getEnv } from '@/lib/config';
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取指定支付渠道的每日全平台限额（0 = 不限制）。
|
|||
|
|
* 优先读 config（Zod 验证），兜底读 process.env，适配未来动态注册的新渠道。
|
|||
|
|
*/
|
|||
|
|
export function getMethodDailyLimit(paymentType: string): number {
|
|||
|
|
const env = getEnv();
|
|||
|
|
const key = `MAX_DAILY_AMOUNT_${paymentType.toUpperCase()}` as keyof typeof env;
|
|||
|
|
const val = env[key];
|
|||
|
|
if (typeof val === 'number') return val;
|
|||
|
|
|
|||
|
|
// 兜底：支持动态渠道（未在 schema 中声明的 MAX_DAILY_AMOUNT_* 变量）
|
|||
|
|
const raw = process.env[`MAX_DAILY_AMOUNT_${paymentType.toUpperCase()}`];
|
|||
|
|
if (raw !== undefined) {
|
|||
|
|
const num = Number(raw);
|
|||
|
|
return Number.isFinite(num) && num >= 0 ? num : 0;
|
|||
|
|
}
|
|||
|
|
return 0; // 默认不限制
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
export interface MethodLimitStatus {
|
|||
|
|
/** 每日限额，0 = 不限 */
|
|||
|
|
dailyLimit: number;
|
|||
|
|
/** 今日已使用金额 */
|
|||
|
|
used: number;
|
|||
|
|
/** 剩余额度，null = 不限 */
|
|||
|
|
remaining: number | null;
|
|||
|
|
/** 是否还可使用（false = 今日额度已满） */
|
|||
|
|
available: boolean;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 批量查询多个支付渠道的今日使用情况。
|
|||
|
|
* 一次 DB groupBy 完成，调用方按需传入渠道列表。
|
|||
|
|
*/
|
|||
|
|
export async function queryMethodLimits(
|
|||
|
|
paymentTypes: string[],
|
|||
|
|
): Promise<Record<string, MethodLimitStatus>> {
|
|||
|
|
const todayStart = new Date();
|
|||
|
|
todayStart.setUTCHours(0, 0, 0, 0);
|
|||
|
|
|
|||
|
|
const usageRows = await prisma.order.groupBy({
|
|||
|
|
by: ['paymentType'],
|
|||
|
|
where: {
|
|||
|
|
paymentType: { in: paymentTypes },
|
|||
|
|
status: { in: ['PAID', 'RECHARGING', 'COMPLETED'] },
|
|||
|
|
paidAt: { gte: todayStart },
|
|||
|
|
},
|
|||
|
|
_sum: { amount: true },
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
const usageMap = Object.fromEntries(
|
|||
|
|
usageRows.map((r) => [r.paymentType, Number(r._sum.amount ?? 0)]),
|
|||
|
|
);
|
|||
|
|
|
|||
|
|
const result: Record<string, MethodLimitStatus> = {};
|
|||
|
|
for (const type of paymentTypes) {
|
|||
|
|
const dailyLimit = getMethodDailyLimit(type);
|
|||
|
|
const used = usageMap[type] ?? 0;
|
|||
|
|
const remaining = dailyLimit > 0 ? Math.max(0, dailyLimit - used) : null;
|
|||
|
|
result[type] = {
|
|||
|
|
dailyLimit,
|
|||
|
|
used,
|
|||
|
|
remaining,
|
|||
|
|
available: dailyLimit === 0 || used < dailyLimit,
|
|||
|
|
};
|
|||
|
|
}
|
|||
|
|
return result;
|
|||
|
|
}
|