筛号余额

src/balance/balance.controller.ts

@@ -41,6 +41,16 @@ export class BalanceController {
         }
     }
 
+    @Get('/screenRecharge/:id/:amount')
+    async screenRecharge(@Req() req, @Param('id') id: string, @Param('amount') amount: number) {
+        if (req.user.roles.includes('user')) {
+            return
+        }
+        const users = await this.balanceService.screenRecharge(parseInt(id), amount)
+        await this.operationLogService.create(req, users, 'User', OperationType.INSERT, '筛号余额充值', amount)
+        return 'ScreenRecharge success!'
+    }
+
     @Post('/records')
     async findRecordsByUserId(@Req() req, @Body() page: PageRequest<BalanceRecord>) {
         if (req.user.roles.includes('user')) {

src/balance/balance.service.ts

@@ -182,6 +182,14 @@ export class BalanceService {
         }
     }
 
+    async screenRecharge(userId: number, amount: number) {
+        const users = await this.usersService.findById(userId)
+        const screenBalance = new Decimal(users.screenBalance)
+        users.screenBalance = screenBalance.plus(amount).toNumber()
+
+        return await this.userRepository.save(users)
+    }
+
     async changeScreenBalance(userId: number, amount: number) {
         const amountDecimal = new Decimal(amount)
         const users = await this.usersService.findById(userId)
@@ -239,7 +247,7 @@ export class BalanceService {
         }
     }
 
-    async feeRefund(userId: number, amount: number, taskId: number){
+    async feeRefund(userId: number, amount: number, taskId: number) {
         try {
             const user = await this.usersService.findById(userId)
             // 获取余额