Added job scheduling for updating db

Author: Ajink7

models/contest.js

@@ -3,10 +3,11 @@ const Schema = mongoose.Schema
 
 const ContestRankingsSchema = new Schema({
     _id: String,
+    startTime: Date,
+    endTime: Date,
     contest_id: Number,
     lastUpdated:{
         type: Date,
-        default: Date.now()
     },
     rankings : [{
         _id: String,

package-lock.json

@@ -16,7 +16,8 @@
     "express": "^4.17.1",
     "express-ejs-layouts": "^2.5.0",
     "mongoose": "^5.12.10",
-    "node-fetch": "^2.6.1"
+    "node-fetch": "^2.6.1",
+    "node-schedule": "^2.0.0"
   },
   "_id": "leetcode-rating-predictor@1.0.0",
   "devDependencies": {

package.json

@@ -2,12 +2,12 @@ const axios = require('axios');
 const fetch = require('node-fetch')
 
 const Contest = require('../models/contest')
-let halfHour = 1000//*60*30
+let halfHour = 1000*60*30
 
 const getContestRankings = async function(contestSlug) {
 
     let contest = await Contest.findById(contestSlug)
-    if(contest === null || Date.now() - contest.lastUpdated >= halfHour){
+    if(contest===null || contest.rankings.length===0){
 
         try {
             rankings = []
@@ -32,12 +32,22 @@ const getContestRankings = async function(contestSlug) {
                 lastUpdated: Date.now(),
                 rankings: rankings
             })
-            if(contest===null){
+            if(contest===null ){
                 await newContest.save()
+                console.log(`Created contest ${contestSlug}`)
+
             }
             else{
-                await Contest.findByIdAndUpdate(contestSlug, newContest)
-                console.log(`Updated contest ${contestSlug}`)
+                let updatedContest = new Contest({
+                    _id: contestSlug,
+                    contest_id: contest_id,
+                    lastUpdated: Date.now(),
+                    rankings: rankings,
+                    startTime: contest.startTime,
+                    endTime: contest.endTime,
+                })
+                await Contest.findByIdAndUpdate(contestSlug, updatedContest)
+                console.log(`Updated Rankings in  contest ${contestSlug}`)
             }
         } 
         catch (error) {
@@ -49,5 +59,67 @@ const getContestRankings = async function(contestSlug) {
     }
 
 }
-module.exports = getContestRankings
+
+
+const fetchContest = async () => {
+
+    try {
+        let res = await fetch("https://leetcode.com/graphql", {
+        "headers": {
+            "accept": "*/*",
+            "accept-language": "en-GB,en-US;q=0.9,en;q=0.8",
+            "content-type": "application/json",
+            "sec-ch-ua": "\" Not A;Brand\";v=\"99\", \"Chromium\";v=\"90\", \"Google Chrome\";v=\"90\"",
+            "sec-ch-ua-mobile": "?0",
+            "sec-fetch-dest": "empty",
+            "sec-fetch-mode": "cors",
+            "sec-fetch-site": "same-origin",
+            "x-newrelic-id": "UAQDVFVRGwEAXVlbBAg="
+            
+        },
+        "referrer": "https://leetcode.com/contest/",
+        "referrerPolicy": "strict-origin-when-cross-origin",
+        "body": "{\"operationName\":null,\"variables\":{},\"query\":\"{\\n  brightTitle\\n  currentTimestamp\\n  allContests {\\n    containsPremium\\n    title\\n    cardImg\\n    titleSlug\\n    description\\n    startTime\\n    duration\\n    originStartTime\\n    isVirtual\\n    company {\\n      watermark\\n      __typename\\n    }\\n    __typename\\n  }\\n}\\n\"}",
+        "method": "POST",
+        "mode": "cors"
+        });
+        res = await res.json()
+        //console.log(res.data.allContests[0])
+        let contestSlug = res.data.allContests[0].titleSlug
+        let startTime = res.data.allContests[0].startTime
+        let endTime = startTime + res.data.allContests[0].duration
+        for(let i=0;i<res.data.allContests.length;i++)
+        {
+            let contest = res.data.allContests[i];
+            let isfound = await Contest.findById(contest.titleSlug)
+            if(isfound){
+                break;
+            }
+            let newContest = new Contest({
+                _id: contest.titleSlug,
+                startTime: contest.startTime,
+                endTime: startTime + contest.duration*1000,
+                lastUpdated: Date.now(),
+            })
+            await newContest.save()
+        }
+        return res.data.allContests
+    }
+    catch(error){
+        console.log(error)
+    }
+
+
+
+}
+
+
+
+
+
+
+
+
+module.exports.getContestRankings = getContestRankings
+module.exports.fetchContest = fetchContest
 

services/contests.js

@@ -0,0 +1,19 @@
+const {getContestRankings , fetchContest} = require('../services/contests')
+const schedule = require('node-schedule');
+
+//Variables For testing
+const jobScheduleTime = {hour: 00,minute: 00}
+const contestScheduleCall = { start: Date.now(), end: Date.now() + 1000*60*60, rule: '*/5 * * * *' }
+
+
+const job = schedule.scheduleJob({hour: 00, minute: 00,},async function(){
+    let contestList = await fetchContest()
+    for(let i=0;i<contestList.length;i++){
+        if(Date.now() + 1000*60*60*24 > contestList[i].startTime && Date.now()<contestList[i].startTime){
+            const event = schedule.scheduleJob({ start: contestList[todaysContest].startTime, end: contestList[todaysContest].endTime, rule: '*/5 * * * *' }, 
+            async function(){
+                getContestRankings(contestList[i].titleSlug)
+            });
+        }
+    }
+});