fix: conflict

This commit is contained in:
archer
2023-03-24 20:18:27 +08:00
15 changed files with 321 additions and 122 deletions

View File

@@ -0,0 +1,48 @@
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();
let billId;
try {
const modelItem = modelList.find((item) => item.model === modelName);
if (!modelItem) return;
const price = modelItem.price * textLen;
try {
// 插入 Bill 记录
const res = await Bill.create({
userId,
type: 'chat',
modelName: modelItem.model,
chatId,
textLen,
price
});
billId = res._id;
// 扣费
await User.findByIdAndUpdate(userId, {
$inc: { balance: -price }
});
} catch (error) {
billId && Bill.findByIdAndDelete(billId);
}
} catch (error) {
console.log(error);
}
};