[DYN-6451] Analytics for ML DataPipeline. (DynamoDS#14936)

* ML DataPipeline analytics * comments * comments * Update DynamoMLDataPipeline.cs
reddyashish · Feb 16, 2024 · dd9c7af · dd9c7af
1 parent 6538bc7
commit dd9c7af
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/src/DynamoMLDataPipeline/DynamoMLDataPipeline.cs b/src/DynamoMLDataPipeline/DynamoMLDataPipeline.cs
@@ -161,7 +161,7 @@ static public string ConvertDynToBase64(string filePath)
             byte[] compressedBuffer = DataUtilities.Compress(stringBuffer);
 
             string base64CompressedBuffer = Convert.ToBase64String(compressedBuffer);
-            LogMessage.Info($"BASE64 string buffer has {base64CompressedBuffer.Length} bytes");
+            LogMessage.Info($"BASE64 string buffer has {base64CompressedBuffer.Length} bytes"); 
 
             // Write to file for testing purposes
             File.WriteAllText(savePath, base64CompressedBuffer);
@@ -184,6 +184,8 @@ static void DataExchangeToForge(string filePath, RestClient client, string token
             // from the response - these will be consumed by the following API calls.
             ExchangeContainerId = exchangeRequestResponseBody["id"].Value;
 
+            Analytics.TrackEvent(Actions.Export, Categories.DynamoMLDataPipelineOperations, "ExchangeContainerID", Convert.ToInt32(ExchangeContainerId));
+
             var schemaNamespaceId = exchangeRequestResponseBody["components"]["data"]["insert"]["autodesk.data:exchange.source.default-1.0.0"]["source"]["String"]["id"].Value;
 
             // STEP 2: START A FULLFILLMENT ---------------------
@@ -225,6 +227,8 @@ static void DataExchangeToForge(string filePath, RestClient client, string token
             }
             LogMessage.Info("Binary upload started!");
 
+            Analytics.TrackEvent(Actions.Export, Categories.DynamoMLDataPipelineOperations, "BinarySize", base64CompressedBuffer.Length);
+
             // STEP 4b: FINISH BINARY UPLOAD -------------------
             // Finish uploading binary assets: Let the system know that the binary assets have been uploaded and are ready for processing. 
             // This call can be made for a single binary or a batch of 25 binaries.
@@ -310,6 +314,7 @@ public void DataExchange(string filePath)
             // Stage collectionId created for Dynamo
             CollectionId = ProductionCollectionID;
 
+            Analytics.TrackEvent(Actions.Export, Categories.DynamoMLDataPipelineOperations, "CollectionID", Convert.ToInt32(CollectionId));
             //ExchangeContainerId = "";
 
             var forgeClient = new RestClient(ProductionClientUrl);

diff --git a/src/NodeServices/IAnalyticsClient.cs b/src/NodeServices/IAnalyticsClient.cs
@@ -138,6 +138,11 @@ public enum Categories
         /// Events Category related to the splash screen
         /// </summary>
         SplashScreenOperations,
+
+        /// <summary>
+        /// Events Category related to DynamoMLDataPipeline
+        /// </summary>
+        DynamoMLDataPipelineOperations
     }
 
     /// <summary>
@@ -443,7 +448,7 @@ public enum Actions
         /// <summary>
         /// Export event, e.g. tracks the ExportSettings event
         /// </summary>
-        Export
+        Export,
     }
 
     public enum HeartBeatType