org.stone.tools.atomic.IntegerFieldUpdaterImpl Maven / Gradle / Ivy

Go to download
/*
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * Copyright(C) Chris2018998,All rights reserved.
 *
 * Project owner contact:[email protected].
 *
 * Project Licensed under Apache License v2.0.
 */
package org.stone.tools.atomic;

import org.stone.tools.UnsafeHolder;
import org.stone.tools.exception.ReflectionOperationException;
import sun.misc.Unsafe;

import java.util.concurrent.atomic.AtomicIntegerFieldUpdater;

/**
 * Atomic Integer Field Updater Implementation(Don't use in other place)
 *
 * @author Chris Liao
 * @version 1.0
 */
public final class IntegerFieldUpdaterImpl extends AtomicIntegerFieldUpdater {
    private final static Unsafe UA = UnsafeHolder.getUnsafe();
    private final long offset;

    private IntegerFieldUpdaterImpl(long offset) {
        this.offset = offset;
    }

    public static  AtomicIntegerFieldUpdater newUpdater(Class beanClass, String fieldName) {
        try {
            return new IntegerFieldUpdaterImpl<>(UA.objectFieldOffset(beanClass.getDeclaredField(fieldName)));
        } catch (NoSuchFieldException e) {
            throw new ReflectionOperationException(e);
        } catch (SecurityException e) {
            throw e;
        } catch (Throwable e) {
            return AtomicIntegerFieldUpdater.newUpdater(beanClass, fieldName);
        }
    }

    public boolean compareAndSet(T bean, int expect, int update) {
        return UA.compareAndSwapInt(bean, this.offset, expect, update);
    }

    public boolean weakCompareAndSet(T bean, int expect, int update) {
        return UA.compareAndSwapInt(bean, this.offset, expect, update);
    }

    public void set(T bean, int newValue) {
        UA.putIntVolatile(bean, this.offset, newValue);
    }

    public void lazySet(T bean, int newValue) {
        UA.putOrderedInt(bean, this.offset, newValue);
    }

    public int get(T bean) {
        return UA.getIntVolatile(bean, this.offset);
    }

}