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.jdk.incubator.concurrent;
25 
26 import jdk.incubator.concurrent.ScopedValue;
27 import java.util.concurrent.atomic.AtomicInteger;
28 import java.util.concurrent.atomic.AtomicReference;
29 
30 @SuppressWarnings("preview")
31 public class ScopedValuesData {
32 
33     static final ScopedValue<Integer> sl1 = ScopedValue.newInstance();
34     static final ThreadLocal<Integer> tl1 = new ThreadLocal<>();
35 
36     static final ScopedValue<Integer> sl2 = ScopedValue.newInstance();
37     static final ScopedValue<Integer> sl3 = ScopedValue.newInstance();
38     static final ScopedValue<Integer> sl4 = ScopedValue.newInstance();
39     static final ScopedValue<Integer> sl5 = ScopedValue.newInstance();
40     static final ScopedValue<Integer> sl6 = ScopedValue.newInstance();
41     static final ScopedValue<AtomicInteger> sl_atomicInt = ScopedValue.newInstance();
42 
43     static final ScopedValue<Integer> unbound = ScopedValue.newInstance();
44 
45     static final ScopedValue<AtomicReference<Integer>> sl_atomicRef = ScopedValue.newInstance();
46 
47     static final ThreadLocal<Integer> tl2 = new ThreadLocal<>();
48     static final ThreadLocal<Integer> tl3 = new ThreadLocal<>();
49     static final ThreadLocal<Integer> tl4 = new ThreadLocal<>();
50     static final ThreadLocal<Integer> tl5 = new ThreadLocal<>();
51     static final ThreadLocal<Integer> tl6 = new ThreadLocal<>();
52     static final ThreadLocal<AtomicInteger> tl_atomicInt = new ThreadLocal<>();
53 
54     static final ScopedValue.Carrier VALUES = ScopedValue
55             .where(sl1, 42).where(sl2, 2).where(sl3, 3)
56             .where(sl4, 4).where(sl5, 5).where(sl6, 6);
57 
58     public static void run(Runnable action) {
59         try {
60             tl1.set(42); tl2.set(2); tl3.set(3); tl4.set(4); tl5.set(5); tl6.set(6);
61             tl1.get();  // Create the ScopedValue cache as a side effect
62             tl_atomicInt.set(new AtomicInteger());
63             VALUES.where(sl_atomicInt, new AtomicInteger())
64                   .where(sl_atomicRef, new AtomicReference<>())
65                   .run(action);
66         } finally {
67             tl1.remove(); tl2.remove(); tl3.remove(); tl4.remove(); tl5.remove(); tl6.remove();
68             tl_atomicInt.remove();
69         }
70     }
71 }
72