Downstream attributes utils

Author: 0marperez

runtime/runtime-core/api/runtime-core.api

@@ -89,6 +89,7 @@ public final class aws/smithy/kotlin/runtime/businessmetrics/BusinessMetricsUtil
 	public static final fun getAccountIdBasedEndpointAccountId ()Laws/smithy/kotlin/runtime/collections/AttributeKey;
 	public static final fun getBusinessMetrics ()Laws/smithy/kotlin/runtime/collections/AttributeKey;
 	public static final fun getServiceEndpointOverride ()Laws/smithy/kotlin/runtime/collections/AttributeKey;
+	public static final fun mergeBusinessMetrics (Laws/smithy/kotlin/runtime/collections/MutableAttributes;Laws/smithy/kotlin/runtime/collections/Attributes;)V
 	public static final fun removeBusinessMetric (Laws/smithy/kotlin/runtime/operation/ExecutionContext;Laws/smithy/kotlin/runtime/businessmetrics/BusinessMetric;)V
 }
 
@@ -139,6 +140,7 @@ public final class aws/smithy/kotlin/runtime/collections/AttributesKt {
 	public static final fun get (Laws/smithy/kotlin/runtime/collections/Attributes;Laws/smithy/kotlin/runtime/collections/AttributeKey;)Ljava/lang/Object;
 	public static final fun isNotEmpty (Laws/smithy/kotlin/runtime/collections/Attributes;)Z
 	public static final fun merge (Laws/smithy/kotlin/runtime/collections/MutableAttributes;Laws/smithy/kotlin/runtime/collections/Attributes;)V
+	public static final fun mergeExcept (Laws/smithy/kotlin/runtime/collections/MutableAttributes;Laws/smithy/kotlin/runtime/collections/Attributes;Ljava/util/Set;)V
 	public static final fun mutableAttributes ()Laws/smithy/kotlin/runtime/collections/MutableAttributes;
 	public static final fun mutableAttributesOf (Lkotlin/jvm/functions/Function1;)Laws/smithy/kotlin/runtime/collections/MutableAttributes;
 	public static final fun putIfAbsent (Laws/smithy/kotlin/runtime/collections/MutableAttributes;Laws/smithy/kotlin/runtime/collections/AttributeKey;Ljava/lang/Object;)V

runtime/runtime-core/common/src/aws/smithy/kotlin/runtime/businessmetrics/BusinessMetricsUtils.kt

@@ -98,3 +98,10 @@ public fun Attributes.copyBusinessMetrics(): MutableSet<String> {
     }
     return copy
 }
+
+/**
+ * Merges another [Attributes] business metrics into this instance of [MutableAttributes].
+ */
+public fun MutableAttributes.mergeBusinessMetrics(other: Attributes) {
+    this[BusinessMetrics] = this[BusinessMetrics].union(other[BusinessMetrics]).toMutableSet()
+}

runtime/runtime-core/common/src/aws/smithy/kotlin/runtime/collections/Attributes.kt

@@ -115,6 +115,18 @@ public fun MutableAttributes.merge(other: Attributes) {
     }
 }
 
+/**
+ * Merge another attributes instance into this set of attributes favoring [other] except for any keys that are
+ * specified in the [exceptions] set
+ */
+public fun MutableAttributes.mergeExcept(other: Attributes, exceptions: Set<AttributeKey<*>>) {
+    other.keys.forEach {
+        if (it in exceptions) return@forEach
+        @Suppress("UNCHECKED_CAST")
+        set(it as AttributeKey<Any>, other[it])
+    }
+}
+
 private class AttributesImpl constructor(seed: Attributes) : MutableAttributes {
     private val map: MutableMap<AttributeKey<*>, Any> = mutableMapOf()
     constructor() : this(emptyAttributes())