Add stats to measure grouping

Author: varadarajan-tw

packages/destination-actions/src/destinations/klaviyo/addProfileToList/index.ts

@@ -66,8 +66,8 @@ const action: ActionDefinition<Settings, Payload> = {
     const profileId = await createProfile(request, email, external_id, phone_number, additionalAttributes)
     return await addProfileToList(request, profileId, list_id)
   },
-  performBatch: async (request, { payload }) => {
-    return sendBatchedProfileImportJobRequest(request, payload)
+  performBatch: async (request, { payload, statsContext }) => {
+    return sendBatchedProfileImportJobRequest(request, payload, statsContext)
   }
 }
 

packages/destination-actions/src/destinations/klaviyo/functions.ts

@@ -7,7 +7,8 @@ import {
   JSONLikeObject,
   HTTPError,
   MultiStatusResponse,
-  ErrorCodes
+  ErrorCodes,
+  StatsContext
 } from '@segment/actions-core'
 import { API_URL, REVISION_DATE } from './config'
 import { Settings } from './generated-types'
@@ -519,7 +520,11 @@ async function updateMultiStatusWithKlaviyoErrors(
   })
 }
 
-export async function removeBulkProfilesFromList(request: RequestClient, payloads: RemoveProfilePayload[]) {
+export async function removeBulkProfilesFromList(
+  request: RequestClient,
+  payloads: RemoveProfilePayload[],
+  statsContext?: StatsContext
+) {
   const multiStatusResponse = new MultiStatusResponse()
 
   const { filteredPayloads, validPayloadIndicesBitmap } = validateAndPrepareRemoveBulkProfilePayloads(
@@ -535,6 +540,13 @@ export async function removeBulkProfilesFromList(request: RequestClient, payload
   const externalIds = extractField(filteredPayloads, 'external_id')
   const phoneNumbers = extractField(filteredPayloads, 'phone_number')
 
+  if (statsContext) {
+    const { tags, statsClient } = statsContext
+    const set = new Set()
+    filteredPayloads.forEach((x) => set.add(x.list_id))
+    statsClient.histogram('actions-klaviyo.remove_profile_from_list.unique_list_id', set.size, tags)
+  }
+
   const listId = filteredPayloads[0]?.list_id as string
 
   try {
@@ -666,7 +678,11 @@ function validateAndPrepareBatchedProfileImportPayloads(
   return { filteredPayloads, validPayloadIndicesBitmap }
 }
 
-export async function sendBatchedProfileImportJobRequest(request: RequestClient, payloads: AddProfileToListPayload[]) {
+export async function sendBatchedProfileImportJobRequest(
+  request: RequestClient,
+  payloads: AddProfileToListPayload[],
+  statsContext?: StatsContext
+) {
   const multiStatusResponse = new MultiStatusResponse()
   const { filteredPayloads, validPayloadIndicesBitmap } = validateAndPrepareBatchedProfileImportPayloads(
     payloads,
@@ -676,6 +692,13 @@ export async function sendBatchedProfileImportJobRequest(request: RequestClient,
   if (!filteredPayloads.length) {
     return multiStatusResponse
   }
+
+  if (statsContext) {
+    const { tags, statsClient } = statsContext
+    const set = new Set()
+    filteredPayloads.forEach((x) => set.add(x.list_id))
+    statsClient.histogram('actions-klaviyo.add_profile_to_list.unique_list_id', set.size, tags)
+  }
   const importJobPayload = constructBulkProfileImportPayload(
     filteredPayloads as unknown as KlaviyoProfile[],
     payloads[0]?.list_id

packages/destination-actions/src/destinations/klaviyo/removeProfile/index.ts

@@ -75,8 +75,8 @@ const action: ActionDefinition<Settings, Payload> = {
     )
     return await removeProfileFromList(request, profileIds, list_id)
   },
-  performBatch: async (request, { payload }) => {
-    return await removeBulkProfilesFromList(request, payload)
+  performBatch: async (request, { payload, statsContext }) => {
+    return await removeBulkProfilesFromList(request, payload, statsContext)
   }
 }
 

packages/destination-actions/src/destinations/klaviyo/removeProfileFromList/index.ts

@@ -41,8 +41,8 @@ const action: ActionDefinition<Settings, Payload> = {
     )
     return await removeProfileFromList(request, profileIds, list_id)
   },
-  performBatch: async (request, { payload }) => {
-    return await removeBulkProfilesFromList(request, payload)
+  performBatch: async (request, { payload, statsContext }) => {
+    return await removeBulkProfilesFromList(request, payload, statsContext)
   }
 }
 

packages/destination-actions/src/destinations/klaviyo/subscribeProfile/index.ts

@@ -116,6 +116,15 @@ const action: ActionDefinition<Settings, Payload> = {
       return profile.email || profile.phone_number
     })
 
+    if (statsContext) {
+      const { statsClient, tags } = statsContext
+      const set = new Set()
+      filteredPayload.forEach((profile) => {
+        set.add(`${profile.list_id}-${profile.custom_source}`)
+      })
+      statsClient.histogram('actions-klaviyo.subscribe_profile.unique_list_id', set.size, tags)
+    }
+
     // if there are no payloads with phone or email throw error
     if (filteredPayload.length === 0) {
       throw new PayloadValidationError('Phone Number or Email is required.')

packages/destination-actions/src/destinations/klaviyo/unsubscribeProfile/index.ts

@@ -83,7 +83,7 @@ const action: ActionDefinition<Settings, Payload> = {
       json: unSubData
     })
   },
-  performBatch: async (request, { payload }) => {
+  performBatch: async (request, { payload, statsContext }) => {
     // remove payloads that have niether email or phone_number
     const filteredPayload = payload.filter((profile) => {
       // Validate and convert the phone number using the provided country code
@@ -100,6 +100,15 @@ const action: ActionDefinition<Settings, Payload> = {
       return profile.email || profile.phone_number
     })
 
+    if (statsContext) {
+      const { statsClient, tags } = statsContext
+      const set = new Set()
+      filteredPayload.forEach((profile) => {
+        set.add(profile.list_id)
+      })
+      statsClient.histogram('actions-klaviyo.unsubscribe_profile.unique_list_id', set.size, tags)
+    }
+
     // if there are no payloads with phone or email throw error
     if (payload.length === 0) {
       throw new PayloadValidationError('Phone Number or Email is required.')

packages/destination-actions/src/destinations/klaviyo/upsertProfile/index.ts

@@ -309,7 +309,7 @@ const action: ActionDefinition<Settings, Payload> = {
     }
   },
 
-  performBatch: async (request, { payload, hookOutputs }) => {
+  performBatch: async (request, { payload, hookOutputs, statsContext }) => {
     payload = payload.filter((profile) => {
       // Validate and convert the phone number using the provided country code
       const validPhoneNumber = validateAndConvertPhoneNumber(profile.phone_number, profile.country_code)
@@ -325,6 +325,13 @@ const action: ActionDefinition<Settings, Payload> = {
       return profile.email || profile.phone_number || profile.external_id
     })
 
+    if (statsContext) {
+      const { tags, statsClient } = statsContext
+      const set = new Set()
+      payload.forEach((x) => set.add(`${x.list_id}-${x.override_list_id}`))
+      statsClient.histogram('actions-klaviyo.remove_profile_from_list.unique_list_id', set.size, tags)
+    }
+
     const profilesWithList: Payload[] = []
     const profilesWithoutList: Payload[] = []