ru.qatools.mongodb.MongoPessimisticLock Maven / Gradle / Ivy

Go to download
package ru.qatools.mongodb;

import ru.qatools.mongodb.error.LockWaitTimeoutException;
import sun.reflect.generics.reflectiveObjects.NotImplementedException;

import java.util.concurrent.TimeUnit;
import java.util.concurrent.locks.Condition;
import java.util.concurrent.locks.Lock;

import static java.util.UUID.randomUUID;

/**
 * @author Ilya Sadykov
 */
public class MongoPessimisticLock implements Lock, PessimisticLock {
    final PessimisticLocking lock;
    final String uuid;


    public MongoPessimisticLock(PessimisticLocking lock, String uuid) {
        this.uuid = uuid;
        this.lock = lock;
    }

    public MongoPessimisticLock(PessimisticLocking lock) {
        this(lock, randomUUID().toString());
    }

    @Override
    public void lock() {
        lock.tryLock(uuid, Long.MAX_VALUE);
    }

    @Override
    public void lockInterruptibly() throws InterruptedException {
        throw new NotImplementedException();
    }

    @Override
    public boolean tryLock() {
        try {
            lock.tryLock(uuid, 0);
            return true;
        } catch (LockWaitTimeoutException e) {
            return false;
        }
    }

    @Override
    public boolean tryLock(long time, TimeUnit unit) throws InterruptedException {
        try {
            lock.tryLock(uuid, unit.toMillis(time));
            return true;
        } catch (LockWaitTimeoutException e) {
            return false;
        }
    }

    @Override
    public void forceUnlock() {
        lock.forceUnlock(uuid);
    }

    @Override
    public boolean isLockedByMe() {
        return lock.isLockedByMe(uuid);
    }

    @Override
    public boolean isLocked() {
        return lock.isLocked(uuid);
    }

    @Override
    public void unlock() {
        lock.unlock(uuid);
    }

    @Override
    public Condition newCondition() {
        throw new NotImplementedException();
    }
}