feat: qa生成

src/service/events/pushChatBill.ts

@@ -0,0 +1,47 @@
+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;
+}) => {
+  try {
+    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,
+        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);
+  }
+};