42 lines
775 B
TypeScript
42 lines
775 B
TypeScript
import { connectToDatabase, Bill, User } from '../mongo';
|
|
import { ModelList } from '@/constants/model';
|
|
|
|
export const pushBill = async ({
|
|
modelName,
|
|
userId,
|
|
chatId,
|
|
textLen
|
|
}: {
|
|
modelName: string;
|
|
userId: string;
|
|
chatId: string;
|
|
textLen: number;
|
|
}) => {
|
|
await connectToDatabase();
|
|
|
|
const modelItem = ModelList.find((item) => item.model === modelName);
|
|
|
|
if (!modelItem) return;
|
|
|
|
const price = modelItem.price * textLen;
|
|
|
|
let billId;
|
|
try {
|
|
// 插入 Bill 记录
|
|
const res = await Bill.create({
|
|
userId,
|
|
chatId,
|
|
textLen,
|
|
price
|
|
});
|
|
billId = res._id;
|
|
|
|
// 扣费
|
|
await User.findByIdAndUpdate(userId, {
|
|
$inc: { balance: -price }
|
|
});
|
|
} catch (error) {
|
|
Bill.findByIdAndDelete(billId);
|
|
}
|
|
};
|