阅读598 返回首页    go 阿里云 go 技术社区[云栖]

JUC中Atomic class之lazySet的一点疑惑

最近再次翻netty和disrupt的源码, 发现一些地方使用AtomicXXX.lazySet()/unsafe.putOrderedXXX系列, 以前一直没有注意lazySet这个方法, 仔细研究一下发现很有意思。我们拿AtomicReferenceFieldUpdater的set()和lazySet()作比较, 其他AtomicXXX类和这个类似。

1 public void set(T obj, V newValue) {
2     // ...
3     unsafe.putObjectVolatile(obj, offset, newValue);
4 }
5  
6 public void lazySet(T obj, V newValue) {
7     // ...
8     unsafe.putOrderedObject(obj, offset, newValue);
9 }

1.首先set()是对volatile变量的一个写操作, 我们知道volatile的write为了保证对其他线程的可见性会追加以下两个Fence(内存屏障)
1)StoreStore // 在intel cpu中, 不存在[写写]重排序, 这个可以直接省略了
2)StoreLoad // 这个是所有内存屏障里最耗性能的
注: 内存屏障相关参考Doug Lea大大的cookbook (https://g.oswego.edu/dl/jmm/cookbook.html)

2.Doug Lea大大又说了, lazySet()省去了StoreLoad屏障, 只留下StoreStore
在这里 https://bugs.java.com/bugdatabase/view_bug.do?bug_id=6275329
把最耗性能的StoreLoad拿掉, 性能必然会提高不少(虽然不能禁止写读的重排序了保证不了可见性, 但给其他应用场景提供了更好的选择, 比如上边连接中Doug Lea举例的场景)。

但是但是, 在好奇心驱使下我翻了下JDK的源码(unsafe.cpp):

01 // 这是unsafe.putObjectVolatile()
02 UNSAFE_ENTRY(void, Unsafe_SetObjectVolatile(JNIEnv *env, jobject unsafe, jobject obj, jlong offset, jobject x_h))
03     UnsafeWrapper("Unsafe_SetObjectVolatile");
04     oop x = JNIHandles::resolve(x_h);
05     oop p = JNIHandles::resolve(obj);
06     void* addr = index_oop_from_field_offset_long(p, offset);
07     OrderAccess::release();
08     if (UseCompressedOops) {
09         oop_store((narrowOop*)addr, x);
10     } else {
11         oop_store((oop*)addr, x);
12     }
13     OrderAccess::fence();
14 UNSAFE_END
15  
16 // 这是unsafe.putOrderedObject()
17 UNSAFE_ENTRY(void, Unsafe_SetOrderedObject(JNIEnv *env, jobject unsafe, jobject obj, jlong offset, jobject x_h))
18     UnsafeWrapper("Unsafe_SetOrderedObject");
19     oop x = JNIHandles::resolve(x_h);
20     oop p = JNIHandles::resolve(obj);
21     void* addr = index_oop_from_field_offset_long(p, offset);
22     OrderAccess::release();
23     if (UseCompressedOops) {
24         oop_store((narrowOop*)addr, x);
25     } else {
26         oop_store((oop*)addr, x);
27     }
28     OrderAccess::fence();
29 UNSAFE_END

仔细看代码是不是有种被骗的感觉, 他喵的一毛一样啊. 难道是JIT做了手脚?生成汇编看看

生成assembly code需要hsdis插件

mac平台从这里下载 https://kenai.com/projects/base-hsdis/downloads/directory/gnu-versions

linux和windows可以从R大的[高级语言虚拟机圈子]下载 https://hllvm.group.iteye.com/

为了测试代码简单, 使用AtomicLong来测:

01 // set()
02 public class LazySetTest {
03     private static final AtomicLong a = new AtomicLong();
04  
05     public static void main(String[] args) {
06         for (int i = 0; i < 100000000; i++) {
07             a.set(i);
08         }
09     }
10 }
11  
12 // lazySet()
13 public class LazySetTest {
14     private static final AtomicLong a = new AtomicLong();
15  
16     public static void main(String[] args) {
17         for (int i = 0; i < 100000000; i++) {
18             a.lazySet(i);
19         }
20     }
21 }

分别执行以下命令:

01 1.export LD_LIBRARY_PATH=~/hsdis插件路径/
02 2.javac LazySetTest.java && java -XX:+UnlockDiagnosticVMOptions -XX:+PrintAssembly LazySetTest
03  
04 // ------------------------------------------------------
05 // set()的assembly code片段:
06 0x000000010ccbfeb3: mov    %r10,0x10(%r9)
07 0x000000010ccbfeb7: lock addl $0x0,(%rsp)     ;*putfield value
08                                             ; - java.util.concurrent.atomic.AtomicLong::set@2 (line 112)
09                                             ; - LazySetTest::main@13 (line 13)
10 0x000000010ccbfebc: inc    %ebp               ;*iinc
11                                             ; - LazySetTest::main@16 (line 12)
12 // ------------------------------------------------------
13 // lazySet()的assembly code片段:
14 0x0000000108766faf: mov    %r10,0x10(%rcx)    ;*invokevirtual putOrderedLong
15                                             ; - java.util.concurrent.atomic.AtomicLong::lazySet@8 (line 122)
16                                             ; - LazySetTest::main@13 (line 13)
17 0x0000000108766fb3: inc    %ebp               ;*iinc
18                                             ; - LazySetTest::main@16 (line 12)

好吧, set()生成的assembly code多了一个lock前缀的指令

查询IA32手册可知道, lock addl $0x0,(%rsp)其实就是StoreLoad屏障了, 而lazySet()确实没生成StoreLoad屏障

这里JIT除了将方法内联, 相同代码生成不同指令是怎么做到的?

https://hg.openjdk.java.net/jdk7u/jdk7u/hotspot/file/6e9aa487055f/src/share/vm/classfile/vmSymbols.hpp

查看如上代码, 812行和868行分别有如下代码:

1 do_intrinsic(_putObjectVolatile,        sun_misc_Unsafe,        putObjectVolatile_name, putObject_signature,   F_RN)
2 do_intrinsic(_putOrderedObject,         sun_misc_Unsafe,        putOrderedObject_name, putOrderedObject_signature, F_RN)

putObjectVolatile与putOrderedObject都在vmSymbols.hpp的宏定义中,jvm会根据instrinsics id生成特定的指令集 putObjectVolatile与putOrderedObject生成的汇编指令不同估计是源于这里了, 继续往下看 hotspot/src/share/vm/opto/libaray_call.cpp这个类:
首先看如下两行代码:

1 case vmIntrinsics::_putObjectVolatile:        return inline_unsafe_access(!is_native_ptr,  is_store, T_OBJECT,   is_volatile);
2 case vmIntrinsics::_putOrderedObject:         return inline_unsafe_ordered_store(T_OBJECT);

再看inline_unsafe_access()和inline_unsafe_ordered_store(), 不贴出全部代码了, 只贴出重要的部分:

01 bool LibraryCallKit::inline_unsafe_ordered_store(BasicType type) {
02   // This is another variant of inline_unsafe_access, differing in
03   // that it always issues store-store ("release") barrier and ensures
04   // store-atomicity (which only matters for "long").
05  
06   // ...
07   if (type == T_OBJECT) // reference stores need a store barrier.
08     store = store_oop_to_unknown(control(), base, adr, adr_type, val, type);
09   else {
10     store = store_to_memory(control(), adr, val, type, adr_type, require_atomic_access);
11   }
12   insert_mem_bar(Op_MemBarCPUOrder);
13   return true;
14 }
15  
16 ---------------------------------------------------------------------------------------------------------
17  
18 bool LibraryCallKit::inline_unsafe_access(bool is_native_ptr, bool is_store, BasicType type, bool is_volatile) {
19   // ....
20  
21   if (is_volatile) {
22     if (!is_store)
23       insert_mem_bar(Op_MemBarAcquire);
24     else
25       insert_mem_bar(Op_MemBarVolatile);
26   }
27  
28   if (need_mem_bar) insert_mem_bar(Op_MemBarCPUOrder);
29  
30   return true;
31 }

我们可以看到 inline_unsafe_access()方法中, 如果是is_volatile为true, 并且是store操作的话, 有这样的一句代码 insert_mem_bar(Op_MemBarVolatile), 而inline_unsafe_ordered_store没有插入这句代码

再继续看/hotspot/src/cpu/x86/vm/x86_64.ad的membar_volatile

01 instruct membar_volatile(rFlagsReg cr) %{
02   match(MemBarVolatile);
03   effect(KILL cr);
04   ins_cost(400);
05  
06   format %{
07     $$template
08     if (os::is_MP()) {
09       $$emit$$"lock addl [rsp + #0], 0\t! membar_volatile"
10     } else {
11       $$emit$$"MEMBAR-volatile ! (empty encoding)"
12     }
13   %}
14   ins_encode %{
15     __ membar(Assembler::StoreLoad);
16   %}
17   ins_pipe(pipe_slow);
18 %}

lock addl [rsp + #0], 0\t! membar_volatile指令原来来自这里

总结:
错过一些细节, 但在主流程上感觉是有一点点明白了, 有错误之处请指正

参考了以下资料:
1.https://g.oswego.edu/dl/jmm/cookbook.html
2.https://wikis.oracle.com/display/HotSpotInternals/PrintAssembly
3.https://www.quora.com/How-does-AtomicLong-lazySet-work
4.https://bad-concurrency.blogspot.ru/2012/10/talk-from-jax-london.html

最后更新：2017-05-22 15:34:08

  上一篇：go  人工智能概念第一股？即将在美国上市的Veritone是怎样一家公司
  下一篇：go  交通数据分析帮你避开99%的酒驾司机