Merge pull request #1106 from Giveth/feat/addFillSnapshotAgain

Added fill snapshot round number

src/repositories/powerSnapshotRepository.ts

@@ -1,6 +1,14 @@
 import { PowerSnapshot } from '../entities/powerSnapshot';
 import { AppDataSource } from '../orm';
 
+export const findInCompletePowerSnapShots = async (): Promise<
+  PowerSnapshot[]
+> => {
+  return PowerSnapshot.createQueryBuilder()
+    .where('"roundNumber" IS NULL')
+    .getMany();
+};
+
 export const updatePowerSnapShots = async (params: {
   roundNumber: number;
   powerSnapshot: PowerSnapshot;

src/services/cronJobs/updatePowerRoundJob.ts

@@ -19,6 +19,7 @@ import {
 } from '../../repositories/previousRoundRankRepository';
 import { getNotificationAdapter } from '../../adapters/adaptersFactory';
 import { sleep } from '../../utils/utils';
+import { fillIncompletePowerSnapshots } from '../powerSnapshotServices';
 
 const cronJobTime =
   (config.get('UPDATE_POWER_ROUND_CRONJOB_EXPRESSION') as string) ||
@@ -30,6 +31,8 @@ export const runUpdatePowerRoundCronJob = () => {
     cronJobTime,
   );
   schedule(cronJobTime, async () => {
+    const fillSnapshotsRoundNumberPromise = fillIncompletePowerSnapshots();
+
     const currentRound = await getPowerRound();
     const powerRound = getRoundNumberByDate(new Date()).round - 1;
     logger.debug('runUpdatePowerRoundCronJob', {
@@ -46,6 +49,9 @@ export const runUpdatePowerRoundCronJob = () => {
       await setPowerRound(powerRound);
       oldBottomRank = await getBottomRank();
     }
+
+    await fillSnapshotsRoundNumberPromise;
+
     await Promise.all([
       refreshProjectPowerView(),
       refreshProjectFuturePowerView(),

src/services/powerSnapshotServices.ts

@@ -0,0 +1,26 @@
+import {
+  findInCompletePowerSnapShots,
+  updatePowerSnapShots,
+} from '../repositories/powerSnapshotRepository';
+import { getTimestampInSeconds } from '../utils/utils';
+import { getRoundNumberByDate } from '../utils/powerBoostingUtils';
+import { logger } from '../utils/logger';
+
+export const fillIncompletePowerSnapshots = async (): Promise<void> => {
+  const incompletePowerSnapshots = await findInCompletePowerSnapShots();
+  logger.debug(
+    'fillIncompletePowerSnapshots incompletePowerSnapshots',
+    JSON.stringify(incompletePowerSnapshots, null, 2),
+  );
+  for (const powerSnapshot of incompletePowerSnapshots) {
+    try {
+      const roundNumber = getRoundNumberByDate(powerSnapshot.time).round;
+      await updatePowerSnapShots({
+        powerSnapshot,
+        roundNumber,
+      });
+    } catch (e) {
+      logger.error('fillIncompletePowerSnapshots error', e);
+    }
+  }
+};