java并发编程--自定义同步组件

concurrent包的组成

看下结构图：

编写一个自定义同步组件来加深对同步器的理解

业务要求：
* 编写一个自定义同步组件来加深对同步器的理解。
* 设计一个同步工具：该工具在同一时刻，只允许至多两个线程同时访问，超过两个线程的
* 访问将被阻塞，我们将这个同步工具命名为TwinsLock。
* 首先，确定访问模式。TwinsLock能够在同一时刻支持多个线程的访问，这显然是共享式
* 访问，因此，需要使用同步器提供的acquireShared(int args)方法等和Shared相关的方法，这就要
* 求TwinsLock必须重写tryAcquireShared(int args)方法和tryReleaseShared(int args)方法，这样才能
* 保证同步器的共享式同步状态的获取与释放方法得以执行。
* 其次，定义资源数。TwinsLock在同一时刻允许至多两个线程的同时访问，表明同步资源
* 数为2，这样可以设置初始状态status为2，当一个线程进行获取，status减1，该线程释放，则
* status加1，状态的合法范围为0、1和2，其中0表示当前已经有两个线程获取了同步资源，此时
* 再有其他线程对同步状态进行获取，该线程只能被阻塞。在同步状态变更时，需要使用
* compareAndSet(int expect,int update)方法做原子性保障。
* 最后，组合自定义同步器。前面的章节提到，自定义同步组件通过组合自定义同步器来完
* 成同步功能，一般情况下自定义同步器会被定义为自定义同步组件的内部类

public class TwinsLock implements Lock {

    private final Sync sync = new Sync(2);

    private static final class Sync extends AbstractQueuedSynchronizer {

        Sync(int count) {
            if (count <= 0) {
                throw new IllegalArgumentException("count must large than zero.");
            }
            setState(count);
        }

        //共享式同步状态的获取
        public int tryAcquireShared(int reduceCount) {
            for (; ; ) {//死循环---自旋
                int current = getState();
                int newCount = current - reduceCount;
                if (newCount < 0 || compareAndSetState(current, newCount)) {
                    return newCount;
                }
            }
        }

        //共享式同步状态的释放
        public boolean tryReleaseShared(int returnCount) {
            for (; ; ) {
                int current = getState();
                int newCount = current + returnCount;
                if (compareAndSetState(current, newCount)) {
                    return true;
                }
            }
        }

        final ConditionObject newCondition() {
            return new ConditionObject();
        }

    }

    public void lock() {
        sync.acquireShared(1);
    }

    public void lockInterruptibly() throws InterruptedException {
        sync.acquireInterruptibly(1);
    }

    public boolean tryLock() {
        return sync.tryAcquireShared(1) >= 0;
    }

    public boolean tryLock(long time, TimeUnit unit) throws InterruptedException {
        return sync.tryAcquireSharedNanos(1,unit.toNanos(time));
    }

    public void unlock() {
        sync.releaseShared(1);
    }

    public Condition newCondition() {
        return sync.newCondition();
    }
}

测试：

import org.junit.Test;
import utils.SleepUtils;

import java.util.concurrent.locks.Lock;

/**
 * 在测试用例中，定义了工作者
 * 线程Worker，该线程在执行过程中获取锁，当获取锁之后使当前线程睡眠1秒（并不释放锁），
 * 随后打印当前线程名称，最后再次睡眠1秒并释放锁
 * Created by tianjun on 2017/2/28.
 */
public class TwinsLockTest {

    @Test
    public void test() {
        final Lock lock = new TwinsLock();
        class Worker extends Thread {
            public void run(){
                while (true){
                    lock.lock();
                    try{
                        SleepUtils.second(1);
                        System.out.println(Thread.currentThread().getName());
                        SleepUtils.second(1);
                    }finally {
                        lock.unlock();
                    }
                }
            }
        }
        //启动10个线程
        for (int i=0;i<10;i++){
            Worker w = new Worker();
            w.setDaemon(true);
            w.start();
        }
        //每隔1秒换行
        for(int i =0 ;i<10;i++){
            SleepUtils.second(1);
            System.out.println();
        }
    }

}

运行结果：(同时只有两个线程能启动)

Thread-1

Thread-0

Thread-1

Thread-0

Thread-1

Thread-0

Thread-1

Thread-0

Thread-1

Thread-0

其实，这里隐藏一个问题，当一个线程重复获取锁时，像上面这样的继承和实现的同步器就会发生，自己阻塞自己的情况。所以就需要实现重入锁的这种功能，可以参考ReentrantLock锁的实现。