From 0011df9366e59c6a05e10235737c20bd012237ac Mon Sep 17 00:00:00 2001
From: CounterFire2023 <136581895+CounterFire2023@users.noreply.github.com>
Date: Thu, 18 Apr 2024 15:49:21 +0800
Subject: [PATCH] =?UTF-8?q?bug=20fix:=20nft=20holder=E7=8A=B6=E6=80=81?=
 =?UTF-8?q?=E9=94=99=E8=AF=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/controllers/nft.controller.ts | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/src/controllers/nft.controller.ts b/src/controllers/nft.controller.ts
index 8d6f73d..31ea429 100644
--- a/src/controllers/nft.controller.ts
+++ b/src/controllers/nft.controller.ts
@@ -78,10 +78,16 @@ class NftController extends BaseController {
     })
     if (user) {
       let records = await NFTHolderRecord.find({ user: user.id })
-      let recordSet = new Set(records.map(o => o.contract.toLowerCase()))
-      let groupSet = new Set(records.map(o => o.group))
+      let recordSet = new Set()
+      let groupSet = new Set()
+      for (let record of records) {
+        recordSet.add(record.contract.toLowerCase())
+        if (record.group) {
+          groupSet.add(record.group)
+        }
+      }
       for (let sub of list) {
-        list.status = recordSet.has(sub.contract.toLowerCase()) || groupSet.has(sub.group) ? 1 : 0
+        sub.status = recordSet.has(sub.contract.toLowerCase()) || groupSet.has(sub.group) ? 1 : 0
       }
     }
     return list