From 3d1476fe4a9858bd87bb3204256ea5209011243c Mon Sep 17 00:00:00 2001
From: Trask Stalnaker <trask.stalnaker@gmail.com>
Date: Wed, 12 Jul 2023 14:46:22 -0700
Subject: [PATCH] Preserve db.cosmosdb.* span attributes

---
 .../exporter/implementation/MappingsBuilder.java      | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/agent/azure-monitor-exporter/src/main/java/com/azure/monitor/opentelemetry/exporter/implementation/MappingsBuilder.java b/agent/azure-monitor-exporter/src/main/java/com/azure/monitor/opentelemetry/exporter/implementation/MappingsBuilder.java
index 6a9b4588352..003c24cdf5b 100644
--- a/agent/azure-monitor-exporter/src/main/java/com/azure/monitor/opentelemetry/exporter/implementation/MappingsBuilder.java
+++ b/agent/azure-monitor-exporter/src/main/java/com/azure/monitor/opentelemetry/exporter/implementation/MappingsBuilder.java
@@ -21,7 +21,16 @@ class MappingsBuilder {
       new HashSet<>(
           asList(
               "http.",
-              "db.",
+
+              // specifically listing out all standard db.* attributes
+              // so that db.cosmosdb.* attributes will be preserved
+              SemanticAttributes.DB_SYSTEM.getKey(),
+              SemanticAttributes.DB_NAME.getKey(),
+              SemanticAttributes.DB_STATEMENT.getKey(),
+              SemanticAttributes.DB_OPERATION.getKey(),
+              SemanticAttributes.DB_CONNECTION_STRING.getKey(),
+              SemanticAttributes.DB_USER.getKey(),
+              SemanticAttributes.DB_SQL_TABLE.getKey(),
               "message.",
               "messaging.",
               "rpc.",