From 1adef48d6bdb6b820298670310b2196010708b40 Mon Sep 17 00:00:00 2001
From: mihir-4116 <mihirb@rudderstack.com>
Date: Wed, 13 Dec 2023 14:44:57 +0530
Subject: [PATCH] feat(hs): chunking data based on batch limit

---
 src/v0/destinations/hs/config.js |   3 +
 src/v0/destinations/hs/util.js   | 134 ++++++++++++++++---------------
 2 files changed, 74 insertions(+), 63 deletions(-)

diff --git a/src/v0/destinations/hs/config.js b/src/v0/destinations/hs/config.js
index 3e91035fac..d0236d9c97 100644
--- a/src/v0/destinations/hs/config.js
+++ b/src/v0/destinations/hs/config.js
@@ -64,6 +64,8 @@ const API_VERSION = {
   v3: 'newApi',
 };
 
+const MAX_CONTACTS_PER_REQUEST = 100;
+
 const ConfigCategory = {
   COMMON: {
     name: 'HSCommonConfig',
@@ -109,5 +111,6 @@ module.exports = {
   SEARCH_LIMIT_VALUE,
   RETL_SOURCE,
   RETL_CREATE_ASSOCIATION_OPERATION,
+  MAX_CONTACTS_PER_REQUEST,
   DESTINATION: 'HS',
 };
diff --git a/src/v0/destinations/hs/util.js b/src/v0/destinations/hs/util.js
index f83ce0b6de..32f664e9f2 100644
--- a/src/v0/destinations/hs/util.js
+++ b/src/v0/destinations/hs/util.js
@@ -1,3 +1,4 @@
+const lodash = require('lodash');
 const get = require('get-value');
 const {
   NetworkInstrumentationError,
@@ -25,6 +26,7 @@ const {
   SEARCH_LIMIT_VALUE,
   hsCommonConfigJson,
   DESTINATION,
+  MAX_CONTACTS_PER_REQUEST,
 } = require('./config');
 
 const tags = require('../../util/tags');
@@ -491,84 +493,90 @@ const getExistingData = async (inputs, destination) => {
   });
 
   values = Array.from(new Set(values));
-  const requestData = {
-    filterGroups: [
-      {
-        filters: [
-          {
-            propertyName: identifierType,
-            values,
-            operator: 'IN',
-          },
-        ],
-      },
-    ],
-    properties: [identifierType],
-    limit: SEARCH_LIMIT_VALUE,
-    after: 0,
-  };
+  const valuesChunk = lodash.chunk(values, MAX_CONTACTS_PER_REQUEST);
 
-  const requestOptions = {
-    headers: {
-      'Content-Type': JSON_MIME_TYPE,
-      Authorization: `Bearer ${Config.accessToken}`,
-    },
-  };
-  let checkAfter = 1; // variable to keep checking if we have more results
-
-  /* eslint-disable no-await-in-loop */
-
-  /* *
-   * This is needed for processing paginated response when searching hubspot.
-   * we can't avoid await in loop as response to the request contains the pagination details
-   * */
-
-  while (checkAfter) {
-    const endpoint = IDENTIFY_CRM_SEARCH_ALL_OBJECTS.replace(':objectType', objectType);
-    const endpointPath = `objects/:objectType/search`;
-
-    const url =
-      Config.authorizationType === 'newPrivateAppApi'
-        ? endpoint
-        : `${endpoint}?hapikey=${Config.apiKey}`;
-    searchResponse =
-      Config.authorizationType === 'newPrivateAppApi'
-        ? await httpPOST(url, requestData, requestOptions, {
+  // eslint-disable-next-line no-restricted-syntax
+  for(const chunk of valuesChunk){
+    const requestData = {
+      filterGroups: [
+        {
+          filters: [
+            {
+              propertyName: identifierType,
+              values: chunk,
+              operator: 'IN',
+            },
+          ],
+        },
+      ],
+      properties: [identifierType],
+      limit: SEARCH_LIMIT_VALUE,
+      after: 0,
+    };
+
+    const requestOptions = {
+      headers: {
+        'Content-Type': JSON_MIME_TYPE,
+        Authorization: `Bearer ${Config.accessToken}`,
+      },
+    };
+    let checkAfter = 1; // variable to keep checking if we have more results
+
+    /* eslint-disable no-await-in-loop */
+
+    /* *
+     * This is needed for processing paginated response when searching hubspot.
+     * we can't avoid await in loop as response to the request contains the pagination details
+     * */
+
+    while (checkAfter) {
+      const endpoint = IDENTIFY_CRM_SEARCH_ALL_OBJECTS.replace(':objectType', objectType);
+      const endpointPath = `objects/:objectType/search`;
+
+      const url =
+        Config.authorizationType === 'newPrivateAppApi'
+          ? endpoint
+          : `${endpoint}?hapikey=${Config.apiKey}`;
+      searchResponse =
+        Config.authorizationType === 'newPrivateAppApi'
+          ? await httpPOST(url, requestData, requestOptions, {
             destType: 'hs',
             feature: 'transformation',
             endpointPath,
           })
-        : await httpPOST(url, requestData, {
+          : await httpPOST(url, requestData, {
             destType: 'hs',
             feature: 'transformation',
             endpointPath,
           });
-    searchResponse = processAxiosResponse(searchResponse);
+      searchResponse = processAxiosResponse(searchResponse);
 
-    if (searchResponse.status !== 200) {
-      throw new NetworkError(
-        `rETL - Error during searching object record. ${searchResponse.response?.message}`,
-        searchResponse.status,
-        {
-          [tags.TAG_NAMES.ERROR_TYPE]: getDynamicErrorType(searchResponse.status),
-        },
-        searchResponse,
-      );
-    }
+      if (searchResponse.status !== 200) {
+        throw new NetworkError(
+          `rETL - Error during searching object record. ${searchResponse.response?.message}`,
+          searchResponse.status,
+          {
+            [tags.TAG_NAMES.ERROR_TYPE]: getDynamicErrorType(searchResponse.status),
+          },
+          searchResponse,
+        );
+      }
 
-    const after = searchResponse.response?.paging?.next?.after || 0;
+      const after = searchResponse.response?.paging?.next?.after || 0;
 
-    requestData.after = after; // assigning to the new value of after
-    checkAfter = after; // assigning to the new value if no after we assign it to 0 and no more calls will take place
+      requestData.after = after; // assigning to the new value of after
+      checkAfter = after; // assigning to the new value if no after we assign it to 0 and no more calls will take place
 
-    const results = searchResponse.response?.results;
-    if (results) {
-      updateHubspotIds = results.map((result) => {
-        const propertyValue = result.properties[identifierType];
-        return { id: result.id, property: propertyValue };
-      });
+      const results = searchResponse.response?.results;
+      if (results) {
+        updateHubspotIds = results.map((result) => {
+          const propertyValue = result.properties[identifierType];
+          return { id: result.id, property: propertyValue };
+        });
+      }
     }
   }
+
   return updateHubspotIds;
 };