fix: 处理新玩家

lovezhangchuangxin · Nov 10, 2024 · 1c47c0a · 1c47c0a
1 parent 283bb33
commit 1c47c0a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/packages/backend/src/service/user/index.ts b/packages/backend/src/service/user/index.ts
@@ -50,6 +50,7 @@ export default class UserService {
     }
     const { password: _, ...safeUserInfo } = userInfo
     data.user.push(userInfo)
+    GameDB.getDB().processUserData(data.user, data.userData, true)
 
     // 生成 jwt
     const token = jwt.sign(

diff --git a/packages/db/src/gamedb.ts b/packages/db/src/gamedb.ts
@@ -42,15 +42,19 @@ export class GameDB extends DB {
     this.set('game', JSON.stringify(data))
   }
 
-  processUserData(users: UserModel[], userDataMap: UseDataMap) {
+  processUserData(
+    users: UserModel[],
+    userDataMap: UseDataMap,
+    newPlayer = false,
+  ) {
     users.forEach(({ id }) => {
       const userData = userDataMap[id] || {}
       if (!userData?.structure) {
         userData.structure = {
           energyStorage: new EnergyStorage({ level: 1, store: 2000 }),
           metalStorage: new MetalStorage({ level: 1, store: 2000 }),
         }
-      } else {
+      } else if (!newPlayer) {
         Object.keys(userData.structure).forEach((key) => {
           let structure =
             userData.structure[key as keyof typeof userData.structure]