fix: added gauge metric

rudderlabs · Dec 6, 2024 · c333688 · c333688
1 parent 35db010
commit c333688
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 12 deletions.
diff --git a/src/util/prometheus.js b/src/util/prometheus.js
@@ -520,6 +520,18 @@ class Prometheus {
         type: 'counter',
         labelNames: ['destination_id'],
       },
+      {
+        name: 'braze_batch_subscription_size',
+        help: 'braze_batch_subscription_size',
+        type: 'gauge',
+        labelNames: ['destination_id'],
+      },
+      {
+        name: 'braze_batch_subscription_combined_size',
+        help: 'braze_batch_subscription_combined_size',
+        type: 'gauge',
+        labelNames: ['destination_id'],
+      },
       {
         name: 'hs_batch_size',
         help: 'hs_batch_size',

diff --git a/src/v0/destinations/braze/util.js b/src/v0/destinations/braze/util.js
@@ -418,13 +418,19 @@ function prepareGroupAndAliasBatch(arrayChunks, responseArray, destination, type
       response.endpoint = getSubscriptionGroupEndPoint(getEndpointFromConfig(destination));
       const subscription_groups = chunk;
       // maketool transformed event
-      if (destination.ID == '2oRdG2shHukATE89DiUovc8wZ3P') {
-        logger.info(`braze subscription chunk ${JSON.stringify(subscription_groups)}`);
-      }
+      logger.info(`braze subscription chunk ${JSON.stringify(subscription_groups)}`);
+
+      stats.gauge('braze_batch_subscription_size', subscription_groups.length, {
+        destination_id: destination.ID,
+      });
 
       // Deduplicate the subscription groups before constructing the response body
       const deduplicatedSubscriptionGroups = combineSubscriptionGroups(subscription_groups);
 
+      stats.gauge('braze_batch_subscription_combined_size', deduplicatedSubscriptionGroups.length, {
+        destination_id: destination.ID,
+      });
+
       response.body.JSON = removeUndefinedAndNullValues({
         subscription_groups: deduplicatedSubscriptionGroups,
       });
@@ -534,10 +540,7 @@ const processBatch = (transformedEvents) => {
   prepareGroupAndAliasBatch(mergeUsersArrayChunks, responseArray, destination, 'merge');
 
   if (successMetadata.length > 0) {
-    // maketool transformed event
-    if (destination.ID == '2oRdG2shHukATE89DiUovc8wZ3P') {
-      logger.info(`Response 1 batchRequest ${JSON.stringify(responseArray)}`);
-    }
+    console.log(`Response 1 batchRequest ${JSON.stringify(responseArray)}`);
     finalResponse.push({
       batchedRequest: responseArray,
       metadata: successMetadata,
@@ -554,11 +557,6 @@ const processBatch = (transformedEvents) => {
     finalResponse.push(...filteredResponses);
   }
 
-  // maketool transformed event
-  if (destination.ID == '2oRdG2shHukATE89DiUovc8wZ3P') {
-    logger.info(`final Response : ${JSON.stringify(finalResponse)}`);
-  }
-
   return finalResponse;
 };