wangqifan 2 жил өмнө
parent
commit
e02ddb190d

+ 7 - 1
src/users/users.service.ts

@@ -21,6 +21,8 @@ import { MembershipService } from '../membership/membership.service'
 import { paginate, Pagination } from 'nestjs-typeorm-paginate'
 import { paginate, Pagination } from 'nestjs-typeorm-paginate'
 import { Role } from 'src/model/role.enum'
 import { Role } from 'src/model/role.enum'
 import { PageRequest } from '../common/dto/page-request'
 import { PageRequest } from '../common/dto/page-request'
+import { BigNumber } from 'bignumber.js'
+import { CommissionService } from '../commission/commission.service'
 
 
 @Injectable()
 @Injectable()
 export class UsersService {
 export class UsersService {
@@ -29,7 +31,8 @@ export class UsersService {
         private readonly userRepository: Repository<Users>,
         private readonly userRepository: Repository<Users>,
         private readonly hashingService: HashingService,
         private readonly hashingService: HashingService,
         private readonly smsService: SmsService,
         private readonly smsService: SmsService,
-        private readonly membershipService: MembershipService
+        private readonly membershipService: MembershipService,
+        private readonly commissionService: CommissionService
     ) {}
     ) {}
 
 
     async findAll(req: PageRequest<Users>): Promise<Pagination<Users>> {
     async findAll(req: PageRequest<Users>): Promise<Pagination<Users>> {
@@ -78,6 +81,9 @@ export class UsersService {
         user = await this.userRepository.save(user)
         user = await this.userRepository.save(user)
         if (newRegister) {
         if (newRegister) {
             await this.membershipService.trial(user.id)
             await this.membershipService.trial(user.id)
+            if(invitor) {
+            await this.commissionService.doCommission(user.id, new BigNumber(0))
+            }
         }
         }
         return user
         return user
     }
     }