提交

my-test/src/main/java/org/example/join/sql/SqlContext.java

@@ -32,30 +32,19 @@ public class SqlContext {
         threadLocal.set(map);
     }
 
-    public static void remove() {
-        threadLocal.remove();
-    }
-
-    public static String getParamKey() {
-        Map<String, Object> map = threadLocal.get();
-        int paramNumber = (int) map.get(PARAM_NUMBER);
-        map.put(PARAM_NUMBER, paramNumber + 1);
-        return SqlConstant.SQL_PARAM + paramNumber;
-    }
-
-    public static Map<String, Object> getParamMap() {
-        Map<String, Object> map = threadLocal.get();
-        return (Map<String, Object>) map.get(PARAM_MAP);
-    }
-
-    public static Map<String, String> getTableAliasMap() {
-        Map<String, Object> map = threadLocal.get();
-        return (Map<String, String>) map.get(TABLE_ALIAS_MAP);
+    public static void putValue(StringJoiner joiner, Object value) {
+        if (value instanceof IFormat iFormat) {
+            joiner.add(iFormat.toSql());
+        } else {
+            String mapKey = getParamKey();
+            joiner.add(StringPool.HASH_LEFT_BRACE + mapKey + StringPool.RIGHT_BRACE);
+            Map<String, Object> map = threadLocal.get();
+            ((Map<String, Object>) map.get(PARAM_MAP)).put(mapKey, value);
+        }
     }
 
-    public static void setTableAliasMap(Sql<?> sql) {
-        Map<String, Object> map = threadLocal.get();
-        map.put(TABLE_ALIAS_MAP, createTableAliasMap(sql));
+    public static void remove() {
+        threadLocal.remove();
     }
 
     public static String getTableAlias(Table table) {
@@ -109,14 +98,21 @@ public class SqlContext {
         return logicDeleteFieldInfo.getColumn();
     }
 
-    public static void putValue(StringJoiner joiner, Object value) {
-        if (value instanceof IFormat iFormat) {
-            joiner.add(iFormat.toSql());
-        } else {
-            String mapKey = getParamKey();
-            joiner.add(StringPool.HASH_LEFT_BRACE + mapKey + StringPool.RIGHT_BRACE);
-            getParamMap().put(mapKey, value);
-        }
+    private static Map<String, String> getTableAliasMap() {
+        Map<String, Object> map = threadLocal.get();
+        return (Map<String, String>) map.get(TABLE_ALIAS_MAP);
+    }
+
+    public static void setTableAliasMap(Sql<?> sql) {
+        Map<String, Object> map = threadLocal.get();
+        map.put(TABLE_ALIAS_MAP, createTableAliasMap(sql));
+    }
+
+    private static String getParamKey() {
+        Map<String, Object> map = threadLocal.get();
+        int paramNumber = (int) map.get(PARAM_NUMBER);
+        map.put(PARAM_NUMBER, paramNumber + 1);
+        return SqlConstant.SQL_PARAM + paramNumber;
     }
 
 }