package com.oracle.truffle.api.test.polyglot;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNotSame;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertSame;
import static org.junit.Assert.assertTrue;
import java.lang.ref.WeakReference;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;
import org.graalvm.polyglot.Context;
import org.graalvm.polyglot.Engine;
import org.graalvm.polyglot.Instrument;
import org.graalvm.polyglot.PolyglotException;
import org.junit.Assert;
import org.junit.Test;
import com.oracle.truffle.api.ContextLocal;
import com.oracle.truffle.api.ContextThreadLocal;
import com.oracle.truffle.api.TruffleContext;
import com.oracle.truffle.api.TruffleLanguage;
import com.oracle.truffle.api.TruffleLanguage.Env;
import com.oracle.truffle.api.frame.VirtualFrame;
import com.oracle.truffle.api.instrumentation.EventContext;
import com.oracle.truffle.api.instrumentation.ExecutionEventListener;
import com.oracle.truffle.api.instrumentation.SourceSectionFilter;
import com.oracle.truffle.api.instrumentation.TruffleInstrument;
import com.oracle.truffle.api.nodes.LanguageInfo;
public class ContextLocalTest extends AbstractPolyglotTest {
private static final int PARALLELISM = 32;
private static final int ITERATIONS = 50;
static final String VALID_EXCLUSIVE_LANGUAGE = "ContextLocalTest_ValidExclusiveLanguage";
static final String VALID_SHARED_LANGUAGE = "ContextLocalTest_ValidSharedLanguage";
static final String VALID_INSTRUMENT = "ContextLocalTest_ValidInstrument";
static final String INVALID_CONTEXT_LOCAL = "ContextLocalTest_InvalidLanguageContextLocal";
static final String INVALID_CONTEXT_THREAD_LOCAL = "ContextLocalTest_InvalidLanguageContextThreadLocal";
@Test
public void customSubclassesDisallowed() {
assertFails(() -> new ContextLocal<String>(null) {
@Override
public String get() {
return null;
}
@Override
public String get(TruffleContext c) {
return null;
}
}, IllegalStateException.class);
assertFails(() -> new ContextThreadLocal<String>(null) {
@Override
public String get() {
return null;
}
@Override
public String get(TruffleContext t) {
return null;
}
@Override
public String get(TruffleContext c, Thread t) {
return null;
}
@Override
public String get(Thread t) {
return null;
}
}, IllegalStateException.class);
}
private static void runInParallel(Runnable callable) throws InterruptedException, ExecutionException {
ExecutorService executor = Executors.newFixedThreadPool(PARALLELISM);
List<Future<?>> futures = new ArrayList<>();
for (int i = 0; i < ITERATIONS; i++) {
futures.add(executor.submit(() -> {
callable.run();
return null;
}));
}
try {
for (Future<?> future : futures) {
future.get();
}
} finally {
executor.shutdown();
executor.awaitTermination(10000, TimeUnit.MILLISECONDS);
}
}
@Test
public void testExclusiveLanguageContextThreadLocal() throws InterruptedException, ExecutionException {
try (Context c0 = Context.create(); Context c1 = Context.create()) {
c0.initialize(VALID_EXCLUSIVE_LANGUAGE);
c0.enter();
Env env0;
TruffleContext t0;
ValidExclusiveLanguage language0;
try {
env0 = ValidExclusiveLanguage.getCurrentContext();
t0 = env0.getContext();
language0 = ValidExclusiveLanguage.getCurrentLanguage();
} finally {
c0.leave();
}
runInParallel(() -> {
c0.enter();
try {
assertSame(env0, language0.contextThreadLocal0.get().env);
assertSame(env0, language0.contextThreadLocal1.get().env);
assertSame(Thread.currentThread(), language0.contextThreadLocal0.get().thread);
assertSame(Thread.currentThread(), language0.contextThreadLocal1.get().thread);
} finally {
c0.leave();
}
});
c1.initialize(VALID_EXCLUSIVE_LANGUAGE);
runInParallel(() -> {
c1.enter();
try {
Env env1 = ValidExclusiveLanguage.getCurrentContext();
TruffleContext t1 = env1.getContext();
ValidExclusiveLanguage language1 = ValidExclusiveLanguage.getCurrentLanguage();
assertSame(env1, language1.contextThreadLocal0.get().env);
assertSame(Thread.currentThread(), language1.contextThreadLocal0.get().thread);
assertSame(env1, language1.contextThreadLocal1.get().env);
assertSame(Thread.currentThread(), language1.contextThreadLocal1.get().thread);
assertFails(() -> language1.contextThreadLocal0.get(t0, Thread.currentThread()),
AssertionError.class,
(e) -> e.getMessage().startsWith("Detected invalid sharing of context locals"));
assertFails(() -> language1.contextThreadLocal1.get(t0, Thread.currentThread()),
AssertionError.class,
(e) -> e.getMessage().startsWith("Detected invalid sharing of context locals"));
assertFails(() -> language0.contextThreadLocal0.get(), AssertionError.class, (e) -> e.getMessage().startsWith("Detected invalid sharing of context locals"));
assertSame(env1, language1.contextThreadLocal0.get(t1, Thread.currentThread()).env);
assertSame(Thread.currentThread(), language1.contextThreadLocal0.get(t1,
Thread.currentThread()).thread);
assertSame(env1, language1.contextThreadLocal1.get(t1, Thread.currentThread()).env);
assertSame(Thread.currentThread(), language1.contextThreadLocal1.get(t1,
Thread.currentThread()).thread);
assertNotSame(env0, env1);
} finally {
c1.leave();
}
});
}
}
@Test
public void testExclusiveLanguageContextLocal() {
try (Context c0 = Context.create();
Context c1 = Context.create()) {
c0.initialize(VALID_EXCLUSIVE_LANGUAGE);
c0.enter();
Env env0;
TruffleContext t0;
ValidExclusiveLanguage language0;
try {
env0 = ValidExclusiveLanguage.getCurrentContext();
t0 = env0.getContext();
language0 = ValidExclusiveLanguage.getCurrentLanguage();
assertSame(env0, language0.contextLocal0.get());
assertSame(env0, language0.contextLocal1.get());
} finally {
c0.leave();
}
c1.initialize(VALID_EXCLUSIVE_LANGUAGE);
c1.enter();
try {
Env env1 = ValidExclusiveLanguage.getCurrentContext();
TruffleContext t1 = env1.getContext();
ValidExclusiveLanguage language1 = ValidExclusiveLanguage.getCurrentLanguage();
assertSame(env1, language1.contextLocal0.get());
assertSame(env1, language1.contextLocal1.get());
assertFails(() -> language1.contextLocal0.get(t0), AssertionError.class, (e) -> e.getMessage().startsWith("Detected invalid sharing of context locals"));
assertFails(() -> language1.contextLocal1.get(t0), AssertionError.class, (e) -> e.getMessage().startsWith("Detected invalid sharing of context locals"));
assertFails(() -> language0.contextLocal0.get(), AssertionError.class, (e) -> e.getMessage().startsWith("Detected invalid sharing of context locals"));
assertSame(env1, language1.contextLocal0.get(t1));
assertSame(env1, language1.contextLocal1.get(t1));
assertNotSame(env0, env1);
} finally {
c1.leave();
}
}
}
@Test
public void testSharedLanguageContextLocal() {
try (Engine engine = Engine.create()) {
try (Context c0 = Context.newBuilder().engine(engine).build();
Context c1 = Context.newBuilder().engine(engine).build()) {
c0.initialize(VALID_SHARED_LANGUAGE);
c0.enter();
Env env0;
TruffleContext t0;
ValidSharedLanguage language0;
try {
env0 = ValidSharedLanguage.getCurrentContext();
t0 = env0.getContext();
language0 = ValidSharedLanguage.getCurrentLanguage();
assertSame(env0, language0.local0.get());
assertSame(env0, language0.local1.get());
} finally {
c0.leave();
}
c1.initialize(VALID_SHARED_LANGUAGE);
c1.enter();
try {
Env env1 = ValidSharedLanguage.getCurrentContext();
TruffleContext t1 = env1.getContext();
ValidSharedLanguage language1 = ValidSharedLanguage.getCurrentLanguage();
assertSame(env1, language1.local0.get());
assertSame(env1, language1.local1.get());
assertSame(env0, language1.local0.get(t0));
assertSame(env0, language1.local1.get(t0));
assertSame(env0, language0.local0.get(t0));
assertSame(env1, language1.local0.get(t1));
assertSame(env1, language1.local1.get(t1));
assertNotSame(env0, env1);
} finally {
c1.leave();
}
}
}
}
@Test
public void testSharedLanguageContextThreadLocal() throws InterruptedException, ExecutionException {
try (Engine engine = Engine.create()) {
try (Context c0 = Context.newBuilder().engine(engine).build();
Context c1 = Context.newBuilder().engine(engine).build()) {
c0.initialize(VALID_SHARED_LANGUAGE);
c0.enter();
Env env0;
TruffleContext t0;
ValidSharedLanguage language0;
try {
env0 = ValidSharedLanguage.getCurrentContext();
t0 = env0.getContext();
language0 = ValidSharedLanguage.getCurrentLanguage();
} finally {
c0.leave();
}
runInParallel(() -> {
c0.enter();
try {
assertSame(env0, language0.contextThreadLocal0.get().env);
assertSame(env0, language0.contextThreadLocal1.get().env);
assertSame(Thread.currentThread(), language0.contextThreadLocal0.get().thread);
assertSame(Thread.currentThread(), language0.contextThreadLocal1.get().thread);
} finally {
c0.leave();
}
});
c1.initialize(VALID_SHARED_LANGUAGE);
runInParallel(() -> {
c1.enter();
try {
Env env1 = ValidSharedLanguage.getCurrentContext();
TruffleContext t1 = env1.getContext();
ValidSharedLanguage language1 = ValidSharedLanguage.getCurrentLanguage();
assertSame(env1, language1.contextThreadLocal0.get().env);
assertSame(env1, language1.contextThreadLocal1.get().env);
assertNull(language1.contextThreadLocal0.get(t0, Thread.currentThread()));
assertNull(language1.contextThreadLocal1.get(t0, Thread.currentThread()));
assertSame(env1, language1.contextThreadLocal0.get(Thread.currentThread()).env);
assertSame(env1, language1.contextThreadLocal1.get(Thread.currentThread()).env);
assertNull(language1.contextThreadLocal0.get(t0, Thread.currentThread()));
assertSame(env1, language1.contextThreadLocal1.get(t1, Thread.currentThread()).env);
assertSame(env1, language1.contextThreadLocal1.get(t1, Thread.currentThread()).env);
assertNotSame(env0, env1);
} finally {
c1.leave();
}
});
}
}
}
@SuppressWarnings("cast")
@Test
public void testContextLocalValidInstrument() {
try (Engine engine = Engine.create()) {
try (Context c0 = Context.newBuilder().engine(engine).build();
Context c1 = Context.newBuilder().engine(engine).build()) {
@SuppressWarnings("unchecked")
ValidInstrument instrument = engine.getInstruments().get(VALID_INSTRUMENT).lookup(ValidInstrument.class);
assertFails(() -> instrument.local0.get(), IllegalStateException.class);
c0.enter();
TruffleContext tc0;
try {
tc0 = instrument.local0.get();
} finally {
c0.leave();
}
c1.enter();
TruffleContext tc1;
try {
tc1 = instrument.local0.get();
} finally {
c1.leave();
}
assertNotSame(tc0, tc1);
assertSame(tc0, instrument.local0.get(tc0));
assertSame(tc1, instrument.local0.get(tc1));
c0.initialize(VALID_SHARED_LANGUAGE);
c0.enter();
try {
assertEquals(tc0, ValidSharedLanguage.getCurrentLanguage().local0.get().getContext());
} finally {
c0.leave();
}
c1.initialize(VALID_SHARED_LANGUAGE);
c1.enter();
try {
assertEquals(tc1, ValidSharedLanguage.getCurrentLanguage().local1.get().getContext());
} finally {
c1.leave();
}
}
}
}
@SuppressWarnings("cast")
@Test
public void testContextThreadLocalValidInstrument() throws InterruptedException, ExecutionException {
try (Engine engine = Engine.create()) {
try (Context c0 = Context.newBuilder().engine(engine).build();
Context c1 = Context.newBuilder().engine(engine).build()) {
ValidInstrument instrument = engine.getInstruments().get(VALID_INSTRUMENT).lookup(ValidInstrument.class);
runInParallel(() -> {
c0.enter();
InstrumentThreadLocalValue tc0;
try {
tc0 = instrument.threadLocal0.get();
} finally {
c0.leave();
}
c1.enter();
InstrumentThreadLocalValue tc1;
try {
tc1 = instrument.threadLocal0.get();
} finally {
c1.leave();
}
assertNotSame(tc0, tc1);
assertEquals(tc0, instrument.threadLocal0.get(tc0.context, Thread.currentThread()));
assertEquals(tc1, instrument.threadLocal0.get(tc1.context, Thread.currentThread()));
c0.initialize(VALID_SHARED_LANGUAGE);
c0.enter();
try {
assertEquals(tc0.context, ValidSharedLanguage.getCurrentLanguage().contextThreadLocal0.get().env.getContext());
} finally {
c0.leave();
}
c1.initialize(VALID_SHARED_LANGUAGE);
c1.enter();
try {
assertEquals(tc1.context, ValidSharedLanguage.getCurrentLanguage().contextThreadLocal1.get().env.getContext());
} finally {
c1.leave();
}
assertFails(() -> instrument.threadLocal0.get(), IllegalStateException.class);
assertFails(() -> instrument.threadLocal0.get(Thread.currentThread()), IllegalStateException.class);
assertFails(() -> instrument.threadLocal0.get(Thread.currentThread()), IllegalStateException.class);
});
}
}
}
private static Object contextLocalDynamicValue = "non-null";
private static Object threadLocalDynamicValue = "non-null";
@Test
public void testInvalidContextLocalLanguage() {
try (Engine engine = Engine.create()) {
try (Context c0 = Context.newBuilder().engine(engine).allowAllAccess(true).build();
Context c1 = Context.newBuilder().engine(engine).allowAllAccess(true).build()) {
c0.initialize(INVALID_CONTEXT_LOCAL);
c1.enter();
try {
c1.initialize(VALID_SHARED_LANGUAGE);
Env env1 = ValidSharedLanguage.getCurrentContext();
LanguageInfo invalid = env1.getInternalLanguages().get(INVALID_CONTEXT_LOCAL);
assertFails(() -> env1.initializeLanguage(invalid), IllegalStateException.class,
(e) -> assertTrue(e.getCause().getMessage(), e.getCause().getMessage().contains("did not create the same number of context locals")));
} finally {
c1.leave();
}
}
}
}
@Test
public void testInvalidContextThreadLocalLanguage() {
try (Engine engine = Engine.create()) {
try (Context c0 = Context.newBuilder().engine(engine).allowAllAccess(true).build();
Context c1 = Context.newBuilder().engine(engine).allowAllAccess(true).build()) {
c0.initialize(INVALID_CONTEXT_THREAD_LOCAL);
c1.enter();
try {
c1.initialize(VALID_SHARED_LANGUAGE);
Env env1 = ValidSharedLanguage.getCurrentContext();
LanguageInfo invalid = env1.getInternalLanguages().get(INVALID_CONTEXT_THREAD_LOCAL);
assertFails(() -> env1.initializeLanguage(invalid), IllegalStateException.class,
(e) -> assertTrue(e.getCause().getMessage(), e.getCause().getMessage().contains("did not create the same number of context thread locals")));
} finally {
c1.leave();
}
}
}
}
@Test
public void testCreateLanguageContextLocalLooLate() {
try (Context c0 = Context.create();) {
c0.initialize(VALID_SHARED_LANGUAGE);
c0.enter();
try {
ValidSharedLanguage lang = ValidSharedLanguage.getCurrentLanguage();
assertFails(() -> lang.createContextLocal0("testString"), IllegalStateException.class,
(e) -> assertEquals(e.getMessage(), "The set of context locals is frozen. " +
"Context locals can only be created during construction of the TruffleLanguage subclass."));
} finally {
c0.leave();
}
}
}
@Test
public void testCreateLanguageContextThreadLocalLooLate() {
try (Context c0 = Context.create();) {
c0.initialize(VALID_SHARED_LANGUAGE);
c0.enter();
try {
ValidSharedLanguage lang = ValidSharedLanguage.getCurrentLanguage();
assertFails(() -> lang.createContextThreadLocal0("testString"), IllegalStateException.class,
(e) -> assertEquals(e.getMessage(), "The set of context thread locals is frozen. " +
"Context thread locals can only be created during construction of the TruffleLanguage subclass."));
} finally {
c0.leave();
}
}
}
@Test
public void testCreateInstrumentContextLocalLooLate() {
try (Context c0 = Context.create();) {
ValidInstrument instrument = c0.getEngine().getInstruments().get(VALID_INSTRUMENT).lookup(ValidInstrument.class);
assertFails(() -> instrument.createContextLocal0(), IllegalStateException.class, (e) -> assertEquals(e.getMessage(), "The set of context locals is frozen. " +
"Context locals can only be created during construction of the TruffleInstrument subclass."));
}
}
@Test
public void testCreateInstrumentContextThreadLocalLooLate() {
try (Context c0 = Context.create();) {
ValidInstrument instrument = c0.getEngine().getInstruments().get(VALID_INSTRUMENT).lookup(ValidInstrument.class);
assertFails(() -> instrument.createContextLocal0(), IllegalStateException.class, (e) -> assertEquals(e.getMessage(), "The set of context locals is frozen. " +
"Context locals can only be created during construction of the TruffleInstrument subclass."));
}
}
@Test
public void testNullContextLocalValue() {
contextLocalDynamicValue = null;
try (Context c0 = Context.create()) {
Instrument instrument = c0.getEngine().getInstruments().get(VALID_INSTRUMENT);
assertFails(() -> instrument.lookup(ValidInstrument.class), PolyglotException.class,
(e) -> assertTrue(e.getMessage(), e.getMessage().contains("ContextLocalFactory.create is not allowed to return null")));
assertFails(() -> c0.initialize(VALID_SHARED_LANGUAGE), PolyglotException.class,
(e) -> assertTrue(e.getMessage(), e.getMessage().contains("ContextLocalFactory.create is not allowed to return null")));
} finally {
contextLocalDynamicValue = "non-null";
}
}
@Test
public void testNullContextThreadLocalValue() {
threadLocalDynamicValue = null;
try (Context c0 = Context.create()) {
Instrument instrument = c0.getEngine().getInstruments().get(VALID_INSTRUMENT);
c0.enter();
try {
assertFails(() -> instrument.lookup(ValidInstrument.class), PolyglotException.class,
(e) -> assertTrue(e.getMessage(), e.getMessage().contains("ContextThreadLocalFactory.create is not allowed to return null")));
assertFails(() -> c0.initialize(VALID_SHARED_LANGUAGE), PolyglotException.class,
(e) -> assertTrue(e.getMessage(), e.getMessage().contains("ContextThreadLocalFactory.create is not allowed to return null")));
} finally {
c0.leave();
}
} finally {
threadLocalDynamicValue = "non-null";
}
}
@Test
public void testUnstableContextLocalValue() {
contextLocalDynamicValue = "foobar";
try (Engine engine = Engine.create()) {
try (Context c0 = Context.newBuilder().engine(engine).allowAllAccess(true).build()) {
c0.getEngine().getInstruments().get(VALID_INSTRUMENT).lookup(ValidInstrument.class);
c0.initialize(VALID_SHARED_LANGUAGE);
}
contextLocalDynamicValue = 42;
assertFails(() -> Context.newBuilder().engine(engine).allowAllAccess(true).build(), PolyglotException.class,
(e) -> assertTrue(e.getMessage(), e.getMessage().contains(
"The return context value type must be stable and exact. " +
"Expected class java.lang.String but got class java.lang.Integer ")));
contextLocalDynamicValue = "42";
try (Context c1 = Context.newBuilder().engine(engine).allowAllAccess(true).build()) {
ValidInstrument instrument = c1.getEngine().getInstruments().get(VALID_INSTRUMENT).lookup(ValidInstrument.class);
c1.enter();
try {
Assert.assertEquals(contextLocalDynamicValue, instrument.localDynamic.get());
contextLocalDynamicValue = 42;
assertFails(() -> c1.initialize(VALID_SHARED_LANGUAGE), PolyglotException.class,
(e) -> assertTrue(e.getMessage(), e.getMessage().contains(
"The return context value type must be stable and exact. " +
"Expected class java.lang.String but got class java.lang.Integer ")));
} finally {
c1.leave();
}
}
} finally {
contextLocalDynamicValue = "non-null";
}
}
@Test
public void testUnstableContextThreadLocalValue() {
threadLocalDynamicValue = "foobar";
try (Engine engine = Engine.create()) {
try (Context c0 = Context.newBuilder().engine(engine).allowAllAccess(true).build()) {
c0.getEngine().getInstruments().get(VALID_INSTRUMENT).lookup(ValidInstrument.class);
c0.initialize(VALID_SHARED_LANGUAGE);
}
threadLocalDynamicValue = 42;
Context c0 = Context.newBuilder().engine(engine).allowAllAccess(true).build();
assertFails(() -> c0.enter(), PolyglotException.class,
(e) -> assertTrue(e.getMessage(), e.getMessage().contains(
"The return context value type must be stable and exact. " +
"Expected class java.lang.String but got class java.lang.Integer ")));
threadLocalDynamicValue = "42";
try (Context c1 = Context.newBuilder().engine(engine).allowAllAccess(true).build()) {
ValidInstrument instrument = c1.getEngine().getInstruments().get(VALID_INSTRUMENT).lookup(ValidInstrument.class);
c1.enter();
try {
Assert.assertEquals(contextLocalDynamicValue, instrument.localDynamic.get());
threadLocalDynamicValue = 42;
assertFails(() -> c1.initialize(VALID_SHARED_LANGUAGE), PolyglotException.class,
(e) -> assertTrue(e.getMessage(), e.getMessage().contains(
"The return context value type must be stable and exact. " +
"Expected class java.lang.String but got class java.lang.Integer ")));
} finally {
c1.leave();
}
}
} finally {
threadLocalDynamicValue = "";
}
}
public void testInstrumentCreatedBeforeContextsInitialized() {
}
@TruffleLanguage.Registration(id = VALID_EXCLUSIVE_LANGUAGE, name = VALID_EXCLUSIVE_LANGUAGE)
public static class ValidExclusiveLanguage extends TruffleLanguage<TruffleLanguage.Env> {
final ContextLocal<Env> contextLocal0 = createContextLocal((e) -> e);
final ContextLocal<Env> contextLocal1 = createContextLocal((e) -> e);
final ContextThreadLocal<LanguageThreadLocalValue> contextThreadLocal0 = createContextThreadLocal(LanguageThreadLocalValue::new);
final ContextThreadLocal<LanguageThreadLocalValue> contextThreadLocal1 = createContextThreadLocal(LanguageThreadLocalValue::new);
@Override
protected TruffleLanguage.Env createContext(TruffleLanguage.Env env) {
return env;
}
@Override
protected boolean isThreadAccessAllowed(Thread thread, boolean singleThreaded) {
return true;
}
static ValidExclusiveLanguage getCurrentLanguage() {
return getCurrentLanguage(ValidExclusiveLanguage.class);
}
static Env getCurrentContext() {
return getCurrentContext(ValidExclusiveLanguage.class);
}
}
static class LanguageThreadLocalValue {
final Env env;
final Thread thread;
LanguageThreadLocalValue(Env env, Thread t) {
this.env = env;
this.thread = t;
}
}
static class InstrumentThreadLocalValue {
final TruffleContext context;
final Thread thread;
InstrumentThreadLocalValue(TruffleContext context, Thread t) {
this.context = context;
this.thread = t;
}
@Override
public boolean equals(Object obj) {
if (!(obj instanceof InstrumentThreadLocalValue)) {
return false;
}
InstrumentThreadLocalValue key = (InstrumentThreadLocalValue) obj;
return this.context == key.context && this.thread == key.thread;
}
@Override
public int hashCode() {
return Objects.hash(context, thread);
}
}
@TruffleLanguage.Registration(id = VALID_SHARED_LANGUAGE, name = VALID_SHARED_LANGUAGE, contextPolicy = TruffleLanguage.ContextPolicy.SHARED)
public static class ValidSharedLanguage extends TruffleLanguage<TruffleLanguage.Env> {
final ContextLocal<Env> local0 = createContextLocal((e) -> e);
final ContextLocal<Env> local1 = createContextLocal((e) -> e);
final ContextLocal<Object> localDynamic = createContextLocal((e) -> contextLocalDynamicValue);
final ContextThreadLocal<LanguageThreadLocalValue> contextThreadLocal0 = createContextThreadLocal(LanguageThreadLocalValue::new);
final ContextThreadLocal<LanguageThreadLocalValue> contextThreadLocal1 = createContextThreadLocal(LanguageThreadLocalValue::new);
final ContextThreadLocal<Object> threadLocalDynamic = createContextThreadLocal((c, t) -> threadLocalDynamicValue);
@Override
protected TruffleLanguage.Env createContext(TruffleLanguage.Env env) {
return env;
}
@Override
protected boolean isThreadAccessAllowed(Thread thread, boolean singleThreaded) {
return true;
}
static ValidSharedLanguage getCurrentLanguage() {
return getCurrentLanguage(ValidSharedLanguage.class);
}
static Env getCurrentContext() {
return getCurrentContext(ValidSharedLanguage.class);
}
public ContextLocal<String> createContextLocal0(String value) {
return createContextLocal((e) -> value);
}
public ContextThreadLocal<String> createContextThreadLocal0(String value) {
return createContextThreadLocal((e, t) -> value);
}
}
@TruffleInstrument.Registration(id = VALID_INSTRUMENT, name = VALID_INSTRUMENT, services = ValidInstrument.class)
public static class ValidInstrument extends TruffleInstrument {
final ContextLocal<TruffleContext> local0 = createContextLocal(this::createInstrumentContextLocal);
final ContextLocal<Object> localDynamic = createContextLocal((e) -> contextLocalDynamicValue);
final ContextThreadLocal<InstrumentThreadLocalValue> threadLocal0 = createContextThreadLocal(this::newInstrumentThreadLocal);
final ContextThreadLocal<Object> threadLocalDynamic = createContextThreadLocal((c, t) -> threadLocalDynamicValue);
private Env environment;
private final ConcurrentHashMap<InstrumentThreadLocalValue, String> seenValues = new ConcurrentHashMap<>();
@Override
protected void onCreate(Env env) {
this.environment = env;
env.registerService(this);
}
InstrumentThreadLocalValue newInstrumentThreadLocal(TruffleContext context, Thread t) {
assertNotNull(environment);
InstrumentThreadLocalValue local = new InstrumentThreadLocalValue(context, t);
if (seenValues.containsKey(local)) {
throw new AssertionError("duplicate thread local factory invocation");
}
seenValues.put(local, "");
return local;
}
TruffleContext createInstrumentContextLocal(TruffleContext context) {
assertNotNull(environment);
return context;
}
public ContextLocal<String> createContextLocal0() {
return createContextLocal((e) -> {
throw new AssertionError();
});
}
public ContextThreadLocal<String> createContextThreadLocal0() {
return createContextThreadLocal((e, t) -> {
throw new AssertionError();
});
}
}
@TruffleLanguage.Registration(id = INVALID_CONTEXT_LOCAL, name = INVALID_CONTEXT_LOCAL)
public static class InvalidLanguageContextLocal extends TruffleLanguage<TruffleLanguage.Env> {
static int effect = 0;
final ContextLocal<Env> local0 = createContextLocal((e) -> e);
final ContextLocal<Env> local1 = (effect++) % 2 == 0 ? createContextLocal((e) -> e) : null;
@Override
protected TruffleLanguage.Env createContext(TruffleLanguage.Env env) {
return env;
}
}
@TruffleLanguage.Registration(id = INVALID_CONTEXT_THREAD_LOCAL, name = INVALID_CONTEXT_THREAD_LOCAL)
public static class InvalidLanguageContextThreadLocal extends TruffleLanguage<TruffleLanguage.Env> {
static int effect = 0;
final ContextThreadLocal<Env> local0 = createContextThreadLocal((e, t) -> e);
final ContextThreadLocal<Env> local1 = (effect++) % 2 == 0 ? createContextThreadLocal((e, t) -> e) : null;
@Override
protected TruffleLanguage.Env createContext(TruffleLanguage.Env env) {
return env;
}
}
@TruffleInstrument.Registration(id = "example", name = "Example Instrument")
public static class ExampleInstrument extends TruffleInstrument {
final ContextThreadLocal<ExampleLocal> local = createContextThreadLocal(ExampleLocal::new);
@Override
protected void onCreate(Env env) {
ExecutionEventListener listener = new ExecutionEventListener() {
public void onEnter(EventContext context, VirtualFrame frame) {
ExampleLocal value = local.get();
assert value.thread.get() == Thread.currentThread();
}
public void onReturnValue(EventContext context, VirtualFrame frame, Object result) {
}
public void onReturnExceptional(EventContext context, VirtualFrame frame, Throwable exception) {
}
};
env.getInstrumenter().attachExecutionEventListener(
SourceSectionFilter.ANY,
listener);
}
static class ExampleLocal {
final TruffleContext context;
final WeakReference<Thread> thread;
ExampleLocal(TruffleContext context, Thread thread) {
this.context = context;
this.thread = new WeakReference<>(thread);
}
}
}
}