open-telemetry
/
opentelemetry-java-instrumentation
mirror of https://github.com/open-telemetry/opentelemetry-java-instrumentation.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Simplify Calldepththreadlocalmap to not use atomic integers

This commit is contained in:
Nikolay Martynov 2018-06-05 14:54:14 -04:00
parent 9174c7804c
commit 3187f7350f
1 changed files with 10 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
dd-java-agent/agent-bootstrap/src/main/java/datadog/trace/bootstrap/CallDepthThreadLocalMap.java
View File

@ -2,7 +2,6 @@ package datadog.trace.bootstrap;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.atomic.AtomicInteger;
/**
* Utility to track nested instrumentation.
@ -11,30 +10,27 @@ import java.util.concurrent.atomic.AtomicInteger;
* #incrementCallDepth at the beginning of each constructor.
*/
public class CallDepthThreadLocalMap {
private static final ThreadLocal<Map<Object, AtomicInteger>> TLS =
new ThreadLocal<Map<Object, AtomicInteger>>() {
private static final ThreadLocal<Map<Object, Integer>> TLS =
new ThreadLocal<Map<Object, Integer>>() {
@Override
public Map<Object, AtomicInteger> initialValue() {
public Map<Object, Integer> initialValue() {
return new HashMap<>();
}
};
public static int incrementCallDepth(final Object k) {
final Map<Object, AtomicInteger> map = TLS.get();
AtomicInteger depth = map.get(k);
final Map<Object, Integer> map = TLS.get();
Integer depth = map.get(k);
if (depth == null) {
depth = new AtomicInteger(0);
map.put(k, depth);
return 0;
depth = 0;
} else {
return depth.incrementAndGet();
depth += 1;
}
map.put(k, depth);
return depth;
}
public static void reset(final Object k) {
final Map<Object, AtomicInteger> map = TLS.get();
if (map != null) {
map.remove(k);
}
TLS.get().remove(k);
}
}