feat: ✨ add date range check

giorgionocera · Apr 11, 2022 · f97dae7 · f97dae7
1 parent 7fa2339
commit f97dae7
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 2 deletions.
diff --git a/code/.env.example b/code/.env.example
@@ -5,4 +5,6 @@ MONGODB_DB_NAME=stream-analyzer
 TWITTER_API_URL=https://api.twitter.com/2/tweets/
 TWITTER_BEARER_TOKEN=
 TWITTER_HASHTAG=BitSong
-TWITTER_BITSONG_ACCOUNT=BitSongOfficial
+TWITTER_BITSONG_ACCOUNT=BitSongOfficial
+ANALYZER_START_DATE=
+ANALYZER_END_DATE=
diff --git a/code/src/tweets/twitter.service.ts b/code/src/tweets/twitter.service.ts
@@ -1,3 +1,4 @@
+import { isBetweenStartEnd } from 'src/utils';
 import { Injectable } from '@nestjs/common';
 import { ConfigService } from '@nestjs/config';
 import { TwitterApi } from 'twitter-api-v2';
@@ -14,6 +15,8 @@ export class TwitterService {
   }
 
   async getSearchStream() {
+    console.log('Starting twitter search stream...');
+
     const stream = this.twitterClient.v2.searchStream({
       expansions: ['author_id', 'entities.mentions.username'],
       'tweet.fields': [
@@ -35,7 +38,18 @@ export class TwitterService {
     });
 
     for await (const { data, includes } of stream) {
-      this.mongo.create(data, includes);
+      if (
+        isBetweenStartEnd(
+          process.env.ANALYZER_START_DATE,
+          process.env.ANALYZER_END_DATE,
+          data.created_at,
+        )
+      ) {
+        this.mongo.create(data, includes);
+      } else {
+        console.log('Out of env dates range, exit...');
+        break;
+      }
     }
   }
 }
diff --git a/code/src/utils/date.ts b/code/src/utils/date.ts
@@ -1,8 +1,19 @@
 import * as dayjs from 'dayjs';
+import * as isBetween from 'dayjs/plugin/isBetween';
+
+dayjs.extend(isBetween);
 
 export const differenceByMonths = (startDate: string, endDate?: string) => {
   const rightDate = endDate ? dayjs(endDate) : dayjs();
   const leftDate = dayjs(startDate);
 
   return rightDate.diff(leftDate, 'M', true);
 };
+
+export const isBetweenStartEnd = (
+  startDate: string,
+  endDate: string,
+  date?: string,
+) => {
+  return dayjs(date ?? new Date()).isBetween(startDate, endDate);
+};