feat(server): support charge reward (#1261)

Author: 0fatal

server/src/account/account.controller.ts

@@ -11,7 +11,11 @@ import {
 } from '@nestjs/common'
 import { ApiBearerAuth, ApiOperation, ApiTags } from '@nestjs/swagger'
 import { IRequest, IResponse } from 'src/utils/interface'
-import { ApiResponseObject, ResponseUtil } from 'src/utils/response'
+import {
+  ApiResponseArray,
+  ApiResponseObject,
+  ResponseUtil,
+} from 'src/utils/response'
 import { AccountService } from './account.service'
 import {
   CreateChargeOrderDto,
@@ -35,6 +39,7 @@ import { SystemDatabase } from 'src/system-database'
 import { Account } from './entities/account'
 import { AccountTransaction } from './entities/account-transaction'
 import { JwtAuthGuard } from 'src/authentication/jwt.auth.guard'
+import { AccountChargeReward } from './entities/account-charge-reward'
 
 @ApiTags('Account')
 @Controller('accounts')
@@ -112,6 +117,18 @@ export class AccountController {
     })
   }
 
+  /**
+   * Get charge reward list
+   */
+  @ApiOperation({ summary: 'Get charge reward list' })
+  @ApiResponseArray(AccountChargeReward)
+  @UseGuards(JwtAuthGuard)
+  @Get('charge-reward')
+  async getChargeRewardList() {
+    const rewards = await this.accountService.findAllChargeRewards()
+    return ResponseUtil.ok(rewards)
+  }
+
   /**
    * WeChat payment notify
    */
@@ -178,12 +195,23 @@ export class AccountController {
           return
         }
 
-        // get & update account balance
+        // get max reward
+        const reward = await db
+          .collection<AccountChargeReward>('AccountChargeReward')
+          .findOne(
+            {
+              amount: { $lte: order.amount },
+            },
+            { sort: { amount: -1 } },
+          )
+
+        const realAmount = reward ? reward.amount + order.amount : order.amount
+
         const ret = await db
           .collection<Account>('Account')
           .findOneAndUpdate(
             { _id: order.accountId },
-            { $inc: { balance: order.amount } },
+            { $inc: { balance: realAmount } },
             { session, returnDocument: 'after' },
           )
 
@@ -194,6 +222,7 @@ export class AccountController {
           {
             accountId: order.accountId,
             amount: order.amount,
+            reward: reward?.reward,
             balance: ret.value.balance,
             message: 'Recharge by WeChat Pay',
             orderId: order._id,

server/src/account/account.service.ts

@@ -13,6 +13,7 @@ import {
   PaymentChannelType,
 } from './entities/account-charge-order'
 import { AccountTransaction } from './entities/account-transaction'
+import { AccountChargeReward } from './entities/account-charge-reward'
 
 @Injectable()
 export class AccountService {
@@ -172,4 +173,20 @@ export class AccountService {
 
     throw new Error('Unsupported payment channel')
   }
+
+  async findAllChargeRewards() {
+    const rewards = await this.db
+      .collection<AccountChargeReward>('AccountChargeReward')
+      .find(
+        {},
+        {
+          sort: {
+            amount: 1,
+          },
+        },
+      )
+      .toArray()
+
+    return rewards
+  }
 }

server/src/account/entities/account-charge-reward.ts

@@ -0,0 +1,19 @@
+import { ApiProperty } from '@nestjs/swagger'
+import { ObjectId } from 'mongodb'
+
+export class AccountChargeReward {
+  @ApiProperty({ type: String })
+  _id?: ObjectId
+
+  @ApiProperty()
+  amount: number
+
+  @ApiProperty({})
+  reward: number
+
+  @ApiProperty()
+  createdAt: Date
+
+  @ApiProperty()
+  updatedAt: Date
+}

server/src/account/entities/account-transaction.ts

@@ -14,6 +14,9 @@ export class AccountTransaction {
   @ApiProperty()
   balance: number
 
+  @ApiProperty()
+  reward?: number
+
   @ApiProperty()
   message: string