1 /*
 2  * Copyright (c) 2021, Red Hat, Inc. All rights reserved.
 3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 4  *
 5  * This code is free software; you can redistribute it and/or modify it
 6  * under the terms of the GNU General Public License version 2 only, as
 7  * published by the Free Software Foundation.
 8  *
 9  * This code is distributed in the hope that it will be useful, but WITHOUT
10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
12  * version 2 for more details (a copy is included in the LICENSE file that
13  * accompanied this code).
14  *
15  * You should have received a copy of the GNU General Public License version
16  * 2 along with this work; if not, write to the Free Software Foundation,
17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18  *
19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20  * or visit www.oracle.com if you need additional information or have any
21  * questions.
22  */
23 
24 package org.openjdk.bench.java.lang;
25 
26 import java.util.concurrent.atomic.AtomicInteger;
27 import java.util.concurrent.atomic.AtomicReference;
28 
29 @SuppressWarnings("preview")
30 public class ScopedValuesData {
31 
32     static final ScopedValue<Integer> sl1 = ScopedValue.newInstance();
33     static final ThreadLocal<Integer> tl1 = new ThreadLocal<>();
34 
35     static final ScopedValue<Integer> sl2 = ScopedValue.newInstance();
36     static final ScopedValue<Integer> sl3 = ScopedValue.newInstance();
37     static final ScopedValue<Integer> sl4 = ScopedValue.newInstance();
38     static final ScopedValue<Integer> sl5 = ScopedValue.newInstance();
39     static final ScopedValue<Integer> sl6 = ScopedValue.newInstance();
40     static final ScopedValue<AtomicInteger> sl_atomicInt = ScopedValue.newInstance();
41 
42     static final ScopedValue<Integer> unbound = ScopedValue.newInstance();
43 
44     static final ScopedValue<AtomicReference<Integer>> sl_atomicRef = ScopedValue.newInstance();
45 
46     static final ThreadLocal<Integer> tl2 = new ThreadLocal<>();
47     static final ThreadLocal<Integer> tl3 = new ThreadLocal<>();
48     static final ThreadLocal<Integer> tl4 = new ThreadLocal<>();
49     static final ThreadLocal<Integer> tl5 = new ThreadLocal<>();
50     static final ThreadLocal<Integer> tl6 = new ThreadLocal<>();
51     static final ThreadLocal<AtomicInteger> tl_atomicInt = new ThreadLocal<>();
52 
53     static final ScopedValue.Carrier VALUES = ScopedValue
54             .where(sl1, 42).where(sl2, 2).where(sl3, 3)
55             .where(sl4, 4).where(sl5, 5).where(sl6, 6);
56 
57     public static void run(Runnable action) {
58         try {
59             tl1.set(42); tl2.set(2); tl3.set(3); tl4.set(4); tl5.set(5); tl6.set(6);
60             tl1.get();  // Create the ScopedValue cache as a side effect
61             tl_atomicInt.set(new AtomicInteger());
62             VALUES.where(sl_atomicInt, new AtomicInteger())
63                   .where(sl_atomicRef, new AtomicReference<>())
64                   .run(action);
65         } finally {
66             tl1.remove(); tl2.remove(); tl3.remove(); tl4.remove(); tl5.remove(); tl6.remove();
67             tl_atomicInt.remove();
68         }
69     }
70 }
71