refactor: ThreadLocal 초기화

woowacourse · Oct 15, 2023 · 7aaacbd · 7aaacbd
1 parent d0f339d
commit 7aaacbd
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/.../main/java/org/springframework/transaction/support/TransactionSynchronizationManager.java b/.../main/java/org/springframework/transaction/support/TransactionSynchronizationManager.java
@@ -4,28 +4,24 @@
 import java.sql.Connection;
 import java.util.HashMap;
 import java.util.Map;
-import java.util.Objects;
 
 public abstract class TransactionSynchronizationManager {
 
-    private static final ThreadLocal<Map<DataSource, Connection>> resources = new ThreadLocal<>();
+    private static final ThreadLocal<Map<DataSource, Connection>> resources = ThreadLocal.withInitial(HashMap::new);
 
     private TransactionSynchronizationManager() {
     }
 
     public static Connection getResource(DataSource key) {
         Map<DataSource, Connection> connectionMap = resources.get();
-        if (Objects.isNull(connectionMap)) {
+        if (!connectionMap.containsKey(key)) {
             return null;
         }
         return connectionMap.get(key);
     }
 
     public static void bindResource(DataSource key, Connection value) {
         Map<DataSource, Connection> connectionMap = resources.get();
-        if (Objects.isNull(connectionMap)) {
-            connectionMap = new HashMap<>();
-        }
         connectionMap.put(key, value);
         resources.set(connectionMap);
     }