1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25 package org.slf4j.helpers;
26
27 import org.slf4j.spi.MDCAdapter;
28
29 import java.util.*;
30 import java.util.Map;
31
32
33
34
35
36
37
38
39
40
41
42
43
44 public class BasicMDCAdapter implements MDCAdapter {
45
46 private InheritableThreadLocal<Map<String, String>> inheritableThreadLocal
47 = new InheritableThreadLocal<Map<String, String>>();
48
49 static boolean isJDK14() {
50 try {
51 String javaVersion = System.getProperty("java.version");
52 return javaVersion.startsWith("1.4");
53 } catch(SecurityException se) {
54
55 return false;
56 }
57 }
58
59 static boolean IS_JDK14 = isJDK14();
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74 public void put(String key, String val) {
75 if (key == null) {
76 throw new IllegalArgumentException("key cannot be null");
77 }
78 Map<String, String> map = (Map<String, String>) inheritableThreadLocal.get();
79 if (map == null) {
80 map = Collections.<String, String>synchronizedMap(new HashMap<String, String>());
81 inheritableThreadLocal.set(map);
82 }
83 map.put(key, val);
84 }
85
86
87
88
89 public String get(String key) {
90 Map<String, String> Map = (Map<String, String>) inheritableThreadLocal.get();
91 if ((Map != null) && (key != null)) {
92 return (String) Map.get(key);
93 } else {
94 return null;
95 }
96 }
97
98
99
100
101 public void remove(String key) {
102 Map<String, String> map = (Map<String, String>) inheritableThreadLocal.get();
103 if (map != null) {
104 map.remove(key);
105 }
106 }
107
108
109
110
111 public void clear() {
112 Map<String, String> map = (Map<String, String>) inheritableThreadLocal.get();
113 if (map != null) {
114 map.clear();
115
116
117 if(isJDK14()) {
118 inheritableThreadLocal.set(null);
119 } else {
120 inheritableThreadLocal.remove();
121 }
122 }
123 }
124
125
126
127
128
129
130
131 public Set<String> getKeys() {
132 Map<String, String> map = (Map<String, String>) inheritableThreadLocal.get();
133 if (map != null) {
134 return map.keySet();
135 } else {
136 return null;
137 }
138 }
139
140
141
142
143
144 public Map<String, String> getCopyOfContextMap() {
145 Map<String, String> oldMap = (Map<String, String>) inheritableThreadLocal.get();
146 if (oldMap != null) {
147 Map<String, String> newMap = Collections.<String, String>synchronizedMap(new HashMap<String, String>());
148 synchronized (oldMap) {
149 newMap.putAll(oldMap);
150 }
151 return newMap;
152 } else {
153 return null;
154 }
155 }
156
157 public void setContextMap(Map<String, String> contextMap) {
158 Map<String, String> map = Collections.<String, String>synchronizedMap(new HashMap<String, String>(contextMap));
159 inheritableThreadLocal.set(map);
160 }
161
162 }