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