Added custom diagnostics handler code

Author: kushagraThapar

src/main/java/com/azure/cosmos/examples/diagnostics/async/CosmosDiagnosticsQuickStartAsync.java

@@ -3,12 +3,14 @@
 
 package com.azure.cosmos.examples.diagnostics.async;
 
+import com.azure.core.util.Context;
 import com.azure.cosmos.ConsistencyLevel;
 import com.azure.cosmos.CosmosAsyncClient;
 import com.azure.cosmos.CosmosAsyncContainer;
 import com.azure.cosmos.CosmosAsyncDatabase;
 import com.azure.cosmos.CosmosClientBuilder;
 import com.azure.cosmos.CosmosDiagnostics;
+import com.azure.cosmos.CosmosDiagnosticsContext;
 import com.azure.cosmos.CosmosDiagnosticsHandler;
 import com.azure.cosmos.CosmosDiagnosticsThresholds;
 import com.azure.cosmos.CosmosException;
@@ -69,7 +71,7 @@ private void diagnosticsDemo() throws Exception {
 
         logger.info("Using Azure Cosmos DB endpoint: {}", AccountSettings.HOST);
 
-        //  Create Diagnostics threshold
+        //  Create diagnostics threshold
         CosmosDiagnosticsThresholds cosmosDiagnosticsThresholds = new CosmosDiagnosticsThresholds();
         //  For demo purposes, we will reduce the threshold so to log all diagnostics
         //  NOTE: Do not use the same thresholds for production
@@ -78,8 +80,17 @@ private void diagnosticsDemo() throws Exception {
         cosmosDiagnosticsThresholds.setNonPointOperationLatencyThreshold(Duration.ofMillis(10));
         cosmosDiagnosticsThresholds.setRequestChargeThreshold(5f);
 
+        //  By default, DEFAULT_LOGGING_HANDLER can be used
         CosmosDiagnosticsHandler cosmosDiagnosticsHandler = CosmosDiagnosticsHandler.DEFAULT_LOGGING_HANDLER;
 
+        //  App developers can also define their own diagnostics handler
+        cosmosDiagnosticsHandler = new CosmosDiagnosticsHandler() {
+            @Override
+            public void handleDiagnostics(CosmosDiagnosticsContext diagnosticsContext, Context traceContext) {
+                logger.info("This is custom diagnostics handler: {}", diagnosticsContext.toJson());
+            }
+        };
+
 
         //  Create Client Telemetry Config
         CosmosClientTelemetryConfig cosmosClientTelemetryConfig =

src/main/java/com/azure/cosmos/examples/diagnostics/sync/CosmosDiagnosticsQuickStart.java

@@ -3,12 +3,14 @@
 
 package com.azure.cosmos.examples.diagnostics.sync;
 
+import com.azure.core.util.Context;
 import com.azure.cosmos.ConsistencyLevel;
 import com.azure.cosmos.CosmosClient;
 import com.azure.cosmos.CosmosClientBuilder;
 import com.azure.cosmos.CosmosContainer;
 import com.azure.cosmos.CosmosDatabase;
 import com.azure.cosmos.CosmosDiagnostics;
+import com.azure.cosmos.CosmosDiagnosticsContext;
 import com.azure.cosmos.CosmosDiagnosticsHandler;
 import com.azure.cosmos.CosmosDiagnosticsThresholds;
 import com.azure.cosmos.CosmosException;
@@ -77,8 +79,17 @@ private void diagnosticsDemo() throws Exception {
         cosmosDiagnosticsThresholds.setNonPointOperationLatencyThreshold(Duration.ofMillis(10));
         cosmosDiagnosticsThresholds.setRequestChargeThreshold(5f);
 
+        //  By default, DEFAULT_LOGGING_HANDLER can be used
         CosmosDiagnosticsHandler cosmosDiagnosticsHandler = CosmosDiagnosticsHandler.DEFAULT_LOGGING_HANDLER;
 
+        //  App developers can also define their own diagnostics handler
+        cosmosDiagnosticsHandler = new CosmosDiagnosticsHandler() {
+            @Override
+            public void handleDiagnostics(CosmosDiagnosticsContext diagnosticsContext, Context traceContext) {
+                logger.info("This is custom diagnostics handler: {}", diagnosticsContext.toJson());
+            }
+        };
+
 
         //  Create Client Telemetry Config
         CosmosClientTelemetryConfig cosmosClientTelemetryConfig =
@@ -115,7 +126,7 @@ private void createDatabaseIfNotExists() throws Exception {
         //  Create database if not exists
         CosmosDatabaseResponse databaseResponse = client.createDatabaseIfNotExists(databaseName);
         CosmosDiagnostics diagnostics = databaseResponse.getDiagnostics();
-        //  logger.info("Create database diagnostics : {}", diagnostics);
+        logger.info("Create database diagnostics : {}", diagnostics);
         database = client.getDatabase(databaseResponse.getProperties().getId());
 
         logger.info("Done.");
@@ -136,7 +147,7 @@ private void createContainerIfNotExists() throws Exception {
         CosmosContainerResponse containerResponse = database.createContainerIfNotExists(containerProperties,
             throughputProperties);
         CosmosDiagnostics diagnostics = containerResponse.getDiagnostics();
-        //  logger.info("Create container diagnostics : {}", diagnostics);
+        logger.info("Create container diagnostics : {}", diagnostics);
         container = database.getContainer(containerResponse.getProperties().getId());
 
         logger.info("Done.");
@@ -157,7 +168,7 @@ private void createDocument() throws Exception {
             new CosmosItemRequestOptions());
 
         CosmosDiagnostics diagnostics = item.getDiagnostics();
-        //  logger.info("Create item diagnostics : {}", diagnostics);
+        logger.info("Create item diagnostics : {}", diagnostics);
 
         logger.info("Done.");
     }
@@ -171,7 +182,7 @@ private void readDocumentById() throws Exception {
             new PartitionKey(documentLastName), Family.class);
 
         CosmosDiagnostics diagnostics = familyCosmosItemResponse.getDiagnostics();
-        //  logger.info("Read item diagnostics : {}", diagnostics);
+        logger.info("Read item diagnostics : {}", diagnostics);
 
         Family family = familyCosmosItemResponse.getItem();
 
@@ -207,13 +218,13 @@ private void queryDocuments() throws Exception {
 
         //  Add handler to capture diagnostics
         filteredFamilies = filteredFamilies.handle(familyFeedResponse -> {
-            //  logger.info("Query Item diagnostics through handler : {}", familyFeedResponse.getCosmosDiagnostics());
+            logger.info("Query Item diagnostics through handler : {}", familyFeedResponse.getCosmosDiagnostics());
         });
 
         //  Or capture diagnostics through iterableByPage() APIs.
         filteredFamilies.iterableByPage().forEach(familyFeedResponse -> {
-            //  logger.info("Query item diagnostics through iterableByPage : {}", familyFeedResponse
-            //  .getCosmosDiagnostics());
+            logger.info("Query item diagnostics through iterableByPage : {}",
+                familyFeedResponse.getCosmosDiagnostics());
         });
 
         logger.info("Done.");
@@ -233,7 +244,7 @@ private void replaceDocument() throws Exception {
                 new CosmosItemRequestOptions());
 
         CosmosDiagnostics diagnostics = itemResponse.getDiagnostics();
-        //  logger.info("Replace item diagnostics : {}", diagnostics);
+        logger.info("Replace item diagnostics : {}", diagnostics);
 
         logger.info("Request charge of replace operation: {} RU", itemResponse.getRequestCharge());
 
@@ -253,7 +264,7 @@ private void upsertDocument() throws Exception {
             container.upsertItem(family, new CosmosItemRequestOptions());
 
         CosmosDiagnostics diagnostics = itemResponse.getDiagnostics();
-        //  logger.info("Upsert item diagnostics : {}", diagnostics);
+        logger.info("Upsert item diagnostics : {}", diagnostics);
 
         logger.info("Done.");
     }
@@ -267,7 +278,7 @@ private void deleteDocument() throws Exception {
             new PartitionKey(documentLastName), new CosmosItemRequestOptions());
 
         CosmosDiagnostics diagnostics = itemResponse.getDiagnostics();
-        //  logger.info("Delete item diagnostics : {}", diagnostics);
+        logger.info("Delete item diagnostics : {}", diagnostics);
 
         logger.info("Done.");
     }
@@ -278,7 +289,7 @@ private void deleteDatabase() throws Exception {
 
         // Delete database
         CosmosDatabaseResponse dbResp = client.getDatabase(databaseName).delete(new CosmosDatabaseRequestOptions());
-        //  logger.info("Status code for database delete: {}", dbResp.getStatusCode());
+        logger.info("Status code for database delete: {}", dbResp.getStatusCode());
 
         logger.info("Done.");
     }