From 05f1ff2398eb1f8e708a8fa739a39d2edbce56eb Mon Sep 17 00:00:00 2001
From: zhl <zhl010101@gmail.com>
Date: Fri, 22 Jan 2021 16:19:38 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E8=8E=B7=E5=8F=96=E5=8D=A1?=
 =?UTF-8?q?=E7=BB=84=E9=9A=8F=E4=BB=8E=E7=9A=84=E9=80=BB=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/rooms/commands/SelectHeroCommand.ts | 14 ++++----------
 1 file changed, 4 insertions(+), 10 deletions(-)

diff --git a/src/rooms/commands/SelectHeroCommand.ts b/src/rooms/commands/SelectHeroCommand.ts
index 729c8c9..a75e650 100644
--- a/src/rooms/commands/SelectHeroCommand.ts
+++ b/src/rooms/commands/SelectHeroCommand.ts
@@ -74,18 +74,12 @@ export class SelectHeroCommand extends Command<CardGameState, { client: Client,
             }
         }
         for (let unitId of units) {
-            let weight = 0;
-            let effectId;
-            for (let [,eff] of effectMap) {
-                if (eff.stageunit_id == unitId) {
-                    weight = eff.weight;
-                    effectId = eff.id;
-                    break;
-                }
-            }
-            if (!effectId) {
+            const eff = effectMap.get(unitId)
+            if (!eff) {
                 error(`未找到效果卡的配置: ${unitId}`);
             }
+            let weight = eff.weight;
+            let effectId = eff.id;
             player.unitCfgs.set(effectId+'', weight);
         }
         this.room.battleMan.updatePlayer(player.id, player);