clear pending reward

ety001 · ety001 · commit c4803e3fb91a · 2023-01-31T04:21:09.000Z
diff --git a/src/db/cache/index.js b/src/db/cache/index.js
@@ -367,6 +367,32 @@ function* getPendingClaimReward(model, username) {
     }
 }
 
+function* clearPendingRewardCache(model, username) {
+    if (env !== 'production') {
+        log('clearPendingRewardCache', { msg: 'none_production' });
+        return false;
+    }
+
+    if (!username) {
+        log('clearPendingRewardCache', { msg: 'username is empty.' });
+        return false;
+    }
+
+    const keyPrefix = model.getCachePrefix();
+    const cacheKey = `${keyPrefix}${username}`;
+
+    let t1, t2, t3;
+    try {
+        if (env === 'production') {
+            yield delAsync(cacheKey);
+        }
+        return true;
+    } catch (e) {
+        log('clearPendingRewardCache', { msg: e.message, cacheKey });
+        return false;
+    }
+}
+
 module.exports = {
     getRecordCache,
     updateRecordCache,
@@ -376,4 +402,5 @@ module.exports = {
     updateRecordCache2,
     clearRecordCache2,
     getPendingClaimReward,
+    clearPendingRewardCache,
 };
diff --git a/src/db/utils/user_utils.js b/src/db/utils/user_utils.js
@@ -1,7 +1,7 @@
 /* eslint-disable arrow-parens */
 import config from 'config';
 import models from 'db/models';
-import { updateRecordCache2 } from 'db/cache';
+import { clearPendingRewardCache } from 'db/cache';
 import { Model } from 'sequelize';
 import { log } from 'server/utils/loggers';
 
@@ -16,59 +16,24 @@ const clearPendingClaimTronReward = function*(username) {
     });
     t2 = process.uptime() * 1000;
     log('[timer] clearPendingClaimTronReward findOne', { t: t2 - t1 });
-    if (user && user.getDataValue('pending_claim_tron_reward') > 0) {
-        const pendingClaimTronReward =
-            user.getDataValue('pending_claim_tron_reward') / 1e6;
-        // transaction
+    if (user) {
         t1 = process.uptime() * 1000;
-        yield models.sequelize.transaction().then(transaction => {
-            // clear pending_claim_tron_reward
-            return user
-                .update(
-                    {
-                        pending_claim_tron_reward: 0,
-                    },
-                    {
-                        transaction,
-                    }
-                )
-                .then(() =>
-                    models.TronReward.create(
-                        {
-                            username,
-                            tron_addr: user.get('tron_addr'),
-                            block_num: 0,
-                            steem_tx_id: '',
-                            reward_vests: `${pendingClaimTronReward *
-                                vestsPerTrx} VESTS`,
-                            reward_steem: '0 STEEM',
-                            reward_sbd: '0 SBD',
-                            vests_per_steem: 0,
-                            reward_type: 1,
-                        },
-                        {
-                            transaction,
-                        }
-                    )
-                )
-                .then(() => transaction.commit())
-                .catch(err => {
-                    transaction.rollback();
-                    log('clear_pending_claim_tron_reward_error:', {
-                        msg: err.message,
-                    });
-                });
-        });
+        yield models.TronReward.update(
+            {
+                tron_addr: user.tron_addr,
+            },
+            {
+                where: {
+                    username,
+                    tron_addr: null,
+                    reward_type: 0,
+                },
+            }
+        );
         t2 = process.uptime() * 1000;
+        // clear redis cache
+        yield clearPendingRewardCache(models.TronReward, username);
         log('[timer] clearPendingClaimTronReward transaction:', { t: t2 - t1 });
-        t1 = process.uptime() * 1000;
-        yield updateRecordCache2(
-            models.TronUser,
-            models.escAttrs({ username })
-        );
-        log('[timer] clearPendingClaimTronReward updateRecordCache2:', {
-            t: process.uptime() * 1000 - t1,
-        });
     }
 };
 
