change new pending_claim_reward reader

ety001 · ety001 · commit 234ff503e285 · 2023-01-30T04:07:31.000Z
diff --git a/src/db/cache/index.js b/src/db/cache/index.js
@@ -302,6 +302,71 @@ function* clearRecordCache2(model, conditions = {}) {
     }
 }
 
+function* getPendingClaimReward(model, username) {
+    if (!model) {
+        return null;
+    }
+    if (!username) {
+        return null;
+    }
+
+    const keyPrefix = model.getCachePrefix();
+    const cacheKey = `${keyPrefix}${username}`;
+
+    let vestsTotal = 0,
+        t1,
+        t2,
+        t3;
+    try {
+        t1 = process.uptime();
+        vestsTotal =
+            env === 'production'
+                ? yield getAsync(cacheKey)
+                : log('getPendingClaimReward', { msg: 'none_production' });
+        t2 = process.uptime();
+        log('[timer] tron_user getPendingClaimReward redis getAsync', {
+            t: (t2 - t1) * 1000,
+        });
+        if (!vestsTotal) {
+            // not hit cache
+            log('getPendingClaimReward', {
+                msg: 'not_hit_cache',
+                cacheKey,
+            });
+            const dbOptions = {
+                where: {
+                    tron_addr: null,
+                    reward_type: 0,
+                    username,
+                },
+            };
+            t1 = process.uptime();
+            vestsTotal = yield model.sum('reward_vests', dbOptions);
+            t2 = process.uptime();
+            log('[timer] tron_user getPendingClaimReward db execute', {
+                t: (t2 - t1) * 1000,
+                vestsTotal,
+            });
+            if (env === 'production') {
+                t1 = process.uptime();
+                yield setAsync(cacheKey, vestsTotal);
+                t2 = process.uptime();
+                yield expireAsync([cacheKey, EXPIRED_TIME]);
+                t3 = process.uptime();
+                log(
+                    '[timer] tron_user getPendingClaimReward redis setAsync, expireAsync:',
+                    { t1: (t2 - t1) * 1000, t2: (t3 - t2) * 1000 }
+                );
+            }
+            return vestsTotal;
+        }
+        return vestsTotal;
+    } catch (e) {
+        log('getPendingClaimReward', { msg: e.message, cacheKey });
+        return null;
+    }
+}
+
 module.exports = {
     getRecordCache,
     updateRecordCache,
@@ -310,4 +375,5 @@ module.exports = {
     getRecordCache2,
     updateRecordCache2,
     clearRecordCache2,
+    getPendingClaimReward,
 };
diff --git a/src/db/models/tron_reward.js b/src/db/models/tron_reward.js
@@ -83,5 +83,6 @@ module.exports = (sequelize, DataTypes) => {
             underscored: true,
         }
     );
+    tronReward.getCachePrefix = () => 'tron_reward_pending_';
     return tronReward;
 };
diff --git a/src/server/api/tron_reward.js b/src/server/api/tron_reward.js
@@ -7,6 +7,7 @@ import config from 'config';
 import {
     clearRecordCache2,
     getRecordCache2,
+    getPendingClaimReward,
     updateRecordCache2,
 } from 'db/cache';
 import models from 'db/models';
@@ -110,10 +111,27 @@ export default function useTronRewardApi(app) {
             }
         }
 
+        let pending_trx;
+
+        try {
+            const vestsPerTrx = config.get('tron_reward.vests_per_trx');
+            const pendingReward = yield getPendingClaimReward(
+                models.TronReward,
+                username
+            );
+            pending_trx = pendingReward / 1000000 / vestsPerTrx;
+        } catch (e) {
+            this.body = JSON.stringify({ error: e.message });
+            log('[timer] get /tron_user all', {
+                t: process.uptime() * 1000 - t1,
+            });
+            return;
+        }
+
         const result = {
             username: tronUser.username,
             tron_addr: tronUser.tron_addr,
-            pending_claim_tron_reward: tronUser.pending_claim_tron_reward,
+            pending_claim_tron_reward: `${pending_trx.toFixed(6)} TRX`,
             tip_count: tronUser.tip_count,
         };
 

Original file line number	Diff line number	Diff line change
`@@ -83,5 +83,6 @@ module.exports = (sequelize, DataTypes) => {`
`83`	`83`	`underscored: true,`
`84`	`84`	`}`
`85`	`85`	`);`
	`86`	`+ tronReward.getCachePrefix = () => 'tron_reward_pending_';`
`86`	`87`	`return tronReward;`
`87`	`88`	`};`