/* * Copyright (c) 2019, 2022, Oracle and/or its affiliates. All rights reserved. * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. * * This code is free software; you can redistribute it and/or modify it * under the terms of the GNU General Public License version 2 only, as * published by the Free Software Foundation. * * This code is distributed in the hope that it will be useful, but WITHOUT * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License * version 2 for more details (a copy is included in the LICENSE file that * accompanied this code). * * You should have received a copy of the GNU General Public License version * 2 along with this work; if not, write to the Free Software Foundation, * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. * * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA * or visit www.oracle.com if you need additional information or have any * questions. */ /** * @test * @summary Test Virtual threads using thread locals * @library /test/lib * @enablePreview * @run testng ThreadLocals */ import jdk.test.lib.thread.VThreadRunner; import org.testng.annotations.Test; import static org.testng.Assert.*; public class ThreadLocals { static final ThreadLocal<Object> LOCAL = new ThreadLocal<>(); static final ThreadLocal<Object> INHERITED_LOCAL = new InheritableThreadLocal<>(); /** * Basic test of thread local set/get. */ @Test public void testThreadLocal1() throws Exception { for (int i = 0; i < 10; i++) { VThreadRunner.run(() -> { assertTrue(LOCAL.get() == null); Object obj = new Object(); LOCAL.set(obj); assertTrue(LOCAL.get() == obj); }); } } /** * Test setting thread local before blocking operation. */ @Test public void testThreadLocal2() throws Exception { VThreadRunner.run(() -> { assertTrue(LOCAL.get() == null); Object obj = new Object(); LOCAL.set(obj); try { Thread.sleep(100); } catch (InterruptedException e) { } assertTrue(LOCAL.get() == obj); }); } /** * Test Thread that cannot set values for its copy of thread-locals. */ @Test public void testThreadLocal3() throws Exception { Object INITIAL_VALUE = new Object(); ThreadLocal<Object> LOCAL2 = new ThreadLocal<>() { @Override protected Object initialValue() { return INITIAL_VALUE; } }; ThreadLocal<Object> INHERITED_LOCAL2 = new InheritableThreadLocal<>() { @Override protected Object initialValue() { return INITIAL_VALUE; } }; VThreadRunner.run(VThreadRunner.NO_THREAD_LOCALS, () -> { assertThrows(UnsupportedOperationException.class, () -> LOCAL.set(null)); assertThrows(UnsupportedOperationException.class, () -> LOCAL.set(new Object())); assertTrue(LOCAL.get() == null); LOCAL.remove(); // should not throw assertThrows(UnsupportedOperationException.class, () -> LOCAL2.set(null)); assertThrows(UnsupportedOperationException.class, () -> LOCAL2.set(new Object())); assertTrue(LOCAL2.get() == INITIAL_VALUE); LOCAL2.remove(); // should not throw assertThrows(UnsupportedOperationException.class, () -> INHERITED_LOCAL.set(null)); assertThrows(UnsupportedOperationException.class, () -> INHERITED_LOCAL.set(new Object())); assertTrue(INHERITED_LOCAL.get() == null); INHERITED_LOCAL.remove(); // should not throw assertThrows(UnsupportedOperationException.class, () -> INHERITED_LOCAL2.set(null)); assertThrows(UnsupportedOperationException.class, () -> INHERITED_LOCAL2.set(new Object())); assertTrue(INHERITED_LOCAL2.get() == INITIAL_VALUE); INHERITED_LOCAL2.remove(); // should not throw }); } /** * Basic test of inheritable thread local set/get, no initial value inherited. */ @Test public void testInheritedThreadLocal1() throws Exception { assertTrue(INHERITED_LOCAL.get() == null); for (int i = 0; i < 10; i++) { VThreadRunner.run(() -> { assertTrue(INHERITED_LOCAL.get() == null); Object obj = new Object(); INHERITED_LOCAL.set(obj); assertTrue(INHERITED_LOCAL.get() == obj); }); } assertTrue(INHERITED_LOCAL.get() == null); } /** * Test inheriting initial value of InheritableThreadLocal from platform thread. */ @Test public void testInheritedThreadLocal2() throws Exception { assertTrue(INHERITED_LOCAL.get() == null); var obj = new Object(); INHERITED_LOCAL.set(obj); try { VThreadRunner.run(() -> { assertTrue(INHERITED_LOCAL.get() == obj); }); } finally { INHERITED_LOCAL.remove(); } } /** * Test inheriting initial value of InheritableThreadLocal from virtual thread. */ @Test public void testInheritedThreadLocal3() throws Exception { assertTrue(INHERITED_LOCAL.get() == null); VThreadRunner.run(() -> { var obj = new Object(); INHERITED_LOCAL.set(obj); VThreadRunner.run(() -> { assertTrue(INHERITED_LOCAL.get() == obj); }); assertTrue(INHERITED_LOCAL.get() == obj); }); assertTrue(INHERITED_LOCAL.get() == null); } /** * Test Thread that does not inherit initial value of InheritableThreadLocals * from platform thread. */ @Test public void testInheritedThreadLocal4() throws Exception { assertTrue(INHERITED_LOCAL.get() == null); var obj = new Object(); INHERITED_LOCAL.set(obj); try { int characteristics = VThreadRunner.NO_INHERIT_THREAD_LOCALS; VThreadRunner.run(characteristics, () -> { assertTrue(INHERITED_LOCAL.get() == null); }); } finally { INHERITED_LOCAL.remove(); } } /** * Test Thread that does not inherit initial value of InheritableThreadLocals * from virtual thread. */ @Test public void testInheritedThreadLocal5() throws Exception { assertTrue(INHERITED_LOCAL.get() == null); VThreadRunner.run(() -> { var obj = new Object(); INHERITED_LOCAL.set(obj); int characteristics = VThreadRunner.NO_INHERIT_THREAD_LOCALS; VThreadRunner.run(characteristics, () -> { assertTrue(INHERITED_LOCAL.get() == null); }); assertTrue(INHERITED_LOCAL.get() == obj); }); assertTrue(INHERITED_LOCAL.get() == null); } }