From 34023517f408072a39f6bffb1e99ad726f2125e7 Mon Sep 17 00:00:00 2001
From: zhl <zhl010101@gmail.com>
Date: Wed, 16 Jun 2021 17:02:12 +0800
Subject: [PATCH] =?UTF-8?q?=E9=A2=86=E5=8F=96=E4=BC=98=E6=83=A0=E5=88=B8?=
 =?UTF-8?q?=E6=97=B6,=20=E6=9B=B4=E6=96=B0=E4=BC=98=E6=83=A0=E5=88=B8?=
 =?UTF-8?q?=E7=9A=84=E6=95=B0=E9=87=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/api/controllers/puzzle.controller.ts | 15 ++++-----------
 src/models/shop/Coupon.ts                |  4 ++++
 src/models/user/UserReward.ts            |  2 ++
 3 files changed, 10 insertions(+), 11 deletions(-)

diff --git a/src/api/controllers/puzzle.controller.ts b/src/api/controllers/puzzle.controller.ts
index 10fc409..8b41431 100644
--- a/src/api/controllers/puzzle.controller.ts
+++ b/src/api/controllers/puzzle.controller.ts
@@ -27,22 +27,15 @@ import { GameUser } from '../../models/user/GameUser'
 import { isObjectId } from '../../utils/string.util'
 import { SINGLE_HELP_COUNT } from '../../constants/BaseConst'
 import { md5 } from '../../utils/security.util'
+import { Coupon } from '../../models/shop/Coupon'
 
 class PuzzleController extends BaseController {
   @role('anon')
   @router('get /api/test')
   async test(req) {
-    // try{
-    //   console.time('ss')
-    //   const nanoid = customAlphabet('2345678abcdefghjkmnpqrstwxy', 10)
-    //   for (let i = 0; i < 100; i++) {
-    //     console.log(nanoid())
-    //   }
-    //   console.timeEnd('ss')
-    // } catch (err) {
-    // }
-    let shop = await Shop.find({}, { _id: 1 }).limit(1)
-    return shop[0]
+    const { id, count } = req.params
+    // const res = await Coupon.updateCount(id, count)
+    return {}
   }
 
   @role('anon')
diff --git a/src/models/shop/Coupon.ts b/src/models/shop/Coupon.ts
index 11cb67b..894d5ac 100644
--- a/src/models/shop/Coupon.ts
+++ b/src/models/shop/Coupon.ts
@@ -89,6 +89,10 @@ class CouponClass extends BaseModule {
     let sort = { _id: -1 }
     return { opt, sort }
   }
+
+  public static updateCount(id: string, count: number) {
+    return Coupon.updateOne({ _id: id }, { $inc: { count: count } })
+  }
 }
 
 export const Coupon = getModelForClass(CouponClass, { existingConnection: CouponClass.db })
diff --git a/src/models/user/UserReward.ts b/src/models/user/UserReward.ts
index 0beb0a8..4175dd1 100644
--- a/src/models/user/UserReward.ts
+++ b/src/models/user/UserReward.ts
@@ -89,6 +89,7 @@ class UserRewardClass extends BaseModule {
     source?: string
   }) {
     let ids: string[] = []
+
     let saveRecord = async function (rewardType: number, cCount: number) {
       let record = new UserReward({
         accountId,
@@ -120,6 +121,7 @@ class UserRewardClass extends BaseModule {
       let cdata = await Coupon.findById(itemId)
       name = cdata.name
       await UserCoupon.addOne({ accountId, shop, item: itemId, count, source, activityId, rewardId })
+      await Coupon.updateCount(itemId, count)
       url = getCouponUrl(shop, itemId)
     }