Add thread context module

Author: RanVaknin

thread-context/pom.xml

@@ -43,6 +43,11 @@
     </dependencyManagement>
 
     <dependencies>
+        <dependency>
+            <groupId>software.amazon.awssdk</groupId>
+            <artifactId>annotations</artifactId>
+            <version>${awsjavasdk.version}</version>
+        </dependency>
         <dependency>
             <groupId>org.junit.jupiter</groupId>
             <artifactId>junit-jupiter</artifactId>

thread-context/src/main/java/software/amazon/awssdk/threadcontext/ThreadStorage.java

@@ -17,29 +17,35 @@
 
 import java.util.HashMap;
 import java.util.Map;
+import software.amazon.awssdk.annotations.SdkProtectedApi;
 
+/**
+ * Utility for thread-local context storage.
+ */
+@SdkProtectedApi
 public final class ThreadStorage {
-    private static final ThreadLocal<Map<String, String>> storage = ThreadLocal.withInitial(HashMap::new);
+    private static final ThreadLocal<Map<String, String>> STORAGE = ThreadLocal.withInitial(HashMap::new);
 
-    private ThreadStorage() {}
+    private ThreadStorage() {
+    }
 
     public static void put(String key, String value) {
-        storage.get().put(key, value);
+        STORAGE.get().put(key, value);
     }
 
     public static String get(String key) {
-        return storage.get().get(key);
+        return STORAGE.get().get(key);
     }
 
     public static String remove(String key) {
-        return storage.get().remove(key);
+        return STORAGE.get().remove(key);
     }
 
     public static void clear() {
-        storage.get().clear();
+        STORAGE.get().clear();
     }
 
     public static boolean containsKey(String key) {
-        return storage.get().containsKey(key);
+        return STORAGE.get().containsKey(key);
     }
 }