From f2d37c30a5ed56242da27b4a207ca7bb482f0745 Mon Sep 17 00:00:00 2001
From: archer <545436317@qq.com>
Date: Mon, 19 Jun 2023 17:28:25 +0800
Subject: [PATCH] feat: baidu statistic

---
 client/public/js/baidutongji.js       |  8 ++++
 client/src/pages/_app.tsx             |  1 +
 client/src/pages/api/chat/saveChat.ts | 62 ++++++++++++++++-----------
 3 files changed, 47 insertions(+), 24 deletions(-)
 create mode 100644 client/public/js/baidutongji.js

diff --git a/client/public/js/baidutongji.js b/client/public/js/baidutongji.js
new file mode 100644
index 000000000..632c5ee9c
--- /dev/null
+++ b/client/public/js/baidutongji.js
@@ -0,0 +1,8 @@
+var _hmt = _hmt || [];
+
+(function () {
+  const hm = document.createElement('script');
+  hm.src = 'https://hm.baidu.com/hm.js?a5357e9dab086658bac0b6faf148882e';
+  const s = document.getElementsByTagName('script')[0];
+  s.parentNode.insertBefore(hm, s);
+})();
diff --git a/client/src/pages/_app.tsx b/client/src/pages/_app.tsx
index 94672a09f..42bdefe87 100644
--- a/client/src/pages/_app.tsx
+++ b/client/src/pages/_app.tsx
@@ -52,6 +52,7 @@ function App({ Component, pageProps }: AppProps) {
       <Script src="/js/qrcode.min.js" strategy="afterInteractive"></Script>
       <Script src="/js/pdf.js" strategy="afterInteractive"></Script>
       <Script src="/js/html2pdf.bundle.min.js" strategy="afterInteractive"></Script>
+      <Script src="/js/baidutongji.js" strategy="afterInteractive"></Script>
       {googleVerKey && (
         <>
           <Script
diff --git a/client/src/pages/api/chat/saveChat.ts b/client/src/pages/api/chat/saveChat.ts
index 7a78cc6f0..238befb9e 100644
--- a/client/src/pages/api/chat/saveChat.ts
+++ b/client/src/pages/api/chat/saveChat.ts
@@ -60,34 +60,48 @@ export async function saveChat({
   }));
 
   if (String(model.userId) === userId) {
-    Model.findByIdAndUpdate(modelId, {
+    await Model.findByIdAndUpdate(modelId, {
       updateTime: new Date()
     });
   }
 
-  const response = await (chatId
-    ? Chat.findByIdAndUpdate(chatId, {
-        $push: {
-          content: {
-            $each: content
-          }
-        },
-        title: content[0].value.slice(0, 20),
-        latestChat: content[1].value,
-        updateTime: new Date()
-      }).then(() => ({
-        newChatId: ''
-      }))
-    : Chat.create({
-        _id: newChatId,
-        userId,
-        modelId,
-        content,
-        title: content[0].value.slice(0, 20),
-        latestChat: content[1].value
-      }).then((res) => ({
-        newChatId: String(res._id)
-      })));
+  const [response] = await Promise.all([
+    ...(chatId
+      ? [
+          Chat.findByIdAndUpdate(chatId, {
+            $push: {
+              content: {
+                $each: content
+              }
+            },
+            title: content[0].value.slice(0, 20),
+            latestChat: content[1].value,
+            updateTime: new Date()
+          }).then(() => ({
+            newChatId: ''
+          }))
+        ]
+      : [
+          Chat.create({
+            _id: newChatId,
+            userId,
+            modelId,
+            content,
+            title: content[0].value.slice(0, 20),
+            latestChat: content[1].value
+          }).then((res) => ({
+            newChatId: String(res._id)
+          }))
+        ]),
+    // update model
+    ...(String(model.userId) === userId
+      ? [
+          Model.findByIdAndUpdate(modelId, {
+            updateTime: new Date()
+          })
+        ]
+      : [])
+  ]);
 
   return {
     ...response