feat: 查询商户数量

Author: kuangshp

src/api/tenant/tenant.module.ts

@@ -2,6 +2,7 @@ import { Module } from '@nestjs/common';
 import { RouterModule } from '@nestjs/core';
 import { TypeOrmModule } from '@nestjs/typeorm';
 import { ADMIN_PREFIX } from '@src/constants';
+import { AccountEntity } from '../account/entities/account.entity';
 import { AreaEntity } from '../area/entities/area.entity';
 import { TenantEntity } from './entities/tenant.entity';
 import { TenantController } from './tenant.controller';
@@ -15,7 +16,7 @@ import { TenantService } from './tenant.service';
         module: TenantModule,
       },
     ]),
-    TypeOrmModule.forFeature([TenantEntity, AreaEntity]),
+    TypeOrmModule.forFeature([TenantEntity, AreaEntity, AccountEntity]),
   ],
   providers: [TenantService],
   controllers: [TenantController],

src/api/tenant/tenant.service.ts

@@ -2,19 +2,22 @@ import { HttpException, HttpStatus, Injectable } from '@nestjs/common';
 import { TenantEntity } from './entities/tenant.entity';
 
 import { InjectRepository } from '@nestjs/typeorm';
-import { Equal, FindOperator, ILike, Repository, SelectQueryBuilder } from 'typeorm';
+import { Equal, FindOperator, ILike, In, Repository, SelectQueryBuilder } from 'typeorm';
 import { TenantDto } from './dto/tenant.dto';
 import { QueryTenantDto } from './dto/tenant.query';
 import { PageEnum, StatusEnum } from '@src/enums';
 import { TenantPageVo, TenantVo } from './vo/tenant.vo';
 import { AreaEntity } from '../area/entities/area.entity';
 import { mapToObj } from '@src/utils';
+import { AccountEntity } from '../account/entities/account.entity';
 
 @Injectable()
 export class TenantService {
   constructor(
     @InjectRepository(TenantEntity)
-    private readonly tenantRepository: Repository<TenantEntity>
+    private readonly tenantRepository: Repository<TenantEntity>,
+    @InjectRepository(AccountEntity)
+    private readonly accountRepository: Repository<AccountEntity>
   ) {}
 
   /**
@@ -149,8 +152,29 @@ export class TenantService {
       .createQueryBuilder('tenant')
       .where(mapToObj(query))
       .getCount();
+    // 根据商户id去查询account表数据
+    const tenantIdList = data.map((item) => item.id);
+    const accountTotalList = await this.accountRepository
+      .createQueryBuilder('account')
+      .select('account.tenantId', 'tenantId')
+      .addSelect('count(*)', 'total')
+      .where({ tenantId: In(tenantIdList) })
+      .groupBy('account.tenantId')
+      .getRawMany();
+    // 将accountTotalList -> map[tenantId] = total
+    const accountTotalMap = new Map();
+    for (const item of accountTotalList) {
+      accountTotalMap.set(item.tenantId, item.total);
+    }
+    const resultData: TenantVo[] = [];
+    for (const item of data) {
+      resultData.push({
+        ...item,
+        accountTotal: +accountTotalMap.get(item.id),
+      });
+    }
     return {
-      data,
+      data: resultData,
       total,
       pageNumber,
       pageSize,

src/api/tenant/vo/tenant.vo.ts

@@ -16,6 +16,7 @@ export class TenantVo extends QueryVo {
   readonly address!: string; // 具体地址
   readonly sort!: number; // 排序
   readonly description!: string; // 描述
+  readonly accountTotal?: number; // 账号数量
 }
 
 export class TenantPageVo extends QueryListVo {