diff --git a/virtdata-lib-basics/src/main/java/io/nosqlbench/virtdata/library/basics/core/threadstate/SharedState.java b/virtdata-lib-basics/src/main/java/io/nosqlbench/virtdata/library/basics/core/threadstate/SharedState.java index e5b8bec6c..4235fc066 100644 --- a/virtdata-lib-basics/src/main/java/io/nosqlbench/virtdata/library/basics/core/threadstate/SharedState.java +++ b/virtdata-lib-basics/src/main/java/io/nosqlbench/virtdata/library/basics/core/threadstate/SharedState.java @@ -39,4 +39,18 @@ public class SharedState { public static ConcurrentHashMap gl_ObjectMap = new ConcurrentHashMap<>(); + public static T put(Scope scope, String name, T value) { + return switch(scope) { + case process -> (T) gl_ObjectMap.put(name, value); + case thread -> (T) tl_ObjectMap.get().put(name, value); + }; + } + + public static T get(Scope scope, String name) { + return switch(scope) { + case process -> (T) gl_ObjectMap.get(name); + case thread -> (T) tl_ObjectMap.get().get(name); + }; + } + }